Merge commit 'f955c625aded244864e83a872b396868a490dbc5' as 'go-enry'

go-enry/internal/code-generator/generator/aliases.go

@@ -0,0 +1,58 @@
+package generator
+
+import (
+	"bytes"
+	"io"
+	"io/ioutil"
+	"strings"
+
+	"gopkg.in/yaml.v2"
+)
+
+// Aliases reads from fileToParse and builds source file from tmplPath. It complies with type File signature.
+func Aliases(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
+	}
+
+	orderedLangList := getAlphabeticalOrderedKeys(languages)
+	languageByAlias := buildAliasLanguageMap(languages, orderedLangList)
+
+	buf := &bytes.Buffer{}
+	if err := executeAliasesTemplate(buf, languageByAlias, tmplPath, tmplName, commit); err != nil {
+		return err
+	}
+
+	return formatedWrite(outPath, buf.Bytes())
+}
+
+func buildAliasLanguageMap(languages map[string]*languageInfo, orderedLangList []string) map[string]string {
+	aliasLangsMap := make(map[string]string)
+	for _, lang := range orderedLangList {
+		langInfo := languages[lang]
+		key := convertToAliasKey(lang)
+		aliasLangsMap[key] = lang
+		for _, alias := range langInfo.Aliases {
+			key := convertToAliasKey(alias)
+			aliasLangsMap[key] = lang
+		}
+	}
+
+	return aliasLangsMap
+}
+
+func convertToAliasKey(s string) (key string) {
+	key = strings.Replace(s, ` `, `_`, -1)
+	key = strings.ToLower(key)
+	return
+}
+
+func executeAliasesTemplate(out io.Writer, languageByAlias map[string]string, aliasesTmplPath, aliasesTmpl, commit string) error {
+	return executeTemplate(out, aliasesTmpl, aliasesTmplPath, commit, nil, languageByAlias)
+}