added filenames_map.go generation

languagesByFilename now is a map[string]string

internal/code-generator/generator/filenames.go

@@ -0,0 +1,50 @@
+package generator
+
+import (
+	"bytes"
+	"io"
+	"text/template"
+
+	yaml "gopkg.in/yaml.v2"
+)
+
+// Filenames reads from buf and builds filenames_map.go file from filenamesTmplPath.
+func Filenames(data []byte, filenamesTmplPath, filenamesTmplName, commit string) ([]byte, error) {
+	languages := make(map[string]*languageInfo)
+	if err := yaml.Unmarshal(data, &languages); err != nil {
+		return nil, err
+	}
+
+	languagesByFilename := buildFilenameLanguageMap(languages)
+
+	buf := &bytes.Buffer{}
+	if err := executeFilenamesTemplate(buf, languagesByFilename, filenamesTmplPath, filenamesTmplName, commit); err != nil {
+		return nil, err
+	}
+
+	return buf.Bytes(), nil
+}
+
+func buildFilenameLanguageMap(languages map[string]*languageInfo) map[string]string {
+	filenameLangMap := make(map[string]string)
+	for lang, langInfo := range languages {
+		for _, filename := range langInfo.Filenames {
+			filenameLangMap[filename] = lang
+		}
+	}
+
+	return filenameLangMap
+}
+
+func executeFilenamesTemplate(out io.Writer, languagesByFilename map[string]string, filenamesTmplPath, filenamesTmpl, commit string) error {
+	fmap := template.FuncMap{
+		"getCommit": func() string { return commit },
+	}
+
+	t := template.Must(template.New(filenamesTmpl).Funcs(fmap).ParseFiles(filenamesTmplPath))
+	if err := t.Execute(out, languagesByFilename); err != nil {
+		return err
+	}
+
+	return nil
+}

internal/code-generator/generator/generator_test.go

@@ -47,130 +47,103 @@ const (
 	interpretersGold         = "test_files/interpreters.gold"
 	interpretersTestTmplPath = "test_files/interpreters.test.go.tmpl"
 	interpretersTestTmplName = "interpreters.test.go.tmpl"
+
+	// Filenames test
+	filenamesTestFile     = "test_files/filenames.test.yml"
+	filenamesGold         = "test_files/filenames.gold"
+	filenamesTestTmplPath = "test_files/filenames.test.go.tmpl"
+	filenamesTestTmplName = "filenames.test.go.tmpl"
 )
 
 func TestFromFile(t *testing.T) {
-	goldLang, err := ioutil.ReadFile(langGold)
-	assert.NoError(t, err)
-
-	goldContent, err := ioutil.ReadFile(contentGold)
-	assert.NoError(t, err)
-
-	goldVendor, err := ioutil.ReadFile(vendorGold)
-	assert.NoError(t, err)
-
-	goldDocumentation, err := ioutil.ReadFile(documentationGold)
-	assert.NoError(t, err)
-
-	goldTypes, err := ioutil.ReadFile(typesGold)
-	assert.NoError(t, err)
-
-	goldInterpreters, err := ioutil.ReadFile(interpretersGold)
-	assert.NoError(t, err)
-
-	outPathLang, err := ioutil.TempFile("/tmp", "generator-test-")
-	assert.NoError(t, err)
-	defer os.Remove(outPathLang.Name())
-
-	outPathContent, err := ioutil.TempFile("/tmp", "generator-test-")
-	assert.NoError(t, err)
-	defer os.Remove(outPathContent.Name())
-
-	outPathVendor, err := ioutil.TempFile("/tmp", "generator-test-")
-	assert.NoError(t, err)
-	defer os.Remove(outPathVendor.Name())
-
-	outPathDocumentation, err := ioutil.TempFile("/tmp", "generator-test-")
-	assert.NoError(t, err)
-	defer os.Remove(outPathDocumentation.Name())
-
-	outPathTypes, err := ioutil.TempFile("/tmp", "generator-test-")
-	assert.NoError(t, err)
-	defer os.Remove(outPathTypes.Name())
-
-	outPathInterpreters, err := ioutil.TempFile("/tmp", "generator-test-")
-	assert.NoError(t, err)
-	defer os.Remove(outPathInterpreters.Name())
-
 	tests := []struct {
 		name        string
 		fileToParse string
-		outPath     string
 		tmplPath    string
 		tmplName    string
 		commit      string
 		generate    Func
-		wantOut     []byte
+		wantOut     string
 	}{
 		{
 			name:        "TestFromFile_Language",
 			fileToParse: ymlTestFile,
-			outPath:     outPathLang.Name(),
 			tmplPath:    languagesTestTmplPath,
 			tmplName:    languagesTestTmplName,
 			commit:      commitTest,
 			generate:    Languages,
-			wantOut:     goldLang,
+			wantOut:     langGold,
 		},
 		{
 			name:        "TestFromFile_Heuristics",
 			fileToParse: heuristicsTestFile,
-			outPath:     outPathContent.Name(),
 			tmplPath:    contentTestTmplPath,
 			tmplName:    contentTestTmplName,
 			commit:      commitTest,
 			generate:    Heuristics,
-			wantOut:     goldContent,
+			wantOut:     contentGold,
 		},
 		{
 			name:        "TestFromFile_Vendor",
 			fileToParse: vendorTestFile,
-			outPath:     outPathVendor.Name(),
 			tmplPath:    vendorTestTmplPath,
 			tmplName:    vendorTestTmplName,
 			commit:      commitTest,
 			generate:    Vendor,
-			wantOut:     goldVendor,
+			wantOut:     vendorGold,
 		},
 		{
 			name:        "TestFromFile_Documentation",
 			fileToParse: documentationTestFile,
-			outPath:     outPathDocumentation.Name(),
 			tmplPath:    documentationTestTmplPath,
 			tmplName:    documentationTestTmplName,
 			commit:      commitTest,
 			generate:    Documentation,
-			wantOut:     goldDocumentation,
+			wantOut:     documentationGold,
 		},
 		{
 			name:        "TestFromFile_Types",
 			fileToParse: typesTestFile,
-			outPath:     outPathTypes.Name(),
 			tmplPath:    typesTestTmplPath,
 			tmplName:    typesTestTmplName,
 			commit:      commitTest,
 			generate:    Types,
-			wantOut:     goldTypes,
+			wantOut:     typesGold,
 		},
 		{
 			name:        "TestFromFile_Interpreters",
 			fileToParse: interpretersTestFile,
-			outPath:     outPathInterpreters.Name(),
 			tmplPath:    interpretersTestTmplPath,
 			tmplName:    interpretersTestTmplName,
 			commit:      commitTest,
 			generate:    Interpreters,
-			wantOut:     goldInterpreters,
+			wantOut:     interpretersGold,
+		},
+		{
+			name:        "TestFromFile_Filenames",
+			fileToParse: filenamesTestFile,
+			tmplPath:    filenamesTestTmplPath,
+			tmplName:    filenamesTestTmplName,
+			commit:      commitTest,
+			generate:    Filenames,
+			wantOut:     filenamesGold,
 		},
 	}
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			err := FromFile(tt.fileToParse, tt.outPath, tt.tmplPath, tt.tmplName, tt.commit, tt.generate)
+			gold, err := ioutil.ReadFile(tt.wantOut)
 			assert.NoError(t, err)
-			out, err := ioutil.ReadFile(tt.outPath)
+
+			outPath, err := ioutil.TempFile("/tmp", "generator-test-")
 			assert.NoError(t, err)
-			assert.EqualValues(t, tt.wantOut, out, fmt.Sprintf("FromFile() = %v, want %v", string(out), string(tt.wantOut)))
+			defer os.Remove(outPath.Name())
+
+			err = FromFile(tt.fileToParse, outPath.Name(), tt.tmplPath, tt.tmplName, tt.commit, tt.generate)
+			assert.NoError(t, err)
+			out, err := ioutil.ReadFile(outPath.Name())
+			assert.NoError(t, err)
+			assert.EqualValues(t, gold, out, fmt.Sprintf("FromFile() = %v, want %v", string(out), string(tt.wantOut)))
 		})
 	}
 }

internal/code-generator/generator/langs.go

@@ -14,6 +14,7 @@ type languageInfo struct {
 	Type         string   `yaml:"type,omitempty"`
 	Extensions   []string `yaml:"extensions,omitempty,flow"`
 	Interpreters []string `yaml:"interpreters,omitempty,flow"`
+	Filenames    []string `yaml:"filenames,omitempty,flow"`
 }
 
 // Languages reads from buf and builds languages.go file from languagesTmplPath.

internal/code-generator/generator/test_files/filenames.gold

@@ -0,0 +1,12 @@
+package slinguist
+
+// CODE GENERATED AUTOMATICALLY WITH gopkg.in/src-d/simple-linguist.v1/internal/code-generator
+// THIS FILE SHOULD NOT BE EDITED BY HAND
+// Extracted from github/linguist commit: fe8b44ab8a225b1ffa75b983b916ea22fee5b6f7
+
+var languagesByFilename = map[string]string{
+	"APKBUILD":       "Alpine Abuild",
+	"CMakeLists.txt": "CMake",
+	"Cakefile":       "CoffeeScript",
+	"mix.lock":       "Elixir",
+}

internal/code-generator/generator/test_files/filenames.test.go.tmpl

@@ -0,0 +1,11 @@
+package slinguist
+
+// CODE GENERATED AUTOMATICALLY WITH gopkg.in/src-d/simple-linguist.v1/internal/code-generator
+// THIS FILE SHOULD NOT BE EDITED BY HAND
+// Extracted from github/linguist commit: {{ getCommit }}
+
+var languagesByFilename = map[string]string{
+	{{range $filename, $language := . -}}
+	"{{ $filename }}":	{{- printf "%q" $language -}},
+	{{end -}}
+}

internal/code-generator/generator/test_files/filenames.test.yml

@@ -0,0 +1,13 @@
+---
+Alpine Abuild:
+  filenames:
+  - APKBUILD
+CMake:
+  filenames:
+  - CMakeLists.txt
+CoffeeScript:
+  filenames:
+  - Cakefile
+Elixir:
+  filenames:
+  - mix.lock