add support for status checks

Author: vistaarjuneja

client/dto/pullrequest.go

@@ -92,6 +92,53 @@ type CreatePullRequest struct {
 	IsDraft      bool   `json:"is_draft,omitempty"`
 }
 
+// PullRequestCheckPayload represents the payload for a pull request check
+type PullRequestCheckPayload struct {
+	Data    interface{} `json:"data"`
+	Kind    string      `json:"kind"`
+	Version string      `json:"version,omitempty"`
+}
+
+// PullRequestCheckReporter represents the entity that reported a check
+type PullRequestCheckReporter struct {
+	Created     int64  `json:"created,omitempty"`
+	DisplayName string `json:"display_name,omitempty"`
+	Email       string `json:"email,omitempty"`
+	ID          int    `json:"id,omitempty"`
+	Type        string `json:"type,omitempty"`
+	UID         string `json:"uid,omitempty"`
+	Updated     int64  `json:"updated,omitempty"`
+}
+
+// PullRequestCheck represents a status check for a pull request
+type PullRequestCheck struct {
+	Created    int64                  `json:"created,omitempty"`
+	Ended      int64                  `json:"ended,omitempty"`
+	ID         int                    `json:"id,omitempty"`
+	Identifier string                 `json:"identifier,omitempty"`
+	Link       string                 `json:"link,omitempty"`
+	Metadata   interface{}            `json:"metadata"`
+	Payload    PullRequestCheckPayload `json:"payload,omitempty"`
+	ReportedBy PullRequestCheckReporter `json:"reported_by,omitempty"`
+	Started    int64                  `json:"started,omitempty"`
+	Status     string                 `json:"status,omitempty"`
+	Summary    string                 `json:"summary,omitempty"`
+	Updated    int64                  `json:"updated,omitempty"`
+}
+
+// PullRequestCheckInfo represents a check with additional information
+type PullRequestCheckInfo struct {
+	Bypassable bool             `json:"bypassable,omitempty"`
+	Check      PullRequestCheck `json:"check,omitempty"`
+	Required   bool             `json:"required,omitempty"`
+}
+
+// PullRequestChecksResponse represents the response from the checks API
+type PullRequestChecksResponse struct {
+	Checks    []PullRequestCheckInfo `json:"checks,omitempty"`
+	CommitSha string                `json:"commit_sha,omitempty"`
+}
+
 // PullRequestOptions represents the options for listing pull requests
 type PullRequestOptions struct {
 	State         []string `json:"state,omitempty"`

client/pullrequest.go

@@ -13,6 +13,7 @@ const (
 	pullRequestGetPath    = pullRequestBasePath + "/%s/pullreq/%d"
 	pullRequestListPath   = pullRequestBasePath + "/%s/pullreq"
 	pullRequestCreatePath = pullRequestBasePath + "/%s/pullreq"
+	pullRequestChecksPath = pullRequestBasePath + "/%s/pullreq/%d/checks"
 )
 
 type PullRequestService struct {
@@ -140,3 +141,18 @@ func (p *PullRequestService) Create(ctx context.Context, scope dto.Scope, repoID
 
 	return pr, nil
 }
+
+// GetChecks retrieves the status checks for a specific pull request
+func (p *PullRequestService) GetChecks(ctx context.Context, scope dto.Scope, repoID string, prNumber int) (*dto.PullRequestChecksResponse, error) {
+	path := fmt.Sprintf(pullRequestChecksPath, repoID, prNumber)
+	params := make(map[string]string)
+	addScope(scope, params)
+
+	checks := new(dto.PullRequestChecksResponse)
+	err := p.client.Get(ctx, path, params, nil, checks)
+	if err != nil {
+		return nil, fmt.Errorf("failed to get pull request checks: %w", err)
+	}
+
+	return checks, nil
+}

pkg/harness/pullreq.go

@@ -215,6 +215,51 @@ func splitAndTrim(s, sep string) []string {
 	return result
 }
 
+// GetPullRequestChecksTool creates a tool for getting pull request status checks
+func GetPullRequestChecksTool(config *config.Config, client *client.Client) (tool mcp.Tool, handler server.ToolHandlerFunc) {
+	return mcp.NewTool("get_pull_request_checks",
+			mcp.WithDescription("Get status checks for a specific pull request in a Harness repository."),
+			mcp.WithString("repo_identifier",
+				mcp.Required(),
+				mcp.Description("The identifier of the repository"),
+			),
+			mcp.WithNumber("pr_number",
+				mcp.Required(),
+				mcp.Description("The number of the pull request"),
+			),
+			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
+			}
+
+			prNumberFloat, err := requiredParam[float64](request, "pr_number")
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+			prNumber := int(prNumberFloat)
+
+			scope, err := fetchScope(config, request, false)
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+
+			data, err := client.PullRequests.GetChecks(ctx, scope, repoIdentifier, prNumber)
+			if err != nil {
+				return nil, fmt.Errorf("failed to get pull request checks: %w", err)
+			}
+
+			r, err := json.Marshal(data)
+			if err != nil {
+				return nil, fmt.Errorf("failed to marshal pull request checks: %w", err)
+			}
+
+			return mcp.NewToolResultText(string(r)), nil
+		}
+}
+
 // 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",

pkg/harness/tools.go

@@ -32,6 +32,7 @@ func InitToolsets(client *client.Client, config *config.Config) (*toolsets.Tools
 		AddReadTools(
 			toolsets.NewServerTool(GetPullRequestTool(config, client)),
 			toolsets.NewServerTool(ListPullRequestsTool(config, client)),
+			toolsets.NewServerTool(GetPullRequestChecksTool(config, client)),
 		).
 		AddWriteTools(
 			toolsets.NewServerTool(CreatePullRequestTool(config, client)),