小重构:

* 支持多目标通知
* 配置文件改用 toml 格式，不再需要手动 trim
* 引入自动 validate，不再手动校验配置错误。
* 增加飞书支持

Author: cupen

build.sh

@@ -0,0 +1,56 @@
+package conf
+
+import (
+	"fmt"
+	"sync/atomic"
+
+	validator "github.com/go-playground/validator/v10"
+)
+
+var (
+	// c = &Config{}
+	cfg atomic.Value
+)
+
+type Config struct {
+	DryRun    *DryRun
+	WebHook   *WebHook
+	Mail      *Mail
+	Slack     *Slack
+	BearyChat *BearyChat
+	Feishu    *Feishu
+}
+
+func (c *Config) Validate() error {
+	if c == nil {
+		return fmt.Errorf("nil config")
+	}
+
+	if c.BearyChat != nil {
+		if c.BearyChat.URL == "" {
+			return fmt.Errorf("Invalid bearcychat config")
+		}
+	}
+	if c.Mail != nil {
+		if len(c.Mail.Receivers) <= 0 {
+			return fmt.Errorf("Invalid mail config")
+		}
+	}
+	if c.Slack != nil {
+		if c.Slack.URL == "" {
+			return fmt.Errorf("Invalid slack config")
+		}
+	}
+
+	if c.WebHook != nil {
+		if c.WebHook.URL == "" {
+			return fmt.Errorf("Invalid slack config")
+		}
+	}
+
+	validate := validator.New()
+	if err := validate.Struct(c); err != nil {
+		return err
+	}
+	return nil
+}

conf/config.go

@@ -0,0 +1,36 @@
+package conf
+
+import (
+	"io"
+	"os"
+
+	"github.com/pelletier/go-toml"
+)
+
+func Init(fpath string) error {
+	f, err := os.Open(fpath)
+	if err != nil {
+		return err
+	}
+	data, err := io.ReadAll(f)
+	if err != nil {
+		return err
+	}
+	c := Config{}
+	if err := toml.Unmarshal(data, &c); err != nil {
+		return err
+	}
+	if err := c.Validate(); err != nil {
+		return err
+	}
+	cfg.Store(&c)
+	return nil
+}
+
+func Reload(fpath string) error {
+	return Init(fpath)
+}
+
+func Get() *Config {
+	return cfg.Load().(*Config)
+}

conf/init.go

@@ -0,0 +1,35 @@
+package conf
+
+type DryRun struct {
+	Text string `toml:"text"`
+}
+
+// 邮件服务器
+type Mail struct {
+	Receivers      []string `toml:"receivers" validate:"required"`
+	ServerUser     string   `toml:"server_user" validate:"required"`
+	ServerPassword string   `toml:"server_password" validate:"required"`
+	ServerHost     string   `toml:"server_host" validate:"required"`
+	ServerPort     int      `toml:"server_port" validate:"required"`
+}
+
+type WebHook struct {
+	URL     string `toml:"url" validate:"required"`
+	Timeout int    `toml:"timeout"`
+}
+
+type Slack struct {
+	URL     string `toml:"url" validate:"required"`
+	Channel string `toml:"channel" validate:"required"`
+}
+
+type BearyChat struct {
+	URL     string `toml:"url" validate:"required"`
+	Channel string `toml:"channel"`
+	Timeout int    `toml:"timeout"`
+}
+
+type Feishu struct {
+	URL     string `toml:"url" validate:"required"`
+	Timeout int    `toml:"timeout"`
+}