nyne/nysoure
1
0
Fork 0
mirror of https://github.com/wgh136/nysoure.git synced 2025-09-27 04:17:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

enhance file download handling to track real users

Browse Source
This commit is contained in:
nyne
2025-09-01 17:51:06 +08:00
parent 44b876ba0e
commit b804741e27
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
server/api/file.go
View File

@@ -205,7 +205,7 @@ func deleteFile(c fiber.Ctx) error {
func downloadFile(c fiber.Ctx) error {
cfToken := c.Query("cf_token")
s, filename, err := service.DownloadFile(c.Params("id"), cfToken)
s, filename, err := service.DownloadFile(c.Params("id"), cfToken, c.Locals("real_user") == true)
if err != nil {
return err
}

4
server/service/file.go
View File

@@ -389,7 +389,7 @@ func GetFile(fid string) (*model.FileView, error) {
}
// DownloadFile handles the file download request. Return a presigned URL or a direct file path.
func DownloadFile(fid, cfToken string) (string, string, error) {
func DownloadFile(fid, cfToken string, isRealUser bool) (string, string, error) {
file, err := dao.GetFile(fid)
if err != nil {
log.Error("failed to get file: ", err)
@@ -436,10 +436,12 @@ func DownloadFile(fid, cfToken string) (string, string, error) {
return "", "", model.NewInternalServerError("failed to download file from storage")
}
if isRealUser {
err = dao.AddResourceDownloadCount(file.ResourceID)
if err != nil {
log.Errorf("failed to add resource download count: %v", err)
}
}
return path, file.Filename, nil
}