mirror of
https://github.com/Luzifer/badge-gen.git
synced 2024-11-09 13:50:03 +00:00
Added GitHub service with license command
This commit is contained in:
parent
f9f41f7900
commit
d2c7c6693e
1 changed files with 76 additions and 0 deletions
76
service_github.go
Normal file
76
service_github.go
Normal file
|
@ -0,0 +1,76 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"net/http"
|
||||
"strings"
|
||||
|
||||
"golang.org/x/net/context"
|
||||
"golang.org/x/net/context/ctxhttp"
|
||||
)
|
||||
|
||||
func init() {
|
||||
registerServiceHandler("github", githubServiceHandler{})
|
||||
}
|
||||
|
||||
type githubServiceHandler struct{}
|
||||
|
||||
func (g githubServiceHandler) GetDocumentation() serviceHandlerDocumentationList {
|
||||
return serviceHandlerDocumentationList{
|
||||
{
|
||||
ServiceName: "GitHub repo license",
|
||||
DemoPath: "/github/license/Luzifer/badge-gen",
|
||||
Arguments: []string{"license", "<user>", "<repo>"},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (g githubServiceHandler) Handle(ctx context.Context, params []string) (title, text, color string, err error) {
|
||||
if len(params) < 2 {
|
||||
err = errors.New("No service-command / parameters were given")
|
||||
return
|
||||
}
|
||||
|
||||
switch params[0] {
|
||||
case "license":
|
||||
title, text, color, err = g.handleLicense(ctx, params[1:])
|
||||
default:
|
||||
err = errors.New("An unknown service command was called")
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func (g githubServiceHandler) handleLicense(ctx context.Context, params []string) (title, text, color string, err error) {
|
||||
path := strings.Join([]string{"repos", params[0], params[1], "license"}, "/")
|
||||
|
||||
req, _ := http.NewRequest("GET", "https://api.github.com/"+path, nil)
|
||||
req.Header.Set("Accept", "application/vnd.github.drax-preview+json")
|
||||
|
||||
var resp *http.Response
|
||||
resp, err = ctxhttp.Do(ctx, http.DefaultClient, req)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
r := struct {
|
||||
License struct {
|
||||
Name string `json:"name"`
|
||||
} `json:"license"`
|
||||
}{}
|
||||
if err = json.NewDecoder(resp.Body).Decode(&r); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
title = "license"
|
||||
text = r.License.Name
|
||||
color = "007ec6"
|
||||
|
||||
if text == "" {
|
||||
text = "None"
|
||||
}
|
||||
|
||||
return
|
||||
}
|
Loading…
Reference in a new issue