aiforge/services/cloudbrain/util.go

package cloudbrain

import (
	"strings"

	"code.gitea.io/gitea/models"
	"code.gitea.io/gitea/modules/context"
	"code.gitea.io/gitea/modules/setting"
)

func GetAiCenterShow(aiCenter string, ctx *context.Context) string {
	aiCenterInfo := strings.Split(aiCenter, "+")

	if len(aiCenterInfo) == 2 {
		if setting.C2NetMapInfo != nil {
			if info, ok := setting.C2NetMapInfo[aiCenterInfo[0]]; ok {
				if ctx.Language() == "zh-CN" {
					return info.Content
				} else {
					return info.ContentEN
				}
			} else {
				return aiCenterInfo[1]
			}

		} else {
			return aiCenterInfo[1]
		}

	}

	return ""

}

func GetAiCenterInfoByCenterCode(aiCenterCode string) *setting.C2NetSequenceInfo {
	if setting.AiCenterCodeAndNameMapInfo != nil {
		if info, ok := setting.AiCenterCodeAndNameMapInfo[aiCenterCode]; ok {
			return info
		} else {
			return nil
		}
	} else {
		return nil
	}
}

func getAiCenterCode(aiCenter string) string {
	aiCenterInfo := strings.Split(aiCenter, "+")
	return aiCenterInfo[0]
}

func UpdateCloudbrainAiCenter(cloudbrain *models.CloudbrainInfo) *models.CloudbrainInfo {
	if cloudbrain.Cloudbrain.Type == models.TypeCloudBrainOne {
		cloudbrain.Cloudbrain.AiCenter = models.AICenterOfCloudBrainOne
		cloudbrain.Cloudbrain.Cluster = models.OpenICluster
	}
	if cloudbrain.Cloudbrain.Type == models.TypeCloudBrainTwo {
		cloudbrain.Cloudbrain.AiCenter = models.AICenterOfCloudBrainTwo
		cloudbrain.Cloudbrain.Cluster = models.OpenICluster
	}
	if cloudbrain.Cloudbrain.Type == models.TypeCDCenter {
		cloudbrain.Cloudbrain.AiCenter = models.AICenterOfChengdu
		cloudbrain.Cloudbrain.Cluster = models.OpenICluster
	}
	if cloudbrain.Cloudbrain.Type == models.TypeC2Net {
		cloudbrain.Cloudbrain.AiCenter = getAiCenterCode(cloudbrain.Cloudbrain.AiCenter)
		cloudbrain.Cloudbrain.Cluster = models.C2NetCluster
	}

	return cloudbrain
}