feat: [FME-4267]: Integrated Cost Categories Name listing (harness#16)

* Removing commented line
* Removing unused import
* Rebase with master branch
* Adding prompt to reinforce parameters format for function Get Cost Overview from CCM
* Fixing account id issue
* Fixed account id field name
* Added List Cloud Cost Management tool
* [CCM-tools] CCM Overview
* [CCM-tools] CCM Overview

Author: javrudsky

README.md

@@ -49,7 +49,8 @@ Toolset Name: `registries`
 
 Toolset Name: `cloudcostmanagement`
 
-- `get_ccm_overview`: Get the ccm overview of a specific account
+- `get_ccm_overview`: Retrieve the cost overview for a specific account.
+- `list_ccm_cost_categories`:  List all cost categories for a specified account.
 
 #### Logs Toolset
 

client/cloudcostmanagement.go

@@ -9,6 +9,7 @@ import (
 const (
 	ccmBasePath        = "ccm/api"
 	ccmGetOverviewPath = ccmBasePath + "/overview?accountIdentifier=%s&startTime=%d&endTime=%d&groupBy=%s"
+	ccmCostCategoryListPath = ccmBasePath + "/business-mapping/filter-panel?accountIdentifier=%s"
 )
 
 type CloudCostManagementService struct {
@@ -27,3 +28,31 @@ func (c *CloudCostManagementService) GetOverview(ctx context.Context, accID stri
 
 	return ccmOverview, nil
 }
+
+func (r *CloudCostManagementService) ListCostCategories(ctx context.Context, scope dto.Scope, opts *dto.CcmListCostCategoriesOptions) (*dto.CCMCostCategoryList, error) {
+	path := ccmCostCategoryListPath
+	params := make(map[string]string)
+	addScope(scope, params)
+
+	// Handle nil options by creating default options
+	if opts == nil {
+		opts = &dto.CcmListCostCategoriesOptions{}
+	}
+
+	if opts.CostCategory != "" {
+		params["costCategory"] = opts.CostCategory
+	}
+	if opts.SearchTerm != "" {
+		params["search"] = opts.SearchTerm
+	}
+
+	// Temporary slice to hold the strings
+	costCategories := new(dto.CCMCostCategoryList)
+
+	err := r.Client.Get(ctx, path, params, nil, costCategories)
+	if err != nil {
+		return nil, fmt.Errorf("failed to list cloud cost managment cost categories: %w", err)
+	}
+
+	return costCategories, nil
+}

client/dto/cloudcostmanagement.go

@@ -9,11 +9,24 @@ const (
 	PeriodTypeYear    string = "YEAR"
 )
 
-// CEView represents a ccm overview
-type CEView struct {
+// CEView represents a basic ccm response.
+// The `data` field contains the response data.
+type CCMBaseResponse struct {
 	Status        string      `json:"state,omitempty"`
-	Data          CCMOverview `json:"data,omitempty"`
+	Message	   string      `json:"message,omitempty"`
 	CorrelationID string      `json:"correlation_id,omitempty"`
+	Error		 []CCMError      `json:"error,omitempty"`
+}
+
+// Response error
+type CCMError struct {
+	FieldId	 string `json:"fieldId,omitempty"`
+	Error string `json:"error,omitempty"`
+}
+
+type CEView struct {
+	CCMBaseResponse
+	Data          CCMOverview `json:"data,omitempty"`
 }
 
 // CCMOverview represents the Overview data from a CCM Overview
@@ -42,3 +55,16 @@ type CCMReference struct {
 	Name string `json:"name,omitempty"`
 	Type string `json:"type,omitempty"`
 }
+
+// CcmCostCategoriesOptions represents options for listing cost categories
+type CcmListCostCategoriesOptions struct {
+	AccountIdentifier string `json:"accountIdentifier,omitempty"`
+	CostCategory string `json:"costCategory,omitempty"`
+	SearchTerm string `json:"search,omitempty"`
+}
+
+// CcmCostCategoryList represents a list of cost categories in CCM
+type CCMCostCategoryList struct {
+	CCMBaseResponse
+	Data []string `json:"data,omitempty"`
+}

cmd/harness-mcp-server/main.go

@@ -247,6 +247,7 @@ func runStdioServer(ctx context.Context, config config.Config) error {
 	}
 
 	slog.Info("Starting server", "url", config.BaseURL)
+	slog.Debug("Using ", "Config ->", config)
 
 	// Define beforeInit function to add client info to user agent
 	beforeInit := func(_ context.Context, _ any, message *mcp.InitializeRequest) {

pkg/harness/cloudcostmanagement.go

@@ -56,11 +56,10 @@ func GetCcmOverviewTool(config *config.Config, client *client.CloudCostManagemen
 			if err != nil {
 				return mcp.NewToolResultError(err.Error()), nil
 			}
-
 			startTime, err :=  utils.FormatMMDDYYYYToUnixMillis(startTimeStr)
 			endTime, err := utils.FormatMMDDYYYYToUnixMillis(endTimeStr)
-
 			groupBy, err := requiredParam[string](request, "groupBy")
+
 			if err != nil {
 				return mcp.NewToolResultError(err.Error()), nil
 			}
@@ -79,6 +78,70 @@ func GetCcmOverviewTool(config *config.Config, client *client.CloudCostManagemen
 		}
 }
 
+
+func ListCcmCostCategoriesTool(config *config.Config, client *client.CloudCostManagementService) (tool mcp.Tool, handler server.ToolHandlerFunc) {
+	return mcp.NewTool("list_ccm_cost_categories",
+			mcp.WithDescription("List the cost categories from an account in Harness Cloud Cost Management"),
+			mcp.WithString("account_id",
+				mcp.Description("The account identifier"),
+			),
+			mcp.WithString("cost_category",
+				mcp.Description("Optional to search for specific category"),
+			),
+			mcp.WithString("search_term",
+				mcp.Description("Optional search term to filter cost categories"),
+			),
+			WithScope(config, false),
+		),
+		func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+			accountId, err := OptionalParam[string](request, "account_id")
+			if accountId == "" {
+				accountId, err = getAccountID(config, request)
+			}
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+
+			params := &dto.CcmListCostCategoriesOptions{}
+			params.AccountIdentifier = accountId
+
+			// Handle cost category parameter 
+			costCategory, ok, err := OptionalParamOK[string](request, "cost_category")
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+			if ok && costCategory != "" {
+				params.CostCategory = costCategory
+			}
+
+			// Handle search parameter
+			searchTerm, ok, err := OptionalParamOK[string](request, "search_term")
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+			if ok && searchTerm != "" {
+				params.SearchTerm = searchTerm
+			}
+
+			scope, err := fetchScope(config, request, false)
+			if err != nil {
+				return mcp.NewToolResultError(err.Error()), nil
+			}
+
+			data, err := client.ListCostCategories(ctx, scope, params)
+			if err != nil {
+				return nil, fmt.Errorf("failed to get CCM Cost Categories: %w", err)
+			}
+
+			r, err := json.Marshal(data)
+			if err != nil {
+				return nil, fmt.Errorf("failed to marshal CCM Cost Category: %w", err)
+			}
+
+			return mcp.NewToolResultText(string(r)), nil
+		}
+	}
+
 // getAccountID retrieves AccountID from the config file
 func getAccountID(config *config.Config, request mcp.CallToolRequest) (string, error) {
 	scope, scopeErr := fetchScope(config, request, true)

pkg/harness/tools.go

@@ -59,7 +59,6 @@ func InitToolsets(config *config.Config) (*toolsets.ToolsetGroup, error) {
 		return nil, err
 	}
 
-
 	// Enable requested toolsets
 	if err := tsg.EnableToolsets(config.Toolsets); err != nil {
 		return nil, err
@@ -312,6 +311,7 @@ func registerCloudCostManagement(config *config.Config, tsg *toolsets.ToolsetGro
 	ccm := toolsets.NewToolset("ccm", "Harness Cloud Cost Management related tools").
 		AddReadTools(
 			toolsets.NewServerTool(GetCcmOverviewTool(config, ccmClient)),
+			toolsets.NewServerTool(ListCcmCostCategoriesTool(config, ccmClient)),
 		)
 
 	// Add toolset to the group