836 lines
27 KiB
Go
836 lines
27 KiB
Go
package app
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"log"
|
||
"mime/multipart"
|
||
"net/http"
|
||
"strings"
|
||
"time"
|
||
"unicode/utf8"
|
||
|
||
"yingce/backend/internal/assets"
|
||
"yingce/backend/internal/model"
|
||
"yingce/backend/internal/repository"
|
||
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
const AnnouncementImageMaxBytes int64 = 10 << 20
|
||
const announcementImageDraftTTL = 24 * time.Hour
|
||
|
||
type CreateAnnouncementRequest struct {
|
||
Title string `json:"title"`
|
||
Content string `json:"content"`
|
||
ImageResourceID string `json:"imageResourceId"`
|
||
Level model.AnnouncementLevel `json:"level"`
|
||
Pinned bool `json:"pinned"`
|
||
}
|
||
|
||
type UpdateAnnouncementRequest = CreateAnnouncementRequest
|
||
|
||
type AnnouncementPage struct {
|
||
Announcements []model.Announcement `json:"announcements"`
|
||
Total int64 `json:"total"`
|
||
Page int `json:"page"`
|
||
Limit int `json:"pageSize"`
|
||
}
|
||
|
||
type UserAnnouncementFeed struct {
|
||
Announcements []model.Announcement `json:"announcements"`
|
||
UnreadCount int64 `json:"unreadCount"`
|
||
}
|
||
|
||
func (s *Service) AdminAnnouncementPage(actor *model.User, query AdminListQuery) (*AnnouncementPage, error) {
|
||
if err := s.RequireAdmin(actor); err != nil {
|
||
return nil, err
|
||
}
|
||
page, limit := normalizeAdminPage(query.Page, query.Limit)
|
||
announcements, total, err := s.repo.AdminAnnouncements(query.Keyword, model.AnnouncementStatus(query.Status), limit, (page-1)*limit)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
for index := range announcements {
|
||
decorateAnnouncement(&announcements[index])
|
||
}
|
||
return &AnnouncementPage{Announcements: announcements, Total: total, Page: page, Limit: limit}, nil
|
||
}
|
||
|
||
func (s *Service) CreateAnnouncement(actor *model.User, req CreateAnnouncementRequest) (*model.Announcement, error) {
|
||
if err := s.RequireAdmin(actor); err != nil {
|
||
return nil, err
|
||
}
|
||
title, content, level, err := normalizeAnnouncementInput(req)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if strings.TrimSpace(req.ImageResourceID) != "" {
|
||
s.storageMu.Lock()
|
||
defer s.storageMu.Unlock()
|
||
}
|
||
imageResourceID, err := s.validateAnnouncementImageDraft(actor, req.ImageResourceID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
now := time.Now()
|
||
announcement := &model.Announcement{
|
||
ID: newID(), Title: title, Content: content, ImageResourceID: imageResourceID, Level: level, Pinned: req.Pinned,
|
||
Status: model.AnnouncementStatusActive, CreatedBy: actor.ID, PublishedAt: now, CreatedAt: now, UpdatedAt: now,
|
||
}
|
||
if err := s.repo.CreateAnnouncementWithImage(announcement); err != nil {
|
||
if errors.Is(err, repository.ErrAnnouncementImageDraftUnavailable) {
|
||
return nil, BadAuthRequest("公告配图草稿已失效,请重新上传")
|
||
}
|
||
return nil, err
|
||
}
|
||
decorateAnnouncement(announcement)
|
||
return announcement, nil
|
||
}
|
||
|
||
func (s *Service) UpdateAnnouncement(actor *model.User, id string, req UpdateAnnouncementRequest) (*model.Announcement, error) {
|
||
if err := s.RequireAdmin(actor); err != nil {
|
||
return nil, err
|
||
}
|
||
s.storageMu.Lock()
|
||
defer s.storageMu.Unlock()
|
||
announcement, err := s.repo.Announcement(strings.TrimSpace(id))
|
||
if err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, BadAuthRequest("公告不存在")
|
||
}
|
||
return nil, err
|
||
}
|
||
title, content, level, err := normalizeAnnouncementInput(req)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
imageResourceID := strings.TrimSpace(req.ImageResourceID)
|
||
newDraftResourceID := ""
|
||
if imageResourceID != announcement.ImageResourceID {
|
||
if imageResourceID != "" {
|
||
imageResourceID, err = s.validateAnnouncementImageDraft(actor, imageResourceID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
newDraftResourceID = imageResourceID
|
||
}
|
||
}
|
||
var oldResource *model.Resource
|
||
var deletionJob *model.ResourceDeletionJob
|
||
if announcement.ImageResourceID != "" && announcement.ImageResourceID != imageResourceID {
|
||
oldResource, err = s.repo.Resource(announcement.ImageResourceID)
|
||
if err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, BadAuthRequest("原公告配图资源不存在,已停止更新以避免数据不一致")
|
||
}
|
||
return nil, err
|
||
}
|
||
if err := s.ensureResourceHasNoBusinessReferences(oldResource, repository.ResourceDirectReference{Kind: "公告", ID: announcement.ID}); err != nil {
|
||
return nil, err
|
||
}
|
||
sharedCount, countErr := s.repo.ResourceStorageReferenceCount(oldResource, []string{oldResource.ID})
|
||
if countErr != nil {
|
||
return nil, countErr
|
||
}
|
||
if sharedCount == 0 {
|
||
jobs := resourceDeletionJobs(oldResource.UserID, map[string]*model.Resource{resourceStorageIdentity(oldResource): oldResource})
|
||
if len(jobs) != 1 {
|
||
return nil, errors.New("无法创建公告配图删除任务")
|
||
}
|
||
deletionJob = &jobs[0]
|
||
}
|
||
}
|
||
now := time.Now()
|
||
announcement.Title = title
|
||
announcement.Content = content
|
||
announcement.ImageResourceID = imageResourceID
|
||
announcement.Level = level
|
||
announcement.Pinned = req.Pinned
|
||
announcement.Status = model.AnnouncementStatusActive
|
||
announcement.ClosedAt = nil
|
||
announcement.PublishedAt = now
|
||
announcement.UpdatedAt = now
|
||
if err := s.repo.UpdateAnnouncementWithImage(announcement, actor.ID, newDraftResourceID, oldResource, deletionJob); err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, BadAuthRequest("公告状态已变化,请刷新后重试")
|
||
}
|
||
if errors.Is(err, repository.ErrAnnouncementImageDraftUnavailable) {
|
||
return nil, BadAuthRequest("公告配图草稿已失效,请重新上传")
|
||
}
|
||
if errors.Is(err, repository.ErrAnnouncementImageReferenced) {
|
||
return nil, BadAuthRequest("原公告配图仍被其他公告引用,已停止更新")
|
||
}
|
||
return nil, err
|
||
}
|
||
decorateAnnouncement(announcement)
|
||
return announcement, nil
|
||
}
|
||
|
||
func (s *Service) CloseAnnouncement(actor *model.User, id string) (*model.Announcement, error) {
|
||
if err := s.RequireAdmin(actor); err != nil {
|
||
return nil, err
|
||
}
|
||
announcement, err := s.repo.Announcement(strings.TrimSpace(id))
|
||
if err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, BadAuthRequest("公告不存在")
|
||
}
|
||
return nil, err
|
||
}
|
||
if announcement.Status == model.AnnouncementStatusClosed {
|
||
return nil, BadAuthRequest("公告已经关闭")
|
||
}
|
||
updated, err := s.repo.CloseAnnouncement(announcement.ID, time.Now())
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if !updated {
|
||
return nil, BadAuthRequest("公告状态已变化,请刷新后重试")
|
||
}
|
||
closed, err := s.repo.Announcement(announcement.ID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
decorateAnnouncement(closed)
|
||
return closed, nil
|
||
}
|
||
|
||
func (s *Service) UserAnnouncements(user *model.User) (*UserAnnouncementFeed, error) {
|
||
if user == nil {
|
||
return nil, Unauthorized("请先登录")
|
||
}
|
||
announcements, unreadCount, err := s.repo.AnnouncementFeed(user.ID)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
for index := range announcements {
|
||
decorateAnnouncement(&announcements[index])
|
||
}
|
||
return &UserAnnouncementFeed{Announcements: announcements, UnreadCount: unreadCount}, nil
|
||
}
|
||
|
||
func (s *Service) UploadAnnouncementImage(actor *model.User, header *multipart.FileHeader) (*model.Resource, error) {
|
||
if err := s.RequireAdmin(actor); err != nil {
|
||
return nil, err
|
||
}
|
||
if err := validateAnnouncementImageUpload(header); err != nil {
|
||
return nil, err
|
||
}
|
||
resource, err := s.UploadResource(actor.ID, header, "image", 0, 0, 0)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
draft := &model.AnnouncementImageDraft{ResourceID: resource.ID, UserID: actor.ID, CreatedAt: time.Now()}
|
||
if err := s.repo.CreateAnnouncementImageDraft(draft); err != nil {
|
||
cleanupErr := s.deleteFreshAnnouncementImageResource(resource)
|
||
if cleanupErr != nil {
|
||
return nil, errors.Join(err, fmt.Errorf("清理未登记的公告配图失败:%w", cleanupErr))
|
||
}
|
||
return nil, err
|
||
}
|
||
resource.PublicURL = ""
|
||
return resource, nil
|
||
}
|
||
|
||
func (s *Service) DiscardAnnouncementImage(actor *model.User, resourceID string) error {
|
||
if err := s.RequireAdmin(actor); err != nil {
|
||
return err
|
||
}
|
||
return s.discardAnnouncementImageDraft(actor.ID, resourceID)
|
||
}
|
||
|
||
func (s *Service) OpenAnnouncementImage(actor *model.User, announcementID string, rangeHeader string) (*ResourceStream, error) {
|
||
if actor == nil {
|
||
return nil, Unauthorized("请先登录")
|
||
}
|
||
announcement, err := s.repo.Announcement(strings.TrimSpace(announcementID))
|
||
if err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, NotFound("公告不存在")
|
||
}
|
||
return nil, err
|
||
}
|
||
if actor.Role != model.UserRoleAdmin && announcement.Status != model.AnnouncementStatusActive {
|
||
return nil, Forbidden("公告不可访问")
|
||
}
|
||
if announcement.ImageResourceID == "" {
|
||
return nil, NotFound("公告配图不存在")
|
||
}
|
||
resource, err := s.repo.Resource(announcement.ImageResourceID)
|
||
if err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, NotFound("公告配图不存在")
|
||
}
|
||
return nil, err
|
||
}
|
||
if resource.Kind != "image" || resource.Status != model.ResourceStatusReady {
|
||
return nil, BadAuthRequest("公告配图资源不可用")
|
||
}
|
||
return s.openResourceRange(resource.UserID, resource, rangeHeader)
|
||
}
|
||
|
||
func (s *Service) PrepareAnnouncementImageDelivery(actor *model.User, announcementID string, options ResourceAccessOptions, rangeHeader string) (*ResourceDelivery, error) {
|
||
if actor == nil {
|
||
return nil, Unauthorized("请先登录")
|
||
}
|
||
announcement, err := s.repo.Announcement(strings.TrimSpace(announcementID))
|
||
if err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, NotFound("公告不存在")
|
||
}
|
||
return nil, err
|
||
}
|
||
if actor.Role != model.UserRoleAdmin && announcement.Status != model.AnnouncementStatusActive {
|
||
return nil, Forbidden("公告不可访问")
|
||
}
|
||
if announcement.ImageResourceID == "" {
|
||
return nil, NotFound("公告配图不存在")
|
||
}
|
||
resource, err := s.repo.Resource(announcement.ImageResourceID)
|
||
if err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, NotFound("公告配图不存在")
|
||
}
|
||
return nil, err
|
||
}
|
||
if resource.Kind != "image" || resource.Status != model.ResourceStatusReady {
|
||
return nil, BadAuthRequest("公告配图资源不可用")
|
||
}
|
||
if options.Purpose == "" {
|
||
options.Purpose = assets.PurposeDisplay
|
||
}
|
||
return s.prepareResourceDelivery(resource.UserID, resource, options, rangeHeader)
|
||
}
|
||
|
||
func (s *Service) cleanupStaleAnnouncementImageDrafts() {
|
||
for {
|
||
drafts, err := s.repo.StaleAnnouncementImageDrafts(time.Now().Add(-announcementImageDraftTTL), 50)
|
||
if err != nil {
|
||
log.Printf("announcement image draft cleanup query failed: %v", err)
|
||
return
|
||
}
|
||
if len(drafts) == 0 {
|
||
return
|
||
}
|
||
cleaned := 0
|
||
for _, draft := range drafts {
|
||
if err := s.discardAnnouncementImageDraft(draft.UserID, draft.ResourceID); err != nil {
|
||
log.Printf("announcement image draft cleanup failed for %s: %v", draft.ResourceID, err)
|
||
continue
|
||
}
|
||
cleaned++
|
||
}
|
||
if len(drafts) < 50 || cleaned == 0 {
|
||
return
|
||
}
|
||
}
|
||
}
|
||
|
||
func (s *Service) discardAnnouncementImageDraft(userID string, resourceID string) error {
|
||
resourceID = strings.TrimSpace(resourceID)
|
||
if resourceID == "" || len(resourceID) > 64 {
|
||
return BadAuthRequest("公告配图资源 ID 无效")
|
||
}
|
||
s.storageMu.Lock()
|
||
defer s.storageMu.Unlock()
|
||
if _, err := s.repo.AnnouncementImageDraftForUser(userID, resourceID); err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return NotFound("公告配图草稿不存在")
|
||
}
|
||
return err
|
||
}
|
||
resource, err := s.repo.ResourceForUser(userID, resourceID)
|
||
if err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
if deleteErr := s.repo.DeleteAnnouncementImageDraft(userID, resourceID); deleteErr != nil {
|
||
return errors.Join(err, fmt.Errorf("清理失效公告配图草稿失败:%w", deleteErr))
|
||
}
|
||
return nil
|
||
}
|
||
return err
|
||
}
|
||
if err := s.ensureResourceHasNoBusinessReferences(resource, repository.ResourceDirectReference{Kind: "公告草稿", ID: resource.ID}); err != nil {
|
||
return err
|
||
}
|
||
sharedCount, err := s.repo.ResourceStorageReferenceCount(resource, []string{resource.ID})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
var deletionJob *model.ResourceDeletionJob
|
||
if sharedCount == 0 {
|
||
jobs := resourceDeletionJobs(userID, map[string]*model.Resource{resourceStorageIdentity(resource): resource})
|
||
if len(jobs) != 1 {
|
||
return errors.New("无法创建公告配图删除任务")
|
||
}
|
||
deletionJob = &jobs[0]
|
||
}
|
||
if err := s.repo.DiscardAnnouncementImageDraft(userID, resource, deletionJob); err != nil {
|
||
if errors.Is(err, repository.ErrAnnouncementImageReferenced) {
|
||
return BadAuthRequest("公告配图已经发布,不能按草稿删除")
|
||
}
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (s *Service) ensureResourceHasNoBusinessReferences(resource *model.Resource, ignoredDirect ...repository.ResourceDirectReference) error {
|
||
if resource == nil {
|
||
return NotFound("资源不存在")
|
||
}
|
||
snapshot, err := s.repo.ResourceReferenceSnapshot(resource.UserID, "", []string{resource.ID})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
hasBlocking := false
|
||
DirectLoop:
|
||
for _, direct := range snapshot.Direct {
|
||
for _, ignored := range ignoredDirect {
|
||
if direct.Kind == ignored.Kind && direct.ID == ignored.ID {
|
||
continue DirectLoop
|
||
}
|
||
}
|
||
hasBlocking = true
|
||
break
|
||
}
|
||
if hasBlocking {
|
||
return BadAuthRequest("公告配图仍被其他业务数据引用,已停止删除")
|
||
}
|
||
resourceIDs := map[string]struct{}{resource.ID: {}}
|
||
for _, document := range snapshot.Documents {
|
||
if documentReferencesResources(document.PrimaryJSON, resourceIDs) || documentReferencesResources(document.SecondaryJSON, resourceIDs) {
|
||
return BadAuthRequest("公告配图仍被其他业务数据引用,已停止删除")
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (s *Service) validateAnnouncementImageDraft(actor *model.User, resourceID string) (string, error) {
|
||
resourceID = strings.TrimSpace(resourceID)
|
||
if resourceID == "" {
|
||
return "", nil
|
||
}
|
||
if len(resourceID) > 64 {
|
||
return "", BadAuthRequest("公告配图资源 ID 无效")
|
||
}
|
||
if _, err := s.repo.AnnouncementImageDraftForUser(actor.ID, resourceID); err != nil {
|
||
return "", BadAuthRequest("公告配图草稿不存在或不属于当前管理员")
|
||
}
|
||
resource, err := s.repo.ResourceForUser(actor.ID, resourceID)
|
||
if err != nil {
|
||
return "", BadAuthRequest("公告配图资源不存在或不属于当前管理员")
|
||
}
|
||
if resource.Kind != "image" || resource.Status != model.ResourceStatusReady || !strings.HasPrefix(strings.ToLower(resource.MimeType), "image/") {
|
||
return "", BadAuthRequest("公告配图必须是上传完成的图片")
|
||
}
|
||
return resourceID, nil
|
||
}
|
||
|
||
func validateAnnouncementImageUpload(header *multipart.FileHeader) error {
|
||
if header == nil {
|
||
return BadAuthRequest("请选择公告配图")
|
||
}
|
||
if header.Size <= 0 || header.Size > AnnouncementImageMaxBytes {
|
||
return BadAuthRequest("公告配图大小必须在 10MB 以内")
|
||
}
|
||
file, err := header.Open()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer file.Close()
|
||
buffer := make([]byte, 512)
|
||
read, err := file.Read(buffer)
|
||
if err != nil && read == 0 {
|
||
return BadAuthRequest("公告配图内容无法读取")
|
||
}
|
||
if detected := http.DetectContentType(buffer[:read]); !strings.HasPrefix(strings.ToLower(detected), "image/") {
|
||
return BadAuthRequest("公告配图必须是真实图片文件")
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (s *Service) deleteFreshAnnouncementImageResource(resource *model.Resource) error {
|
||
if resource == nil {
|
||
return nil
|
||
}
|
||
if err := s.deleteStoredResourceObject(resource.UserID, resource); err != nil {
|
||
return err
|
||
}
|
||
return s.repo.DeleteResource(resource.UserID, resource.ID)
|
||
}
|
||
|
||
func decorateAnnouncement(announcement *model.Announcement) {
|
||
if announcement == nil || strings.TrimSpace(announcement.ImageResourceID) == "" {
|
||
return
|
||
}
|
||
announcement.ImageURL = "/api/announcements/" + announcement.ID + "/image"
|
||
}
|
||
|
||
func (s *Service) MarkAnnouncementsRead(user *model.User, announcementIDs []string) (int64, error) {
|
||
if user == nil {
|
||
return 0, Unauthorized("请先登录")
|
||
}
|
||
if len(announcementIDs) > 5000 {
|
||
return 0, BadAuthRequest("单次已读公告数量过多")
|
||
}
|
||
ids := uniqueNonEmpty(announcementIDs)
|
||
for _, id := range ids {
|
||
if len(id) > 64 {
|
||
return 0, BadAuthRequest("公告 ID 无效")
|
||
}
|
||
}
|
||
if err := s.repo.MarkAnnouncementsRead(user.ID, ids, time.Now()); err != nil {
|
||
return 0, err
|
||
}
|
||
_, unreadCount, err := s.repo.AnnouncementFeed(user.ID)
|
||
return unreadCount, err
|
||
}
|
||
|
||
func validAnnouncementLevel(level model.AnnouncementLevel) bool {
|
||
return level == model.AnnouncementLevelInfo || level == model.AnnouncementLevelSuccess || level == model.AnnouncementLevelWarning || level == model.AnnouncementLevelCritical
|
||
}
|
||
|
||
func normalizeAnnouncementInput(req CreateAnnouncementRequest) (string, string, model.AnnouncementLevel, error) {
|
||
title := strings.TrimSpace(req.Title)
|
||
content := strings.TrimSpace(req.Content)
|
||
if title == "" {
|
||
return "", "", "", BadAuthRequest("请填写公告标题")
|
||
}
|
||
if utf8.RuneCountInString(title) > 120 {
|
||
return "", "", "", BadAuthRequest("公告标题不能超过 120 个字符")
|
||
}
|
||
if utf8.RuneCountInString(content) > 4000 {
|
||
return "", "", "", BadAuthRequest("公告正文不能超过 4000 个字符")
|
||
}
|
||
if !validAnnouncementLevel(req.Level) {
|
||
return "", "", "", BadAuthRequest("公告级别无效")
|
||
}
|
||
if len(strings.TrimSpace(req.ImageResourceID)) > 64 {
|
||
return "", "", "", BadAuthRequest("公告配图资源 ID 无效")
|
||
}
|
||
return title, content, req.Level, nil
|
||
}
|
||
|
||
type CreateBannerAnnouncementRequest struct {
|
||
Title string `json:"title"`
|
||
// TitleRuns 是标题的样式分段;非空时 Title 由分段文本拼接得出,请求里的 Title 只作兼容兜底。
|
||
TitleRuns []model.BannerTitleRun `json:"titleRuns"`
|
||
// NoticeType 通知类型,决定通知条底色;空值按默认类型(公告)处理。
|
||
// 图标素材(emoji)内嵌在 TitleRuns 的文本里,不单独存字段。
|
||
NoticeType string `json:"noticeType"`
|
||
Link string `json:"link"`
|
||
Status string `json:"status"` // "active" | "disabled"
|
||
StartsAt *time.Time `json:"startsAt"`
|
||
EndsAt *time.Time `json:"endsAt"`
|
||
}
|
||
|
||
type UpdateBannerAnnouncementRequest = CreateBannerAnnouncementRequest
|
||
|
||
type BannerAnnouncementPage struct {
|
||
Banners []model.BannerAnnouncement `json:"banners"`
|
||
Total int64 `json:"total"`
|
||
Page int `json:"page"`
|
||
Limit int `json:"pageSize"`
|
||
}
|
||
|
||
func (s *Service) ActiveBannerAnnouncements() ([]model.BannerAnnouncement, error) {
|
||
banners, err := s.repo.ActiveBannerAnnouncements(time.Now())
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if banners == nil {
|
||
banners = []model.BannerAnnouncement{}
|
||
}
|
||
return banners, nil
|
||
}
|
||
|
||
func (s *Service) AdminBannerAnnouncementPage(actor *model.User, query AdminListQuery) (*BannerAnnouncementPage, error) {
|
||
if err := s.RequireAdmin(actor); err != nil {
|
||
return nil, err
|
||
}
|
||
page, limit := normalizeAdminPage(query.Page, query.Limit)
|
||
banners, total, err := s.repo.AdminBannerAnnouncements(query.Keyword, query.Status, limit, (page-1)*limit)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
if banners == nil {
|
||
banners = []model.BannerAnnouncement{}
|
||
}
|
||
return &BannerAnnouncementPage{Banners: banners, Total: total, Page: page, Limit: limit}, nil
|
||
}
|
||
|
||
// 标题样式分段的边界:字号与字重只接受固定档位,字体与颜色进白名单,避免把任意 CSS 写进通知条。
|
||
const (
|
||
bannerTitleMinFontSize = 10
|
||
bannerTitleMaxFontSize = 20
|
||
bannerTitleMaxRuns = 60
|
||
bannerTitleMaxTextRunes = 120
|
||
)
|
||
|
||
var bannerTitleFontWeights = map[int]struct{}{400: {}, 500: {}, 600: {}, 700: {}}
|
||
var bannerTitleFontFamilies = map[string]struct{}{"sans": {}, "serif": {}, "mono": {}}
|
||
|
||
// normalizeBannerTitle 统一处理标题与样式分段:分段非空时以分段文本为准,分段为空时回落到纯文本标题。
|
||
func normalizeBannerTitle(title string, runs []model.BannerTitleRun) (string, []model.BannerTitleRun, error) {
|
||
normalized, err := normalizeBannerTitleRuns(runs)
|
||
if err != nil {
|
||
return "", nil, err
|
||
}
|
||
if len(normalized) == 0 {
|
||
plain := strings.TrimSpace(title)
|
||
if plain == "" {
|
||
return "", nil, BadAuthRequest("请填写通知标题")
|
||
}
|
||
if utf8.RuneCountInString(plain) > bannerTitleMaxTextRunes {
|
||
return "", nil, BadAuthRequest("通知标题不能超过 120 个字符")
|
||
}
|
||
return plain, nil, nil
|
||
}
|
||
plain := strings.TrimSpace(bannerTitleRunsText(normalized))
|
||
if plain == "" {
|
||
return "", nil, BadAuthRequest("请填写通知标题")
|
||
}
|
||
if utf8.RuneCountInString(plain) > bannerTitleMaxTextRunes {
|
||
return "", nil, BadAuthRequest("通知标题不能超过 120 个字符")
|
||
}
|
||
return plain, normalized, nil
|
||
}
|
||
|
||
func normalizeBannerTitleRuns(runs []model.BannerTitleRun) ([]model.BannerTitleRun, error) {
|
||
if len(runs) == 0 {
|
||
return nil, nil
|
||
}
|
||
if len(runs) > bannerTitleMaxRuns {
|
||
return nil, BadAuthRequest("通知标题样式片段过多")
|
||
}
|
||
normalized := make([]model.BannerTitleRun, 0, len(runs))
|
||
for _, run := range runs {
|
||
if run.FontSize != nil {
|
||
value := *run.FontSize
|
||
if value < bannerTitleMinFontSize || value > bannerTitleMaxFontSize {
|
||
return nil, BadAuthRequest("标题字号需在 10 到 20 之间")
|
||
}
|
||
run.FontSize = &value
|
||
}
|
||
if run.FontWeight != nil {
|
||
value := *run.FontWeight
|
||
if _, ok := bannerTitleFontWeights[value]; !ok {
|
||
return nil, BadAuthRequest("标题字重仅支持 400 / 500 / 600 / 700")
|
||
}
|
||
run.FontWeight = &value
|
||
}
|
||
family := strings.TrimSpace(run.FontFamily)
|
||
if family != "" {
|
||
if _, ok := bannerTitleFontFamilies[family]; !ok {
|
||
return nil, BadAuthRequest("标题字体仅支持默认、衬线或等宽")
|
||
}
|
||
}
|
||
run.FontFamily = family
|
||
color := strings.ToUpper(strings.TrimSpace(run.Color))
|
||
if color != "" && !isBannerTitleHexColor(color) {
|
||
return nil, BadAuthRequest("标题颜色需为 #RRGGBB 格式")
|
||
}
|
||
run.Color = color
|
||
if run.Text == "" {
|
||
continue
|
||
}
|
||
normalized = append(normalized, run)
|
||
}
|
||
return mergeBannerTitleRuns(normalized), nil
|
||
}
|
||
|
||
// mergeBannerTitleRuns 合并相邻的同样式分段,避免前端反复编辑后存下大量零碎片段。
|
||
func mergeBannerTitleRuns(runs []model.BannerTitleRun) []model.BannerTitleRun {
|
||
merged := make([]model.BannerTitleRun, 0, len(runs))
|
||
for _, run := range runs {
|
||
if len(merged) > 0 && sameBannerTitleRunStyle(merged[len(merged)-1], run) {
|
||
merged[len(merged)-1].Text += run.Text
|
||
continue
|
||
}
|
||
merged = append(merged, run)
|
||
}
|
||
return merged
|
||
}
|
||
|
||
func sameBannerTitleRunStyle(left model.BannerTitleRun, right model.BannerTitleRun) bool {
|
||
return equalIntPointer(left.FontSize, right.FontSize) &&
|
||
equalIntPointer(left.FontWeight, right.FontWeight) &&
|
||
left.FontFamily == right.FontFamily &&
|
||
left.Color == right.Color
|
||
}
|
||
|
||
func equalIntPointer(left *int, right *int) bool {
|
||
if left == nil || right == nil {
|
||
return left == nil && right == nil
|
||
}
|
||
return *left == *right
|
||
}
|
||
|
||
func isBannerTitleHexColor(value string) bool {
|
||
if len(value) != 7 || value[0] != '#' {
|
||
return false
|
||
}
|
||
for _, char := range value[1:] {
|
||
if (char >= '0' && char <= '9') || (char >= 'A' && char <= 'F') {
|
||
continue
|
||
}
|
||
return false
|
||
}
|
||
return true
|
||
}
|
||
|
||
func bannerTitleRunsText(runs []model.BannerTitleRun) string {
|
||
var text strings.Builder
|
||
for _, run := range runs {
|
||
text.WriteString(run.Text)
|
||
}
|
||
return text.String()
|
||
}
|
||
|
||
func normalizeBannerLink(link string) (string, error) {
|
||
link = strings.TrimSpace(link)
|
||
if link == "" {
|
||
return "", nil
|
||
}
|
||
lower := strings.ToLower(link)
|
||
if strings.HasPrefix(lower, "http://") || strings.HasPrefix(lower, "https://") {
|
||
if utf8.RuneCountInString(link) > 500 {
|
||
return "", BadAuthRequest("跳转链接不能超过 500 个字符")
|
||
}
|
||
return link, nil
|
||
}
|
||
if strings.HasPrefix(link, "/") {
|
||
if utf8.RuneCountInString(link) > 500 {
|
||
return "", BadAuthRequest("站内路径不能超过 500 个字符")
|
||
}
|
||
return link, nil
|
||
}
|
||
return "", BadAuthRequest("跳转链接仅支持 http(s) 外链或以 / 开头的站内路径")
|
||
}
|
||
|
||
// 通知类型的取值白名单。
|
||
//
|
||
// 必须与前端 web/src/lib/announcements/banner-notice.ts 的 BANNER_NOTICE_TYPES 逐项一致:
|
||
// 后端多一项会让前端的合法取值被拒,少一项会让前端选中的值被静默丢弃;
|
||
// web/test/banner-notice.test.ts 会读本文件做双向校验。
|
||
// 图标素材(emoji)是普通文本,随 TitleRuns 走,不需要单独白名单。
|
||
var bannerNoticeTypes = map[string]struct{}{
|
||
"notice": {}, "activity": {}, "update": {}, "warning": {},
|
||
}
|
||
|
||
const bannerNoticeDefaultType = "notice"
|
||
|
||
// normalizeBannerNoticeType 把空值补成默认类型,其余值必须在白名单内,避免存进无法渲染的类型。
|
||
func normalizeBannerNoticeType(value string) (string, error) {
|
||
normalized := strings.ToLower(strings.TrimSpace(value))
|
||
if normalized == "" {
|
||
return bannerNoticeDefaultType, nil
|
||
}
|
||
if _, ok := bannerNoticeTypes[normalized]; !ok {
|
||
return "", BadAuthRequest("通知类型无效,仅支持公告、活动、更新或警告")
|
||
}
|
||
return normalized, nil
|
||
}
|
||
|
||
func (s *Service) CreateBannerAnnouncement(actor *model.User, req CreateBannerAnnouncementRequest) (*model.BannerAnnouncement, error) {
|
||
if err := s.RequireAdmin(actor); err != nil {
|
||
return nil, err
|
||
}
|
||
title, titleRuns, err := normalizeBannerTitle(req.Title, req.TitleRuns)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
link, err := normalizeBannerLink(req.Link)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
noticeType, err := normalizeBannerNoticeType(req.NoticeType)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
status := strings.TrimSpace(req.Status)
|
||
if status != "active" && status != "disabled" {
|
||
return nil, BadAuthRequest("通知状态无效,仅支持 active 或 disabled")
|
||
}
|
||
if req.StartsAt != nil && req.EndsAt != nil && req.EndsAt.Before(*req.StartsAt) {
|
||
return nil, BadAuthRequest("有效期结束时间不能早于开始时间")
|
||
}
|
||
now := time.Now()
|
||
banner := &model.BannerAnnouncement{
|
||
ID: newID(),
|
||
Title: title,
|
||
TitleRuns: titleRuns,
|
||
NoticeType: noticeType,
|
||
Link: link,
|
||
Status: status,
|
||
StartsAt: req.StartsAt,
|
||
EndsAt: req.EndsAt,
|
||
CreatedBy: actor.ID,
|
||
CreatedAt: now,
|
||
UpdatedAt: now,
|
||
}
|
||
if err := s.repo.CreateBannerAnnouncement(banner); err != nil {
|
||
return nil, err
|
||
}
|
||
return banner, nil
|
||
}
|
||
|
||
func (s *Service) UpdateBannerAnnouncement(actor *model.User, id string, req UpdateBannerAnnouncementRequest) (*model.BannerAnnouncement, error) {
|
||
if err := s.RequireAdmin(actor); err != nil {
|
||
return nil, err
|
||
}
|
||
banner, err := s.repo.BannerAnnouncement(strings.TrimSpace(id))
|
||
if err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, BadAuthRequest("常驻通知不存在")
|
||
}
|
||
return nil, err
|
||
}
|
||
title, titleRuns, err := normalizeBannerTitle(req.Title, req.TitleRuns)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
link, err := normalizeBannerLink(req.Link)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
noticeType, err := normalizeBannerNoticeType(req.NoticeType)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
status := strings.TrimSpace(req.Status)
|
||
if status != "active" && status != "disabled" {
|
||
return nil, BadAuthRequest("通知状态无效,仅支持 active 或 disabled")
|
||
}
|
||
if req.StartsAt != nil && req.EndsAt != nil && req.EndsAt.Before(*req.StartsAt) {
|
||
return nil, BadAuthRequest("有效期结束时间不能早于开始时间")
|
||
}
|
||
banner.Title = title
|
||
banner.TitleRuns = titleRuns
|
||
banner.NoticeType = noticeType
|
||
banner.Link = link
|
||
banner.Status = status
|
||
banner.StartsAt = req.StartsAt
|
||
banner.EndsAt = req.EndsAt
|
||
banner.UpdatedAt = time.Now()
|
||
|
||
if err := s.repo.UpdateBannerAnnouncement(banner); err != nil {
|
||
return nil, err
|
||
}
|
||
return banner, nil
|
||
}
|
||
|
||
func (s *Service) DeleteBannerAnnouncement(actor *model.User, id string) error {
|
||
if err := s.RequireAdmin(actor); err != nil {
|
||
return err
|
||
}
|
||
_, err := s.repo.BannerAnnouncement(strings.TrimSpace(id))
|
||
if err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return BadAuthRequest("常驻通知不存在")
|
||
}
|
||
return err
|
||
}
|
||
return s.repo.DeleteBannerAnnouncement(strings.TrimSpace(id))
|
||
}
|