迁移公告，邮件

2026-08-13 22:04:27 +08:00
parent 21c264fcd7
commit 9044bafdff
18 changed files with 946 additions and 28 deletions
@@ -0,0 +1,127 @@
+package data
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"time"
+
+	"kra/internal/biz"
+
+	"gorm.io/gorm"
+)
+
+type announcementPO struct {
+	ID          uint `gorm:"primaryKey"`
+	CreatedAt   time.Time
+	UpdatedAt   time.Time
+	DeletedAt   gorm.DeletedAt `gorm:"index"`
+	Title       string
+	Content     string `gorm:"type:text"`
+	UserID      *uint  `gorm:"column:user_id"`
+	Attachments []byte `gorm:"type:json"`
+}
+
+func (announcementPO) TableName() string { return "gva_announcements_info" }
+
+type announcementRepo struct{ data *Data }
+
+func NewAnnouncementRepo(data *Data) biz.AnnouncementRepo { return &announcementRepo{data: data} }
+
+func newAnnouncement(item *biz.Announcement) announcementPO {
+	attachments := []byte(item.Attachments)
+	if len(attachments) == 0 || !json.Valid(attachments) {
+		attachments = []byte("[]")
+	}
+	return announcementPO{ID: item.ID, Title: item.Title, Content: item.Content, UserID: item.UserID, Attachments: attachments}
+}
+
+func announcementToBiz(item announcementPO) *biz.Announcement {
+	attachments := json.RawMessage(item.Attachments)
+	if len(attachments) == 0 || !json.Valid(attachments) {
+		attachments = json.RawMessage("[]")
+	}
+	return &biz.Announcement{ID: item.ID, CreatedAt: item.CreatedAt, UpdatedAt: item.UpdatedAt, Title: item.Title, Content: item.Content, UserID: item.UserID, Attachments: attachments}
+}
+
+func (r *announcementRepo) Create(ctx context.Context, item *biz.Announcement) error {
+	po := newAnnouncement(item)
+	if err := r.data.gormDB.WithContext(ctx).Create(&po).Error; err != nil {
+		return err
+	}
+	item.ID, item.CreatedAt, item.UpdatedAt = po.ID, po.CreatedAt, po.UpdatedAt
+	return nil
+}
+
+func (r *announcementRepo) Delete(ctx context.Context, id uint) error {
+	return r.data.gormDB.WithContext(ctx).Delete(&announcementPO{}, id).Error
+}
+
+func (r *announcementRepo) DeleteByIDs(ctx context.Context, ids []uint) error {
+	return r.data.gormDB.WithContext(ctx).Where("id IN ?", ids).Delete(&announcementPO{}).Error
+}
+
+func (r *announcementRepo) Update(ctx context.Context, item *biz.Announcement) error {
+	po := newAnnouncement(item)
+	result := r.data.gormDB.WithContext(ctx).Model(&announcementPO{}).Where("id = ?", item.ID).Updates(map[string]any{
+		"title": po.Title, "content": po.Content, "user_id": po.UserID, "attachments": po.Attachments,
+	})
+	if result.Error != nil {
+		return result.Error
+	}
+	if result.RowsAffected == 0 {
+		return gorm.ErrRecordNotFound
+	}
+	return nil
+}
+
+func (r *announcementRepo) Find(ctx context.Context, id uint) (*biz.Announcement, error) {
+	var po announcementPO
+	if err := r.data.gormDB.WithContext(ctx).First(&po, id).Error; err != nil {
+		return nil, err
+	}
+	return announcementToBiz(po), nil
+}
+
+func (r *announcementRepo) List(ctx context.Context, filter biz.AnnouncementFilter) ([]*biz.Announcement, int64, error) {
+	db := r.data.gormDB.WithContext(ctx).Model(&announcementPO{})
+	if filter.StartCreatedAt != nil && filter.EndCreatedAt != nil {
+		db = db.Where("created_at BETWEEN ? AND ?", filter.StartCreatedAt, filter.EndCreatedAt)
+	}
+	var total int64
+	if err := db.Count(&total).Error; err != nil {
+		return nil, 0, err
+	}
+	if filter.PageSize > 0 {
+		page := filter.Page
+		if page < 1 {
+			page = 1
+		}
+		db = db.Offset((page - 1) * filter.PageSize).Limit(filter.PageSize)
+	}
+	var pos []announcementPO
+	if err := db.Find(&pos).Error; err != nil {
+		return nil, 0, err
+	}
+	items := make([]*biz.Announcement, 0, len(pos))
+	for _, po := range pos {
+		items = append(items, announcementToBiz(po))
+	}
+	return items, total, nil
+}
+
+func (r *announcementRepo) UserOptions(ctx context.Context) ([]biz.UserOption, error) {
+	var rows []struct {
+		Label string
+		Value uint
+	}
+	err := r.data.gormDB.WithContext(ctx).Table("sys_users").Select("nick_name AS label, id AS value").Where("deleted_at IS NULL").Scan(&rows).Error
+	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, err
+	}
+	items := make([]biz.UserOption, 0, len(rows))
+	for _, row := range rows {
+		items = append(items, biz.UserOption{Label: row.Label, Value: row.Value})
+	}
+	return items, nil
+}
@@ -18,7 +18,7 @@ import (
 )

 // ProviderSet is data providers.
