From fb1112f3a2adcc15217afebc19aac6793ad1e76b Mon Sep 17 00:00:00 2001
From: Derrick Hammer <derrick@derrickhammer.com>
Date: Mon, 22 Jan 2024 19:08:56 -0500
Subject: [PATCH] fix: we need to pin the file after creating the upload

---
 storage/storage.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/storage/storage.go b/storage/storage.go
index c111d02..f26bb78 100644
--- a/storage/storage.go
+++ b/storage/storage.go
@@ -564,12 +564,18 @@ func (s *StorageServiceImpl) buildNewTusUploadTask(upload *models.TusUpload) (jo
 				return err
 			}
 
-			_, err = s.CreateUpload(dbHash, upload.UploaderID, upload.UploaderIP, uint64(byteCount), upload.Protocol)
+			newUpload, err := s.CreateUpload(dbHash, upload.UploaderID, upload.UploaderIP, uint64(byteCount), upload.Protocol)
 			if err != nil {
 				s.portal.Logger().Error("Could not create upload", zap.Error(err))
 				return err
 			}
 
+			err = s.portal.Accounts().PinByID(newUpload.ID, upload.UploaderID)
+			if err != nil {
+				s.portal.Logger().Error("Could not pin upload", zap.Error(err))
+				return err
+			}
+
 			return nil
 		}, upload)