feat: [CCM-24080]: Add support for commitment coverage in CCM (#34)

* Address review comments

* Add support for commitment coverage in CCM

Author: sandyydk

README.md

@@ -60,6 +60,7 @@ Toolset Name: `cloudcostmanagement`
 - `list_ccm_cost_categories`:  List all cost categories names for a specified account.
 - `list_ccm_cost_categories_detail`:  List all cost categories details for a specified account.
 - `get_ccm_cost_category`:  Retrieve a cost category detail by Id  for a specified account.
+- `get_ccm_commitment_coverage`: Get commitment coverage information for an account in Harness Cloud Cost Management
 
 #### Chaos Engineering Toolset
 

client/cloudcostmanagement.go

@@ -4,17 +4,22 @@ import (
 	"context"
 	"fmt"
 	"log/slog"
+	"time"
 
 	"github.com/harness/harness-mcp/client/dto"
 	"github.com/harness/harness-mcp/pkg/utils"
 )
 
 const (
-	ccmBasePath                   = "ccm/api"
-	ccmGetOverviewPath            = ccmBasePath + "/overview?accountIdentifier=%s&startTime=%d&endTime=%d&groupBy=%s"
-	ccmCostCategoryListPath       = ccmBasePath + "/business-mapping/filter-panel?accountIdentifier=%s"
-	ccmCostCategoryDetailListPath = ccmBasePath + "/business-mapping?accountIdentifier=%s"    // This endpoint lists cost categories
-	ccmGetCostCategoryPath        = ccmBasePath + "/business-mapping/%s?accountIdentifier=%s" // This endpoint lists cost categories
+	ccmBasePath                      = "ccm/api"
+	ccmCommitmentBasePath            = "/lw/co/api"
+	ccmGetOverviewPath               = ccmBasePath + "/overview?accountIdentifier=%s&startTime=%d&endTime=%d&groupBy=%s"
+	ccmCostCategoryListPath          = ccmBasePath + "/business-mapping/filter-panel?accountIdentifier=%s"
+	ccmCostCategoryDetailListPath    = ccmBasePath + "/business-mapping?accountIdentifier=%s"    // This endpoint lists cost categories
+	ccmGetCostCategoryPath           = ccmBasePath + "/business-mapping/%s?accountIdentifier=%s" // This endpoint lists cost categories
+	ccmCommitmentCoverageDetailsPath = ccmCommitmentBasePath + "/accounts/%s/v1/detail/compute_coverage?accountIdentifier=%s"
+
+	ccmCommitmentComputeService string = "Amazon Elastic Compute Cloud - Compute"
 )
 
 type CloudCostManagementService struct {
@@ -131,3 +136,50 @@ func setCCMPaginationDefault(opts *dto.CCMPaginationOptions) {
 		opts.Limit = safeMaxPageSize
 	}
 }
+
+func (r *CloudCostManagementService) GetComputeCoverage(ctx context.Context, scope dto.Scope, opts *dto.CCMCommitmentOptions) (*dto.CCMCommitmentBaseResponse, error) {
+	path := fmt.Sprintf(ccmCommitmentCoverageDetailsPath, scope.AccountID, scope.AccountID)
+	params := make(map[string]string)
+	addScope(scope, params)
+
+	// Handle nil options by creating default options
+	if opts == nil {
+		opts = &dto.CCMCommitmentOptions{}
+	}
+
+	if opts.StartDate != nil && *opts.StartDate != "" {
+		params["start_date"] = *opts.StartDate
+	} else {
+		// Default to last 30 days
+		params["start_date"] = utils.FormatUnixToMMDDYYYY(time.Now().AddDate(0, 0, -30).Unix())
+	}
+	if opts.EndDate != nil && *opts.EndDate != "" {
+		params["end_date"] = *opts.EndDate
+	} else {
+		// Default to last 30 days
+		params["end_date"] = utils.CurrentMMDDYYYY()
+	}
+
+	var requestPayload = dto.CCMCommitmentAPIFilter{
+		Service: ccmCommitmentComputeService, // Default value
+
+	}
+
+	if opts.Service != nil && *opts.Service != "" {
+		requestPayload.Service = *opts.Service
+	}
+
+	if len(opts.CloudAccountIDs) > 0 {
+		requestPayload.CloudAccounts = opts.CloudAccountIDs
+	}
+
+	// Temporary slice to hold the strings
+	coverageRespone := new(dto.CCMCommitmentBaseResponse)
+
+	err := r.Client.Post(ctx, path, params, requestPayload, coverageRespone)
+	if err != nil {
+		return nil, fmt.Errorf("failed to get cloud cost managment compute coverage with path %s: %w", path, err)
+	}
+
+	return coverageRespone, nil
+}

client/dto/cloudcostmanagement.go

@@ -78,6 +78,14 @@ type CCMListCostCategoriesOptions struct {
 	SearchTerm        string `json:"search,omitempty"`
 }
 
+type CCMCommitmentOptions struct {
+	AccountIdentifier *string  `json:"accountIdentifier,omitempty"`
+	CloudAccountIDs   []string `json:"cloudAccountId,omitempty"`
+	Service           *string  `json:"service,omitempty"`
+	StartDate         *string  `json:"startDate,omitempty"`
+	EndDate           *string  `json:"endDate,omitempty"`
+}
+
 // CcmCostCategoryList represents a list of cost categories in CCM
 type CCMCostCategoryList struct {
 	CCMBaseResponse
@@ -206,3 +214,15 @@ type CCMGetCostCategoryOptions struct {
 	AccountIdentifier string `json:"accountIdentifier,omitempty"`
 	CostCategoryId    string `json:"id,omitempty"`
 }
+
+type CCMCommitmentBaseResponse struct {
+	Ts       int64    `json:"ts"`
+	Success  bool     `json:"success"`
+	Errors   []string `json:"errors"`
+	Response any      `json:"response"`
+}
+
+type CCMCommitmentAPIFilter struct {
+	CloudAccounts []string `json:"cloud_account_ids,omitempty"`
+	Service       string   `json:"service,omitempty"`
+}

pkg/harness/cloudcostmanagement.go

@@ -284,3 +284,89 @@ func getAccountID(config *config.Config, request mcp.CallToolRequest) (string, e
 	}
 	return "", fmt.Errorf("Account ID is required")
 }
+
+func FetchCommitmentCoverageTool(config *config.Config, client *client.CloudCostManagementService) (tool mcp.Tool, handler server.ToolHandlerFunc) {
+	return mcp.NewTool("get_ccm_commitment_coverage",
+			mcp.WithDescription("Get commitment coverage information for an account in Harness Cloud Cost Management"),
+			mcp.WithString("start_date",
+				mcp.Required(),
+				mcp.Description("Start date to filter commitment coverage"),
+			),
+			mcp.WithString("end_date",
+				mcp.Required(),
+				mcp.Description("End date to filter commitment coverage"),
+			),
+			mcp.WithString("service",
+				mcp.Description("Optional service to filter commitment coverage"),
+			),
+			mcp.WithArray("cloud_account_ids",
+				mcp.Description("Optional cloud account IDs to filter commitment coverage"),
+				mcp.Items(map[string]any{
+					"type": "string",
+				}),
+			),
+			WithScope(config, false),
+		),
+		func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+			accountId, err := getAccountID(config, request)
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+
+			params := &dto.CCMCommitmentOptions{}
+			params.AccountIdentifier = &accountId
+
+			// Handle service parameter
+			service, ok, err := OptionalParamOK[string](request, "service")
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+			if ok && service != "" {
+				params.Service = &service
+			}
+
+			// Handle cloud account IDs parameter
+			cloudAccountIDs, ok, err := OptionalParamOK[[]string](request, "cloud_account_ids")
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+			if ok && len(cloudAccountIDs) > 0 {
+				params.CloudAccountIDs = cloudAccountIDs
+			}
+
+			// Handle start date parameter
+			startDate, ok, err := OptionalParamOK[string](request, "start_date")
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+			if ok && startDate != "" {
+				params.StartDate = &startDate
+			}
+
+			// Handle end date parameter
+			endDate, ok, err := OptionalParamOK[string](request, "end_date")
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+			if ok && endDate != "" {
+				params.EndDate = &endDate
+			}
+
+			scope, err := fetchScope(config, request, false)
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+
+			data, err := client.GetComputeCoverage(ctx, scope, params)
+			if err != nil {
+				return nil, fmt.Errorf("failed to get commitment coverage: %w", err)
+			}
+
+			r, err := json.Marshal(data)
+			if err != nil {
+				return nil, fmt.Errorf("failed to marshal commitment coverage: %w", err)
+			}
+
+			return mcp.NewToolResultText(string(r)), nil
+		}
+}

pkg/harness/tools.go

@@ -479,6 +479,7 @@ func registerCloudCostManagement(config *config.Config, tsg *toolsets.ToolsetGro
 			toolsets.NewServerTool(ListCcmCostCategoriesTool(config, ccmClient)),
 			toolsets.NewServerTool(ListCcmCostCategoriesDetailTool(config, ccmClient)),
 			toolsets.NewServerTool(GetCcmCostCategoryTool(config, ccmClient)),
+			toolsets.NewServerTool(FetchCommitmentCoverageTool(config, ccmClient)),
 		)
 
 	// Add toolset to the group