add ACM support in external mode (#140)

* fixed last run type

* add ACM support in external mode

Author: vistaarjuneja

client/acm.go

@@ -0,0 +1,89 @@
+package client
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+
+	"github.com/harness/harness-mcp/client/dto"
+)
+
+const (
+	acmCreateTaskPath = "/api/v1/accounts/%s/tasks"
+	acmRunTaskPath    = "/api/v1/accounts/%s/executions"
+)
+
+// ACMService provides methods to interact with the Autonomous Code Maintenance API
+type ACMService struct {
+	Client *Client
+}
+
+// CreateTask creates a new autonomous code maintenance task
+func (s *ACMService) CreateTask(ctx context.Context, scope dto.Scope, params *dto.CreateACMTaskRequest) (*dto.ACMTaskResponse, error) {
+	// Build the path
+	path := fmt.Sprintf(acmCreateTaskPath, scope.AccountID)
+
+	// Set up query parameters
+	queryParams := make(map[string]string)
+
+	headers := make(map[string]string)
+	addHarnessAccountToHeaders(scope, headers)
+
+	// Make the request
+	result := new(dto.ACMTaskResponse)
+	err := s.Client.Post(ctx, path, queryParams, params, headers, result)
+	if err != nil {
+		return nil, fmt.Errorf("failed to create ACM task: %w", err)
+	}
+
+	return result, nil
+}
+
+// ListTaskExecutions lists executions of an autonomous code maintenance task
+func (s *ACMService) ListTaskExecutions(ctx context.Context, scope dto.Scope, params *dto.GetACMExecutionsRequest) (*dto.ACMExecutionsListResponse, error) {
+	// Build the path
+	path := fmt.Sprintf(acmRunTaskPath, scope.AccountID)
+
+	// Set up query parameters
+	queryParams := make(map[string]string)
+
+	headers := make(map[string]string)
+	addHarnessAccountToHeaders(scope, headers)
+
+	// Add specific query parameters for this endpoint
+	queryParams["task_id"] = params.TaskID
+	queryParams["page"] = fmt.Sprintf("%d", params.Page)
+	queryParams["limit"] = fmt.Sprintf("%d", params.Limit)
+
+	// Make the request
+	result := new(dto.ACMExecutionsListResponse)
+	err := s.Client.Get(ctx, path, queryParams, headers, result)
+	if err != nil {
+		return nil, fmt.Errorf("failed to list ACM task executions: %w", err)
+	}
+
+	return result, nil
+}
+
+// TriggerTaskExecution triggers execution of an autonomous code maintenance task
+func (s *ACMService) TriggerTaskExecution(ctx context.Context, scope dto.Scope, params *dto.TriggerACMTaskExecutionRequest) (*dto.ACMExecution, error) {
+	// Build the path
+	path := fmt.Sprintf(acmRunTaskPath, scope.AccountID)
+
+	// Set up query parameters
+	queryParams := make(map[string]string)
+
+	headers := make(map[string]string)
+	addHarnessAccountToHeaders(scope, headers)
+
+	slog.Info("triggering ACM task execution", "task_id", params.TaskID, "repository_id", params.RepositoryID, "source_branch", params.SourceBranch)
+
+	// Make the request
+	result := new(dto.ACMExecution)
+	err := s.Client.Post(ctx, path, queryParams, params, headers, result)
+	if err != nil {
+		return nil, fmt.Errorf("failed to trigger ACM task execution: %w", err)
+	}
+
+	return result, nil
+}

client/dto/acm.go

@@ -0,0 +1,75 @@
+package dto
+
+// CreateACMTaskRequest represents the request to create an autonomous code maintenance task
+type CreateACMTaskRequest struct {
+	Name         string `json:"name"`
+	Identifier   string `json:"identifier"`
+	Instructions string `json:"instructions"`
+	Path         string `json:"path"` // project and org path
+}
+
+// ACMTaskResponse represents the response from creating an autonomous code maintenance task
+type ACMTaskResponse struct {
+	ID             string        `json:"id"`
+	SpaceID        string        `json:"space_id"`
+	Identifier     string        `json:"identifier"`
+	Name           string        `json:"name"`
+	CreatedBy      TriggeredBy   `json:"created_by"`
+	Created        int64         `json:"created"`
+	Updated        int64         `json:"updated"`
+	Instructions   string        `json:"instructions"`
+	ExecutionCount int           `json:"execution_count"`
+	LastRun        *ACMExecution `json:"last_run"`
+}
+
+// TriggerACMTaskExecutionRequest represents the request to trigger execution of an autonomous code maintenance task
+type TriggerACMTaskExecutionRequest struct {
+	TaskID       string `json:"task_id"`
+	RepositoryID string `json:"repository_id"`
+	SourceBranch string `json:"source_branch"`
+}
+
+// GetACMExecutionsRequest represents the request to get executions of an autonomous code maintenance task
+type GetACMExecutionsRequest struct {
+	TaskID string `json:"task_id"`
+	Page   int    `json:"page"`
+	Limit  int    `json:"limit"`
+}
+
+// ACMExecution represents a single execution of an autonomous code maintenance task
+type ACMExecution struct {
+	ID          string      `json:"id"`
+	TaskID      string      `json:"task_id"`
+	TaskType    string      `json:"task_type"`
+	TriggeredBy TriggeredBy `json:"triggered_by"`
+	Triggered   int64       `json:"triggered"`
+	Started     int64       `json:"started"`
+	Completed   int64       `json:"completed,omitempty"`
+	Input       TaskInput   `json:"input"`
+	Output      TaskOutput  `json:"output"`
+	Status      string      `json:"status"`
+}
+
+// TriggeredBy represents the entity who triggered the execution
+type TriggeredBy struct {
+	UUID  string `json:"uuid"`
+	Name  string `json:"name"`
+	Email string `json:"email"`
+	Type  string `json:"type"`
+}
+
+// TaskInput represents the input parameters for a task execution
+type TaskInput struct {
+	Name         string `json:"name"`
+	Instructions string `json:"instructions"`
+	RepositoryID string `json:"repository_id"`
+	SourceBranch string `json:"source_branch"`
+	TargetBranch string `json:"target_branch"`
+}
+
+type TaskOutput struct {
+	// Can be expanded with specific fields as needed
+}
+
+// ACMExecutionsListResponse represents the response from listing executions of an autonomous code maintenance task
+type ACMExecutionsListResponse []ACMExecution

pkg/harness/tools.go

@@ -271,6 +271,9 @@ func initLegacyToolsets(config *config.Config, tsg *toolsets.ToolsetGroup) error
 			if err := modules.RegisterInfrastructure(config, tsg); err != nil {
 				return err
 			}
+			if err := modules.RegisterACM(config, tsg); err != nil {
+				return err
+			}
 		} else {
 			// Register specified toolsets
 			for _, toolset := range config.Toolsets {
@@ -367,6 +370,10 @@ func initLegacyToolsets(config *config.Config, tsg *toolsets.ToolsetGroup) error
 					if err := modules.RegisterInfrastructure(config, tsg); err != nil {
 						return err
 					}
+				case "acm":
+					if err := modules.RegisterACM(config, tsg); err != nil {
+						return err
+					}
 				}
 			}
 		}
@@ -424,4 +431,3 @@ func RegisterDefault(config *config.Config, tsg *toolsets.ToolsetGroup) error {
 	tsg.AddToolset(defaultToolset)
 	return nil
 }
-