272 lines
8.2 KiB
Go
272 lines
8.2 KiB
Go
|
package setting
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"code.gitea.io/gitea/modules/log"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
KUBERNETES = "kubernetes" // 支持 "k8s" 和 "kubernetes"
|
|||
|
|
DOCKER = "docker"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 检查用户输入的 DevContainer Agent 是否有效
|
|||
|
|
func isValidAgent(agent string) bool {
|
|||
|
|
return agent == "k8s" || agent == KUBERNETES || agent == DOCKER
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
CLOUD_PROVIDER_TENCENT = "tencent"
|
|||
|
|
DEVCONTAINER_CLOUD_NAT_RULE_DESCRIPTION_PREFIX = "DevContainer: "
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// validCloudProviderSet 私有 Set 结构，标识目前系统所有支持的 Cloud Provider 类型
|
|||
|
|
var validCloudProviderSet = map[string]struct{}{
|
|||
|
|
CLOUD_PROVIDER_TENCENT: {},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type DevcontainerType struct {
|
|||
|
|
Enabled bool
|
|||
|
|
Host string
|
|||
|
|
Agent string
|
|||
|
|
Namespace string
|
|||
|
|
TimeoutSeconds int64
|
|||
|
|
|
|||
|
|
DefaultGitBranchName string
|
|||
|
|
DefaultDevcontainerImageName string
|
|||
|
|
DockerHost string
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type SSHKeyPairType struct {
|
|||
|
|
KeySize int
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type CloudType struct {
|
|||
|
|
Enabled bool
|
|||
|
|
Provider string
|
|||
|
|
Tencent CloudProviderTencentType // 移除 ini 标签，通过代码处理
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type CloudProviderTencentType struct {
|
|||
|
|
Endpoint string
|
|||
|
|
Region string
|
|||
|
|
NatGatewayId string
|
|||
|
|
PublicIpAddress string
|
|||
|
|
PrivateIpAddress string
|
|||
|
|
IpProtocol string
|
|||
|
|
SecretId string
|
|||
|
|
SecretKey string
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var Devcontainer = DevcontainerType{
|
|||
|
|
Enabled: false,
|
|||
|
|
Namespace: "default",
|
|||
|
|
TimeoutSeconds: 900, // Max wait time for DevContainer to be ready (blocking), default is 15 minutes, can be overridden by app.ini
|
|||
|
|
|
|||
|
|
DefaultGitBranchName: "main", // Default branch name for .devcontainer/devcontainer.json
|
|||
|
|
|
|||
|
|
DefaultDevcontainerImageName: "devstar.cn/public/base-ssh-devcontainer:ubuntu-20.04-20241014", // Default image if not specified
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var SSHKeypair = SSHKeyPairType{
|
|||
|
|
KeySize: 2048, // Size of the SSH key
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var Cloud = CloudType{
|
|||
|
|
Enabled: false, // Cloud feature toggle
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// validateDevcontainerSettings 检查从 ini 配置文件中读取 DevStar DevContainer 配置信息，若数据无效，则自动禁用 DevContainer
|
|||
|
|
func validateDevcontainerSettings() {
|
|||
|
|
|
|||
|
|
// 检查 Host 是否为空，若为空，则自动将 DevContainer 设置为禁用
|
|||
|
|
if len(Devcontainer.Host) == 0 {
|
|||
|
|
log.Warn("INVALID config 'host' for DevStar DevContainer")
|
|||
|
|
Devcontainer.Enabled = false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查用户输入的 DevContainer Agent 是否存在支持列表，若不支持，则将 DevContainer 设置为禁用
|
|||
|
|
if !isValidAgent(Devcontainer.Agent) {
|
|||
|
|
log.Warn("Invalid config 'agent' for DevStar DevContainer")
|
|||
|
|
Devcontainer.Enabled = false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查默认分支名称设置
|
|||
|
|
if len(Devcontainer.DefaultGitBranchName) == 0 {
|
|||
|
|
log.Warn("INVALID config 'DefaultGitBranchName' for DevStar DevContainer")
|
|||
|
|
Devcontainer.Enabled = false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查默认 DevContainer Image
|
|||
|
|
if len(Devcontainer.DefaultDevcontainerImageName) == 0 {
|
|||
|
|
log.Warn("INVALID config 'DefaultGitBranchNameDefaultDevcontainerImageName' for DevStar DevContainer")
|
|||
|
|
Devcontainer.Enabled = false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if Devcontainer.Enabled == false {
|
|||
|
|
log.Warn("DevStar DevContainer Service Disabled")
|
|||
|
|
} else {
|
|||
|
|
log.Info("DevStar DevContainer Service Enabled")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// validateSSHKeyPairSettings 检查从 ini 配置文件中读取 DevStar SSH Key Pair 配置信息
|
|||
|
|
func validateSSHKeyPairSettings() {
|
|||
|
|
if SSHKeypair.KeySize < 1024 {
|
|||
|
|
SSHKeypair.KeySize = 1024
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// validateDevcontainerCloudSettings 检查从 ini 配置文件中读取 DevStar Cloud 配置信息
|
|||
|
|
func validateDevcontainerCloudSettings() {
|
|||
|
|
switch Cloud.Provider {
|
|||
|
|
case CLOUD_PROVIDER_TENCENT:
|
|||
|
|
// 腾讯云配置检查
|
|||
|
|
|
|||
|
|
if len(Cloud.Tencent.NatGatewayId) < 4 {
|
|||
|
|
log.Warn("INVALID NAT Gateway ID '%v' for DevStar Cloud Provider Tencent", Cloud.Tencent.NatGatewayId)
|
|||
|
|
Cloud.Enabled = false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if Cloud.Tencent.IpProtocol != "TCP" && Cloud.Tencent.IpProtocol != "UDP" && Cloud.Tencent.IpProtocol != "tcp" && Cloud.Tencent.IpProtocol != "udp" {
|
|||
|
|
log.Warn("INVALID IP Protocol '%v' for DevStar Cloud Provider Tencent", Cloud.Tencent.IpProtocol)
|
|||
|
|
Cloud.Enabled = false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if len(Cloud.Tencent.Region) < 3 || len(Cloud.Tencent.Endpoint) == 0 {
|
|||
|
|
log.Warn("INVALID (Region, Endpoint) pair ('%v', '%v') for DevStar Cloud Provider Tencent",
|
|||
|
|
Cloud.Tencent.Region, Cloud.Tencent.Endpoint)
|
|||
|
|
Cloud.Enabled = false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if len(Cloud.Tencent.PrivateIpAddress) == 0 || len(Cloud.Tencent.PublicIpAddress) == 0 {
|
|||
|
|
log.Warn("INVALID (PublicIpAddress, PrivateIpAddress) pair ('%v', '%v') for DevStar Cloud Provider Tencent",
|
|||
|
|
Cloud.Tencent.PublicIpAddress, Cloud.Tencent.PrivateIpAddress)
|
|||
|
|
Cloud.Enabled = false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if len(Cloud.Tencent.SecretId) == 0 || len(Cloud.Tencent.SecretKey) == 0 {
|
|||
|
|
log.Warn("INVALID (SecretId, SecretKey) pair for DevStar Cloud Provider Tencent")
|
|||
|
|
Cloud.Enabled = false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
default:
|
|||
|
|
// 无效 Cloud Provider 名称
|
|||
|
|
log.Warn("INVALID config '%v' for DevStar Cloud", Cloud.Provider)
|
|||
|
|
Cloud.Enabled = false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if Cloud.Enabled == false {
|
|||
|
|
log.Warn("DevStar Cloud Provider Service Disabled")
|
|||
|
|
} else {
|
|||
|
|
log.Info("DevStar Cloud Provider '%v' Enabled", Cloud.Provider)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 修改 loadDevcontainerFrom 函数以支持新旧配置节
|
|||
|
|
func loadDevcontainerFrom(rootCfg ConfigProvider) {
|
|||
|
|
// 检查是否存在新的配置节
|
|||
|
|
hasNewConfig := true
|
|||
|
|
if _, err := rootCfg.GetSection("devcontainer"); err != nil {
|
|||
|
|
hasNewConfig = false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查是否存在旧的配置节
|
|||
|
|
hasOldConfig := true
|
|||
|
|
if _, err := rootCfg.GetSection("devstar.devcontainer"); err != nil {
|
|||
|
|
hasOldConfig = false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 根据存在的配置节处理
|
|||
|
|
if hasNewConfig {
|
|||
|
|
// 新配置节存在，直接使用
|
|||
|
|
mustMapSetting(rootCfg, "devcontainer", &Devcontainer)
|
|||
|
|
log.Info("从 [devcontainer] 节加载配置")
|
|||
|
|
} else if hasOldConfig {
|
|||
|
|
// 只有旧配置节存在，直接从旧配置节加载
|
|||
|
|
mustMapSetting(rootCfg, "devstar.devcontainer", &Devcontainer)
|
|||
|
|
log.Info("从 [devstar.devcontainer] 节加载配置")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 进行配置验证
|
|||
|
|
validateDevcontainerSettings()
|
|||
|
|
|
|||
|
|
// 加载其他配置
|
|||
|
|
mustMapSetting(rootCfg, "ssh_key_pair", &SSHKeypair)
|
|||
|
|
validateSSHKeyPairSettings()
|
|||
|
|
|
|||
|
|
if Devcontainer.Agent == "k8s" || Devcontainer.Agent == KUBERNETES {
|
|||
|
|
// 调用新的云配置加载函数
|
|||
|
|
loadCloudConfigWithFallback(rootCfg)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 打印最终使用的命名空间
|
|||
|
|
log.Info("DevContainer 将在命名空间 '%s' 中创建", Devcontainer.Namespace)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 新增: 处理云配置加载的函数，支持新旧两种配置节
|
|||
|
|
func loadCloudConfigWithFallback(rootCfg ConfigProvider) {
|
|||
|
|
// 1. 先尝试加载主配置节
|
|||
|
|
hasDevcontainerCloud := true
|
|||
|
|
if _, err := rootCfg.GetSection("devcontainer.cloud"); err != nil {
|
|||
|
|
hasDevcontainerCloud = false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
hasDevstarCloud := true
|
|||
|
|
if _, err := rootCfg.GetSection("devstar.cloud"); err != nil {
|
|||
|
|
hasDevstarCloud = false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 2. 优先使用新配置节，不存在则使用旧配置节
|
|||
|
|
var cloudSectionName string
|
|||
|
|
if hasDevcontainerCloud {
|
|||
|
|
cloudSectionName = "devcontainer.cloud"
|
|||
|
|
log.Info("从 [devcontainer.cloud] 节加载云配置")
|
|||
|
|
} else if hasDevstarCloud {
|
|||
|
|
cloudSectionName = "devstar.cloud"
|
|||
|
|
log.Info("从 [devstar.cloud] 节加载云配置")
|
|||
|
|
} else {
|
|||
|
|
log.Warn("未找到云配置节，Cloud 功能将被禁用")
|
|||
|
|
Cloud.Enabled = false
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 3. 加载基本云配置
|
|||
|
|
if err := rootCfg.Section(cloudSectionName).MapTo(&Cloud); err != nil {
|
|||
|
|
log.Error("加载云配置时出错: %v", err)
|
|||
|
|
Cloud.Enabled = false
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 4. 根据选择的配置节路径，决定腾讯云配置节路径
|
|||
|
|
var tencentSectionName string
|
|||
|
|
if cloudSectionName == "devcontainer.cloud" {
|
|||
|
|
tencentSectionName = "devcontainer.cloud.tencent"
|
|||
|
|
} else {
|
|||
|
|
tencentSectionName = "devstar.cloud.tencent"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 5. 检查腾讯云配置节是否存在
|
|||
|
|
if _, err := rootCfg.GetSection(tencentSectionName); err != nil {
|
|||
|
|
log.Warn("未找到腾讯云配置节 [%s]", tencentSectionName)
|
|||
|
|
if Cloud.Provider == CLOUD_PROVIDER_TENCENT {
|
|||
|
|
log.Error("虽然指定使用腾讯云，但未找到对应配置，Cloud 功能将被禁用")
|
|||
|
|
Cloud.Enabled = false
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 6. 加载腾讯云配置
|
|||
|
|
if Cloud.Provider == CLOUD_PROVIDER_TENCENT {
|
|||
|
|
log.Info("从 [%s] 节加载腾讯云配置", tencentSectionName)
|
|||
|
|
if err := rootCfg.Section(tencentSectionName).MapTo(&Cloud.Tencent); err != nil {
|
|||
|
|
log.Error("加载腾讯云配置时出错: %v", err)
|
|||
|
|
Cloud.Enabled = false
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 7. 验证云配置
|
|||
|
|
validateDevcontainerCloudSettings()
|
|||
|
|
}
|