154 lines
3.5 KiB
Go
154 lines
3.5 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/xanzy/go-gitlab"
|
|
)
|
|
|
|
const mrVersionsUrl = "https://gitlab.com/api/v4/projects/%s/merge_requests/%d/versions"
|
|
|
|
type MRVersion struct {
|
|
ID int `json:"id"`
|
|
HeadCommitSHA string `json:"head_commit_sha"`
|
|
BaseCommitSHA string `json:"base_commit_sha"`
|
|
StartCommitSHA string `json:"start_commit_sha"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
MergeRequestID int `json:"merge_request_id"`
|
|
State string `json:"state"`
|
|
RealSize string `json:"real_size"`
|
|
}
|
|
|
|
func (c *Client) Comment() error {
|
|
if len(os.Args) < 6 {
|
|
c.Usage("comment")
|
|
}
|
|
|
|
lineNumber, fileName, comment := os.Args[3], os.Args[4], os.Args[5]
|
|
if lineNumber == "" || fileName == "" || comment == "" {
|
|
c.Usage("comment")
|
|
}
|
|
|
|
lineNumberInt, err := strconv.Atoi(lineNumber)
|
|
if err != nil {
|
|
return fmt.Errorf("Not a valid line number: %w", err)
|
|
}
|
|
|
|
err, response := getMRVersions(c.projectId, c.mergeId)
|
|
if err != nil {
|
|
log.Fatalf("Error making diff thread: %s", err)
|
|
}
|
|
defer response.Body.Close()
|
|
|
|
body, err := ioutil.ReadAll(response.Body)
|
|
|
|
var diffVersionInfo []MRVersion
|
|
err = json.Unmarshal(body, &diffVersionInfo)
|
|
|
|
time := time.Now()
|
|
|
|
/* This is necessary since we do not know whether the comment is on a line that
|
|
has been changed or not. Making all three of these API calls will let us leave
|
|
the comment regardless. See the Gitlab documentation: https://docs.gitlab.com/ee/api/discussions.html#create-a-new-thread-in-the-merge-request-diff */
|
|
wg := sync.WaitGroup{}
|
|
wg.Add(3)
|
|
|
|
resultChannel := make(chan *gitlab.Discussion, 3)
|
|
|
|
for i := 0; i < 3; i++ {
|
|
ii := i
|
|
go func() {
|
|
defer wg.Done()
|
|
options := gitlab.CreateMergeRequestDiscussionOptions{
|
|
Body: &comment,
|
|
CreatedAt: &time,
|
|
Position: &gitlab.NotePosition{
|
|
PositionType: "text",
|
|
BaseSHA: diffVersionInfo[0].BaseCommitSHA,
|
|
HeadSHA: diffVersionInfo[0].HeadCommitSHA,
|
|
StartSHA: diffVersionInfo[0].StartCommitSHA,
|
|
NewPath: fileName,
|
|
OldPath: fileName,
|
|
},
|
|
}
|
|
|
|
if ii == 0 {
|
|
options.Position.NewLine = lineNumberInt
|
|
} else if ii == 1 {
|
|
options.Position.OldLine = lineNumberInt
|
|
} else {
|
|
options.Position.NewLine = lineNumberInt
|
|
options.Position.OldLine = lineNumberInt
|
|
}
|
|
|
|
discussion, _, _ := c.git.Discussions.CreateMergeRequestDiscussion(c.projectId, c.mergeId, &options)
|
|
resultChannel <- discussion
|
|
|
|
}()
|
|
}
|
|
|
|
go func() {
|
|
wg.Wait()
|
|
close(resultChannel)
|
|
}()
|
|
|
|
var createdDiscussion *gitlab.Discussion
|
|
for discussion := range resultChannel {
|
|
if discussion != nil {
|
|
createdDiscussion = discussion
|
|
}
|
|
}
|
|
|
|
if createdDiscussion == nil {
|
|
return fmt.Errorf("Could not leave comment")
|
|
}
|
|
|
|
fmt.Println("Left Comment: " + comment[0:min(len(comment), 25)] + "...")
|
|
|
|
return nil
|
|
}
|
|
|
|
func min(a int, b int) int {
|
|
if a < b {
|
|
return a
|
|
}
|
|
return b
|
|
}
|
|
|
|
/* Gets the latest merge request revision data */
|
|
func getMRVersions(projectId string, mergeId int) (e error, response *http.Response) {
|
|
|
|
gitlabToken := os.Getenv("GITLAB_TOKEN")
|
|
url := fmt.Sprintf(mrVersionsUrl, projectId, mergeId)
|
|
|
|
req, err := http.NewRequest(http.MethodGet, url, nil)
|
|
|
|
req.Header.Add("PRIVATE-TOKEN", gitlabToken)
|
|
|
|
if err != nil {
|
|
return err, nil
|
|
}
|
|
|
|
client := &http.Client{}
|
|
response, err = client.Do(req)
|
|
|
|
if err != nil {
|
|
return err, nil
|
|
}
|
|
|
|
if response.StatusCode != 200 {
|
|
return errors.New("Non-200 status code: " + response.Status), nil
|
|
}
|
|
|
|
return nil, response
|
|
}
|