gitea-comment-plugin/main.go

107 lines
2.7 KiB
Go

package main
import (
"bytes"
"encoding/json"
"flag"
"fmt"
"log"
"net/http"
"os"
"strconv"
)
type Payload struct {
Body string `json:"body"`
}
func LookupEnvOrString(key string, defaultVal string) string {
if val, ok := os.LookupEnv(key); ok {
return val
}
return defaultVal
}
func LookupEnvOrInt(key string, defaultVal int) int {
if val, ok := os.LookupEnv(key); ok {
v, err := strconv.Atoi(val)
if err != nil {
log.Fatalf("LookupEnvOrInt[%s]: %v", key, err)
}
return v
}
return defaultVal
}
func main() {
var giteaToken string
var giteaAddress string
var comment string
var commentFile string
var repoOwner string
var repoName string
var prIndex int
flag.StringVar(&giteaToken, "gitea-token", LookupEnvOrString("PLUGIN_GITEA_TOKEN", giteaToken), "API token for Gitea")
flag.StringVar(&giteaAddress, "gitea-address", LookupEnvOrString("PLUGIN_GITEA_ADDRESS", giteaAddress), "Gitea URL")
flag.StringVar(&comment, "comment", LookupEnvOrString("PLUGIN_COMMENT", comment), "Comment for Gitea")
flag.StringVar(&commentFile, "comment-file", LookupEnvOrString("PLUGIN_COMMENT_FILE", commentFile), "Comment file for Gitea")
flag.StringVar(&repoOwner, "repo-owner", LookupEnvOrString("CI_REPO_OWNER", repoOwner), "Owner of the repository")
flag.StringVar(&repoName, "repo-name", LookupEnvOrString("CI_REPO_NAME", repoName), "Name of the repository")
flag.IntVar(&prIndex, "pr-index", LookupEnvOrInt("CI_COMMIT_PULL_REQUEST", prIndex), "Index of the PR")
flag.Parse()
if comment == "" && commentFile == "" {
panic("You must provide a comment")
}
if commentFile != "" && comment != "" {
panic("You must provide only one of comment or comment-file")
}
if commentFile != "" {
commentBytes, err := os.ReadFile(commentFile)
if err != nil {
panic(err)
}
comment = string(commentBytes)
}
if giteaToken == "" {
panic("You must provide a Gitea API Token")
}
if giteaAddress == "" {
panic("You must provide a Gitea URL")
}
if repoOwner == "" {
panic("You must provide an repo owner")
}
if repoName == "" {
panic("You must provide a repo name")
}
if prIndex == 0 {
panic("You must provide an index for PR")
}
data := Payload{
Body: comment,
}
payloadBytes, err := json.Marshal(data)
if err != nil {
panic(err)
}
body := bytes.NewReader(payloadBytes)
url := fmt.Sprintf("%s/api/v1/repos/%s/%s/issues/%d/comments?access_token=%s", giteaAddress, repoOwner, repoName, prIndex, giteaToken)
req, err := http.NewRequest("POST", url, body)
if err != nil {
panic(err)
}
req.Header.Set("Accept", "application/json")
req.Header.Set("Content-Type", "application/json")
resp, err := http.DefaultClient.Do(req)
if err != nil {
panic(err)
}
defer resp.Body.Close()
}