Adding upstream version 0.8.9.
Signed-off-by: Daniel Baumann <daniel@debian.org>
This commit is contained in:
parent
3b2c48b5e4
commit
c0c4addb85
285 changed files with 25880 additions and 0 deletions
94
pkg/services/telegram/telegram_config.go
Normal file
94
pkg/services/telegram/telegram_config.go
Normal file
|
@ -0,0 +1,94 @@
|
|||
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
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue