mirror of
https://github.com/Luzifer/badge-gen.git
synced 2024-11-09 13:50:03 +00:00
89 lines
1.8 KiB
Go
89 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"golang.org/x/net/context"
|
|
"golang.org/x/net/context/ctxhttp"
|
|
)
|
|
|
|
func init() {
|
|
registerServiceHandler("gratipay", gratipayServiceHandler{})
|
|
}
|
|
|
|
type gratipayServiceHandler struct{}
|
|
|
|
func (s gratipayServiceHandler) GetDocumentation() serviceHandlerDocumentationList {
|
|
return serviceHandlerDocumentationList{
|
|
{
|
|
ServiceName: "Gratipay user",
|
|
DemoPath: "/gratipay/user/js_bin",
|
|
Arguments: []string{"user", "<username>"},
|
|
},
|
|
{
|
|
ServiceName: "Gratipay team",
|
|
DemoPath: "/gratipay/team/Gratipay",
|
|
Arguments: []string{"team", "<teamname>"},
|
|
},
|
|
}
|
|
}
|
|
|
|
func (s gratipayServiceHandler) Handle(ctx context.Context, params []string) (title, text, color string, err error) {
|
|
if len(params) < 2 {
|
|
err = errors.New("You need to provide type and user/teamname")
|
|
return
|
|
}
|
|
|
|
if params[0] == "user" {
|
|
params[1] = "~" + params[1]
|
|
}
|
|
|
|
path := strings.Join([]string{params[1], "public.json"}, "/")
|
|
text, err = cacheStore.Get("gratipay", path)
|
|
|
|
if err != nil {
|
|
apiURL := "https://gratipay.com/" + path
|
|
|
|
r := struct {
|
|
Receiving float64 `json:"receiving"`
|
|
Taking float64 `json:"taking,string"`
|
|
}{}
|
|
|
|
var resp *http.Response
|
|
resp, err = ctxhttp.Get(ctx, http.DefaultClient, apiURL)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if err = json.NewDecoder(resp.Body).Decode(&r); err != nil {
|
|
return
|
|
}
|
|
|
|
sum := r.Taking + r.Receiving
|
|
color = "brightgreen"
|
|
|
|
if sum == 0.0 {
|
|
color = "red"
|
|
} else if sum < 10 {
|
|
color = "yellow"
|
|
} else if sum < 100 {
|
|
color = "green"
|
|
}
|
|
|
|
text = fmt.Sprintf("$%.2f / week::%s", sum, color)
|
|
cacheStore.Set("gratipay", path, text, 10*time.Minute)
|
|
}
|
|
|
|
tmp := strings.Split(text, "::")
|
|
text = tmp[0]
|
|
color = tmp[1]
|
|
|
|
title = "gratipay"
|
|
return
|
|
}
|