tgdrive/teldrive
1
1
Fork 0
mirror of https://github.com/tgdrive/teldrive.git synced 2024-09-20 08:15:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: file extension handling in GetCategory function

Browse source
This commit is contained in:
divyam234 2024-04-19 23:57:39 +05:30
parent 3ca28cc06d
commit a7734821a1
2 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
internal/category/category.go
View file

@ -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

2
pkg/services/file.go
View file

@ -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"