docs.go

package pork

import (
	"encoding/base64"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
	"strings"

	"github.com/mspaulding06/nap"

	"github.com/spf13/cobra"
)

type ReadResponse struct {
	Content string `json:"content"`
}

var DocsCmd = &cobra.Command{
	Use:   "docs",
	Short: "read the documentation for a repository",
	Run: func(cmd *cobra.Command, args []string) {
		if len(args) <= 0 {
			log.Fatalln("You must supply repository argument")
		}
		if err := GetRepositoryReadme(args[0]); err != nil {
			log.Fatalln("Failed to get docs: ", err)
		}
	},
}

func GetRepositoryReadme(repository string) error {
	values := strings.Split(repository, "/")
	return GitHubAPI().Call("docs", map[string]string{
		"owner":   values[0],
		"project": values[1],
	}, nil)
}

func ReadmeSuccess(resp *http.Response) error {
	defer resp.Body.Close()
	content, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return err
	}
	respContent := ReadResponse{}
	json.Unmarshal(content, &respContent)
	buff, err := base64.StdEncoding.DecodeString(respContent.Content)
	if err != nil {
		return err
	}
	fmt.Println(string(buff))
	return nil
}

func ReadmeDefaultRouter(resp *http.Response) error {
	return fmt.Errorf("status code %d", resp.StatusCode)
}

func GetReadmeResource() *nap.RestResource {
	router := nap.NewRouter()
	router.RegisterFunc(200, ReadmeSuccess)
	router.DefaultRouter = ReadmeDefaultRouter
	resource := nap.NewResource("/repos/{{.owner}}/{{.project}}/readme", "GET", router)
	return resource
}