Use languages from heartbeat package

wakatime · Aug 29, 2023 · 2232c8b · 2232c8b
1 parent d7422d6
commit 2232c8b
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 5 deletions.
diff --git a/pkg/lexer/c_test.go b/pkg/lexer/c_test.go
@@ -5,6 +5,7 @@ import (
 	"testing"
 
 	"github.com/alecthomas/chroma/v2/lexers"
+	"github.com/wakatime/wakatime-cli/pkg/heartbeat"
 
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
@@ -29,7 +30,7 @@ func TestC_AnalyseText(t *testing.T) {
 		},
 	}
 
-	l := lexers.Get("C")
+	l := lexers.Get(heartbeat.LanguageC.StringChroma())
 	require.NotNil(t, l)
 
 	for name, test := range tests {

diff --git a/pkg/lexer/cpp_test.go b/pkg/lexer/cpp_test.go
@@ -5,6 +5,7 @@ import (
 	"testing"
 
 	"github.com/alecthomas/chroma/v2/lexers"
+	"github.com/wakatime/wakatime-cli/pkg/heartbeat"
 
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
@@ -25,7 +26,7 @@ func TestCpp_AnalyseText(t *testing.T) {
 		},
 	}
 
-	l := lexers.Get("C++")
+	l := lexers.Get(heartbeat.LanguageCPP.StringChroma())
 	require.NotNil(t, l)
 
 	for name, test := range tests {

diff --git a/pkg/lexer/cuda.go b/pkg/lexer/cuda.go
@@ -27,7 +27,7 @@ func (l CUDA) Lexer() chroma.Lexer {
 	)
 
 	lexer.SetAnalyser(func(text string) float32 {
-		c := lexers.Get("C")
+		c := lexers.Get(heartbeat.LanguageC.StringChroma())
 		if c == nil {
 			return 0
 		}

diff --git a/pkg/lexer/ec.go b/pkg/lexer/ec.go
@@ -27,7 +27,7 @@ func (l EC) Lexer() chroma.Lexer {
 	)
 
 	lexer.SetAnalyser(func(text string) float32 {
-		c := lexers.Get("C")
+		c := lexers.Get(heartbeat.LanguageC.StringChroma())
 		if c == nil {
 			return 0
 		}

diff --git a/pkg/lexer/nesc.go b/pkg/lexer/nesc.go
@@ -27,7 +27,7 @@ func (l NesC) Lexer() chroma.Lexer {
 	)
 
 	lexer.SetAnalyser(func(text string) float32 {
-		c := lexers.Get("C")
+		c := lexers.Get(heartbeat.LanguageC.StringChroma())
 		if c == nil {
 			return 0
 		}