diff options
author | Lauris BH <lauris@nix.lv> | 2020-02-20 21:53:55 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-02-20 16:53:55 -0300 |
commit | 3c45cf8494fcd29e1a99b0ee6f253808eb607053 (patch) | |
tree | cd0e6347bcd2bfc42c18408169a8757ca1dda920 /modules/analyze | |
parent | efbd7ca39bde69ff84d4b309bee99edfe2977521 (diff) | |
download | gitea-3c45cf8494fcd29e1a99b0ee6f253808eb607053.tar.gz gitea-3c45cf8494fcd29e1a99b0ee6f253808eb607053.zip |
Add detected file language to code search (#10256)
Move langauge detection to separate module to be more reusable
Add option to disable vendored file exclusion from file search
Allways show all language stats for search
Diffstat (limited to 'modules/analyze')
-rw-r--r-- | modules/analyze/code_langauge.go | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/modules/analyze/code_langauge.go b/modules/analyze/code_langauge.go new file mode 100644 index 0000000000..f7dd3e7cfc --- /dev/null +++ b/modules/analyze/code_langauge.go @@ -0,0 +1,36 @@ +// Copyright 2020 The Gitea Authors. All rights reserved. +// Use of this source code is governed by a MIT-style +// license that can be found in the LICENSE file. + +package analyze + +import ( + "path/filepath" + + "github.com/src-d/enry/v2" +) + +// GetCodeLanguageWithCallback detects code language based on file name and content using callback +func GetCodeLanguageWithCallback(filename string, contentFunc func() ([]byte, error)) string { + if language, ok := enry.GetLanguageByExtension(filename); ok { + return language + } + + if language, ok := enry.GetLanguageByFilename(filename); ok { + return language + } + + content, err := contentFunc() + if err != nil { + return enry.OtherLanguage + } + + return enry.GetLanguage(filepath.Base(filename), content) +} + +// GetCodeLanguage detects code language based on file name and content +func GetCodeLanguage(filename string, content []byte) string { + return GetCodeLanguageWithCallback(filename, func() ([]byte, error) { + return content, nil + }) +} |