// Package sys // @Link https://github.com/bufanyun/hotgo // @Copyright Copyright (c) 2026 HotGo // @Author Ms <133814250@qq.com> // @License https://github.com/bufanyun/hotgo/blob/master/LICENSE package sys import ( "context" "crypto/sha256" "encoding/hex" "encoding/json" "fmt" "hotgo/internal/library/imtenant" "hotgo/internal/model/input/sysin" "hotgo/internal/service" "strings" "time" "github.com/gogf/gf/v2/database/gdb" "github.com/gogf/gf/v2/errors/gerror" "github.com/gogf/gf/v2/os/gctx" "github.com/gogf/gf/v2/util/guid" "golang.org/x/crypto/bcrypt" ) type sSysImUser struct{} func NewSysImUser() *sSysImUser { return &sSysImUser{} } func init() { service.RegisterSysImUser(NewSysImUser()) } // List 获取IM用户列表 func (s *sSysImUser) List(ctx context.Context, in *sysin.ImUserListInp) (list []*sysin.ImUserListModel, totalCount int, err error) { if err = in.Filter(ctx); err != nil { return } scope, err := resolveImAdminScope(ctx, in.TenantId) if err != nil { return } db, tenant, err := imtenant.DB(ctx, scope.TenantId) if err != nil { return } mod := s.listModel(ctx, db) mod = s.applyListWhere(mod, in) mod = mod.Page(in.Page, in.PerPage).OrderDesc("u.id") if err = mod.ScanAndCount(&list, &totalCount, false); err != nil { err = gerror.Wrap(err, "获取IM用户列表失败,请稍后重试!") return } if len(list) == 0 { return } for _, item := range list { item.TenantId = tenant.Id item.TenantName = tenant.Name } if err = s.fillAuthAndCredentials(ctx, db, list); err != nil { return } return } func (s *sSysImUser) listModel(ctx context.Context, db gdb.DB) *gdb.Model { return db.Model("im_users u").Safe().Ctx(ctx). LeftJoin("im_user_profiles up", "up.user_id = u.id"). LeftJoin("im_user_security us", "us.user_id = u.id"). Fields( "CAST(u.id AS CHAR) user_id", "u.username", "COALESCE(u.phone, '') phone", "u.state", "u.lastseen last_seen", "u.useragent user_agent", "u.createdat created_at", "u.updatedat updated_at", "up.nickname", "up.public_account", "up.gender", "up.region", "us.auth_version", "us.password_updated_at", ) } func (s *sSysImUser) applyListWhere(mod *gdb.Model, in *sysin.ImUserListInp) *gdb.Model { if userId, _ := sysin.ParseOptionalInt64(in.UserId); userId > 0 { mod = mod.Where("u.id", userId) } if in.Nickname != "" { mod = mod.WhereLike("up.nickname", "%"+in.Nickname+"%") } if in.AuthName != "" { mod = mod.Where("u.username = ? OR u.phone = ? OR up.email = ?", in.AuthName, in.AuthName, in.AuthName) } if in.State != nil { mod = mod.Where("u.state", *in.State) } if len(in.CreatedAt) == 2 { mod = mod.WhereBetween("u.createdat", in.CreatedAt[0], in.CreatedAt[1]) } if len(in.LastSeen) == 2 { mod = mod.WhereBetween("u.lastseen", in.LastSeen[0], in.LastSeen[1]) } return mod } func (s *sSysImUser) fillAuthAndCredentials(ctx context.Context, db gdb.DB, list []*sysin.ImUserListModel) (err error) { ids := make([]int64, 0, len(list)) userMap := make(map[int64]*sysin.ImUserListModel, len(list)) for _, item := range list { userId, parseErr := sysin.ParseOptionalInt64(item.UserId) if parseErr != nil { return parseErr } ids = append(ids, userId) userMap[userId] = item } credentialRows, err := db.Model("im_credentials").Safe().Ctx(ctx). Fields("userid", "method", "value", "done"). WhereIn("userid", ids). WhereNull("deletedat"). OrderAsc("method").OrderAsc("value"). All() if err != nil { return gerror.Wrap(err, "获取IM用户凭证失败,请稍后重试!") } credentialMap := make(map[int64][]string) for _, row := range credentialRows { userId := row["userid"].Int64() method := row["method"].String() value := row["value"].String() done := row["done"].Bool() if !done { value += "(未验证)" } credentialMap[userId] = append(credentialMap[userId], method+":"+value) } for userId, item := range userMap { item.AuthNames = item.Username if item.Phone != "" && item.Phone != item.Username { item.AuthNames += ", " + item.Phone } item.Credentials = strings.Join(credentialMap[userId], ", ") } return } // View 获取IM用户指定信息 func (s *sSysImUser) View(ctx context.Context, in *sysin.ImUserViewInp) (res *sysin.ImUserViewModel, err error) { if err = in.Filter(ctx); err != nil { return } scope, err := resolveImAdminScope(ctx, in.TenantId) if err != nil { return } db, tenant, err := imtenant.DB(ctx, scope.TenantId) if err != nil { return } userId, err := sysin.ParseOptionalInt64(in.UserId) if err != nil { return nil, err } fields := append(s.viewFields(), "u.access access_json", "u.public profile_json", "u.trusted trusted_json") err = db.Model("im_users u").Safe().Ctx(ctx). LeftJoin("im_user_profiles up", "up.user_id = u.id"). LeftJoin("im_user_security us", "us.user_id = u.id"). Fields(fields...). Where("u.id", userId). Scan(&res) if err != nil { return nil, gerror.Wrap(err, "获取IM用户信息失败,请稍后重试!") } if res == nil { return nil, nil } res.TenantId = tenant.Id res.TenantName = tenant.Name list := []*sysin.ImUserListModel{&res.ImUserListModel} if err = s.fillAuthAndCredentials(ctx, db, list); err != nil { return nil, err } return } func (s *sSysImUser) viewFields() []interface{} { return []interface{}{ "CAST(u.id AS CHAR) user_id", "u.username", "COALESCE(u.phone, '') phone", "u.state", "u.lastseen last_seen", "u.useragent user_agent", "u.createdat created_at", "u.updatedat updated_at", "up.nickname", "up.public_account", "up.gender", "up.region", "up.signature", "up.avatar_file_id", "CAST(up.birth_date AS CHAR) birth_date", "up.version profile_version", "us.auth_version", "us.password_updated_at", } } func (s *sSysImUser) ResetPassword(ctx context.Context, in *sysin.ImUserPasswordResetInp) (err error) { if err = in.Filter(ctx); err != nil { return } scope, err := resolveImAdminScope(ctx, in.TenantId) if err != nil { return err } userId, err := sysin.ParseOptionalInt64(in.UserId) if err != nil { return err } db, _, err := imtenant.DB(ctx, scope.TenantId) if err != nil { return err } err = resetIMUserPassword(ctx, db, scope, userId, in.NewPassword) result := "success" if err != nil { result = "failed" } _ = writeImAdminAudit(ctx, scope.TenantId, "user_password_reset", "user", userId, result, nil) return err } func resetIMUserPassword(ctx context.Context, db gdb.DB, scope imAdminScope, userId int64, newPassword string) error { passwordHash, err := bcrypt.GenerateFromPassword([]byte(newPassword), bcrypt.DefaultCost) if err != nil { return gerror.Wrap(err, "生成密码摘要失败") } securityId, err := imtenant.NextSnowflakeID(ctx) if err != nil { return gerror.Wrap(err, "生成安全记录ID失败") } eventId, err := imtenant.NextSnowflakeID(ctx) if err != nil { return gerror.Wrap(err, "生成安全事件ID失败") } now := time.Now().UTC().Round(time.Millisecond) requestId := scope.RequestId if requestId == "" { requestId = gctx.CtxId(ctx) } if requestId == "" { requestId = guid.S() } eventKeyBytes := sha256.Sum256([]byte(fmt.Sprintf("password-reset:%d:%d:%s", scope.TenantId, userId, requestId))) eventKey := hex.EncodeToString(eventKeyBytes[:]) payload, _ := json.Marshal(map[string]any{ "user_id": userId, "operator_id": scope.OperatorId, "request_id": requestId, }) return db.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error { exists, txErr := tx.GetValue("SELECT EXISTS(SELECT 1 FROM im_users WHERE id=? LIMIT 1)", userId) if txErr != nil { return txErr } if !exists.Bool() { return gerror.New("IM用户不存在") } result, txErr := tx.Exec("UPDATE im_auth SET secret=? WHERE userid=?", passwordHash, userId) if txErr != nil { return txErr } if affected, _ := result.RowsAffected(); affected == 0 { return gerror.New("用户认证记录不存在") } if _, txErr = tx.Exec(`INSERT INTO im_user_security (id,user_id,auth_version,password_updated_at,created_at,updated_at) VALUES(?,?,?,?,?,?) ON DUPLICATE KEY UPDATE auth_version=auth_version+1, password_updated_at=VALUES(password_updated_at),updated_at=VALUES(updated_at)`, securityId, userId, 2, now, now, now); txErr != nil { return txErr } version, txErr := tx.GetValue("SELECT auth_version FROM im_user_security WHERE user_id=?", userId) if txErr != nil { return txErr } var eventPayload map[string]any _ = json.Unmarshal(payload, &eventPayload) eventPayload["auth_version"] = version.Int() payload, _ = json.Marshal(eventPayload) _, txErr = tx.Exec(`INSERT INTO im_business_events_outbox (id,event_id,target_type,target,actor_user_id,event_type,payload,status,created_at,updated_at) VALUES(?,?,?,?,?,?,CAST(? AS JSON),'pending',?,?) ON DUPLICATE KEY UPDATE event_id=VALUES(event_id)`, eventId, eventKey, "user", fmt.Sprintf("%d", userId), 0, "security.password_reset", string(payload), now, now) return txErr }) }