Migrate all modules to their own subfolder

Handles #375

modules/travisci/client.go

@@ -0,0 +1,95 @@
+package travisci
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"net/http"
+	"net/url"
+	"os"
+
+	"github.com/wtfutil/wtf/wtf"
+)
+
+var TRAVIS_HOSTS = map[bool]string{
+	false: "travis-ci.org",
+	true:  "travis-ci.com",
+}
+
+func BuildsFor() (*Builds, error) {
+	builds := &Builds{}
+
+	pro := wtf.Config.UBool("wtf.mods.travisci.pro", false)
+	travisAPIURL.Host = "api." + TRAVIS_HOSTS[pro]
+
+	resp, err := travisRequest("builds")
+	if err != nil {
+		return builds, err
+	}
+
+	parseJson(&builds, resp.Body)
+
+	return builds, nil
+}
+
+/* -------------------- Unexported Functions -------------------- */
+
+var (
+	travisAPIURL = &url.URL{Scheme: "https", Path: "/"}
+)
+
+func travisRequest(path string) (*http.Response, error) {
+	params := url.Values{}
+	params.Add("limit", "10")
+
+	requestUrl := travisAPIURL.ResolveReference(&url.URL{Path: path, RawQuery: params.Encode()})
+
+	req, err := http.NewRequest("GET", requestUrl.String(), nil)
+	req.Header.Add("Accept", "application/json")
+	req.Header.Add("Content-Type", "application/json")
+	req.Header.Add("Travis-API-Version", "3")
+
+	bearer := fmt.Sprintf("token %s", apiToken())
+	req.Header.Add("Authorization", bearer)
+	if err != nil {
+		return nil, err
+	}
+
+	httpClient := &http.Client{}
+	resp, err := httpClient.Do(req)
+	if err != nil {
+		return nil, err
+	}
+
+	if resp.StatusCode < 200 || resp.StatusCode > 299 {
+		return nil, fmt.Errorf(resp.Status)
+	}
+
+	return resp, nil
+}
+
+func apiToken() string {
+	return wtf.Config.UString(
+		"wtf.mods.travisci.apiKey",
+		os.Getenv("WTF_TRAVIS_API_TOKEN"),
+	)
+}
+
+func parseJson(obj interface{}, text io.Reader) {
+	jsonStream, err := ioutil.ReadAll(text)
+	if err != nil {
+		panic(err)
+	}
+
+	decoder := json.NewDecoder(bytes.NewReader(jsonStream))
+
+	for {
+		if err := decoder.Decode(obj); err == io.EOF {
+			break
+		} else if err != nil {
+			panic(err)
+		}
+	}
+}

modules/travisci/travis.go

@@ -0,0 +1,32 @@
+package travisci
+
+type Builds struct {
+	Builds []Build `json:"builds"`
+}
+
+type Build struct {
+	ID         int        `json:"id"`
+	CreatedBy  Owner      `json:"created_by"`
+	Branch     Branch     `json:"branch"`
+	Number     string     `json:"number"`
+	Repository Repository `json:"repository"`
+	Commit     Commit     `json:"commit"`
+	State      string     `json:"state"`
+}
+
+type Owner struct {
+	Login string `json:"login"`
+}
+
+type Branch struct {
+	Name string `json:"name"`
+}
+
+type Repository struct {
+	Name string `json:"name"`
+	Slug string `json:"slug"`
+}
+
+type Commit struct {
+	Message string `json:"message"`
+}

modules/travisci/widget.go