-var ProviderSet = wire.NewSet(NewData, NewAdminRepo, NewSystemRepo, NewAccessRepo, NewSettingsRepo, NewVersionRepo, NewExportRepo, NewAuditRepo, NewTaskRepo, NewMediaRepo, NewCache, NewFileStorage)
+var ProviderSet = wire.NewSet(NewData, NewAdminRepo, NewSystemRepo, NewAccessRepo, NewSettingsRepo, NewVersionRepo, NewExportRepo, NewAuditRepo, NewTaskRepo, NewMediaRepo, NewAnnouncementRepo, NewEmailRepo, NewCache, NewFileStorage)

 // Data is a struct that contains the database client.
 type Data struct {
@@ -41,6 +41,12 @@ func NewData(c *conf.Data) (*Data, func(), error) {
 		_ = db.Close()
 		return nil, nil, err
 	}
+	// The announcement module owns its table lifecycle independently of the
+	// first-run database initializer.
+	if err = gormDB.AutoMigrate(&announcementPO{}); err != nil {
+		_ = db.Close()
+		return nil, nil, fmt.Errorf("migrate announcement table: %w", err)
+	}
 	var redisClient *redis.Client
 	if c.Redis != nil && c.Redis.Addr != "" {
 		options := &redis.Options{Addr: c.Redis.Addr}
@@ -0,0 +1,149 @@
+package data
+
+import (
+	"context"
+	"crypto/tls"
+	"errors"
+	"fmt"
+	"io"
+	"mime"
+	"net"
+	"net/smtp"
+	"strings"
+	"time"
+
+	"kra/internal/biz"
+	"kra/internal/conf"
+)
+
+type emailRepo struct {
+	config *conf.AdminBackend_Email
+}
+
+func NewEmailRepo(config *conf.AdminBackend) biz.EmailRepo {
+	if config == nil {
+		return &emailRepo{}
+	}
+	return &emailRepo{config: config.Email}
+}
+
+func (r *emailRepo) Enabled() bool {
+	return r.config != nil && r.config.Host != "" && r.config.From != "" && r.config.Secret != "" && r.config.Port > 0
+}
+
+func (r *emailRepo) DefaultRecipients() []string {
+	if r.config == nil {
+		return nil
+	}
+	parts := strings.Split(r.config.To, ",")
+	result := make([]string, 0, len(parts))
+	for _, part := range parts {
+		if recipient := strings.TrimSpace(part); recipient != "" {
+			result = append(result, recipient)
+		}
+	}
+	return result
+}
+
+func cleanHeader(value string) string {
+	return strings.NewReplacer("\r", "", "\n", "").Replace(value)
+}
+
+func (r *emailRepo) Send(ctx context.Context, to []string, subject, body string) error {
+	if !r.Enabled() {
+		return errors.New("邮件服务未配置")
+	}
+	if len(to) == 0 {
+		return errors.New("收件人不能为空")
+	}
+	config := r.config
+	address := net.JoinHostPort(config.Host, fmt.Sprint(config.Port))
+	dialer := &net.Dialer{Timeout: 10 * time.Second}
+	var conn net.Conn
+	var err error
+	if config.IsSsl {
+		conn, err = tls.DialWithDialer(dialer, "tcp", address, &tls.Config{ServerName: config.Host, MinVersion: tls.VersionTLS12})
+	} else {
+		conn, err = dialer.DialContext(ctx, "tcp", address)
+	}
+	if err != nil {
+		return err
+	}
+	defer conn.Close()
+	_ = conn.SetDeadline(time.Now().Add(15 * time.Second))
+	client, err := smtp.NewClient(conn, config.Host)
+	if err != nil {
+		return err
+	}
+	defer client.Close()
+	if !config.IsSsl {
+		if supported, _ := client.Extension("STARTTLS"); supported {
+			if err = client.StartTLS(&tls.Config{ServerName: config.Host, MinVersion: tls.VersionTLS12}); err != nil {
+				return err
+			}
+		}
+	}
+	var auth smtp.Auth
+	if config.IsLoginAuth {
+		auth = &loginAuth{username: config.From, password: config.Secret}
+	} else {
+		auth = smtp.PlainAuth("", config.From, config.Secret, config.Host)
+	}
+	if ok, _ := client.Extension("AUTH"); ok {
+		if err = client.Auth(auth); err != nil {
+			return err
+		}
+	}
+	if err = client.Mail(config.From); err != nil {
+		return err
+	}
+	for _, recipient := range to {
+		if err = client.Rcpt(recipient); err != nil {
+			return err
+		}
+	}
+	writer, err := client.Data()
+	if err != nil {
+		return err
+	}
+	fromHeader := cleanHeader(config.From)
+	if config.Nickname != "" {
+		fromHeader = fmt.Sprintf("%s <%s>", mime.QEncoding.Encode("UTF-8", cleanHeader(config.Nickname)), fromHeader)
+	}
+	toHeader := make([]string, 0, len(to))
+	for _, recipient := range to {
+		toHeader = append(toHeader, cleanHeader(recipient))
+	}
+	message := "From: " + fromHeader + "\r\n" +
+		"To: " + strings.Join(toHeader, ",") + "\r\n" +
+		"Subject: " + mime.QEncoding.Encode("UTF-8", cleanHeader(subject)) + "\r\n" +
+		"MIME-Version: 1.0\r\nContent-Type: text/html; charset=UTF-8\r\n\r\n" + body
+	if _, err = io.WriteString(writer, message); err != nil {
+		_ = writer.Close()
+		return err
+	}
+	if err = writer.Close(); err != nil {
+		return err
+	}
+	return client.Quit()
+}
+
+type loginAuth struct{ username, password string }
+
+func (a *loginAuth) Start(*smtp.ServerInfo) (string, []byte, error) {
+	return "LOGIN", nil, nil
+}
+
+func (a *loginAuth) Next(challenge []byte, more bool) ([]byte, error) {
+	if !more {
+		return nil, nil
+	}
+	prompt := strings.ToLower(strings.TrimSpace(string(challenge)))
+	if strings.Contains(prompt, "username") || strings.Contains(prompt, "user") {
+		return []byte(a.username), nil
+	}
+	if strings.Contains(prompt, "password") || strings.Contains(prompt, "pass") {
+		return []byte(a.password), nil
+	}
+	return nil, fmt.Errorf("unsupported SMTP LOGIN challenge %q", prompt)
+}
@@ -95,7 +95,7 @@ func (r *systemRepo) IsInitialized(ctx context.Context) (bool, error) {

 func (r *systemRepo) Initialize(ctx context.Context) error {
 	db := r.data.gormDB.WithContext(ctx)
-	if err := db.AutoMigrate(&userPO{}, &authorityPO{}, &menuPO{}, &userAuthorityPO{}, &authorityMenuPO{}, &apiPO{}, &ignoredAPIPO{}, &authorityAPIPO{}, &menuButtonPO{}, &authorityButtonPO{}, &departmentPO{}, &positionPO{}, &userDepartmentPO{}, &userPositionPO{}, &authorityDepartmentPO{}, &dictionaryPO{}, &dictionaryDetailPO{}, &parameterPO{}, &apiTokenPO{}, &securityConfigPO{}, &versionPO{}, &exportTemplatePO{}, &exportConditionPO{}, &exportJoinPO{}, &operationPO{}, &loginLogPO{}, &dataAccessLogPO{}, &errorRecordPO{}, &taskPO{}, &taskLogPO{}, &mediaPO{}, &categoryPO{}, &uploadSessionPO{}, &uploadChunkPO{}); err != nil {
+	if err := db.AutoMigrate(&userPO{}, &authorityPO{}, &menuPO{}, &userAuthorityPO{}, &authorityMenuPO{}, &apiPO{}, &ignoredAPIPO{}, &authorityAPIPO{}, &menuButtonPO{}, &authorityButtonPO{}, &departmentPO{}, &positionPO{}, &userDepartmentPO{}, &userPositionPO{}, &authorityDepartmentPO{}, &dictionaryPO{}, &dictionaryDetailPO{}, &parameterPO{}, &apiTokenPO{}, &securityConfigPO{}, &versionPO{}, &exportTemplatePO{}, &exportConditionPO{}, &exportJoinPO{}, &operationPO{}, &loginLogPO{}, &dataAccessLogPO{}, &errorRecordPO{}, &taskPO{}, &taskLogPO{}, &mediaPO{}, &categoryPO{}, &uploadSessionPO{}, &uploadChunkPO{}, &announcementPO{}); err != nil {
 		return err
 	}
 	return db.Transaction(func(tx *gorm.DB) error {
@@ -419,12 +419,13 @@ func defaultMenus() []menuPO {
 	}
 	return []menuPO{
 		{Path: "dashboard", Name: "dashboard", Component: "view/dashboard/index.vue", Title: "仪表盘", Icon: "odometer", Sort: 1},
-		root("permission", "permission", "权限管理", "perm-gva", 2), root("org", "org", "组织管理", "share", 3), root("systemConfig", "systemConfig", "系统设置", "config-gva", 4), root("monitor", "monitor", "运维监控", "monitor-gva", 5), root("media", "media", "媒体管理", "folder-opened", 6),
+		root("permission", "permission", "权限管理", "perm-gva", 2), root("org", "org", "组织管理", "share", 3), root("systemConfig", "systemConfig", "系统设置", "config-gva", 4), root("monitor", "monitor", "运维监控", "monitor-gva", 5), root("media", "media", "媒体管理", "folder-opened", 6), root("plugin", "plugin", "插件系统", "cherry", 10),
 		{Path: "person", Name: "person", Component: "view/person/person.vue", Title: "个人信息", Icon: "postcard", Hidden: true, Sort: 13},
 		child("permission", "authority", "authority", "view/superAdmin/authority/authority.vue", "角色管理", "role-gva", 1), child("permission", "menu", "menu", "view/superAdmin/menu/menu.vue", "菜单管理", "tickets", 2), child("permission", "api", "api", "view/superAdmin/api/api.vue", "api管理", "api-gva", 3), child("permission", "apiToken", "apiToken", "view/systemTools/apiToken/index.vue", "API Token", "key", 4),
 		child("org", "user", "user", "view/superAdmin/user/user.vue", "用户管理", "user", 1), child("org", "department", "department", "view/superAdmin/department/department.vue", "部门管理", "office-building", 2), child("org", "position", "position", "view/superAdmin/position/position.vue", "岗位管理", "postcard", 3),
 		child("systemConfig", "system", "system", "view/systemTools/system/system.vue", "配置文件", "config-file-gva", 1), child("systemConfig", "dictionary", "dictionary", "view/superAdmin/dictionary/sysDictionary.vue", "字典管理", "notebook", 2), child("systemConfig", "sysParams", "sysParams", "view/superAdmin/params/sysParams.vue", "参数管理", "set-up", 3), child("systemConfig", "security", "security", "view/system/security/index.vue", "安全配置", "security-gva", 4),
 		child("monitor", "operation", "operation", "view/superAdmin/operation/sysOperationRecord.vue", "操作历史", "document", 1), child("monitor", "loginLog", "loginLog", "view/systemTools/loginLog/index.vue", "登录日志", "clock", 2), child("monitor", "sysError", "sysError", "view/systemTools/sysError/sysError.vue", "错误日志", "error-gva", 3), child("monitor", "sysVersion", "sysVersion", "view/systemTools/version/version.vue", "版本管理", "version-gva", 4), child("monitor", "state", "state", "view/system/state.vue", "服务器状态", "server", 5), child("monitor", "dataAccessLog", "dataAccessLog", "view/superAdmin/dataAccessLog/dataAccessLog.vue", "数据权限审计", "warning", 6), child("monitor", "timedTask", "timedTask", "view/systemTools/timedTask/index.vue", "定时任务", "timer", 7), child("monitor", "logViewer", "logViewer", "view/systemTools/logViewer/index.vue", "文件日志", "document", 8),
 		child("media", "upload", "upload", "view/media/upload.vue", "媒体库（上传下载）", "upload", 1), child("media", "chunkUpload", "chunkUpload", "view/media/chunkUpload.vue", "大文件上传", "folder-add", 2),
+		child("plugin", "plugin-email", "plugin-email", "plugin/email/view/index.vue", "邮件插件", "message", 4), child("plugin", "anInfo", "anInfo", "plugin/announcement/view/info.vue", "公告管理[示例]", "bell", 5),
 	}
 }