helpers.go

package main

import (
	"encoding/json"

	"bufio"
	"bytes"
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
	"os"
	"os/exec"
	"regexp"
	"strings"
	"sort"

	"github.com/agnivade/levenshtein"
	"gopkg.in/yaml.v2"
  "github.com/riywo/loginshell"
  "github.com/mitchellh/go-homedir"
)

//DownloadFileToStirng downloads a file from a given URL and returns it's
//contents as a string if successful
func DownloadFileToStirng(url string) (string, error) {

	// fmt.Println("Downloading ", url)

	var client http.Client
	resp, err := client.Get(url + "?access_token=" + appConf.GithubToken)
	if err != nil {
		return "", err
	}
	defer resp.Body.Close()

	if resp.StatusCode == http.StatusOK {
		bodyBytes, err := ioutil.ReadAll(resp.Body)
		if err != nil {
			return "", err
		}
		return string(bodyBytes), nil
	}

	return "", err
}

type GitHubFile struct {
	Name        string `json:"name"`
	Path        string `json:"path"`
	Sha         string `json:"sha"`
	Size        int    `json:"size"`
	URL         string `json:"url"`
	HTMLURL     string `json:"html_url"`
	GitURL      string `json:"git_url"`
	DownloadURL string `json:"download_url"`
	Type        string `json:"type"`
	Links       struct {
		Self string `json:"self"`
		Git  string `json:"git"`
		HTML string `json:"html"`
	} `json:"_links"`
}

type GitHubFilesCollection struct {
	Collection []GitHubFile
}

func findYAMLinRepo(repoURL string) []GitHubFile {

	parts := strings.Split(repoURL, "/")
	ApiUrl := ("https://api.github.com/repos/" + parts[3] + "/" + parts[4] + "/contents/")
	// fmt.Println("generated api URL: ", ApiUrl)

	// Get all files from repo
	// repoFiles, err := DownloadFileToStirng("https://api.github.com/repos/atelierbram/base16-atelier-schemes/contents/")
	repoFiles, err := DownloadFileToStirng(ApiUrl)
	check(err)
	keys := make([]GitHubFile, 0)
	json.Unmarshal([]byte(repoFiles), &keys)

	// Create a list of .yaml files
	var colorSchemes []GitHubFile
	for _, v := range keys {
		re := regexp.MustCompile(".*yaml")
		if re.MatchString(v.Name) {
			colorSchemes = append(colorSchemes, v)
		}
	}

	// fmt.Println("Found ", len(colorSchemes), "in repo ", repoFiles)
	return colorSchemes
}

func LoadStringMap(path string) map[string]string {

	f, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE, 0600)
	check(err)
	yamlFile, err := ioutil.ReadAll(f)
	check(err)
	data := make(map[string]string)
	err = yaml.Unmarshal(yamlFile, data)
	check(err)
	return data
}

func SaveStringMap(data map[string]string, path string) {

	yamlData, err := yaml.Marshal(data)
	check(err)
	saveFile, err := os.Create(path)
	defer saveFile.Close()
	saveFile.Write(yamlData)
	saveFile.Close()
	fmt.Println("wrote to: ", saveFile.Name())
}

func FindMatchInMap(choices map[string]string, input string) string {

	if len(choices) == 0 {
		panic("cannot select from empty choices")

	}
	var match string
	distance := 1000

	for k := range choices {
		tempDistance := levenshtein.ComputeDistance(input, k)
		if tempDistance < distance {
			match = k
			distance = tempDistance
		}
	}

	return match
}

func exe_cmd(cmd string) {

	if len(cmd) == 0 {
		return
	}
  shell, err := loginshell.Shell()
  if err != nil { panic(err) }
	fmt.Printf("[HOOK]: Running in shell %v: %v", shell, cmd)
	out, err := exec.Command(shell,"-c",cmd).Output()

	if err != nil {
		fmt.Printf("%s\n", err)
	}
	fmt.Printf("%s\n", out)
}

func WriteFile(path string, data string) {
	f, err := os.Create(path)
	defer f.Close()
	check(err)
	f.Write([]byte(data))
	f.Close()
}

func AppendFile(path string, data string) {
	f, err := os.OpenFile(path, os.O_APPEND|os.O_WRONLY, 0600)
	check(err)
	defer f.Close()
	_, err = f.WriteString(data)
	check(err)
}

func ReplaceMultiline(input string, replacement string, blockStart, blockEnd string) string {
	r := regexp.MustCompile("(?s)" + blockStart + ".*" + blockEnd)
	return blockStart + r.ReplaceAllString(input, replacement) + blockEnd
}

func deepCompareFiles(file1, file2 string) bool {
	sf, err := os.Open(file1)
	if err != nil {
		log.Fatal(err)
	}

	df, err := os.Open(file2)
	if err != nil {
		log.Fatal(err)
	}

	sscan := bufio.NewScanner(sf)
	dscan := bufio.NewScanner(df)

	for sscan.Scan() {
		dscan.Scan()
		if !bytes.Equal(sscan.Bytes(), dscan.Bytes()) {
			return false
		}
	}

	return true
}

func sortMapKeys(in map[string]string) []string {
  sortedKeys := make([]string, 0, len(in))
	for k := range in {
		sortedKeys = append(sortedKeys, k)
	}
	sort.Strings(sortedKeys)

  return sortedKeys
}

func expandPath(path string) string {
  result,err := homedir.Expand(path)
	check(err)
  return result
}