ACP / MCP

2026-06-12 11:44:19 +08:00
parent 2a9661a6ef
commit 4ea4adfd8e
47 changed files with 2558 additions and 240 deletions
@@ -0,0 +1,105 @@
+// Code generated by sqlc. DO NOT EDIT.
+// versions:
+//   sqlc v1.31.1
+// source: agent_kind_config_files.sql
+
+package acpsqlc
+
+import (
+	"context"
+
+	"github.com/jackc/pgx/v5/pgtype"
+)
+
+const deleteAgentKindConfigFile = `-- name: DeleteAgentKindConfigFile :execrows
+DELETE FROM acp_agent_kind_config_files
+WHERE agent_kind_id = $1 AND rel_path = $2
+`
+
+type DeleteAgentKindConfigFileParams struct {
+	AgentKindID pgtype.UUID `json:"agent_kind_id"`
+	RelPath     string      `json:"rel_path"`
+}
+
+func (q *Queries) DeleteAgentKindConfigFile(ctx context.Context, arg DeleteAgentKindConfigFileParams) (int64, error) {
+	result, err := q.db.Exec(ctx, deleteAgentKindConfigFile, arg.AgentKindID, arg.RelPath)
+	if err != nil {
+		return 0, err
+	}
+	return result.RowsAffected(), nil
+}
+
+const listAgentKindConfigFiles = `-- name: ListAgentKindConfigFiles :many
+SELECT id, agent_kind_id, rel_path, encrypted_content, updated_by, created_at, updated_at
+FROM acp_agent_kind_config_files
+WHERE agent_kind_id = $1
+ORDER BY rel_path ASC
+`
+
+func (q *Queries) ListAgentKindConfigFiles(ctx context.Context, agentKindID pgtype.UUID) ([]AcpAgentKindConfigFile, error) {
+	rows, err := q.db.Query(ctx, listAgentKindConfigFiles, agentKindID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	var items []AcpAgentKindConfigFile
+	for rows.Next() {
+		var i AcpAgentKindConfigFile
+		if err := rows.Scan(
+			&i.ID,
+			&i.AgentKindID,
+			&i.RelPath,
+			&i.EncryptedContent,
+			&i.UpdatedBy,
+			&i.CreatedAt,
+			&i.UpdatedAt,
+		); err != nil {
+			return nil, err
+		}
+		items = append(items, i)
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+	return items, nil
+}
+
+const upsertAgentKindConfigFile = `-- name: UpsertAgentKindConfigFile :one
+INSERT INTO acp_agent_kind_config_files (
+  id, agent_kind_id, rel_path, encrypted_content, updated_by
+) VALUES ($1, $2, $3, $4, $5)
+ON CONFLICT (agent_kind_id, rel_path) DO UPDATE
+SET encrypted_content = EXCLUDED.encrypted_content,
+    updated_by        = EXCLUDED.updated_by,
+    updated_at        = now()
+RETURNING id, agent_kind_id, rel_path, encrypted_content, updated_by, created_at, updated_at
+`
+
+type UpsertAgentKindConfigFileParams struct {
+	ID               pgtype.UUID `json:"id"`
+	AgentKindID      pgtype.UUID `json:"agent_kind_id"`
+	RelPath          string      `json:"rel_path"`
+	EncryptedContent []byte      `json:"encrypted_content"`
+	UpdatedBy        pgtype.UUID `json:"updated_by"`
+}
+
+func (q *Queries) UpsertAgentKindConfigFile(ctx context.Context, arg UpsertAgentKindConfigFileParams) (AcpAgentKindConfigFile, error) {
+	row := q.db.QueryRow(ctx, upsertAgentKindConfigFile,
+		arg.ID,
+		arg.AgentKindID,
+		arg.RelPath,
+		arg.EncryptedContent,
+		arg.UpdatedBy,
+	)
+	var i AcpAgentKindConfigFile
+	err := row.Scan(
+		&i.ID,
+		&i.AgentKindID,
+		&i.RelPath,
+		&i.EncryptedContent,
+		&i.UpdatedBy,
+		&i.CreatedAt,
+		&i.UpdatedAt,
+	)
+	return i, err
+}
@@ -24,23 +24,25 @@ func (q *Queries) CountAgentKindUsage(ctx context.Context, agentKindID pgtype.UU

 const createAgentKind = `-- name: CreateAgentKind :one
 INSERT INTO acp_agent_kinds (
-  id, name, display_name, description, binary_path, args, encrypted_env, enabled, created_by, tool_allowlist
-) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
+  id, name, display_name, description, binary_path, args, encrypted_env, enabled, created_by, tool_allowlist, client_type, encrypted_mcp_servers
+) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12)
 RETURNING id, name, display_name, description, binary_path, args, encrypted_env,
-          enabled, created_by, created_at, updated_at, tool_allowlist
+          enabled, created_by, created_at, updated_at, tool_allowlist, client_type, encrypted_mcp_servers
 `

 type CreateAgentKindParams struct {
-	ID            pgtype.UUID `json:"id"`
-	Name          string      `json:"name"`
-	DisplayName   string      `json:"display_name"`
-	Description   string      `json:"description"`
-	BinaryPath    string      `json:"binary_path"`
-	Args          []string    `json:"args"`
-	EncryptedEnv  []byte      `json:"encrypted_env"`
-	Enabled       bool        `json:"enabled"`
-	CreatedBy     pgtype.UUID `json:"created_by"`
-	ToolAllowlist []string    `json:"tool_allowlist"`
+	ID                  pgtype.UUID `json:"id"`
+	Name                string      `json:"name"`
+	DisplayName         string      `json:"display_name"`
+	Description         string      `json:"description"`
+	BinaryPath          string      `json:"binary_path"`
+	Args                []string    `json:"args"`
+	EncryptedEnv        []byte      `json:"encrypted_env"`
+	Enabled             bool        `json:"enabled"`
+	CreatedBy           pgtype.UUID `json:"created_by"`
+	ToolAllowlist       []string    `json:"tool_allowlist"`
+	ClientType          string      `json:"client_type"`
+	EncryptedMcpServers []byte      `json:"encrypted_mcp_servers"`
 }

 func (q *Queries) CreateAgentKind(ctx context.Context, arg CreateAgentKindParams) (AcpAgentKind, error) {
@@ -55,6 +57,8 @@ func (q *Queries) CreateAgentKind(ctx context.Context, arg CreateAgentKindParams
 		arg.Enabled,
 		arg.CreatedBy,
 		arg.ToolAllowlist,
+		arg.ClientType,
+		arg.EncryptedMcpServers,
 	)
 	var i AcpAgentKind
 	err := row.Scan(
@@ -70,6 +74,8 @@ func (q *Queries) CreateAgentKind(ctx context.Context, arg CreateAgentKindParams
 		&i.CreatedAt,
 		&i.UpdatedAt,
 		&i.ToolAllowlist,
+		&i.ClientType,
+		&i.EncryptedMcpServers,
 	)
 	return i, err
 }
@@ -85,7 +91,7 @@ func (q *Queries) DeleteAgentKind(ctx context.Context, id pgtype.UUID) error {

 const getAgentKindByID = `-- name: GetAgentKindByID :one
 SELECT id, name, display_name, description, binary_path, args, encrypted_env,
-       enabled, created_by, created_at, updated_at, tool_allowlist
+       enabled, created_by, created_at, updated_at, tool_allowlist, client_type, encrypted_mcp_servers
 FROM acp_agent_kinds
 WHERE id = $1
 `
@@ -106,13 +112,15 @@ func (q *Queries) GetAgentKindByID(ctx context.Context, id pgtype.UUID) (AcpAgen
 		&i.CreatedAt,
 		&i.UpdatedAt,
 		&i.ToolAllowlist,
+		&i.ClientType,
+		&i.EncryptedMcpServers,
 	)
 	return i, err
 }

 const getAgentKindByName = `-- name: GetAgentKindByName :one
 SELECT id, name, display_name, description, binary_path, args, encrypted_env,
-       enabled, created_by, created_at, updated_at, tool_allowlist
+       enabled, created_by, created_at, updated_at, tool_allowlist, client_type, encrypted_mcp_servers
 FROM acp_agent_kinds
 WHERE name = $1
 `
@@ -133,13 +141,15 @@ func (q *Queries) GetAgentKindByName(ctx context.Context, name string) (AcpAgent
 		&i.CreatedAt,
 		&i.UpdatedAt,
 		&i.ToolAllowlist,
+		&i.ClientType,
+		&i.EncryptedMcpServers,
 	)
 	return i, err
 }

 const listAgentKinds = `-- name: ListAgentKinds :many
 SELECT id, name, display_name, description, binary_path, args, encrypted_env,
-       enabled, created_by, created_at, updated_at, tool_allowlist
+       enabled, created_by, created_at, updated_at, tool_allowlist, client_type, encrypted_mcp_servers
 FROM acp_agent_kinds
 ORDER BY created_at DESC
 `
@@ -166,6 +176,8 @@ func (q *Queries) ListAgentKinds(ctx context.Context) ([]AcpAgentKind, error) {
 			&i.CreatedAt,
 			&i.UpdatedAt,
 			&i.ToolAllowlist,
+			&i.ClientType,
+			&i.EncryptedMcpServers,
 		); err != nil {
 			return nil, err
 		}
@@ -179,7 +191,7 @@ func (q *Queries) ListAgentKinds(ctx context.Context) ([]AcpAgentKind, error) {

 const listEnabledAgentKinds = `-- name: ListEnabledAgentKinds :many
 SELECT id, name, display_name, description, binary_path, args, encrypted_env,
-       enabled, created_by, created_at, updated_at, tool_allowlist
+       enabled, created_by, created_at, updated_at, tool_allowlist, client_type, encrypted_mcp_servers
 FROM acp_agent_kinds
 WHERE enabled = TRUE
 ORDER BY name ASC
@@ -207,6 +219,8 @@ func (q *Queries) ListEnabledAgentKinds(ctx context.Context) ([]AcpAgentKind, er
 			&i.CreatedAt,
 			&i.UpdatedAt,
 			&i.ToolAllowlist,
+			&i.ClientType,
+			&i.EncryptedMcpServers,
 		); err != nil {
 			return nil, err
 		}
@@ -220,28 +234,32 @@ func (q *Queries) ListEnabledAgentKinds(ctx context.Context) ([]AcpAgentKind, er

 const updateAgentKind = `-- name: UpdateAgentKind :one
 UPDATE acp_agent_kinds
-SET display_name   = $2,
-    description    = $3,
-    binary_path    = $4,
-    args           = $5,
-    encrypted_env  = $6,
-    enabled        = $7,
-    tool_allowlist = $8,
-    updated_at     = now()
+SET display_name          = $2,
+    description           = $3,
+    binary_path           = $4,
+    args                  = $5,
+    encrypted_env         = $6,
+    enabled               = $7,
+    tool_allowlist        = $8,
+    client_type           = $9,
+    encrypted_mcp_servers = $10,
+    updated_at            = now()
 WHERE id = $1
 RETURNING id, name, display_name, description, binary_path, args, encrypted_env,
-          enabled, created_by, created_at, updated_at, tool_allowlist
+          enabled, created_by, created_at, updated_at, tool_allowlist, client_type, encrypted_mcp_servers
 `

 type UpdateAgentKindParams struct {
-	ID            pgtype.UUID `json:"id"`
-	DisplayName   string      `json:"display_name"`
-	Description   string      `json:"description"`
-	BinaryPath    string      `json:"binary_path"`
-	Args          []string    `json:"args"`
-	EncryptedEnv  []byte      `json:"encrypted_env"`
-	Enabled       bool        `json:"enabled"`
-	ToolAllowlist []string    `json:"tool_allowlist"`
+	ID                  pgtype.UUID `json:"id"`
+	DisplayName         string      `json:"display_name"`
+	Description         string      `json:"description"`
+	BinaryPath          string      `json:"binary_path"`
+	Args                []string    `json:"args"`
+	EncryptedEnv        []byte      `json:"encrypted_env"`
+	Enabled             bool        `json:"enabled"`
+	ToolAllowlist       []string    `json:"tool_allowlist"`
+	ClientType          string      `json:"client_type"`
+	EncryptedMcpServers []byte      `json:"encrypted_mcp_servers"`
 }

 func (q *Queries) UpdateAgentKind(ctx context.Context, arg UpdateAgentKindParams) (AcpAgentKind, error) {
@@ -254,6 +272,8 @@ func (q *Queries) UpdateAgentKind(ctx context.Context, arg UpdateAgentKindParams
 		arg.EncryptedEnv,
 		arg.Enabled,
 		arg.ToolAllowlist,
+		arg.ClientType,
+		arg.EncryptedMcpServers,
 	)
 	var i AcpAgentKind
 	err := row.Scan(
@@ -269,6 +289,8 @@ func (q *Queries) UpdateAgentKind(ctx context.Context, arg UpdateAgentKindParams
 		&i.CreatedAt,
 		&i.UpdatedAt,
 		&i.ToolAllowlist,
+		&i.ClientType,
+		&i.EncryptedMcpServers,
 	)
 	return i, err
 }
@@ -11,18 +11,30 @@ import (
 )

 type AcpAgentKind struct {
-	ID            pgtype.UUID        `json:"id"`
-	Name          string             `json:"name"`
-	DisplayName   string             `json:"display_name"`
-	Description   string             `json:"description"`
-	BinaryPath    string             `json:"binary_path"`
-	Args          []string           `json:"args"`
-	EncryptedEnv  []byte             `json:"encrypted_env"`
-	Enabled       bool               `json:"enabled"`
-	CreatedBy     pgtype.UUID        `json:"created_by"`
-	CreatedAt     pgtype.Timestamptz `json:"created_at"`
-	UpdatedAt     pgtype.Timestamptz `json:"updated_at"`
-	ToolAllowlist []string           `json:"tool_allowlist"`
+	ID                  pgtype.UUID        `json:"id"`
+	Name                string             `json:"name"`
+	DisplayName         string             `json:"display_name"`
+	Description         string             `json:"description"`
+	BinaryPath          string             `json:"binary_path"`
+	Args                []string           `json:"args"`
+	EncryptedEnv        []byte             `json:"encrypted_env"`
+	Enabled             bool               `json:"enabled"`
+	CreatedBy           pgtype.UUID        `json:"created_by"`
+	CreatedAt           pgtype.Timestamptz `json:"created_at"`
+	UpdatedAt           pgtype.Timestamptz `json:"updated_at"`
+	ToolAllowlist       []string           `json:"tool_allowlist"`
+	ClientType          string             `json:"client_type"`
+	EncryptedMcpServers []byte             `json:"encrypted_mcp_servers"`
+}
+
+type AcpAgentKindConfigFile struct {
+	ID               pgtype.UUID        `json:"id"`
+	AgentKindID      pgtype.UUID        `json:"agent_kind_id"`
+	RelPath          string             `json:"rel_path"`
+	EncryptedContent []byte             `json:"encrypted_content"`
+	UpdatedBy        pgtype.UUID        `json:"updated_by"`
+	CreatedAt        pgtype.Timestamptz `json:"created_at"`
+	UpdatedAt        pgtype.Timestamptz `json:"updated_at"`
 }

 type AcpEvent struct {