@@ -0,0 +1,192 @@
+package travisci
+
+import (
+	"fmt"
+	"github.com/gdamore/tcell"
+	"github.com/rivo/tview"
+	"github.com/wtfutil/wtf/wtf"
+	"strings"
+)
+
+const HelpText = `
+ Keyboard commands for Travis CI:
+
+   /: Show/hide this help window
+   j: Select the next build in the list
+   k: Select the previous build in the list
+   r: Refresh the data
+
+   arrow down: Select the next build in the list
+   arrow up:   Select the previous build in the list
+
+   return: Open the selected build in a browser
+`
+
+type Widget struct {
+	wtf.HelpfulWidget
+	wtf.TextWidget
+
+	builds   *Builds
+	selected int
+}
+
+func NewWidget(app *tview.Application, pages *tview.Pages) *Widget {
+	widget := Widget{
+		HelpfulWidget: wtf.NewHelpfulWidget(app, pages, HelpText),
+		TextWidget:    wtf.NewTextWidget(app, "TravisCI", "travisci", true),
+	}
+
+	widget.HelpfulWidget.SetView(widget.View)
+	widget.unselect()
+
+	widget.View.SetInputCapture(widget.keyboardIntercept)
+
+	return &widget
+}
+
+/* -------------------- Exported Functions -------------------- */
+
+func (widget *Widget) Refresh() {
+	if widget.Disabled() {
+		return
+	}
+
+	builds, err := BuildsFor()
+
+	if err != nil {
+		widget.View.SetWrap(true)
+		widget.View.SetTitle(widget.Name)
+		widget.View.SetText(err.Error())
+	} else {
+		widget.builds = builds
+	}
+
+	widget.display()
+}
+
+/* -------------------- Unexported Functions -------------------- */
+
+func (widget *Widget) display() {
+	if widget.builds == nil {
+		return
+	}
+
+	widget.View.SetWrap(false)
+
+	widget.View.SetTitle(widget.ContextualTitle(fmt.Sprintf("%s - Builds", widget.Name)))
+	widget.View.SetText(widget.contentFrom(widget.builds))
+}
+
+func (widget *Widget) contentFrom(builds *Builds) string {
+	var str string
+	for idx, build := range builds.Builds {
+
+		str = str + fmt.Sprintf(
+			"[%s] [%s] %s-%s (%s) [%s]%s - [blue]%s\n",
+			widget.rowColor(idx),
+			buildColor(&build),
+			build.Repository.Name,
+			build.Number,
+			build.Branch.Name,
+			widget.rowColor(idx),
+			strings.Split(build.Commit.Message, "\n")[0],
+			build.CreatedBy.Login,
+		)
+	}
+
+	return str
+}
+
+func (widget *Widget) rowColor(idx int) string {
+	if widget.View.HasFocus() && (idx == widget.selected) {
+		return wtf.DefaultFocussedRowColor()
+	}
+	return "White"
+}
+
+func buildColor(build *Build) string {
+	switch build.State {
+	case "broken":
+		return "red"
+	case "failed":
+		return "red"
+	case "failing":
+		return "red"
+	case "pending":
+		return "yellow"
+	case "started":
+		return "yellow"
+	case "fixed":
+		return "green"
+	case "passed":
+		return "green"
+	default:
+		return "white"
+	}
+}
+
+func (widget *Widget) next() {
+	widget.selected++
+	if widget.builds != nil && widget.selected >= len(widget.builds.Builds) {
+		widget.selected = 0
+	}
+
+	widget.display()
+}
+
+func (widget *Widget) prev() {
+	widget.selected--
+	if widget.selected < 0 && widget.builds != nil {
+		widget.selected = len(widget.builds.Builds) - 1
+	}
+
+	widget.display()
+}
+
+func (widget *Widget) openBuild() {
+	sel := widget.selected
+	if sel >= 0 && widget.builds != nil && sel < len(widget.builds.Builds) {
+		build := &widget.builds.Builds[widget.selected]
+		travisHost := TRAVIS_HOSTS[wtf.Config.UBool("wtf.mods.travisci.pro", false)]
+		wtf.OpenFile(fmt.Sprintf("https://%s/%s/%s/%d", travisHost, build.Repository.Slug, "builds", build.ID))
+	}
+}
+
+func (widget *Widget) unselect() {
+	widget.selected = -1
+	widget.display()
+}
+
+func (widget *Widget) keyboardIntercept(event *tcell.EventKey) *tcell.EventKey {
+	switch string(event.Rune()) {
+	case "/":
+		widget.ShowHelp()
+	case "j":
+		widget.next()
+		return nil
+	case "k":
+		widget.prev()
+		return nil
+	case "r":
+		widget.Refresh()
+		return nil
+	}
+
+	switch event.Key() {
+	case tcell.KeyDown:
+		widget.next()
+		return nil
+	case tcell.KeyEnter:
+		widget.openBuild()
+		return nil
+	case tcell.KeyEsc:
+		widget.unselect()
+		return event
+	case tcell.KeyUp:
+		widget.prev()
+		widget.display()
+		return nil
+	default:
+		return event
+	}
+}