mirror of
https://github.com/thegeeklab/wp-gitea-release.git
synced 2024-11-10 03:40:40 +00:00
263 lines
4.7 KiB
Go
263 lines
4.7 KiB
Go
|
package main
|
||
|
|
||
|
import (
|
||
|
"errors"
|
||
|
"fmt"
|
||
|
"net/url"
|
||
|
"os"
|
||
|
"path"
|
||
|
"path/filepath"
|
||
|
"strings"
|
||
|
|
||
|
"github.com/drone/drone-go/drone"
|
||
|
"github.com/drone/drone-go/plugin"
|
||
|
"github.com/google/go-github/github"
|
||
|
"golang.org/x/oauth2"
|
||
|
)
|
||
|
|
||
|
type Params struct {
|
||
|
BaseUrl string `json:"base_url"`
|
||
|
UploadUrl string `json:"upload_url"`
|
||
|
APIKey string `json:"api_key"`
|
||
|
Files []string `json:"files"`
|
||
|
}
|
||
|
|
||
|
func main() {
|
||
|
r := drone.Repo{}
|
||
|
b := drone.Build{}
|
||
|
w := drone.Workspace{}
|
||
|
v := Params{}
|
||
|
|
||
|
plugin.Param("repo", &r)
|
||
|
plugin.Param("build", &b)
|
||
|
plugin.Param("workspace", &w)
|
||
|
plugin.Param("vargs", &v)
|
||
|
|
||
|
plugin.MustParse()
|
||
|
|
||
|
if b.Event != "tag" {
|
||
|
fmt.Printf("The GitHub Release plugin is only available for tags\n")
|
||
|
os.Exit(0)
|
||
|
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if len(v.BaseUrl) == 0 {
|
||
|
v.BaseUrl = "https://api.github.com/"
|
||
|
} else {
|
||
|
if !strings.HasSuffix(v.BaseUrl, "/") {
|
||
|
v.BaseUrl = v.BaseUrl + "/"
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if len(v.UploadUrl) == 0 {
|
||
|
v.UploadUrl = "https://uploads.github.com/"
|
||
|
} else {
|
||
|
if !strings.HasSuffix(v.UploadUrl, "/") {
|
||
|
v.UploadUrl = v.UploadUrl + "/"
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if len(v.APIKey) == 0 {
|
||
|
fmt.Printf("You must provide an API key\n")
|
||
|
os.Exit(1)
|
||
|
|
||
|
return
|
||
|
}
|
||
|
|
||
|
files := make([]string, 0)
|
||
|
|
||
|
for _, glob := range v.Files {
|
||
|
globed, err := filepath.Glob(glob)
|
||
|
|
||
|
if err != nil {
|
||
|
fmt.Printf("Failed to glob %s\n", glob)
|
||
|
os.Exit(1)
|
||
|
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if globed != nil {
|
||
|
files = append(files, globed...)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
baseUrl, err := url.Parse(v.BaseUrl)
|
||
|
|
||
|
if err != nil {
|
||
|
fmt.Printf("Failed to parse base URL\n")
|
||
|
os.Exit(1)
|
||
|
|
||
|
return
|
||
|
}
|
||
|
|
||
|
uploadUrl, err := url.Parse(v.UploadUrl)
|
||
|
|
||
|
if err != nil {
|
||
|
fmt.Printf("Failed to parse upload URL\n")
|
||
|
os.Exit(1)
|
||
|
|
||
|
return
|
||
|
}
|
||
|
|
||
|
ts := oauth2.StaticTokenSource(
|
||
|
&oauth2.Token{
|
||
|
AccessToken: v.APIKey,
|
||
|
})
|
||
|
|
||
|
tc := oauth2.NewClient(
|
||
|
oauth2.NoContext,
|
||
|
ts)
|
||
|
|
||
|
client := github.NewClient(tc)
|
||
|
client.BaseURL = baseUrl
|
||
|
client.UploadURL = uploadUrl
|
||
|
|
||
|
release, releaseErr := retrieveRelease(
|
||
|
client,
|
||
|
r.Owner,
|
||
|
r.Name,
|
||
|
filepath.Base(b.Ref))
|
||
|
|
||
|
if releaseErr != nil {
|
||
|
fmt.Println(releaseErr)
|
||
|
os.Exit(1)
|
||
|
|
||
|
return
|
||
|
}
|
||
|
|
||
|
uploadErr := appendFiles(
|
||
|
client,
|
||
|
r.Owner,
|
||
|
r.Name,
|
||
|
*release.ID,
|
||
|
files)
|
||
|
|
||
|
if uploadErr != nil {
|
||
|
fmt.Println(uploadErr)
|
||
|
os.Exit(1)
|
||
|
|
||
|
return
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func prepareRelease(client *github.Client, owner string, repo string, tag string) (*github.RepositoryRelease, error) {
|
||
|
var release *github.RepositoryRelease
|
||
|
var releaseErr error
|
||
|
|
||
|
release, releaseErr = retrieveRelease(
|
||
|
client,
|
||
|
owner,
|
||
|
repo,
|
||
|
tag)
|
||
|
|
||
|
if releaseErr != nil {
|
||
|
return nil, releaseErr
|
||
|
}
|
||
|
|
||
|
if release != nil {
|
||
|
return release, nil
|
||
|
}
|
||
|
|
||
|
release, releaseErr = createRelease(
|
||
|
client,
|
||
|
owner,
|
||
|
repo,
|
||
|
tag)
|
||
|
|
||
|
if releaseErr != nil {
|
||
|
return nil, releaseErr
|
||
|
}
|
||
|
|
||
|
if release != nil {
|
||
|
return release, nil
|
||
|
}
|
||
|
|
||
|
return nil, errors.New(
|
||
|
"Failed to retrieve or create a release")
|
||
|
}
|
||
|
|
||
|
func retrieveRelease(client *github.Client, owner string, repo string, tag string) (*github.RepositoryRelease, error) {
|
||
|
release, _, err := client.Repositories.GetReleaseByTag(
|
||
|
owner,
|
||
|
repo,
|
||
|
tag)
|
||
|
|
||
|
if err != nil {
|
||
|
return nil, errors.New(
|
||
|
"Failed to retrieve release")
|
||
|
}
|
||
|
|
||
|
fmt.Printf("Successfully retrieved %s release\n", tag)
|
||
|
return release, nil
|
||
|
}
|
||
|
|
||
|
func createRelease(client *github.Client, owner string, repo string, tag string) (*github.RepositoryRelease, error) {
|
||
|
release, _, err := client.Repositories.CreateRelease(
|
||
|
owner,
|
||
|
repo,
|
||
|
&github.RepositoryRelease{TagName: github.String(tag)})
|
||
|
|
||
|
if err != nil {
|
||
|
return nil, errors.New(
|
||
|
"Failed to create release")
|
||
|
}
|
||
|
|
||
|
fmt.Printf("Successfully created %s release\n", tag)
|
||
|
return release, nil
|
||
|
}
|
||
|
|
||
|
func appendFiles(client *github.Client, owner string, repo string, id int, files []string) error {
|
||
|
assets, _, err := client.Repositories.ListReleaseAssets(
|
||
|
owner,
|
||
|
repo,
|
||
|
id,
|
||
|
&github.ListOptions{})
|
||
|
|
||
|
if err != nil {
|
||
|
return errors.New(
|
||
|
"Failed to fetch existing assets")
|
||
|
}
|
||
|
|
||
|
for _, file := range files {
|
||
|
handle, err := os.Open(file)
|
||
|
|
||
|
if err != nil {
|
||
|
return errors.New(
|
||
|
fmt.Sprintf("Failed to read %s artifact", file))
|
||
|
}
|
||
|
|
||
|
for _, asset := range assets {
|
||
|
if *asset.Name == path.Base(file) {
|
||
|
_, deleteErr := client.Repositories.DeleteReleaseAsset(
|
||
|
owner,
|
||
|
repo,
|
||
|
*asset.ID)
|
||
|
|
||
|
if deleteErr != nil {
|
||
|
return errors.New(
|
||
|
fmt.Sprintf("Failed to delete %s artifact", file))
|
||
|
} else {
|
||
|
fmt.Printf("Successfully deleted old %s artifact\n", *asset.Name)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
_, _, uploadErr := client.Repositories.UploadReleaseAsset(
|
||
|
owner,
|
||
|
repo,
|
||
|
id,
|
||
|
&github.UploadOptions{Name: path.Base(file)},
|
||
|
handle)
|
||
|
|
||
|
if uploadErr != nil {
|
||
|
return errors.New(
|
||
|
fmt.Sprintf("Failed to upload %s artifact", file))
|
||
|
} else {
|
||
|
fmt.Printf("Successfully uploaded %s artifact\n", file)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|