Add special cases detection for go.mod and CMmakeLists.txt files

wakatime · Dec 16, 2020 · 72e6261 · 72e6261
1 parent d8b487e
commit 72e6261
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pkg/language/language.go b/pkg/language/language.go
@@ -97,6 +97,13 @@ func detectSpecialCases(fp string) (heartbeat.Language, bool) {
 	dir, file := filepath.Split(fp)
 	ext := strings.ToLower(filepath.Ext(file))
 
+	switch file {
+	case "go.mod":
+		return heartbeat.LanguageGo, true
+	case "CMmakeLists.txt":
+		return heartbeat.LanguageCMake, true
+	}
+
 	// nolint
 	if strings.HasPrefix(ext, ".h") || strings.HasPrefix(ext, ".c") {
 		if correspondingFileExists(fp, ".c") {