ssh-keys.go

package main

import (
	"context"
	"fmt"
	"io/ioutil"
	"os"
	"strings"

	"github.com/google/go-github/github"
	"github.com/raguay/goAlfred"
	"gopkg.in/yaml.v2"
)

func recordHit(user string) error {
	cache, err := readCache()
	if err != nil {
		return err
	}

	cache[user]++

	err = writeCache(cache)
	return err
}

func readCache() (map[string]int, error) {
	src, err := ioutil.ReadFile(goAlfred.Cache() + "/cache.yml")
	if err != nil {
		if os.IsNotExist(err) {
			return map[string]int{}, nil
		}
		return nil, err
	}

	var cache map[string]int
	err = yaml.Unmarshal(src, &cache)
	if err != nil {
		return nil, err
	}

	return cache, nil
}

func writeCache(cache map[string]int) error {
	src, err := yaml.Marshal(cache)
	if err != nil {
		return err
	}

	err = ioutil.WriteFile(goAlfred.Cache()+"/cache.yml", src, 0644)
	if err != nil {
		return err
	}
	return nil
}

func getKeys(ctx context.Context, user string, gh *github.Client) string {
	var keys []string

	err := recordHit(user)
	if err != nil {
		panic(err)
	}

	page := 0
	for {
		results, response, err := gh.Users.ListKeys(ctx, user, &github.ListOptions{Page: page, PerPage: 500})
		if err != nil {
			panic(err)
		}
		page = response.NextPage
		for i := 0; i < len(results); i++ {
			keys = append(keys, *results[i].Key)
		}
		if response.NextPage == response.LastPage {
			break
		}
	}

	return (strings.Join(keys, "\n") + "\n")

}

func findUser(ctx context.Context, user string, gh *github.Client) string {
	results, _, err := gh.Search.Users(ctx, user, &github.SearchOptions{})
	if err != nil {
		panic(err)
	}

	cache, err := readCache()
	if err != nil {
		panic(err)
	}

	for i := 0; i < len(results.Users); i++ {
		user := results.Users[i]
		id := *user.ID
		login := *user.Login
		var name string
		if user.Name != nil {
			name = *user.Name
		} else {
			name = *user.Login
		}
		priority := cache[login]

		goAlfred.AddResult(
			fmt.Sprintf("%d", id), // uid
			login, // arg string
			login, // title
			"Copy SSH keys for "+name+" to clipboard", // subtitle
			"icon.png", // icon
			"yes",      // valid
			"",         // auto
			"",         // rtype
			priority,
		)
	}
	return (goAlfred.ToXML())
}

func main() {

	if len(os.Args) > 2 {
		// Load up creds if present
		gh := github.NewClient(nil)
		ctx := context.Background()
		user := os.Args[2]
		switch os.Args[1] {
		case "login":
			// TODO: save creds for authenticating to github
		case "logout":
			// TODO: delete creds
		case "keys-for":
			fmt.Print(getKeys(ctx, user, gh))
		case "find-user":
			fmt.Print(findUser(ctx, user, gh))
		}
	}

}