Implement getting HTML color code for languages

internal/code-generator/assets/colors.go.tmpl

@@ -0,0 +1,7 @@
+package data
+
+var LanguagesColor = map[string]string{
+  {{range $language, $color := . -}}
+  "{{$language}}": "{{$color -}}",
+  {{end -}}
+}

internal/code-generator/generator/colors.go

@@ -0,0 +1,47 @@
+package generator
+
+import (
+	"bytes"
+	"io"
+	"io/ioutil"
+
+	"gopkg.in/yaml.v2"
+)
+
+// Colors generates a map in Go with language name -> color string.
+// It is of generator.File type.
+func Colors(fileToParse, samplesDir, outPath, tmplPath, tmplName, commit string) error {
+	data, err := ioutil.ReadFile(fileToParse)
+	if err != nil {
+		return err
+	}
+
+	languages := make(map[string]*languageInfo)
+	if err := yaml.Unmarshal(data, &languages); err != nil {
+		return err
+	}
+
+	langColorMap := buildLanguageColorMap(languages)
+
+	buf := &bytes.Buffer{}
+	if err := executeMimeTemplate(buf, langColorMap, tmplPath, tmplName, commit); err != nil {
+		return err
+	}
+
+	return formatedWrite(outPath, buf.Bytes())
+}
+
+func buildLanguageColorMap(languages map[string]*languageInfo) map[string]string {
+	langColorMap := make(map[string]string)
+	for lang, info := range languages {
+		if len(info.Color) != 0 {
+			langColorMap[lang] = info.Color
+		}
+	}
+
+	return langColorMap
+}
+
+func executeColorTemplate(out io.Writer, langColorMap map[string]string, tmplPath, tmplName, commit string) error {
+	return executeTemplate(out, tmplName, tmplPath, commit, nil, langColorMap)
+}

internal/code-generator/generator/generator_test.go

@@ -80,6 +80,11 @@ const (
 	mimeTypeGold         = testDir + "/mimeType.gold"
 	mimeTypeTestTmplPath = assetsDir + "/mimeType.go.tmpl"
 	mimeTypeTestTmplName = "mimeType.go.tmpl"
+
+	// colors test
+	colorsGold         = testDir + "/colors.gold"
+	colorsTestTmplPath = assetsDir + "/colors.go.tmpl"
+	colorsTestTmplName = "colors.go.tmpl"
 )
 
 type GeneratorTestSuite struct {
@@ -243,6 +248,16 @@ func (s *GeneratorTestSuite) SetupSuite() {
 			generate:    MimeType,
 			wantOut:     mimeTypeGold,
 		},
+		{
+			name:        "Colors()",
+			fileToParse: filepath.Join(s.tmpLinguist, languagesFile),
+			samplesDir:  "",
+			tmplPath:    colorsTestTmplPath,
+			tmplName:    colorsTestTmplName,
+			commit:      commit,
+			generate:    Colors,
+			wantOut:     colorsGold,
+		},
 	}
 }
 

internal/code-generator/generator/langinfo.go

@@ -4,6 +4,7 @@ import "sort"
 
 type languageInfo struct {
 	Type         string   `yaml:"type,omitempty"`
+	Color        string   `yaml:"color,omitempty"`
 	Aliases      []string `yaml:"aliases,omitempty"`
 	Extensions   []string `yaml:"extensions,omitempty,flow"`
 	Interpreters []string `yaml:"interpreters,omitempty,flow"`