fix: precedence in case of multiple matching language mappings (fix #172)

models/heartbeat.go

@@ -34,10 +34,11 @@ func (h *Heartbeat) Valid() bool {
 }
 
 func (h *Heartbeat) Augment(languageMappings map[string]string) {
+	maxPrec := -1 // precision / mapping complexity -> more concrete ones shall take precedence
 	for ending, value := range languageMappings {
-		if strings.HasSuffix(h.Entity, "."+ending) {
+		if ok, prec := strings.HasSuffix(h.Entity, "."+ending), strings.Count(ending, "."); ok && prec > maxPrec {
 			h.Language = value
-			return
+			maxPrec = prec
 		}
 	}
 }

models/heartbeat_test.go

@@ -28,22 +28,28 @@ func TestHeartbeat_Augment(t *testing.T) {
 	testMappings := map[string]string{
 		"py":        "Python3",
 		"foo":       "Foo Script",
+		"php":       "PHP 8",
 		"blade.php": "Blade",
 	}
 
-	sut1, sut2 := &Heartbeat{
+	sut1, sut2, sut3 := &Heartbeat{
 		Entity:   "~/dev/file.py",
 		Language: "Python",
 	}, &Heartbeat{
 		Entity:   "~/dev/file.blade.php",
 		Language: "unknown",
+	}, &Heartbeat{
+		Entity:   "~/dev/file.php",
+		Language: "PHP",
 	}
 
 	sut1.Augment(testMappings)
 	sut2.Augment(testMappings)
+	sut3.Augment(testMappings)
 
 	assert.Equal(t, "Python3", sut1.Language)
 	assert.Equal(t, "Blade", sut2.Language)
+	assert.Equal(t, "PHP 8", sut3.Language)
 }
 
 func TestHeartbeat_GetKey(t *testing.T) {

scripts/sample_data.py

@@ -17,7 +17,10 @@ LANGUAGES = {
     'Go': 'go',
     'Java': 'java',
     'JavaScript': 'js',
-    'Python': 'py'
+    'Python': 'py',
+    # https://github.com/muety/wakapi/issues/172
+    'PHP': 'php',
+    'Blade': 'blade.php'
 }