fix: file extension handling in GetCategory function

tgdrive · Apr 19, 2024 · a773482 · a773482
1 parent 3ca28cc
commit a773482
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/internal/category/category.go b/internal/category/category.go
@@ -25,7 +25,11 @@ var (
 )
 
 func GetCategory(fileName string) Category {
-	fileExtension := strings.Split(strings.ToLower(filepath.Ext(fileName)), ".")[1]
+	fileExtension := filepath.Ext(fileName)
+	if fileExtension != "" {
+		fileExtension = strings.ToLower(fileExtension[1:])
+
+	}
 
 	if contains(documentExtensions, fileExtension) {
 		return Document

diff --git a/pkg/services/file.go b/pkg/services/file.go
@@ -11,7 +11,7 @@ import (
 	"strings"
 
 	"github.com/divyam234/teldrive/internal/cache"
-	category "github.com/divyam234/teldrive/internal/category"
+	"github.com/divyam234/teldrive/internal/category"
 	"github.com/divyam234/teldrive/internal/config"
 	"github.com/divyam234/teldrive/internal/database"
 	"github.com/divyam234/teldrive/internal/http_range"