feat: support push notification clients

pkg/notifier/clients.go

@@ -0,0 +1,70 @@
+package notifier
+
+import (
+	"encoding/json"
+
+	"github.com/nikoksr/notify/service/dingding"
+	po "github.com/nikoksr/notify/service/pushover"
+	"github.com/nikoksr/notify/service/telegram"
+	"github.com/pkg/errors"
+)
+
+type PushoverConfig struct {
+	UserKey  string `json:"user_key"`
+	GroupKey string `json:"group_key"`
+	AppToken    string `json:"app_token"`
+}
+
+func NewPushoverClient(s string) (NotificationClient, error) {
+	var cfg PushoverConfig
+	if err := json.Unmarshal([]byte(s), &cfg); err != nil {
+		return nil, errors.Wrap(err, "json")
+	}
+
+	c := po.New(cfg.AppToken)
+	if cfg.UserKey != "" {
+		c.AddReceivers(cfg.UserKey)
+	}
+	if cfg.GroupKey != "" {
+		c.AddReceivers(cfg.GroupKey)
+	}
+
+	return &Notifier{service: c}, nil
+}
+
+type DingTalkConfig struct {
+	Token  string `json:"token"`
+	Secret string `json:"secret"`
+}
+
+func NewDingTalkClient(s string) (NotificationClient, error) {
+	var cfg DingTalkConfig
+	if err := json.Unmarshal([]byte(s), &cfg); err != nil {
+		return nil, errors.Wrap(err, "json")
+	}
+
+	svc := dingding.New(&dingding.Config{
+		Token:  cfg.Token,
+		Secret: cfg.Secret,
+	})
+	return &Notifier{service: svc}, nil
+}
+
+type TelegramConfig struct {
+	Token  string `json:"token"`
+	ChatID int64  `json:"chat_id"`
+}
+
+func NewTelegramClient(s string) (NotificationClient, error) {
+	var cfg TelegramConfig
+	if err := json.Unmarshal([]byte(s), &cfg); err != nil {
+		return nil, errors.Wrap(err, "json")
+	}
+
+	svc, err := telegram.New(cfg.Token)
+	if err != nil {
+		panic(err)
+	}
+	svc.AddReceivers(cfg.ChatID)
+	return &Notifier{service: svc}, nil
+}

pkg/notifier/doc.go

@@ -0,0 +1,36 @@
+package notifier
+
+import (
+	"context"
+	"polaris/pkg/utils"
+
+	"github.com/nikoksr/notify"
+)
+
+type HandlerFunc func(string) (NotificationClient, error)
+
+type NotificationClient interface {
+	SendMsg(msg string) error
+}
+
+type Notifier struct {
+	service notify.Notifier
+}
+
+func (s *Notifier) SendMsg(msg string) error {
+	notifier := notify.New()
+	notifier.UseServices(s.service)
+	return notifier.Send(context.TODO(), "polaris", msg)
+}
+
+var handler = utils.Map[string, HandlerFunc]{}
+
+func init() {
+	handler.Store("pushover", NewPushoverClient)
+	handler.Store("dingtalk", NewDingTalkClient)
+	handler.Store("telegram", NewTelegramClient)
+}
+
+func Gethandler(name string) (HandlerFunc, bool) {
+	return handler.Load(name)
+}

pkg/notifier/message/message.go

@@ -0,0 +1,8 @@
+package message
+
+const (
+	BeginDownload = "开始下载：%s"
+	DownloadComplete = "下载完成：%s"
+	ProcessingComplete = "文件处理完成：%s"
+	ProcessingFailed = "文件处理失败：%s"
+)