feat: 实时消息与管理端运营能力

新增群聊/私信与公共 WebSocket 总线,站长用户内容档案,并将通知与待审角标改为推送驱动;同步精简管理仪表盘并修复主题换肤 DOM 冲突。

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
2026-09-15 22:58:16 +08:00
parent 29db9ae9b5
commit 42c395a864
55 changed files with 7607 additions and 346 deletions

View File

@@ -3,6 +3,7 @@ package model
import (
"crypto/sha256"
"encoding/hex"
"errors"
"fmt"
"log"
@@ -29,9 +30,15 @@ func InitDB(dsn string) error {
return fmt.Errorf("refresh token 旧数据迁移失败: %w", err)
}
// 清理前次失败迁移遗留的 FK 约束(constraint:- 标签防止新建,但不删旧的)
if err := dropStaleChatFKConstraints(db); err != nil {
return fmt.Errorf("清理遗留 FK 约束失败: %w", err)
}
if err := db.AutoMigrate(
&User{}, &Board{}, &Post{}, &Comment{}, &RefreshToken{}, &Like{}, &Notification{}, &Checkin{},
&Announcement{}, &SiteSetting{}, &Attachment{}, &UserBoard{}, &LoginLog{},
&ChatRoom{}, &ChatRoomMember{}, &ChatMessage{},
); err != nil {
return fmt.Errorf("自动迁移失败: %w", err)
}
@@ -54,7 +61,20 @@ func InitDB(dsn string) error {
}
DB = db
if err := db.Exec(`
CREATE UNIQUE INDEX IF NOT EXISTS idx_chat_direct_key
ON chat_rooms (direct_key)
WHERE room_type = 'direct' AND deleted_at IS NULL AND direct_key <> ''
`).Error; err != nil {
return fmt.Errorf("私聊唯一索引创建失败: %w", err)
}
seedDefaultBoards(db)
if err := seedDefaultChatRoom(db); err != nil {
return fmt.Errorf("默认群聊大厅初始化失败: %w", err)
}
if err := ensureDefaultChatMemberships(db); err != nil {
return fmt.Errorf("默认群聊入群补齐失败: %w", err)
}
log.Println("[model] PostgreSQL 数据库初始化完成")
return nil
}
@@ -71,6 +91,22 @@ func PingDB() error {
return sqlDB.Ping()
}
// dropStaleChatFKConstraints 清理前次迁移失败遗留的 FK 约束。
// GORM AutoMigrate 先建表再加约束:首次迁移时 Notification.Room 的 FK 添加失败,
// 但 chat_rooms 表已建好且带 fk_chat_rooms_last_message 约束。
// 模型已改用 constraint:- 防止新建,但 GORM 不会删除已存在的约束,需手动清理。
func dropStaleChatFKConstraints(db *gorm.DB) error {
// 对每个遗留约束名,在可能的两张表上都尝试删除(IF EXISTS 保证幂等)
pairs := []struct{ table, constraint string }{
{"chat_rooms", "fk_chat_rooms_last_message"},
{"notifications", "fk_notifications_room"},
}
for _, p := range pairs {
db.Exec(fmt.Sprintf(`ALTER TABLE IF EXISTS %s DROP CONSTRAINT IF EXISTS %s`, p.table, p.constraint))
}
return nil
}
// prepareRefreshTokenMigration 旧版 refresh_tokens 表把明文存在 token 列,
// 新版改为 token_hash(SHA-256,NOT NULL+唯一索引)。在 AutoMigrate 之前:
// 1. 新增可带默认值的 token_hash 列(避免对存量行加 NOT NULL 列失败)
@@ -193,3 +229,92 @@ func seedDefaultBoards(db *gorm.DB) {
}
}
}
// seedDefaultChatRoom 确保存在唯一的全站大厅(公开默认群)
func seedDefaultChatRoom(db *gorm.DB) error {
var n int64
if err := db.Model(&ChatRoom{}).Where("is_default = ?", true).Count(&n).Error; err != nil {
return err
}
if n > 0 {
return nil
}
var owner User
if err := db.Where("role = ?", RoleOwner).Order("id ASC").First(&owner).Error; err != nil {
// 尚无用户时跳过,等有站长后再由下次启动补齐
if errors.Is(err, gorm.ErrRecordNotFound) {
return nil
}
// 尝试任意用户
if err2 := db.Order("id ASC").First(&owner).Error; err2 != nil {
return nil
}
}
room := ChatRoom{
Name: "全站大厅",
Description: "所有成员默认加入的公开群聊",
OwnerID: owner.ID,
RoomType: ChatRoomTypeGroup,
IsPrivate: false,
IsDefault: true,
MemberCount: 0,
}
if err := db.Create(&room).Error; err != nil {
return err
}
log.Println("[model] 已创建默认群聊「全站大厅」")
return nil
}
// ensureDefaultChatMemberships 把未封禁用户补进默认大厅(幂等)
func ensureDefaultChatMemberships(db *gorm.DB) error {
var room ChatRoom
if err := db.Where("is_default = ?", true).First(&room).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return nil
}
return err
}
var userIDs []uint
if err := db.Model(&User{}).
Where("deleted_at IS NULL AND banned = ?", false).
Pluck("id", &userIDs).Error; err != nil {
return err
}
if len(userIDs) == 0 {
return nil
}
var existing []uint
if err := db.Model(&ChatRoomMember{}).
Where("room_id = ?", room.ID).
Pluck("user_id", &existing).Error; err != nil {
return err
}
have := make(map[uint]struct{}, len(existing))
for _, id := range existing {
have[id] = struct{}{}
}
added := 0
for _, uid := range userIDs {
if _, ok := have[uid]; ok {
continue
}
role := ChatRoleMember
if uid == room.OwnerID {
role = ChatRoleOwner
}
m := ChatRoomMember{RoomID: room.ID, UserID: uid, Role: role}
if err := db.Create(&m).Error; err != nil {
continue
}
added++
}
if added > 0 {
_ = db.Model(&ChatRoom{}).Where("id = ?", room.ID).
UpdateColumn("member_count", gorm.Expr(
"(SELECT count(*) FROM chat_room_members WHERE room_id = ?)", room.ID,
)).Error
log.Printf("[model] 默认群聊补齐入群 %d 人", added)
}
return nil
}