add create repo support

Author: vistaarjuneja

client/dto/pullrequest.go

@@ -83,6 +83,15 @@ type PullRequestStats struct {
 	UnresolvedCount int `json:"unresolved_count,omitempty"`
 }
 
+// CreatePullRequest represents the request body for creating a new pull request
+type CreatePullRequest struct {
+	Title        string `json:"title"`
+	Description  string `json:"description,omitempty"`
+	SourceBranch string `json:"source_branch"`
+	TargetBranch string `json:"target_branch,omitempty"`
+	IsDraft      bool   `json:"is_draft,omitempty"`
+}
+
 // PullRequestOptions represents the options for listing pull requests
 type PullRequestOptions struct {
 	State         []string `json:"state,omitempty"`

client/pullrequest.go

@@ -9,9 +9,10 @@ import (
 )
 
 const (
-	pullRequestBasePath = "code/api/v1/repos"
-	pullRequestGetPath  = pullRequestBasePath + "/%s/pullreq/%d"
-	pullRequestListPath = pullRequestBasePath + "/%s/pullreq"
+	pullRequestBasePath   = "code/api/v1/repos"
+	pullRequestGetPath    = pullRequestBasePath + "/%s/pullreq/%d"
+	pullRequestListPath   = pullRequestBasePath + "/%s/pullreq"
+	pullRequestCreatePath = pullRequestBasePath + "/%s/pullreq"
 )
 
 type PullRequestService struct {
@@ -120,3 +121,22 @@ func (p *PullRequestService) List(ctx context.Context, scope dto.Scope, repoID s
 
 	return prs, nil
 }
+
+// Create creates a new pull request in the specified repository
+func (p *PullRequestService) Create(ctx context.Context, scope dto.Scope, repoID string, createPR *dto.CreatePullRequest) (*dto.PullRequest, error) {
+	path := fmt.Sprintf(pullRequestCreatePath, repoID)
+	params := make(map[string]string)
+	addScope(scope, params)
+
+	if createPR == nil {
+		createPR = &dto.CreatePullRequest{}
+	}
+
+	pr := new(dto.PullRequest)
+	err := p.client.Post(ctx, path, params, createPR, pr)
+	if err != nil {
+		return nil, fmt.Errorf("failed to create pull request: %w", err)
+	}
+
+	return pr, nil
+}

pkg/harness/pullreq.go

@@ -214,3 +214,90 @@ func splitAndTrim(s, sep string) []string {
 
 	return result
 }
+
+// CreatePullRequestTool creates a tool for creating a new pull request
+func CreatePullRequestTool(config *config.Config, client *client.Client) (tool mcp.Tool, handler server.ToolHandlerFunc) {
+	return mcp.NewTool("create_pull_request",
+			mcp.WithDescription("Create a new pull request in a Harness repository."),
+			mcp.WithString("repo_identifier",
+				mcp.Required(),
+				mcp.Description("The identifier of the repository"),
+			),
+			mcp.WithString("title",
+				mcp.Required(),
+				mcp.Description("The title of the pull request"),
+			),
+			mcp.WithString("description",
+				mcp.Description("The description of the pull request"),
+			),
+			mcp.WithString("source_branch",
+				mcp.Required(),
+				mcp.Description("The source branch for the pull request"),
+			),
+			mcp.WithString("target_branch",
+				mcp.Description("The target branch for the pull request"),
+				mcp.DefaultString("main"),
+			),
+			mcp.WithBoolean("is_draft",
+				mcp.Description("Whether the pull request should be created as a draft"),
+				mcp.DefaultBool(false),
+			),
+			WithScope(config, false),
+		),
+		func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+			repoIdentifier, err := requiredParam[string](request, "repo_identifier")
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+
+			title, err := requiredParam[string](request, "title")
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+
+			description, err := OptionalParam[string](request, "description")
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+
+			sourceBranch, err := requiredParam[string](request, "source_branch")
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+
+			isDraft, err := OptionalParam[bool](request, "is_draft")
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+
+			targetBranch, err := OptionalParam[string](request, "target_branch")
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+
+			scope, err := fetchScope(config, request, false)
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+
+			createRequest := &dto.CreatePullRequest{
+				Title:        title,
+				SourceBranch: sourceBranch,
+				TargetBranch: targetBranch,
+				IsDraft:      isDraft,
+				Description:  description,
+			}
+
+			data, err := client.PullRequests.Create(ctx, scope, repoIdentifier, createRequest)
+			if err != nil {
+				return nil, fmt.Errorf("failed to create pull request: %w", err)
+			}
+
+			r, err := json.Marshal(data)
+			if err != nil {
+				return nil, fmt.Errorf("failed to marshal pull request: %w", err)
+			}
+
+			return mcp.NewToolResultText(string(r)), nil
+		}
+}

pkg/harness/tools.go

@@ -32,6 +32,9 @@ func InitToolsets(client *client.Client, config *config.Config) (*toolsets.Tools
 		AddReadTools(
 			toolsets.NewServerTool(GetPullRequestTool(config, client)),
 			toolsets.NewServerTool(ListPullRequestsTool(config, client)),
+		).
+		AddWriteTools(
+			toolsets.NewServerTool(CreatePullRequestTool(config, client)),
 		)
 
 	// Create the repositories toolset