feat: implement hgctl agent module (#3267)

hgctl/pkg/agent/agent.go

@@ -15,32 +15,161 @@
 package agent
 
 import (
+	"fmt"
 	"io"
 
+	"github.com/alibaba/higress/hgctl/pkg/agent/services"
 	"github.com/spf13/cobra"
 	cmdutil "k8s.io/kubectl/pkg/cmd/util"
 )
 
+// API Type
+const (
+	A2A   = "a2a"
+	REST  = "restful"
+	MODEL = "model"
+)
+
 func NewAgentCmd() *cobra.Command {
 	agentCmd := &cobra.Command{
 		Use:   "agent",
-		Short: "start the interactive agent window",
+		Short: "Start the interactive agent window",
 		Run: func(cmd *cobra.Command, args []string) {
-			cmdutil.CheckErr(handleAgentInvoke(cmd.OutOrStdout()))
+			cmdutil.CheckErr(invokeAgentCore(cmd.OutOrStdout()))
 		},
 	}
 
+	agentCmd.AddCommand(createAgentCmd())
+	agentCmd.AddCommand(deployAgentCmd())
+	agentCmd.AddCommand(newAgentAddCmd())
+
 	return agentCmd
 }
 
-func handleAgentInvoke(w io.Writer) error {
-
-	return getAgent().Start()
+func invokeAgentCore(w io.Writer) error {
+	core, err := getCore()
+	if err != nil {
+		return fmt.Errorf("failed to get core: %s", err)
+	}
+	return core.Start()
 }
 
-// Sub-Agent1:
-// 1. Parse the url provided by user to MCP server configuration.
-// 2. Publish the parsed MCP Server to Higress
-func addPrequisiteSubAgent() error {
+type AgentAddArg struct {
+	HigressConsoleAuthArg
+	HimarketAdminAuthArg
+
+	name  string
+	url   string
+	typ   string
+	scope string
+
+	asProduct bool
+	noPublish bool
+}
+
+func newAgentAddCmd() *cobra.Command {
+	arg := &AgentAddArg{}
+
+	cmd := &cobra.Command{
+		Use:   "add [name] [url]",
+		Short: "add agent to local interactive window and publish it to higress (optional)",
+		Run: func(cmd *cobra.Command, args []string) {
+			arg.name = args[0]
+			arg.url = args[1]
+
+			resolveHigressConsoleAuth(&arg.HigressConsoleAuthArg)
+			resolveHimarketAdminAuth(&arg.HimarketAdminAuthArg)
+			cmdutil.CheckErr(handleAddAgent(cmd.OutOrStdout(), *arg))
+		},
+		Args: cobra.ExactArgs(2),
+	}
+
+	cmd.PersistentFlags().StringVarP(&arg.typ, "type", "t", MODEL, "Determine the agent's API type (a2a, model, restful) default is model")
+	cmd.PersistentFlags().StringVarP(&arg.scope, "scope", "s", "project", `Configuration scope (project or global)`)
+	cmd.PersistentFlags().BoolVar(&arg.noPublish, "no-publish", false, "If it's set then the agent API will not be plubished to Higress")
+	cmd.PersistentFlags().BoolVar(&arg.asProduct, "as-product", false, "If it's set then the agent API will be published to Himarket (no-publish must be false)")
+
+	addHigressConsoleAuthFlag(cmd, &arg.HigressConsoleAuthArg)
+	addHimarketAdminAuthFlag(cmd, &arg.HimarketAdminAuthArg)
+	return cmd
+}
+
+func handleAddAgent(writer io.Writer, arg AgentAddArg) error {
+	if err := validateArg(arg); err != nil {
+		return err
+	}
+
+	if !arg.noPublish {
+		if err := publishAgentAPIToHigress(arg); err != nil {
+			fmt.Printf("failed to publish agent api to higress: %s\n", err)
+			return err
+		}
+
+		fmt.Printf("Agent %s is published to Higress successfully\n", arg.name)
+
+		if arg.asProduct {
+			if err := publishAPIToHimarket(arg.typ, arg.name, arg.HimarketAdminAuthArg); err != nil {
+				fmt.Println("failed to publish it to himarket, please do it mannually")
+				return err
+			}
+			fmt.Printf("Agent %s is published to Himarket successfully\n", arg.name)
+		}
+		// TODO: pop up higress window
+	}
+
+	return nil
+}
+
+func publishAgentAPIToHigress(arg AgentAddArg) error {
+	client := services.NewHigressClient(arg.hgURL, arg.hgUser, arg.hgPassword)
+
+	switch arg.typ {
+	case A2A:
+	case MODEL:
+		// add ai service
+		body := services.BuildAIProviderServiceBody(arg.name, arg.url)
+		// Debug
+		// fmt.Printf("services: body: %v\n", body)
+		if resp, err := services.HandleAddAIProviderService(client, body); err != nil {
+			fmt.Println(string(resp))
+			return err
+		}
+
+		// add ai route
+		body = services.BuildAddAIRouteBody(arg.name, arg.url)
+		// fmt.Printf("Route body: %v\n", body)
+		if res, err := services.HandleAddAIRoute(client, body); err != nil {
+			fmt.Println(string(res))
+			return err
+		}
+
+	case REST:
+		srvName := fmt.Sprintf("agent-%s", arg.name)
+		body, targetSrvName, _, err := services.BuildServiceBodyAndSrv(srvName, arg.url)
+		if err != nil {
+			return fmt.Errorf("invalid url format: %s", err)
+		}
+
+		if resp, err := services.HandleAddServiceSource(client, body); err != nil {
+			fmt.Println(string(resp))
+			return err
+		}
+
+		if resp, err := services.HandleAddRoute(client, services.BuildAPIRouteBody(arg.name, targetSrvName)); err != nil {
+			fmt.Println(string(resp))
+			return err
+		}
+	default:
+		return fmt.Errorf("unsupported agent protocol type: %s", arg.typ)
+
+	}
+
+	return nil
+}
+
+func validateArg(arg AgentAddArg) error {
+	if !arg.noPublish {
+		return arg.HigressConsoleAuthArg.validate()
+	}
 	return nil
 }