package store import ( "context" "errors" "time" "gorm.io/gorm" ) // ErrUserExists 表示邮箱已注册(唯一约束冲突)。 var ErrUserExists = errors.New("email already registered") // CreateUser 新建用户(密码已在调用方 bcrypt 哈希)。邮箱重复返回 ErrUserExists。 func (p *Postgres) CreateUser(ctx context.Context, email, name, passwordHash string) (*User, error) { if p.db == nil { return nil, errStoreDisabled } var existed int64 p.db.WithContext(ctx).Model(&User{}).Where("email = ?", email).Count(&existed) if existed > 0 { return nil, ErrUserExists } u := &User{Email: email, Name: name, PasswordHash: passwordHash} if err := p.db.WithContext(ctx).Create(u).Error; err != nil { return nil, err } return u, nil } // GetUserByEmail 按邮箱取用户(登录用)。 func (p *Postgres) GetUserByEmail(ctx context.Context, email string) (*User, error) { if p.db == nil { return nil, errStoreDisabled } var u User if err := p.db.WithContext(ctx).Where("email = ?", email).First(&u).Error; err != nil { return nil, err } return &u, nil } // GetUserByID 按主键取用户(鉴权后取当前用户)。 func (p *Postgres) GetUserByID(ctx context.Context, id string) (*User, error) { if p.db == nil { return nil, errStoreDisabled } var u User if err := p.db.WithContext(ctx).First(&u, "id = ?", id).Error; err != nil { if errors.Is(err, gorm.ErrRecordNotFound) { return nil, nil } return nil, err } return &u, nil } // GetUserByWechatOpenID 按微信 openid 查用户;不存在返回 (nil, nil)。 func (p *Postgres) GetUserByWechatOpenID(ctx context.Context, openID string) (*User, error) { if p.db == nil { return nil, errStoreDisabled } if openID == "" { return nil, nil } var u User err := p.db.WithContext(ctx).Where("wechat_openid = ?", openID).First(&u).Error if errors.Is(err, gorm.ErrRecordNotFound) { return nil, nil } if err != nil { return nil, err } return &u, nil } // CreateWechatUser 为首次微信登录的用户建号(无邮箱/密码)。name 是展示名(微信昵称或默认)。 func (p *Postgres) CreateWechatUser(ctx context.Context, openID, name string) (*User, error) { if p.db == nil { return nil, errStoreDisabled } if openID == "" { return nil, errors.New("openid 必填") } u := &User{WechatOpenID: openID, Name: name} if err := p.db.WithContext(ctx).Create(u).Error; err != nil { return nil, err } return u, nil } // (测试辅助见 user_wechat_test.go) // WechatUserRow 是后台微信用户列表一行:账号 + 其个人租户名 + 余额。 type WechatUserRow struct { ID string `json:"id"` Name string `json:"name"` // gorm 默认把 WechatOpenID 断成列名 wechat_open_id,与 SQL alias wechat_openid 对不上 → 空。 // 显式指定列名(BalanceMicro 同理,避免 balance_micro 被断成 balance_micro 之外的名字)。 WechatOpenID string `gorm:"column:wechat_openid" json:"wechat_openid"` CreatedAt time.Time `json:"created_at"` TenantName string `json:"tenant_name"` BalanceMicro int64 `gorm:"column:balance_micro" json:"balance_micro"` } // ListWechatUsers 列出所有微信登录用户(wechat_openid 非空),带其 owner 租户名与余额。 // 系统级读,跨租户。倒序(新注册在前)。 func (p *Postgres) ListWechatUsers(ctx context.Context, limit int) []WechatUserRow { if p.db == nil { return nil } if limit <= 0 || limit > 500 { limit = 200 } var out []WechatUserRow // 用户 → 其作为 owner 的租户(EnsureDefaultTenant 建的个人租户) p.db.WithContext(WithoutTenant(ctx)).Table("sundynix_user as u"). Select("u.id, u.name, u.wechat_openid, u.created_at, " + "coalesce(t.name,'') as tenant_name, coalesce(t.credit_balance_micro,0) as balance_micro"). Joins("left join sundynix_tenant_member m on m.user_id = u.id and m.role = 'owner'"). Joins("left join sundynix_tenant t on t.id = m.tenant_id"). Where("u.wechat_openid <> '' and u.deleted_at is null"). Order("u.created_at desc").Limit(limit).Scan(&out) return out }