mirror of
https://github.com/Luzifer/badge-gen.git
synced 2024-09-16 13:58:32 +00:00
88 lines
2.1 KiB
Go
88 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/sirupsen/logrus"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
const travisCacheDuration = 5 * time.Minute
|
|
|
|
func init() {
|
|
registerServiceHandler("travis", travisServiceHandler{})
|
|
}
|
|
|
|
type travisServiceHandler struct{}
|
|
|
|
func (travisServiceHandler) GetDocumentation() serviceHandlerDocumentationList {
|
|
return serviceHandlerDocumentationList{{
|
|
ServiceName: "Travis-CI",
|
|
DemoPath: "/travis/Luzifer/password",
|
|
Arguments: []string{"<user>", "<repo>", "[branch]"},
|
|
}}
|
|
}
|
|
|
|
func (travisServiceHandler) IsEnabled() bool { return true }
|
|
|
|
func (travisServiceHandler) Handle(ctx context.Context, params []string) (title, text, color string, err error) {
|
|
if len(params) < 2 { //nolint:gomnd
|
|
err = errors.New("you need to provide user and repo")
|
|
return title, text, color, err
|
|
}
|
|
|
|
if len(params) < 3 { //nolint:gomnd
|
|
params = append(params, "master")
|
|
}
|
|
|
|
path := strings.Join([]string{"repos", params[0], params[1], "branches", params[2]}, "/")
|
|
|
|
var state string
|
|
state, err = cacheStore.Get("travis", path)
|
|
|
|
if err != nil {
|
|
var resp *http.Response
|
|
req, _ := http.NewRequestWithContext(ctx, "GET", "https://api.travis-ci.org/"+path, nil)
|
|
resp, err = http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
return title, text, color, errors.Wrap(err, "executing request")
|
|
}
|
|
defer func() {
|
|
if err := resp.Body.Close(); err != nil {
|
|
logrus.WithError(err).Error("closing request body (leaked fd)")
|
|
}
|
|
}()
|
|
|
|
r := struct {
|
|
File string `json:"file"`
|
|
Branch struct {
|
|
State string `json:"state"`
|
|
} `json:"branch"`
|
|
}{}
|
|
|
|
if err = json.NewDecoder(resp.Body).Decode(&r); err != nil {
|
|
return title, text, color, errors.Wrap(err, "decoding JSON response")
|
|
}
|
|
state = r.Branch.State
|
|
logErr(cacheStore.Set("travis", path, state, travisCacheDuration), "writing Travis status to cache")
|
|
}
|
|
|
|
title = "travis"
|
|
text = state
|
|
if text == "" {
|
|
text = "unknown"
|
|
}
|
|
color = map[string]string{
|
|
"unknown": "9f9f9f",
|
|
"passed": "4c1",
|
|
"failed": "e05d44",
|
|
"canceled": "9f9f9f",
|
|
}[text]
|
|
|
|
return title, text, color, nil
|
|
}
|