package imconfig import ( "context" api "hotgo/api/admin/imconfiginternal" "hotgo/internal/dao" "hotgo/internal/library/imtenant" "hotgo/internal/model/entity" "hotgo/internal/service" "strings" "github.com/gogf/gf/v2/errors/gerror" "github.com/gogf/gf/v2/os/gtime" "github.com/gogf/gf/v2/util/gconv" ) var Config = cConfig{} type cConfig struct{} func (c *cConfig) Tenant(ctx context.Context, req *api.TenantReq) (res *api.TenantRes, err error) { tenant, err := getTenant(ctx, dao.ImTenant.Columns().Code, req.Code) if err != nil { return nil, err } return &api.TenantRes{TenantModel: tenantToModel(tenant)}, nil } func (c *cConfig) TenantByID(ctx context.Context, req *api.TenantByIDReq) (res *api.TenantRes, err error) { tenant, err := getTenant(ctx, dao.ImTenant.Columns().Id, req.Id) if err != nil { return nil, err } return &api.TenantRes{TenantModel: tenantToModel(tenant)}, nil } func (c *cConfig) TenantList(ctx context.Context, req *api.TenantListReq) (res *api.TenantListRes, err error) { limit := req.Limit if limit <= 0 || limit > 1000 { limit = 1000 } mod := imtenant.RegistryDB().Model(dao.ImTenant.Table()).Safe().Ctx(ctx).Limit(limit) if req.State != "" { if req.State == "active" { mod = mod.Where(dao.ImTenant.Columns().State, 1) } else { mod = mod.Where(dao.ImTenant.Columns().State, req.State) } } var tenants []*entity.ImTenant if err = mod.OrderAsc(dao.ImTenant.Columns().Id).Scan(&tenants); err != nil { return nil, gerror.Wrap(err, "获取IM租户配置失败") } res = &api.TenantListRes{Items: make([]*api.TenantModel, 0, len(tenants))} for _, tenant := range tenants { res.Items = append(res.Items, tenantToModel(tenant)) } return res, nil } func (c *cConfig) RegistrationSms(ctx context.Context, req *api.RegistrationSmsReq) (res *api.RegistrationSmsRes, err error) { var record *entity.SysConfig if err = dao.SysConfig.Ctx(ctx). Where(dao.SysConfig.Columns().Group, "im"). Where(dao.SysConfig.Columns().Key, "register_sms_enabled"). Where(dao.SysConfig.Columns().Status, 1). Scan(&record); err != nil { return nil, gerror.Wrap(err, "获取IM注册短信配置失败") } enabled := true if record != nil { enabled = gconv.String(record.Value) != "0" } return &api.RegistrationSmsRes{Enabled: enabled}, nil } func (c *cConfig) ThirdParty(ctx context.Context, req *api.ThirdPartyReq) (res *api.ThirdPartyRes, err error) { values := map[string]string{} keys := splitKeys(req.Keys) mod := dao.SysConfig.Ctx(ctx).Where(dao.SysConfig.Columns().Group, req.Group).Where(dao.SysConfig.Columns().Status, 1) if len(keys) > 0 { mod = mod.WhereIn(dao.SysConfig.Columns().Key, keys) } var rows []*entity.SysConfig if err = mod.Scan(&rows); err != nil { return nil, gerror.Wrap(err, "获取IM第三方配置失败") } for _, row := range rows { values[row.Key] = row.Value } return &api.ThirdPartyRes{Values: values}, nil } func (c *cConfig) TenantLoginIPWhitelist(ctx context.Context, req *api.TenantLoginIPWhitelistReq) (res *api.TenantLoginIPWhitelistRes, err error) { if _, err = getTenant(ctx, dao.ImTenant.Columns().Id, req.Id); err != nil { return nil, err } data, err := service.SysImUserPolicyGroup().LoginIPWhitelistByTenant(ctx, req.Id) if err != nil { return nil, err } return &api.TenantLoginIPWhitelistRes{TenantId: data.TenantId, AppList: data.AppList, WebList: data.WebList}, nil } func (c *cConfig) AppReleaseLatest(ctx context.Context, req *api.AppReleaseLatestReq) (res *api.AppReleaseLatestRes, err error) { platform := strings.ToLower(strings.TrimSpace(req.Platform)) channel := strings.ToLower(strings.TrimSpace(req.Channel)) if platform != "ios" && platform != "android" { return nil, gerror.New("客户端平台不合法") } if channel == "" { channel = "stable" } if strings.ContainsAny(channel, " \t\r\n/") || len(channel) > 32 { return nil, gerror.New("客户端渠道不合法") } var row struct { ID int64 `json:"id"` Platform string `json:"platform"` Channel string `json:"channel"` LatestVersion string `json:"latest_version"` MinSupportedVersion string `json:"min_supported_version"` UpdateType string `json:"update_type"` ReleaseNotes string `json:"release_notes"` StoreURL string `json:"store_url"` PackageURL string `json:"package_url"` Checksum string `json:"checksum"` RemindAfterSeconds int `json:"remind_after_seconds"` PublishedAt *gtime.Time `json:"published_at"` CreatedAt *gtime.Time `json:"created_at"` UpdatedAt *gtime.Time `json:"updated_at"` } err = imtenant.RegistryDB().Model("im_app_releases").Safe().Ctx(ctx). Where("platform", platform).Where("channel", channel).Limit(1).Scan(&row) if err != nil { return nil, gerror.Wrap(err, "获取IM客户端版本策略失败") } if row.ID <= 0 { return nil, gerror.New("IM客户端版本策略不存在") } return &api.AppReleaseLatestRes{ ID: row.ID, Platform: row.Platform, Channel: row.Channel, LatestVersion: row.LatestVersion, MinSupportedVersion: row.MinSupportedVersion, UpdateType: row.UpdateType, ReleaseNotes: row.ReleaseNotes, StoreURL: row.StoreURL, PackageURL: row.PackageURL, Checksum: row.Checksum, RemindAfterSeconds: row.RemindAfterSeconds, PublishedAt: gtimeToRFC3339(row.PublishedAt), CreatedAt: gtimeToRFC3339(row.CreatedAt), UpdatedAt: gtimeToRFC3339(row.UpdatedAt), }, nil } func getTenant(ctx context.Context, field string, value any) (*entity.ImTenant, error) { var tenant *entity.ImTenant if err := imtenant.RegistryDB().Model(dao.ImTenant.Table()).Safe().Ctx(ctx).Where(field, value).Scan(&tenant); err != nil { return nil, gerror.Wrap(err, "获取IM租户配置失败") } if tenant == nil || tenant.Id <= 0 { return nil, gerror.New("IM租户不存在") } return tenant, nil } func tenantToModel(tenant *entity.ImTenant) *api.TenantModel { if tenant == nil { return nil } desc := tenant.TenantDesc return &api.TenantModel{ ID: tenant.Id, Code: tenant.Code, Name: tenant.Name, TenantDesc: &desc, DBName: tenant.DbName, DBDSN: tenant.DbDsn, State: tenant.State, CreatedAt: gtimeToRFC3339(tenant.CreatedAt), CreatedBy: tenant.CreatedBy, UpdatedAt: gtimeToRFC3339(tenant.UpdatedAt), UpdatedBy: tenant.UpdatedBy, } } func gtimeToRFC3339(t *gtime.Time) string { if t == nil { return "" } return t.Time.Format("2006-01-02T15:04:05Z07:00") } func splitKeys(raw string) []string { if raw == "" { return nil } parts := strings.Split(raw, ",") keys := make([]string, 0, len(parts)) for _, part := range parts { if part = strings.TrimSpace(part); part != "" { keys = append(keys, part) } } return keys }