feat: [ML-1172]: Adding support for internal mode (#33)

* feat: [ML-1172]: skip log service for now

* feat: [ML-1172]: fix idp from rebase

* feat: [ML-1172]: Adding better formatting for baseURLs

* feat: [ML-1172]: Adding support for internal mode

* feat: [ML-1172]: fix internal mode

* feat: [ML-1172]: Expose more tools for internal mode

Author: David-Kreiner

client/chaos.go

@@ -3,6 +3,7 @@ package client
 import (
 	"context"
 	"fmt"
+
 	"github.com/harness/harness-mcp/client/dto"
 )
 
@@ -12,28 +13,16 @@ const (
 	chaosGetExperimentPath    = "api/rest/v2/experiments/%s"
 	chaosGetExperimentRunPath = "api/rest/v2/experiments/%s/run"
 	chaosExperimentRunPath    = "api/rest/v2/experiments/%s/run"
-
-	// Prefix to prepend for external API calls
-	externalChaosManagerPathPrefix = "chaos/manager/"
 )
 
 type ChaosService struct {
-	Client           *Client
-	UseInternalPaths bool
-}
-
-func (c *ChaosService) buildPath(basePath string) string {
-	if c.UseInternalPaths {
-		return basePath
-	}
-	return externalChaosManagerPathPrefix + basePath
+	Client *Client
 }
 
 func (c *ChaosService) ListExperiments(ctx context.Context, scope dto.Scope, pagination *dto.PaginationOptions) (*dto.ListExperimentResponse, error) {
 	var (
-		pathTemplate = c.buildPath(chaosListExperimentsPath)
-		path         = fmt.Sprintf(pathTemplate)
-		params       = make(map[string]string)
+		path   = chaosListExperimentsPath
+		params = make(map[string]string)
 	)
 
 	// Set default pagination
@@ -56,9 +45,8 @@ func (c *ChaosService) ListExperiments(ctx context.Context, scope dto.Scope, pag
 
 func (c *ChaosService) GetExperiment(ctx context.Context, scope dto.Scope, experimentID string) (*dto.GetExperimentResponse, error) {
 	var (
-		pathTemplate = c.buildPath(chaosGetExperimentPath)
-		path         = fmt.Sprintf(pathTemplate, experimentID)
-		params       = make(map[string]string)
+		path   = fmt.Sprintf(chaosGetExperimentPath, experimentID)
+		params = make(map[string]string)
 	)
 
 	// Add scope parameters
@@ -75,9 +63,8 @@ func (c *ChaosService) GetExperiment(ctx context.Context, scope dto.Scope, exper
 
 func (c *ChaosService) GetExperimentRun(ctx context.Context, scope dto.Scope, experimentID, experimentRunID string) (*dto.ChaosExperimentRun, error) {
 	var (
-		pathTemplate = c.buildPath(chaosGetExperimentRunPath)
-		path         = fmt.Sprintf(pathTemplate, experimentID)
-		params       = make(map[string]string)
+		path   = fmt.Sprintf(chaosGetExperimentRunPath, experimentID)
+		params = make(map[string]string)
 	)
 
 	params["experimentRunId"] = experimentRunID
@@ -95,9 +82,8 @@ func (c *ChaosService) GetExperimentRun(ctx context.Context, scope dto.Scope, ex
 
 func (c *ChaosService) RunExperiment(ctx context.Context, scope dto.Scope, experimentID string) (*dto.RunChaosExperimentResponse, error) {
 	var (
-		pathTemplate = c.buildPath(chaosExperimentRunPath)
-		path         = fmt.Sprintf(pathTemplate, experimentID)
-		params       = make(map[string]string)
+		path   = fmt.Sprintf(chaosExperimentRunPath, experimentID)
+		params = make(map[string]string)
 	)
 
 	// Add scope parameters

client/connectors.go

@@ -8,14 +8,19 @@ import (
 	pkgDTO "github.com/harness/harness-mcp/pkg/harness/dto" // Alias for the other DTOs
 )
 
+const (
+	listConnectorCataloguePath = "/api/connectors/catalogue"
+	getConnectorPath           = "/api/connectors/%s"
+)
+
 type ConnectorService struct {
 	Client *Client
 }
 
 // ListConnectorCatalogue fetches the connector catalogue.
 // API Documentation: https://apidocs.harness.io/tag/Connectors#operation/getConnectorCatalogue
 func (c *ConnectorService) ListConnectorCatalogue(ctx context.Context, scope dto.Scope) ([]pkgDTO.ConnectorCatalogueItem, error) {
-	path := "/ng/api/connectors/catalogue"
+	path := listConnectorCataloguePath
 	params := make(map[string]string)
 	params["accountIdentifier"] = scope.AccountID
 	if scope.OrgID != "" {
@@ -61,7 +66,7 @@ func (c *ConnectorService) ListConnectorCatalogue(ctx context.Context, scope dto
 // GetConnector retrieves a connector by its identifier
 // https://apidocs.harness.io/tag/Connectors#operation/getConnector
 func (c *ConnectorService) GetConnector(ctx context.Context, scope dto.Scope, connectorIdentifier string) (*pkgDTO.ConnectorDetail, error) {
-	path := fmt.Sprintf("/ng/api/connectors/%s", connectorIdentifier)
+	path := fmt.Sprintf(getConnectorPath, connectorIdentifier)
 	params := make(map[string]string)
 	// Ensure accountIdentifier is always set
 	if scope.AccountID == "" {

client/dashboard.go

@@ -14,8 +14,8 @@ import (
 )
 
 const (
-	dashboardSearchPath = "dashboard/v1/search"
-	dashboardDataPath   = "dashboard/download/dashboards/%s/csv"
+	dashboardSearchPath = "/v1/search"
+	dashboardDataPath   = "/download/dashboards/%s/csv"
 )
 
 // DashboardService handles all dashboard-related API interactions

client/environments.go

@@ -8,10 +8,9 @@ import (
 )
 
 const (
-	environmentBasePath        = "ng/api/environments"
-	environmentGetPath         = environmentBasePath + "/%s"
-	environmentListPath        = environmentBasePath
-	environmentMoveConfigsPath = environmentBasePath + "V2/move-config/%s"
+	environmentListPath        = "/api/environments"
+	environmentGetPath         = "/api/environments/%s"
+	environmentMoveConfigsPath = "/api/environments/V2/move-config/%s"
 )
 
 type EnvironmentClient struct {

client/idp.go

@@ -27,8 +27,7 @@ const (
 )
 
 type IDPService struct {
-	Client           *Client
-	UseInternalPaths bool
+	Client *Client
 }
 
 func (i *IDPService) GetEntity(ctx context.Context, scope dto.Scope, kind string, identifier string) (*dto.EntityResponse, error) {

client/infrastructure.go

@@ -8,9 +8,8 @@ import (
 )
 
 const (
-	infrastructureBasePath        = "ng/api/infrastructures"
-	infrastructureListPath        = infrastructureBasePath
-	infrastructureMoveConfigsPath = infrastructureBasePath + "/move-config/%s"
+	infrastructureListPath        = "/api/infrastructures"
+	infrastructureMoveConfigsPath = "/api/infrastructures/move-config/%s"
 )
 
 type InfrastructureClient struct {

client/logs.go

@@ -9,18 +9,19 @@ import (
 )
 
 const (
-	logDownloadPath = "log-service/blob/download"
+	logDownloadPath = "/blob/download"
 )
 
 // LogService handles operations related to pipeline logs
 type LogService struct {
-	Client *Client
+	LogServiceClient *Client
+	PipelineClient   *Client
 }
 
 // DownloadLogs fetches a download URL for pipeline execution logs
 func (l *LogService) DownloadLogs(ctx context.Context, scope dto.Scope, planExecutionID string) (string, error) {
 	// First, get the pipeline execution details to determine the prefix format
-	pipelineService := &PipelineService{Client: l.Client} // TODO: needs to be changed for internal case, we should move this above
+	pipelineService := &PipelineService{Client: l.PipelineClient} // TODO: needs to be changed for internal case, we should move this above
 	execution, err := pipelineService.GetExecution(ctx, scope, planExecutionID)
 	if err != nil {
 		return "", fmt.Errorf("failed to get execution details: %w", err)
@@ -61,7 +62,7 @@ func (l *LogService) DownloadLogs(ctx context.Context, scope dto.Scope, planExec
 
 	for attempt := 1; attempt <= maxAttempts; attempt++ {
 		response = &dto.LogDownloadResponse{}
-		err = l.Client.Post(ctx, logDownloadPath, params, nil, response)
+		err = l.LogServiceClient.Post(ctx, logDownloadPath, params, nil, response)
 		if err != nil {
 			lastErr = fmt.Errorf("failed to fetch log download URL (attempt %d): %w", attempt, err)
 			if attempt == maxAttempts {

client/pipelines.go

@@ -8,35 +8,22 @@ import (
 )
 
 const (
-	// Base API paths
-	pipelinePath                 = "api/pipelines/%s"
-	pipelineListPath             = "api/pipelines/list"
-	pipelineExecutionPath        = "api/pipelines/execution/url"
-	pipelineExecutionGetPath     = "api/pipelines/execution/v2/%s"
-	pipelineExecutionSummaryPath = "api/pipelines/execution/summary"
-
-	// Prefix to prepend for external API calls
-	externalPathPrefix = "pipeline/"
+	pipelinePath                 = "/api/pipelines/%s"
+	pipelineListPath             = "/api/pipelines/list"
+	pipelineExecutionPath        = "/api/pipelines/execution/url"
+	pipelineExecutionGetPath     = "/api/pipelines/execution/v2/%s"
+	pipelineExecutionSummaryPath = "/api/pipelines/execution/summary"
 )
 
 type PipelineService struct {
-	Client           *Client
-	UseInternalPaths bool
-}
-
-func (p *PipelineService) buildPath(basePath string) string {
-	if p.UseInternalPaths {
-		return basePath
-	}
-	return externalPathPrefix + basePath
+	Client *Client
 }
 
 func (p *PipelineService) Get(ctx context.Context, scope dto.Scope, pipelineID string) (
 	*dto.Entity[dto.PipelineData],
 	error,
 ) {
-	pathTemplate := p.buildPath(pipelinePath)
-	path := fmt.Sprintf(pathTemplate, pipelineID)
+	path := fmt.Sprintf(pipelinePath, pipelineID)
 
 	// Prepare query parameters
 	params := make(map[string]string)
@@ -59,7 +46,7 @@ func (p *PipelineService) List(
 	scope dto.Scope,
 	opts *dto.PipelineListOptions,
 ) (*dto.ListOutput[dto.PipelineListItem], error) {
-	path := p.buildPath(pipelineListPath)
+	path := pipelineListPath
 	// Prepare query parameters
 	params := make(map[string]string)
 	addScope(scope, params)
@@ -103,7 +90,7 @@ func (p *PipelineService) ListExecutions(
 	scope dto.Scope,
 	opts *dto.PipelineExecutionOptions,
 ) (*dto.ListOutput[dto.PipelineExecution], error) {
-	path := p.buildPath(pipelineExecutionSummaryPath)
+	path := pipelineExecutionSummaryPath
 	// Prepare query parameters
 	params := make(map[string]string)
 	addScope(scope, params)
@@ -161,8 +148,7 @@ func (p *PipelineService) GetExecution(
 	scope dto.Scope,
 	planExecutionID string,
 ) (*dto.Entity[dto.PipelineExecution], error) {
-	pathTemplate := p.buildPath(pipelineExecutionGetPath)
-	path := fmt.Sprintf(pathTemplate, planExecutionID)
+	path := fmt.Sprintf(pipelineExecutionGetPath, planExecutionID)
 
 	// Prepare query parameters
 	params := make(map[string]string)
@@ -191,7 +177,7 @@ func (p *PipelineService) FetchExecutionURL(
 	scope dto.Scope,
 	pipelineID, planExecutionID string,
 ) (string, error) {
-	path := p.buildPath(pipelineExecutionPath)
+	path := pipelineExecutionPath
 
 	// Prepare query parameters
 	params := make(map[string]string)

client/pullrequest.go

@@ -9,12 +9,11 @@ import (
 )
 
 const (
-	pullRequestBasePath       = "code/api/v1/repos"
-	pullRequestGetPath        = pullRequestBasePath + "/%s/pullreq/%d"
-	pullRequestListPath       = pullRequestBasePath + "/%s/pullreq"
-	pullRequestCreatePath     = pullRequestBasePath + "/%s/pullreq"
-	pullRequestChecksPath     = pullRequestBasePath + "/%s/pullreq/%d/checks"
-	pullRequestActivitiesPath = pullRequestBasePath + "/%s/pullreq/%d/activities"
+	pullRequestGetPath        = "/api/v1/repos/%s/pullreq/%d"
+	pullRequestListPath       = "/api/v1/repos/%s/pullreq"
+	pullRequestCreatePath     = "/api/v1/repos/%s/pullreq"
+	pullRequestChecksPath     = "/api/v1/repos/%s/pullreq/%d/checks"
+	pullRequestActivitiesPath = "/api/v1/repos/%s/pullreq/%d/activities"
 )
 
 type PullRequestService struct {

client/repositories.go

@@ -8,9 +8,8 @@ import (
 )
 
 const (
-	repositoryBasePath = "code/api/v1/repos"
-	repositoryGetPath  = repositoryBasePath + "/%s"
-	repositoryListPath = repositoryBasePath
+	repositoryBasePath = "/api/v1/repos"
+	repositoryGetPath  = "/api/v1/repos/%s"
 )
 
 type RepositoryService struct {
@@ -48,7 +47,7 @@ func setDefaultPaginationForRepo(opts *dto.RepositoryOptions) {
 }
 
 func (r *RepositoryService) List(ctx context.Context, scope dto.Scope, opts *dto.RepositoryOptions) ([]*dto.Repository, error) {
-	path := repositoryListPath
+	path := repositoryBasePath
 	params := make(map[string]string)
 	addScope(scope, params)