mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-30 06:16:37 +08:00
ryan
9af84c8ed6
Replace c.JSON(200, response.Err) and middleware gin.H bypasses with response.Abort* helpers so errors flow through Gin Error chain and ErrorHandlerMiddleware for OTel trace correlation. Split oauth/sources.go into domain-focused files and decompose upload into handler/filesrv/stats/task/cache/storage/util subpackages with a root facade preserving existing import paths.
43 lines
1.1 KiB
Go
43 lines
1.1 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Package stats maintains incremental upload statistics and aggregations.
|
|
package stats
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/apps/upload/util"
|
|
)
|
|
|
|
const (
|
|
catImage = "图片"
|
|
catVideo = "视频"
|
|
catAudio = "音频"
|
|
catDocument = "文档"
|
|
catArchive = "压缩包"
|
|
catOther = "其他"
|
|
)
|
|
|
|
// GetFileCategory classifies a file by mime type and extension.
|
|
func GetFileCategory(mimeType, ext string) string {
|
|
mimeType = strings.ToLower(mimeType)
|
|
ext = strings.ToLower(ext)
|
|
|
|
if strings.HasPrefix(mimeType, "image/") || util.IsImageExtension(ext) {
|
|
return catImage
|
|
}
|
|
if strings.HasPrefix(mimeType, "video/") {
|
|
return catVideo
|
|
}
|
|
if strings.HasPrefix(mimeType, "audio/") {
|
|
return catAudio
|
|
}
|
|
if util.IsArchiveExtension(ext) || strings.Contains(mimeType, "zip") || strings.Contains(mimeType, "tar") || strings.Contains(mimeType, "gzip") {
|
|
return catArchive
|
|
}
|
|
if util.IsDocumentExtension(ext) || strings.HasPrefix(mimeType, "text/") || mimeType == "application/pdf" {
|
|
return catDocument
|
|
}
|
|
return catOther
|
|
} |