sundynix-agentix/sundynix-gateway/internal/store/user.go

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 是展示名（微信昵称或默认）。
//
// 邮箱填一个合成的、按 openid 唯一的占位值（wx-<openid>@wx.local）：User.Email 上有整列
// 唯一索引，微信用户若都留空邮箱，第二个就会撞唯一约束建号失败。合成邮箱唯一且一望即知非真实，
// 既绕开冲突又不动索引（改成部分唯一索引需删存量索引，无法在无库可见时稳妥进行）。
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, Email: "wx-" + openID + "@wx.local"}
	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
}