yunzerwebsiteallinone/go/services/tenant_user.go

package services

import (
	"strings"

	"server/models"
	authsvc "server/services/auth"
)

// ResolveTenantUID 把「令牌中的 user_id（认证中心 identity_id）」换算成
// 老表 yz_system_tenant_user.uid。
//
// 背景：uid 全量迁移脚本（scripts/uidmigrate）刻意跳过 yz_system_tenant_user
// （见其 skipTables），业务表列迁成 identity_id，而老表 uid 仍是老值。
// 因此凡是用 identity_id 去查老表的场景都必须先经过这里换算；
// 两者已经对齐的环境（或传入的本来就是老 uid）原值返回。
func ResolveTenantUID(tid, userID uint64) uint64 {
	if tid == 0 || userID == 0 {
		return 0
	}
	if legacy := authsvc.FindLegacyUID(tid, userID); legacy > 0 {
		return legacy
	}
	return userID
}

// BindTenantUser 绑定用户到租户（若已存在则更新状态/默认值）
func BindTenantUser(tid, uid uint64, account, name, phone, email *string, sex *uint8, birth *string, password *string, isDefault, status int8, remark *string) (uint64, error) {
	var existed models.SystemTenantUser
	err := models.Orm.QueryTable(new(models.SystemTenantUser)).
		Filter("tid", tid).
		Filter("uid", uid).
		One(&existed)
	if err == nil {
		update := map[string]interface{}{
			"account":    account,
			"name":       name,
			"phone":      phone,
			"email":      email,
			"password":   password,
			"status":     status,
			"is_default": isDefault,
			"remark":     remark,
		}
		if sex != nil {
			update["sex"] = *sex
		}
		if birth != nil {
			trimmedBirth := strings.TrimSpace(*birth)
			if trimmedBirth == "" {
				update["birth"] = nil
			} else {
				update["birth"] = trimmedBirth
			}
		}
		_, uErr := models.Orm.QueryTable(new(models.SystemTenantUser)).Filter("id", existed.ID).Update(update)
		return existed.ID, uErr
	}

	m := &models.SystemTenantUser{
		Tid:       tid,
		Uid:       uid,
		Account:   account,
		Name:      name,
		Phone:     phone,
		Email:     email,
		Password:  password,
		IsDefault: isDefault,
		Status:    status,
		Remark:    remark,
	}
	if sex != nil {
		m.Sex = *sex
	}
	if birth != nil {
		trimmedBirth := strings.TrimSpace(*birth)
		if trimmedBirth != "" {
			m.Birth = &trimmedBirth
		}
	}
	id, iErr := models.Orm.Insert(m)
	return uint64(id), iErr
}

// UnbindTenantUser 删除绑定关系
func UnbindTenantUser(id uint64) error {
	_, err := models.Orm.QueryTable(new(models.SystemTenantUser)).Filter("id", id).Delete()
	return err
}

// ListTenantUsersByTid 根据租户ID查询绑定关系
func ListTenantUsersByTid(tid uint64) ([]models.SystemTenantUser, error) {
	var rows []models.SystemTenantUser
	_, err := models.Orm.QueryTable(new(models.SystemTenantUser)).
		Filter("tid", tid).
		OrderBy("-is_default", "-id").
		All(&rows)
	return rows, err
}

// ListTenantBindingsByUid 根据用户ID查询绑定关系
func ListTenantBindingsByUid(uid uint64) ([]models.SystemTenantUser, error) {
	var rows []models.SystemTenantUser
	_, err := models.Orm.QueryTable(new(models.SystemTenantUser)).
		Filter("uid", uid).
		OrderBy("-is_default", "-id").
		All(&rows)
	return rows, err
}

// GetTenantUserByUidAndTid 根据用户ID和租户ID查询租户用户绑定关系
func GetTenantUserByUidAndTid(uid, tid uint64) (*models.SystemTenantUser, error) {
	var row models.SystemTenantUser
	err := models.Orm.QueryTable(new(models.SystemTenantUser)).
		Filter("uid", uid).
		Filter("tid", tid).
		One(&row)
	if err != nil {
		return nil, err
	}
	return &row, nil
}

// GetTenantUserByUid 根据用户ID查询默认/最新租户用户绑定关系
func GetTenantUserByUid(uid uint64) (*models.SystemTenantUser, error) {
	var row models.SystemTenantUser
	err := models.Orm.QueryTable(new(models.SystemTenantUser)).
		Filter("uid", uid).
		OrderBy("-is_default", "-id").
		One(&row)
	if err != nil {
		return nil, err
	}
	return &row, nil
}

// GetTenantByID 根据租户ID查询租户信息
func GetTenantByID(id uint64) (*models.SystemTenant, error) {
	var row models.SystemTenant
	err := models.Orm.QueryTable(new(models.SystemTenant)).
		Filter("id", id).
		One(&row)
	if err != nil {
		return nil, err
	}
	return &row, nil
}

// SetDefaultTenant 设置用户默认租户（同一用户仅一个默认）
func SetDefaultTenant(uid, tid uint64) error {
	_, err := models.Orm.QueryTable(new(models.SystemTenantUser)).Filter("uid", uid).Update(map[string]interface{}{
		"is_default": 0,
	})
	if err != nil {
		return err
	}
	_, err = models.Orm.QueryTable(new(models.SystemTenantUser)).
		Filter("uid", uid).
		Filter("tid", tid).
		Update(map[string]interface{}{"is_default": 1})
	return err
}