Add files via upload

backend/internal/report/report.go

@@ -0,0 +1,189 @@
+package report
+
+import (
+	"ALLinSSL/backend/public"
+	"crypto/tls"
+	"encoding/json"
+	"fmt"
+	"github.com/jordan-wright/email"
+	"net/smtp"
+	"time"
+)
+
+func GetSqlite() (*public.Sqlite, error) {
+	s, err := public.NewSqlite("data/data.db", "")
+	if err != nil {
+		return nil, err
+	}
+	s.Connect()
+	s.TableName = "report"
+	return s, nil
+}
+
+func GetList(search string, p, limit int64) ([]map[string]any, int, error) {
+	var data []map[string]any
+	var count int64
+	s, err := GetSqlite()
+	if err != nil {
+		return data, 0, err
+	}
+	defer s.Close()
+	
+	var limits []int64
+	if p >= 0 && limit >= 0 {
+		limits = []int64{0, limit}
+		if p > 1 {
+			limits[0] = (p - 1) * limit
+			limits[1] = p * limit
+		}
+	}
+	
+	if search != "" {
+		count, err = s.Where("name like ?", []interface{}{"%" + search + "%"}).Count()
+		data, err = s.Where("name like ?", []interface{}{"%" + search + "%"}).Limit(limits).Order("update_time", "desc").Select()
+	} else {
+		count, err = s.Count()
+		data, err = s.Order("update_time", "desc").Limit(limits).Select()
+	}
+	if err != nil {
+		return data, 0, err
+	}
+	return data, int(count), nil
+}
+
+func GetReport(id string) (map[string]any, error) {
+	s, err := GetSqlite()
+	if err != nil {
+		return nil, err
+	}
+	defer s.Close()
+	data, err := s.Where("id=?", []interface{}{id}).Select()
+	if err != nil {
+		return nil, err
+	}
+	if len(data) == 0 {
+		return nil, fmt.Errorf("没有找到此通知配置")
+	}
+	return data[0], nil
+	
+}
+
+func AddReport(Type, config, name string) error {
+	s, err := GetSqlite()
+	if err != nil {
+		return err
+	}
+	defer s.Close()
+	now := time.Now().Format("2006-01-02 15:04:05")
+	_, err = s.Insert(map[string]interface{}{
+		"name":        name,
+		"type":        Type,
+		"config":      config,
+		"create_time": now,
+		"update_time": now,
+	})
+	return err
+}
+
+func UpdReport(id, config, name string) error {
+	s, err := GetSqlite()
+	if err != nil {
+		return err
+	}
+	defer s.Close()
+	_, err = s.Where("id=?", []interface{}{id}).Update(map[string]interface{}{
+		"name":   name,
+		"config": config,
+	})
+	return err
+}
+
+func DelReport(id string) error {
+	s, err := GetSqlite()
+	if err != nil {
+		return err
+	}
+	defer s.Close()
+	_, err = s.Where("id=?", []interface{}{id}).Delete()
+	return err
+}
+
+func NotifyTest(id string) error {
+	if id == "" {
+		return fmt.Errorf("缺少参数")
+	}
+	providerData, err := GetReport(id)
+	if err != nil {
+		return err
+	}
+	params := map[string]any{
+		"provider_id": id,
+		"body":        "测试消息通道",
+		"subject":     "测试消息通道",
+	}
+	switch providerData["type"] {
+	case "mail":
+		err = NotifyMail(params)
+	}
+	return err
+}
+
+func Notify(params map[string]any) error {
+	if params == nil {
+		return fmt.Errorf("缺少参数")
+	}
+	providerName, ok := params["provider"].(string)
+	if !ok {
+		return fmt.Errorf("通知类型错误")
+	}
+	switch providerName {
+	case "mail":
+		return NotifyMail(params)
+	// case "btpanel-site":
+	// 	return NotifyBt(params)
+	default:
+		return fmt.Errorf("不支持的通知类型")
+	}
+}
+
+func NotifyMail(params map[string]any) error {
+	
+	if params == nil {
+		return fmt.Errorf("缺少参数")
+	}
+	providerID := params["provider_id"].(string)
+	// fmt.Println(providerID)
+	providerData, err := GetReport(providerID)
+	if err != nil {
+		return err
+	}
+	configStr := providerData["config"].(string)
+	var config map[string]string
+	err = json.Unmarshal([]byte(configStr), &config)
+	if err != nil {
+		return fmt.Errorf("解析配置失败: %v", err)
+	}
+	
+	e := email.NewEmail()
+	e.From = config["sender"]
+	e.To = []string{config["receiver"]}
+	e.Subject = params["subject"].(string)
+	
+	e.Text = []byte(params["body"].(string))
+	
+	addr := fmt.Sprintf("%s:%s", config["smtpHost"], config["smtpPort"])
+	
+	auth := smtp.PlainAuth("", config["sender"], config["password"], config["smtpHost"])
+	
+	// 使用 SSL（通常是 465）
+	if config["smtpPort"] == "465" {
+		tlsConfig := &tls.Config{
+			InsecureSkipVerify: true, // 开发阶段跳过证书验证，生产建议关闭
+			ServerName:         config["smtpHost"],
+		}
+		return e.SendWithTLS(addr, auth, tlsConfig)
+	}
+	
+	// 普通明文发送（25端口，非推荐）
+	return e.Send(addr, auth)
+}

backend/internal/report/report_test.go

@@ -0,0 +1,17 @@
+package report
+
+import (
+	"fmt"
+	"testing"
+)
+
+func TestMail(t *testing.T) {
+	config := map[string]any{
+		"provider":    "mail",
+		"provider_id": "4",
+		"subject":     "执行结束",
+		"body":        "执行结束",
+	}
+	err := NotifyMail(config)
+	fmt.Println(err)
+}