diff --git a/pkg/lexer/c_test.go b/pkg/lexer/c_test.go index b933e69e..f06aecb4 100644 --- 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 index 46ece49c..70e5a7a0 100644 --- 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 index 37c14b5b..307f4e02 100644 --- 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 index e0cafb32..6de0eca3 100644 --- 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 index f306db51..1c1b1b17 100644 --- 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 }