jenkins.go

package jenkins

import (
	"fmt"
	"net/url"
	"os"
	"strconv"
	"strings"

	"github.com/aquasecurity/go-git-pr-commenter/pkg/commenter/github"
	"github.com/aquasecurity/go-git-pr-commenter/pkg/commenter/gitlab"

	"github.com/aquasec-com/go-environments/enums"
	"github.com/aquasec-com/go-environments/environments/jenkins"
	env_utils "github.com/aquasec-com/go-environments/environments/utils"

	"github.com/aquasecurity/go-git-pr-commenter/pkg/commenter"
	"github.com/aquasecurity/go-git-pr-commenter/pkg/commenter/bitbucket"
	bitbucket_server "github.com/aquasecurity/go-git-pr-commenter/pkg/commenter/bitbucket-server"
	"github.com/aquasecurity/go-git-pr-commenter/pkg/commenter/utils"
	bitbucketutils "github.com/aquasecurity/go-git-pr-commenter/pkg/commenter/utils/bitbucket"
)

func NewJenkins(baseRef string) (commenter.Repository, error) {
	cloneUrl, _ := utils.GetRepositoryCloneURL()
	sanitizedCloneUrl := env_utils.StripCredentialsFromUrl(cloneUrl)
	scmSource, scmApiUrl := jenkins.GetRepositorySource(sanitizedCloneUrl)

	if _, exists := bitbucketutils.GetBitbucketPayload(); strings.Contains(cloneUrl, "bitbucket") || exists {
		username, ok := os.LookupEnv("USERNAME")
		if !ok {
			return nil, fmt.Errorf("USERNAME env var is not set")
		}
		password, ok := os.LookupEnv("PASSWORD")
		if !ok {
			return nil, fmt.Errorf("PASSWORD env var is not set")
		}

		if strings.Contains(cloneUrl, "bitbucket.org") {
			return bitbucket.CreateClient(username, password, bitbucketutils.GetPrId(), bitbucketutils.GetRepositoryName(cloneUrl))
		} else { // bitbucket server
			repoName := bitbucketutils.GetRepositoryName(cloneUrl)
			project, repo := bitbucketutils.GetProjectAndRepo(repoName)
			return bitbucket_server.NewBitbucketServer(scmApiUrl, username, password, bitbucketutils.GetPrId(), project, repo, baseRef)
		}
	} else if scmSource == enums.GithubServer || scmSource == enums.Github {
		_, org, repoName, _, err := env_utils.ParseDataFromCloneUrl(cloneUrl, scmApiUrl, scmSource)
		if err != nil {
			return nil, fmt.Errorf("failed parsing url with error: %s", err.Error())
		}
		token := os.Getenv("GITHUB_TOKEN")
		prNumber := os.Getenv("CHANGE_ID")
		// for gh single jenkins pipeline
		if prNumber == "" {
			prNumber = os.Getenv("ghprbPullId")
		}
		prNumberInt, err := strconv.Atoi(prNumber)
		if err != nil {
			return nil, fmt.Errorf("failed converting prNumber to int, %s: %s", prNumber, err.Error())
		}

		if scmSource == enums.Github {
			return github.NewGithub(
				token,
				org,
				repoName,
				prNumberInt)
		} else { //github server
			return github.NewGithubServer(scmApiUrl, token, org, repoName, prNumberInt)
		}

	} else if scmSource == enums.GitlabServer || scmSource == enums.Gitlab {
		_, org, repoName, _, err := env_utils.ParseDataFromCloneUrl(cloneUrl, scmApiUrl, scmSource)
		if err != nil {
			return nil, fmt.Errorf("failed parsing url with error: %s", err.Error())
		}
		token := os.Getenv("GITLAB_TOKEN")
		prNumber := os.Getenv("CHANGE_ID")

		return gitlab.NewGitlab(token, scmApiUrl, url.PathEscape(fmt.Sprintf("%s/%s", org, repoName)), prNumber)
	}
	return nil, nil
}