main.go

package main

import (
	"fmt"
	"os"
	"strings"

	"github.com/bitrise-io/go-utils/log"
	"github.com/bitrise-tools/go-steputils/stepconf"
	"github.com/kvvzr/bitrise-step-comment-on-github-pull-request/github"
)

type Config struct {
	AuthToken        stepconf.Secret `env:"personal_access_token,required"`
	Body             string          `env:"body,required"`
	RepositoryURL    string          `env:"repository_url,required"`
	IssueNumber      int             `env:"issue_number,required"`
	APIBaseURL       string          `env:"api_base_url,required"`
	UpdateCommentTag string          `env:"update_comment_tag"`
}

func ownerAndRepo(url string) (string, string) {
	url = strings.TrimPrefix(strings.TrimPrefix(url, "https://"), "git@")
	paths := strings.FieldsFunc(url, func(r rune) bool { return r == '/' || r == ':' })
	return paths[1], strings.TrimSuffix(paths[2], ".git")
}

// wraps the tag in some special characters to avoid colliding with random text
func decoratedTag(tag string) string {
	return fmt.Sprintf("<!-- %s -->", tag)
}

func main() {
	var conf Config
	if err := stepconf.Parse(&conf); err != nil {
		log.Errorf("Error: %s\n", err)
		os.Exit(1)
	}
	stepconf.Print(conf)

	owner, repo := ownerAndRepo(conf.RepositoryURL)
	commentBody := conf.Body

	var githubClient *github.GithubClient

	if conf.APIBaseURL == "" {
		githubClient = github.NewClient(string(conf.AuthToken))
	} else {
		githubClient = github.NewEnterpriseClient(conf.APIBaseURL, string(conf.AuthToken))
	}

	// if tag is set, try to find and update existing comment
	if conf.UpdateCommentTag != "" {
		commentBody = fmt.Sprintf("%s\n\n%s", conf.Body, decoratedTag(conf.UpdateCommentTag))
		taggedComment, err := githubClient.GetFirstCommentWithTag(owner, repo, conf.IssueNumber, decoratedTag(conf.UpdateCommentTag))

		if err == nil { // comment with the given tag found
			comment, err := githubClient.UpdateComment(owner, repo, *taggedComment.ID, commentBody)

			if err != nil {
				log.Errorf("Github API call failed when updating comment: %w\n", err)
				os.Exit(1)
			} else {
				log.Successf("Success: %v\n", comment)
				os.Exit(0)
			}
		}
	}

	// creating a new comment (either no update tag is set or no existing comment with the given tag found)
	comment, err := githubClient.CreateComment(owner, repo, conf.IssueNumber, commentBody)
	if err != nil {
		log.Errorf("Github API call failed: %w\n", conf.IssueNumber, err)
		os.Exit(1)
	}

	log.Successf("Success: %v\n", comment)
	os.Exit(0)
}