94 lines
2.9 KiB
Go
94 lines
2.9 KiB
Go
package telegram
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/nicholas-fedor/shoutrrr/pkg/format"
|
|
"github.com/nicholas-fedor/shoutrrr/pkg/types"
|
|
)
|
|
|
|
// Scheme identifies this service in configuration URLs.
|
|
const (
|
|
Scheme = "telegram"
|
|
)
|
|
|
|
// ErrInvalidToken indicates an invalid Telegram token format or content.
|
|
var (
|
|
ErrInvalidToken = errors.New("invalid telegram token")
|
|
ErrNoChannelsDefined = errors.New("no channels defined in config URL")
|
|
)
|
|
|
|
// Config holds settings for the Telegram notification service.
|
|
type Config struct {
|
|
Token string `url:"user"`
|
|
Preview bool ` default:"Yes" desc:"If disabled, no web page preview will be displayed for URLs" key:"preview"`
|
|
Notification bool ` default:"Yes" desc:"If disabled, sends Message silently" key:"notification"`
|
|
ParseMode parseMode ` default:"None" desc:"How the text Message should be parsed" key:"parsemode"`
|
|
Chats []string ` desc:"Chat IDs or Channel names (using @channel-name)" key:"chats,channels"`
|
|
Title string ` default:"" desc:"Notification title, optionally set by the sender" key:"title"`
|
|
}
|
|
|
|
// Enums returns the fields that use an EnumFormatter for their values.
|
|
func (config *Config) Enums() map[string]types.EnumFormatter {
|
|
return map[string]types.EnumFormatter{
|
|
"ParseMode": ParseModes.Enum,
|
|
}
|
|
}
|
|
|
|
// GetURL generates a URL from the current configuration values.
|
|
func (config *Config) GetURL() *url.URL {
|
|
resolver := format.NewPropKeyResolver(config)
|
|
|
|
return config.getURL(&resolver)
|
|
}
|
|
|
|
// SetURL updates the configuration from a URL representation.
|
|
func (config *Config) SetURL(url *url.URL) error {
|
|
resolver := format.NewPropKeyResolver(config)
|
|
|
|
return config.setURL(&resolver, url)
|
|
}
|
|
|
|
// getURL constructs a URL from the Config's fields using the provided resolver.
|
|
func (config *Config) getURL(resolver types.ConfigQueryResolver) *url.URL {
|
|
tokenParts := strings.Split(config.Token, ":")
|
|
|
|
return &url.URL{
|
|
User: url.UserPassword(tokenParts[0], tokenParts[1]),
|
|
Host: Scheme,
|
|
Scheme: Scheme,
|
|
ForceQuery: true,
|
|
RawQuery: format.BuildQuery(resolver),
|
|
}
|
|
}
|
|
|
|
// setURL updates the Config from a URL using the provided resolver.
|
|
func (config *Config) setURL(resolver types.ConfigQueryResolver, url *url.URL) error {
|
|
password, _ := url.User.Password()
|
|
|
|
token := url.User.Username() + ":" + password
|
|
if url.String() != "telegram://dummy@dummy.com" {
|
|
if !IsTokenValid(token) {
|
|
return fmt.Errorf("%w: %s", ErrInvalidToken, token)
|
|
}
|
|
}
|
|
|
|
for key, vals := range url.Query() {
|
|
if err := resolver.Set(key, vals[0]); err != nil {
|
|
return fmt.Errorf("setting config property %q from URL query: %w", key, err)
|
|
}
|
|
}
|
|
|
|
if url.String() != "telegram://dummy@dummy.com" {
|
|
if len(config.Chats) < 1 {
|
|
return ErrNoChannelsDefined
|
|
}
|
|
}
|
|
|
|
config.Token = token
|
|
|
|
return nil
|
|
}
|