feat: implement hgctl agent module (#3267)

2026-06-08 20:27:31 +08:00 · 2025-12-26 13:47:32 +08:00
parent e7e3ab5ff6
commit 17e80b30fe
31 changed files with 5858 additions and 652 deletions
--- a/hgctl/pkg/agent/config.go
+++ b/hgctl/pkg/agent/config.go
@@ -0,0 +1,129 @@
+// Copyright (c) 2025 Alibaba Group Holding Ltd.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package agent
+
+import (
+	"fmt"
+	"log"
+	"os"
+
+	"github.com/mitchellh/go-homedir"
+	"github.com/spf13/viper"
+)
+
+type CoreType string
+
+const (
+	CORE_CLAUDE   CoreType = "claude"
+	CORE_QODERCLI CoreType = "qodercli"
+)
+
+const (
+	// AgentBinaryName  = "claude"
+	// BinaryVersion    = "0.1.0"
+	// DevVersion       = "dev"
+	// NodeLeastVersion = 18
+	// AgentInstallCmd  = "npm install -g @anthropic-ai/claude-code"
+	// AgentReleasePage = "https://docs.claude.com/en/docs/claude-code/setup"
+
+	HGCTL_AGENT_CORE         = "hgctl-agent-core"
+	AGENT_MODEL_PROVIDER     = "agent-model-provider"
+	AGENT_CHAT_MODEL         = "agent-chat-model"
+	HIGRESS_CONSOLE_URL      = "higress-console-url"
+	HIGRESS_CONSOLE_USER     = "higress-console-user"
+	HIGRESS_CONSOLE_PASSWORD = "higress-console-password"
+	HIGRESS_GATEWAY_URL      = "higress-gateway-url"
+
+	HIMARKET_ADMIN_URL         = "himarket-admin-url"
+	HIMARKET_ADMIN_USER        = "himarket-admin-user"
+	HIMARKET_ADMIN_PASSWORD    = "himarket-admin-password"
+	HIMARKET_TARGET_HIGRESS_ID = "himarket-target-higress-id"
+
+	HIMARKET_DEVELOPER_URL      = "himarket-developer-url"
+	HIMARKET_DEVELOPER_USER     = "himarket-developer-user"
+	HIMARKET_DEVELOPER_PASSWORD = "himarket-developer-password"
+
+	// --- AgentRun ---
+	AGENTRUN_MODEL_NAME             = "agentrun-model-name"
+	AGENTRUN_SANDBOX_NAME           = "agentrun-sandbox-name"
+	ALIBABA_CLOUD_ACCESS_KEY_ID     = "alibaba-cloud-access-key-id"
+	ALIBABA_CLOUD_ACCESS_KEY_SECRET = "alibaba-cloud-access-key-secret"
+	ALIBABA_CLOUD_SECURITY_TOK      = "alibaba-cloud-security-tok"
+	AGENTRUN_ACCOUNT_ID             = "agentrun-account-id"
+	AGENTRUN_REGION                 = "agentrun-region"
+	AGENTRUN_SDK_DEB                = "agentrun-sdk-deb"
+)
+
+var GlobalConfig HgctlAgentConfig
+
+type HgctlAgentConfig struct {
+	AgenticCore        CoreType `mapstructure:"hgctl-agent-core"`
+	AgentChatModel     string   `mapstructure:"agent-chat-model"`
+	AgentModelProvider string   `mapstructure:"agent-model-provider"`
+
+	// Higress Console credentials
+	HigressConsoleURL      string `mapstructure:"higress-console-url"`
+	HigressConsoleUser     string `mapstructure:"higress-console-user"`
+	HigressConsolePassword string `mapstructure:"higress-console-password"`
+	HigressGatewayURL      string `mapstructure:"higress-gateway-url"`
+	// Himarket Admin credentials
+	HimarketAdminURL        string `mapstructure:"himarket-admin-url"`
+	HimarketAdminUser       string `mapstructure:"himarket-admin-user"`
+	HimarketAdminPassword   string `mapstructure:"himarket-admin-password"`
+	HimarketTargetHigressID string `mapstructure:"himarket-target-higress-id"`
+
+	// Himarket Developer credentials
+	HimarketDeveloperURL      string `mapstructure:"himarket-developer-url"`
+	HimarketDeveloperUser     string `mapstructure:"himarket-developer-user"`
+	HimarketDeveloperPassword string `mapstructure:"himarket-developer-password"`
+
+	// AgentRun Configuration
+	AgentRunModelName           string `mapstructure:"agentrun-model-name"`
+	AgentRunSandboxName         string `mapstructure:"agentrun-sandbox-name"`
+	AlibabaCloudAccessKeyID     string `mapstructure:"alibaba-cloud-access-key-id"`
+	AlibabaCloudAccessKeySecret string `mapstructure:"alibaba-cloud-access-key-secret"`
+	AlibabaCloudSecurityTok     string `mapstructure:"alibaba-cloud-security-tok"`
+	AgentRunAccountID           string `mapstructure:"agentrun-account-id"`
+	AgentRunRegion              string `mapstructure:"agentrun-region"`
+}
+
+func InitConfig() {
+	viper.SetConfigName(".hgctl")
+	viper.SetConfigType("json")
+
+	home, err := homedir.Dir()
+	if err != nil {
+		log.Fatalf("Error finding home directory: %v", err)
+	}
+
+	viper.AddConfigPath(home)
+
+	if err := viper.ReadInConfig(); err != nil {
+		if _, ok := err.(viper.ConfigFileNotFoundError); !ok {
+			fmt.Fprintf(os.Stderr, "Fatal error reading config file: %v\n", err)
+		}
+	}
+
+	// Unmarshal into the GlobalConfig variable
+	_ = viper.Unmarshal(&GlobalConfig)
+
+	// Validate supported AgentCore currently
+	switch viper.GetString(HGCTL_AGENT_CORE) {
+	case string(CORE_CLAUDE), string(CORE_QODERCLI):
+		return
+	default:
+		viper.SetDefault(HGCTL_AGENT_CORE, string(CORE_QODERCLI))
+	}
+}