kevinandClaude
eff4972668
把工程根目录中和"配置加载"、"数据存储"两个领域相关的全部 Go 文件迁到 internal/ 下的子包，按功能分组的第一阶段成果。 internal/config/ - 从 config.go 抽出 Config / MQTTConfig / WebConfig / DatabaseConfig 等 类型并大写导出，函数改名 Default/Load/Write/Validate/BuildTLS 等。 - 测试随被测代码迁移成 package config 的内部测试。 - 根目录留 config.go 作桥接：用 type alias 让旧的小写名（config / mqttConfig / databaseConfig 等）继续可用，避免修改 30+ 处调用方。 internal/store/ - 把 db.go、store_query.go、db_write_queue.go 与 13 个 *_store.go 一并 迁入；26 个 *Record 类型与 store 同包以避免循环依赖。 - store -> Store；50+ 标识符从小写未导出改为大写导出（包括 record、 ListOptions、错误变量、bot/llm/runtime 常量、helpers 等）。 - 新增 DB() / Driver() 访问器供 ai 子系统使用，避免直接访问私有字段。 - bot_pki_store.go 独立出来，把 PKI 解密所需的 store 方法集中归类。 - helpers.go 提供 hashPassword / uint32FromRecord / printJSON 等以前在 其他根目录文件中的辅助；test_helpers_test.go 提供 verifyPassword 与 publicMapTileSourceDTO 让测试可以本地运行而不依赖 main 包。 根目录新增： - store_bridge.go：完整 type-alias / 函数包装层，把 internal/store 的 导出名映射回旧的小写名，让 admin_*_routes.go、web.go、bot_service.go 等仍未迁出的文件继续编译。后续步骤把它们迁到各自领域包后可逐步删除。 - test_helpers_test.go：根目录测试沿用 openTestStore 的入口。 go build ./... 与 go test ./... 全部通过；测试数量与重构前一致。 Co-Authored-By: Claude <noreply@anthropic.com>
100 lines
2.5 KiB
Go
100 lines
2.5 KiB
Go
package store
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var ErrUserAlreadyExists = errors.New("user already exists")
|
|
|
|
func (s *Store) GetUserByUsername(username string) (*UserRecord, error) {
|
|
var user UserRecord
|
|
if err := s.db.Where("username = ?", username).Take(&user).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
func (s *Store) GetUserByID(id uint64) (*UserRecord, error) {
|
|
var user UserRecord
|
|
if err := s.db.Where("id = ?", id).Take(&user).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
func (s *Store) ListUsers() ([]UserRecord, error) {
|
|
var users []UserRecord
|
|
return users, s.db.Order("id ASC").Find(&users).Error
|
|
}
|
|
|
|
func (s *Store) CreateAdminUser(username, password string) (*UserRecord, error) {
|
|
username = strings.TrimSpace(username)
|
|
if username == "" {
|
|
return nil, fmt.Errorf("username is required")
|
|
}
|
|
if password == "" {
|
|
return nil, fmt.Errorf("password is required")
|
|
}
|
|
if _, err := s.GetUserByUsername(username); err == nil {
|
|
return nil, ErrUserAlreadyExists
|
|
} else if !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, err
|
|
}
|
|
hash, err := hashPassword(password)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("hash user password: %w", err)
|
|
}
|
|
user := UserRecord{Username: username, PasswordHash: hash, Role: AdminRole}
|
|
if err := s.db.Create(&user).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
func (s *Store) UpdateUserPassword(id uint64, password string) (*UserRecord, error) {
|
|
if id == 0 {
|
|
return nil, fmt.Errorf("user id is required")
|
|
}
|
|
if password == "" {
|
|
return nil, fmt.Errorf("password is required")
|
|
}
|
|
user, err := s.GetUserByID(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
hash, err := hashPassword(password)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("hash user password: %w", err)
|
|
}
|
|
if err := s.db.Model(&UserRecord{}).Where("id = ?", id).Updates(map[string]any{"password_hash": hash, "updated_at": time.Now()}).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
user.PasswordHash = hash
|
|
return s.GetUserByID(id)
|
|
}
|
|
|
|
func (s *Store) EnsureDefaultAdmin(username, password string) error {
|
|
var existing UserRecord
|
|
err := s.db.Where("username = ?", username).Take(&existing).Error
|
|
if err == nil {
|
|
return nil
|
|
}
|
|
if !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return err
|
|
}
|
|
hash, err := hashPassword(password)
|
|
if err != nil {
|
|
return fmt.Errorf("hash admin password: %w", err)
|
|
}
|
|
user := UserRecord{Username: username, PasswordHash: hash, Role: AdminRole}
|
|
if err := s.db.Create(&user).Error; err != nil {
|
|
return fmt.Errorf("create default admin user: %w", err)
|
|
}
|
|
return nil
|
|
}
|