Extract hasPermittedExtension function

hrs · Jun 8, 2023 · 21e8fad · 21e8fad
1 parent 0b4b3b6
commit 21e8fad
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/corpus/document.go b/corpus/document.go
@@ -135,7 +135,7 @@ func (doc *Document) calcNorm() float64 {
 
 func isTextFile(path string) bool {
 	// If the file's extension is explicitly permitted, just use that.
-	if permitted, _ := explicitlyPermittedExtensions[filepath.Ext(path)]; permitted {
+	if hasPermittedExtension(path) {
 		return true
 	}
 
@@ -169,3 +169,9 @@ func isTextFile(path string) bool {
 
 	return strings.HasPrefix(mimeType, "text/")
 }
+
+func hasPermittedExtension(path string) bool {
+	ext := strings.TrimPrefix(filepath.Ext(path), ".")
+	permitted, ok := explicitlyPermittedExtensions[ext]
+	return ok && permitted
+}