初始提交: Gitea 项目代码

modules/analyze/code_language.go

@@ -0,0 +1,28 @@
+// Copyright 2020 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package analyze
+
+import (
+	"path"
+
+	"github.com/go-enry/go-enry/v2"
+)
+
+// GetCodeLanguage detects code language based on file name and content
+// It can be slow when the content is used for detection
+func GetCodeLanguage(filename string, content []byte) string {
+	if language, ok := enry.GetLanguageByExtension(filename); ok {
+		return language
+	}
+
+	if language, ok := enry.GetLanguageByFilename(filename); ok {
+		return language
+	}
+
+	if len(content) == 0 {
+		return enry.OtherLanguage
+	}
+
+	return enry.GetLanguage(path.Base(filename), content)
+}