148 lines
4.2 KiB
Go
148 lines
4.2 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"strings"
|
|
"net/http"
|
|
nomad "github.com/hashicorp/nomad/api"
|
|
//"code.gitea.io/sdk/gitea"
|
|
)
|
|
|
|
type GitUser struct {
|
|
Name string `json:"name"`
|
|
Email string `json:"email"`
|
|
Username string `json:username"`
|
|
}
|
|
|
|
type GiteaCommit struct {
|
|
Id string `json:"id"`
|
|
Message string `json:"message"`
|
|
Url string `json:"string"`
|
|
Author GitUser `json:"author"`
|
|
Committer GitUser `json:"committer"`
|
|
Timestamp string `json:"timestamp"`
|
|
}
|
|
|
|
type GiteaAccount struct {
|
|
Id int64 `json:"id"`
|
|
Login string `json:"login"`
|
|
FullName string `json:"full_name"`
|
|
Email string `json:"email"`
|
|
AvatarUrl string `json:"avatar_url"`
|
|
Username string `json:"username"`
|
|
}
|
|
|
|
type GiteaRepository struct {
|
|
Id int64 `json:"id"`
|
|
Owner GiteaAccount `json:"owner"`
|
|
Description string `json:"description"`
|
|
Private bool `json:"private"`
|
|
Fork bool `json:"private"`
|
|
HtmlUrl string `json:"html_url"`
|
|
SshUrl string `json:"ssh_url"`
|
|
CloneUrl string `json:"clone_url"`
|
|
Website string `json:"website"`
|
|
StarsCount int64 `json:"stars_count"`
|
|
ForksCount int64 `json:"forks_count"`
|
|
WatchersCount int64 `json:"watchers_count"`
|
|
OpenIssuesCount int64 `json:"open_issues_count"`
|
|
DefaultBranch string `json:"default_branch"`
|
|
CreatedAt string `json:"created_at"`
|
|
UpdatedAt string `json:"updated_at"`
|
|
}
|
|
|
|
type GiteaNotification struct {
|
|
Secret string `json:"secret"`
|
|
Ref string `json:"ref"`
|
|
Before string `json:"before"`
|
|
After string `json:"after"`
|
|
CompareUrl string `json:"compare_url"`
|
|
Commits []GiteaCommit `json:"commits"`
|
|
Repository GiteaRepository `json:"repository"`
|
|
Pusher GiteaAccount `json:"pusher"`
|
|
Sender GiteaAccount `json:"sender"`
|
|
}
|
|
|
|
func hook(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodPost {
|
|
http.Error(w, "Hook only support POST requests", http.StatusBadRequest)
|
|
}
|
|
|
|
//@FIXME check for key (in consul?)
|
|
|
|
var notification GiteaNotification
|
|
dec := json.NewDecoder(r.Body)
|
|
if err := dec.Decode(¬ification); err != nil {
|
|
http.Error(w, "Can't parse your request JSON", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
log.Printf("Gitea notification: %+v\n", notification)
|
|
meta := map[string]string{
|
|
"REPO_URL": notification.Repository.CloneUrl,
|
|
"COMMIT": notification.After,
|
|
// @FIXME: this code is not correct, this is a hack
|
|
"BRANCH": strings.ReplaceAll(notification.Ref, "refs/heads/", ""),
|
|
}
|
|
|
|
// @FIXME logic on how to inject secrets securely
|
|
|
|
jobs := NomadClient.Jobs()
|
|
dres, dmeta, err := jobs.Dispatch("builder", meta, []byte{}, "albatros", &nomad.WriteOptions{})
|
|
if err != nil {
|
|
http.Error(w, "Can't submit your job to Nomad", http.StatusInternalServerError)
|
|
}
|
|
log.Printf("Query info: %+v\n", dmeta)
|
|
log.Printf("Job info: %+v\n", dres)
|
|
|
|
io.WriteString(w, dres.DispatchedJobID)
|
|
}
|
|
|
|
func build(w http.ResponseWriter, r *http.Request) {
|
|
q := r.URL.Query()
|
|
jobID, ok := q["job"]
|
|
if !ok || len(jobID) < 1 {
|
|
http.Error(w, "Missing query parameter 'job'. Try adding '?job=xxx'", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
jobs := NomadClient.Jobs()
|
|
allocList, query, err := jobs.Allocations(jobID[0], true, &nomad.QueryOptions{})
|
|
if err != nil {
|
|
http.Error(w, "Unable to fetch your job on Nomad.", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
log.Printf("Query info: %+v\n", query)
|
|
if len(allocList) < 1 {
|
|
http.Error(w, "Job does not contain at least 1 allocation, can't display logs", http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
alloc := allocList[0]
|
|
log.Printf("Alloc: %+v\n", alloc)
|
|
|
|
}
|
|
|
|
var NomadClient *nomad.Client
|
|
|
|
func main() {
|
|
var err error
|
|
nomadConfig := nomad.DefaultConfig()
|
|
nomadConfig.Namespace = "ci"
|
|
// @TODO read env for encrypted parameters
|
|
NomadClient, err = nomad.NewClient(nomadConfig)
|
|
if err != nil {
|
|
log.Fatal("Unable to connect to Nomad, check your config and setup")
|
|
}
|
|
|
|
http.HandleFunc("/hook", hook)
|
|
http.HandleFunc("/build", build)
|
|
|
|
fmt.Println("albatros (:8080)")
|
|
if err = http.ListenAndServe(":8080", nil); err != nil {
|
|
log.Fatal("Can't start HTTP server")
|
|
}
|
|
}
|