From 1a1b9eb55671bcff94335eb4667398ffd60d6698 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Mon, 24 Mar 2025 21:39:21 +0800
Subject: [PATCH] :sparkles: Allow to set preferred storage destination

---
 pkg/internal/services/analyzer.go | 6 +++++-
 settings.toml                     | 2 ++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/pkg/internal/services/analyzer.go b/pkg/internal/services/analyzer.go
index 29795e4..dae1a5a 100644
--- a/pkg/internal/services/analyzer.go
+++ b/pkg/internal/services/analyzer.go
@@ -291,7 +291,11 @@ func AnalyzeAttachment(file models.Attachment) error {
 	if !linked {
 		go func() {
 			start = time.Now()
-			if err := ReUploadFile(file, 1); err != nil {
+			preferred := viper.GetInt("preferred_destination")
+			if preferred == 0 {
+				preferred = 1
+			}
+			if err := ReUploadFile(file, preferred); err != nil {
 				log.Warn().Any("file", file).Err(err).Msg("Unable to move file to permanet storage...")
 			} else {
 				// Recycle the temporary file
diff --git a/settings.toml b/settings.toml
index 3af723e..a782f13 100644
--- a/settings.toml
+++ b/settings.toml
@@ -5,6 +5,8 @@ grpc_bind = "0.0.0.0:7004"
 
 nexus_addr = "localhost:7001"
 
+preferred_destination = 1
+
 [workers]
 files_deletion = 4
 files_analyze = 4