main.go

package main

import (
	"bytes"
	"encoding/base64"
	"html/template"
	"io/ioutil"
	"log"
	"net/http"
	"sort"
	"strings"

	colorful "github.com/lucasb-eyer/go-colorful"
	yaml "gopkg.in/yaml.v2"
)

type image struct {
	LangName  string
	LangColor string
	TextColor string
}

type imageLink struct {
	EncodedName string
	ImageName   string
}

type readme []imageLink

func main() {

	// parse templates
	tmpl, err := template.ParseFiles("./image.tmpl", "./readme.tmpl")
	if err != nil {
		log.Fatal(err)
	}

	// get and decode yaml
	baseURL := "https://raw.githubusercontent.com"
	req, err := http.Get(baseURL + "/github/linguist/master/lib/linguist/languages.yml")
	if err != nil {
		log.Fatal(err)
	}

	defer req.Body.Close()

	var languageData map[string]map[string]interface{}
	err = yaml.NewDecoder(req.Body).Decode(&languageData)
	if err != nil {
		log.Fatal(err)
	}

	// generate svgs and readme
	readmeData := readme{}

	// sort languages by name
	keys := make([]string, len(languageData))
	for key := range languageData {
		keys = append(keys, key)
	}
	less := func(i, j int) bool {
		return strings.ToLower(keys[i]) < strings.ToLower(keys[j])
	}
	sort.Slice(keys, less)

	// find languages with colors
	for _, lang := range keys {
		meta := languageData[lang]
		if meta["color"] != nil {
			color, ok := meta["color"].(string)
			if ok {

				// check if color is light or dark to determine text color
				c, err := colorful.Hex(color)
				if err != nil {
					log.Fatal(err)
				}
				_, _, l := c.Hcl()
				textColor := "#FFF"
				if l > 0.7 {
					textColor = "#000"
				}

				// create svg images with file name as base64
				imageName := base64.StdEncoding.EncodeToString([]byte(lang))

				svgBuffer := bytes.Buffer{}

				img := image{
					LangName:  lang,
					LangColor: color,
					TextColor: textColor,
				}

				err = tmpl.ExecuteTemplate(&svgBuffer, "image.tmpl", img)
				if err != nil {
					log.Fatal(err)
				}

				err = ioutil.WriteFile("./svgs/"+imageName+".svg", svgBuffer.Bytes(), 0644)
				if err != nil {
					log.Fatal(err)
				}

				// encode any spaces
				encodedName := strings.Replace(lang, " ", "%20", -1)

				// encode any single quotes
				encodedName = strings.Replace(encodedName, "'", "&apos;", -1)

				// add language to readme
				readmeData = append(readmeData, imageLink{
					EncodedName: encodedName,
					ImageName:   imageName,
				})
			}
		}
	}

	readmeBuffer := bytes.Buffer{}
	err = tmpl.ExecuteTemplate(&readmeBuffer, "readme.tmpl", readmeData)
	if err != nil {
		log.Fatal(err)
	}

	// create README.md
	err = ioutil.WriteFile("./README.md", readmeBuffer.Bytes(), 0644)
	if err != nil {
		log.Fatal(err)
	}

}