drone-github-comment/plugin/impl.go

97 lines
2.2 KiB
Go

package plugin
import (
"context"
"crypto/sha256"
"fmt"
"net/url"
"strings"
"github.com/google/go-github/github"
"golang.org/x/oauth2"
)
// Settings for the Plugin.
type Settings struct {
BaseURL string
IssueNum int
Key string
Message string
Update bool
APIKey string
baseURL *url.URL
}
// Validate handles the settings validation of the plugin.
func (p *Plugin) Validate() error {
var err error
if p.pipeline.Build.Event != "pull" {
return fmt.Errorf("github comment plugin is only available for pull requests")
}
if p.settings.APIKey == "" {
return fmt.Errorf("no api key provided")
}
if p.settings.Message == "" {
return fmt.Errorf("no message provides")
}
if p.settings.Message != "" {
if p.settings.Message, err = readStringOrFile(p.settings.Message); err != nil {
return fmt.Errorf("error while reading %s: %w", p.settings.Message, err)
}
}
if !strings.HasSuffix(p.settings.BaseURL, "/") {
p.settings.BaseURL = p.settings.BaseURL + "/"
}
p.settings.baseURL, err = url.Parse(p.settings.BaseURL)
if err != nil {
return fmt.Errorf("failed to parse base url: %w", err)
}
if p.settings.Key == "" {
key := fmt.Sprintf("%s/%s/%d", p.pipeline.Repo.Owner, p.pipeline.Repo.Name, p.settings.IssueNum)
hash := sha256.Sum256([]byte(key))
p.settings.Key = fmt.Sprintf("%x", hash)
}
if p.settings.Key, err = readStringOrFile(p.settings.Key); err != nil {
return fmt.Errorf("error while reading %s: %w", p.settings.Key, err)
}
return nil
}
// Execute provides the implementation of the plugin.
func (p *Plugin) Execute() error {
ts := oauth2.StaticTokenSource(&oauth2.Token{AccessToken: p.settings.APIKey})
tc := oauth2.NewClient(
context.WithValue(context.Background(), oauth2.HTTPClient, p.network.Client),
ts,
)
client := github.NewClient(tc)
client.BaseURL = p.settings.baseURL
cc := commentClient{
Client: client,
Context: p.network.Context,
Repo: p.pipeline.Repo.Name,
Owner: p.pipeline.Repo.Owner,
Message: p.settings.Message,
Update: p.settings.Update,
Key: p.settings.Key,
IssueNum: p.pipeline.Build.PullRequest,
}
err := cc.issueComment()
if err != nil {
return fmt.Errorf("failed to create or update comment: %w", err)
}
return nil
}