[preprocessor/folder] add project based file attribute

likecoin · May 30, 2024 · 49e9f94 · 49e9f94
1 parent 01f3a29
commit 49e9f94
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/preprocessor/folder/file.go b/preprocessor/folder/file.go
@@ -11,6 +11,7 @@ import (
 	"os"
 	"path/filepath"
 	"slices"
+	"strings"
 	"time"
 
 	"github.com/jackc/pgx/v5/pgxpool"
@@ -53,6 +54,8 @@ func getFileMetadata(filePath string) (map[string]any, error) {
 	}
 	mediaType := http.DetectContentType(bytes)
 
+	syncRoot := config.GetConfig().FolderPreprocessor.SyncFolderRoot
+
 	return map[string]any{
 		"sha256":        hex.EncodeToString(sha.Sum(nil)),
 		"md5":           hex.EncodeToString(md.Sum(nil)),
@@ -62,6 +65,7 @@ func getFileMetadata(filePath string) (map[string]any, error) {
 		"file_name":     fileInfo.Name(),
 		"last_modified": fileInfo.ModTime().Format(time.RFC3339),
 		"time_created":  fileInfo.ModTime().Format(time.RFC3339),
+		"asset_origin":  strings.TrimPrefix(filePath, syncRoot),
 	}, nil
 }