44 lines
1.2 KiB
Go
44 lines
1.2 KiB
Go
package generators
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/nicholas-fedor/shoutrrr/pkg/generators/basic"
|
|
"github.com/nicholas-fedor/shoutrrr/pkg/generators/xouath2"
|
|
"github.com/nicholas-fedor/shoutrrr/pkg/services/telegram"
|
|
"github.com/nicholas-fedor/shoutrrr/pkg/types"
|
|
)
|
|
|
|
var ErrUnknownGenerator = errors.New("unknown generator")
|
|
|
|
var generatorMap = map[string]func() types.Generator{
|
|
"basic": func() types.Generator { return &basic.Generator{} },
|
|
"oauth2": func() types.Generator { return &xouath2.Generator{} },
|
|
"telegram": func() types.Generator { return &telegram.Generator{} },
|
|
}
|
|
|
|
// NewGenerator creates an instance of the generator that corresponds to the provided identifier.
|
|
func NewGenerator(identifier string) (types.Generator, error) {
|
|
generatorFactory, valid := generatorMap[strings.ToLower(identifier)]
|
|
if !valid {
|
|
return nil, fmt.Errorf("%w: %q", ErrUnknownGenerator, identifier)
|
|
}
|
|
|
|
return generatorFactory(), nil
|
|
}
|
|
|
|
// ListGenerators lists all available generators.
|
|
func ListGenerators() []string {
|
|
generators := make([]string, len(generatorMap))
|
|
|
|
i := 0
|
|
|
|
for key := range generatorMap {
|
|
generators[i] = key
|
|
i++
|
|
}
|
|
|
|
return generators
|
|
}
|