connector.go

package main

import (
	"fmt"
	"regexp"
	"strings"

	"github.com/fatih/color"
	"github.com/urfave/cli/v2"
	"gopkg.in/yaml.v3"
)

// apply(create or update) connector
func applyConnector(c *cli.Context) error {
	filePath := c.String("file")
	githubUsername := c.String("git-user")
	delegateName := c.String("delegate-name")
	awsCrossAccountRoleArn := c.String("aws-cross-account-role-arn")
	awsAccessKey := c.String("aws-access-key")
	awsRegion := c.String("cloud-region")
	hostIpOrFqdn := c.String("host-ip")
	hostPort := c.String("port")

	baseURL := getNGBaseURL(c)

	if filePath == "" {
		fmt.Println("Please enter valid filename")
		return nil
	}
	fmt.Println("Trying to create or update connector using the yaml=",
		getColoredText(filePath, color.FgCyan))

	// Getting the account details
	createConnectorURL := GetUrlWithQueryParams("", baseURL, CONNECTOR_ENDPOINT, map[string]string{
		"accountIdentifier": cliCdRequestData.Account,
	})

	var content, _ = readFromFile(filePath)
	if isGithubConnectorYAML(content) {
		if githubUsername == "" || githubUsername == GITHUB_USERNAME_PLACEHOLDER {
			githubUsername = TextInput("Enter valid github username:")
		}
		content = replacePlaceholderValues(content, GITHUB_USERNAME_PLACEHOLDER, githubUsername)
	}
	if isK8sConnectorYAML(content) {
		if delegateName == "" || delegateName == DELEGATE_NAME_PLACEHOLDER {
			delegateName = TextInput("Enter valid delegate name:")
		}
		content = replacePlaceholderValues(content, DELEGATE_NAME_PLACEHOLDER, delegateName)
	}
	if isAwsConnectorYAML(content) {
		if awsCrossAccountRoleArn == "" || awsCrossAccountRoleArn == AWS_CROSS_ACCOUNT_ROLE_ARN {
			awsCrossAccountRoleArn = TextInput("Enter valid aws cross account role arn:")
		}
		if awsAccessKey == "" || awsAccessKey == AWS_ACCESS_KEY {
			awsAccessKey = TextInput("Enter valid aws access key:")
		}
		if awsRegion == "" || awsRegion == REGION_NAME_PLACEHOLDER {
			awsRegion = TextInput("Enter valid aws region:")
		}
		if delegateName == "" || delegateName == DELEGATE_NAME_PLACEHOLDER {
			delegateName = TextInput("Enter valid delegate name:")
		}

		//TODO: find a better way to resolve placeholders, dont depend on fixed placeholders
		content = replacePlaceholderValues(content, AWS_CROSS_ACCOUNT_ROLE_ARN, awsCrossAccountRoleArn)
		content = replacePlaceholderValues(content, AWS_ACCESS_KEY, awsAccessKey)
		content = replacePlaceholderValues(content, REGION_NAME_PLACEHOLDER, awsRegion)
		content = replacePlaceholderValues(content, DELEGATE_NAME_PLACEHOLDER, delegateName)
	}
	if isPdcConnectorYAML(content) {
		hasPortNumber := strings.Contains(content, HOST_PORT_PLACEHOLDER)
		if hostIpOrFqdn == "" || hostIpOrFqdn == HOST_IP_PLACEHOLDER {
			hostIpOrFqdn = TextInput("Enter valid host ip / fqdn:")
		}
		if hasPortNumber && (hostPort == "" || hostPort == HOST_PORT_PLACEHOLDER) {
			hostPort = TextInput("Enter valid host port:")
		}
		if delegateName == "" || delegateName == DELEGATE_NAME_PLACEHOLDER {
			delegateName = TextInput("Enter valid delegate name:")
		}
		content = replacePlaceholderValues(content, HOST_IP_PLACEHOLDER, hostIpOrFqdn)
		content = replacePlaceholderValues(content, DELEGATE_NAME_PLACEHOLDER, delegateName)
		content = replacePlaceholderValues(content, HOST_PORT_PLACEHOLDER, hostPort)
	}
	requestBody := make(map[string]interface{})
	if err := yaml.Unmarshal([]byte(content), requestBody); err != nil {
		return err
	}
	identifier := valueToString(GetNestedValue(requestBody, "connector", "identifier").(string))
	projectIdentifier := valueToString(GetNestedValue(requestBody, "connector", "projectIdentifier").(string))
	orgIdentifier := valueToString(GetNestedValue(requestBody, "connector", "orgIdentifier").(string))
	entityExists := getEntity(baseURL, fmt.Sprintf("connectors/%s", identifier),
		projectIdentifier, orgIdentifier, map[string]string{})

	var err error
	if !entityExists {
		println("Creating connector with id: ", getColoredText(identifier, color.FgGreen))
		_, err = Post(createConnectorURL, cliCdRequestData.AuthToken, requestBody, CONTENT_TYPE_JSON, nil)
		if err == nil {
			println(getColoredText("Successfully created connector with id= ", color.FgGreen) +
				getColoredText(identifier, color.FgBlue))

			return nil
		}

	} else {
		println("Found connector with id=", getColoredText(identifier, color.FgCyan))
		println("Updating details of connector with id=", getColoredText(identifier, color.FgBlue))

		_, err = Put(createConnectorURL, cliCdRequestData.AuthToken, requestBody, CONTENT_TYPE_JSON, nil)
		if err == nil {
			println(getColoredText("Successfully updated connector with id= ", color.FgGreen) +
				getColoredText(identifier, color.FgBlue))
			return nil
		}

	}
	return nil
}

func isGithubConnectorYAML(str string) bool {
	regexPattern := `type:\s+Github`
	match, _ := regexp.MatchString(regexPattern, str)
	return match
}

func isK8sConnectorYAML(str string) bool {
	regexPattern := `type:\s+K8sCluster`
	match, _ := regexp.MatchString(regexPattern, str)
	return match
}

func isAwsConnectorYAML(str string) bool {
	regexPattern := `type:\s+Aws`
	match, _ := regexp.MatchString(regexPattern, str)
	return match
}

func isPdcConnectorYAML(str string) bool {
	regexPattern := `type:\s+Pdc`
	match, _ := regexp.MatchString(regexPattern, str)
	return match
}

// Delete an existing connector
func deleteConnector(*cli.Context) error {
	fmt.Println(NOT_IMPLEMENTED)
	return nil
}

// Delete an existing connector
func listConnector(*cli.Context) error {
	fmt.Println(NOT_IMPLEMENTED)
	return nil
}