im_manager/server/internal/controller/internalapi/imconfig/im_config.go

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 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
}