From 205cb0bc4df8d014b6bbaecae690bf76cabea90f Mon Sep 17 00:00:00 2001
From: divyam234 <47589864+divyam234@users.noreply.github.com>
Date: Fri, 8 Dec 2023 03:35:40 +0530
Subject: [PATCH] save encrypt state in uploads

---
 pkg/database/migrations/20231208020544_alter.sql | 1 +
 pkg/mapper/mapper.go                             | 1 +
 pkg/models/upload.go                             | 1 +
 pkg/schemas/upload.go                            | 2 ++
 pkg/services/upload.go                           | 1 +
 5 files changed, 6 insertions(+)

diff --git a/pkg/database/migrations/20231208020544_alter.sql b/pkg/database/migrations/20231208020544_alter.sql
index 3a50143..3c72956 100644
--- a/pkg/database/migrations/20231208020544_alter.sql
+++ b/pkg/database/migrations/20231208020544_alter.sql
@@ -1,4 +1,5 @@
 -- +goose Up
 -- +goose StatementBegin
 ALTER TABLE teldrive.files ADD COLUMN "encrypted" BOOLEAN NOT NULL DEFAULT FALSE;
+ALTER TABLE teldrive.uploads ADD COLUMN "encrypted" BOOLEAN NOT NULL DEFAULT FALSE;
 -- +goose StatementEnd
\ No newline at end of file
diff --git a/pkg/mapper/mapper.go b/pkg/mapper/mapper.go
index 9d8d85b..8149df8 100644
--- a/pkg/mapper/mapper.go
+++ b/pkg/mapper/mapper.go
@@ -46,6 +46,7 @@ func ToUploadOut(in *models.Upload) *schemas.UploadPartOut {
 		ChannelID: in.ChannelID,
 		PartNo:    in.PartNo,
 		Size:      in.Size,
+		Encrypted: in.Encrypted,
 	}
 	return out
 }
diff --git a/pkg/models/upload.go b/pkg/models/upload.go
index 222b39c..417cf9a 100644
--- a/pkg/models/upload.go
+++ b/pkg/models/upload.go
@@ -10,6 +10,7 @@ type Upload struct {
 	Name      string    `gorm:"type:text"`
 	PartNo    int       `gorm:"type:integer"`
 	PartId    int       `gorm:"type:integer"`
+	Encrypted bool      `gorm:"default:false"`
 	ChannelID int64     `gorm:"type:bigint"`
 	Size      int64     `gorm:"type:bigint"`
 	CreatedAt time.Time `gorm:"default:timezone('utc'::text, now())"`
diff --git a/pkg/schemas/upload.go b/pkg/schemas/upload.go
index e412af5..5e4f7ab 100644
--- a/pkg/schemas/upload.go
+++ b/pkg/schemas/upload.go
@@ -13,6 +13,7 @@ type UploadPartOut struct {
 	PartNo    int    `json:"partNo"`
 	ChannelID int64  `json:"channelId"`
 	Size      int64  `json:"size"`
+	Encrypted bool   `json:"encrypted"`
 }
 
 type UploadOut struct {
@@ -26,4 +27,5 @@ type UploadPart struct {
 	PartNo    int    `json:"partNo"`
 	ChannelID int64  `json:"channelId"`
 	Size      int64  `json:"size"`
+	Encrypted bool   `json:"encrypted"`
 }
diff --git a/pkg/services/upload.go b/pkg/services/upload.go
index 58a482d..93167b5 100644
--- a/pkg/services/upload.go
+++ b/pkg/services/upload.go
@@ -201,6 +201,7 @@ func (us *UploadService) UploadFile(c *gin.Context) (*schemas.UploadPartOut, *ty
 			Size:      fileSize,
 			PartNo:    uploadQuery.PartNo,
 			UserId:    userId,
+			Encrypted: uploadQuery.Encrypted,
 		}
 
 		if err := us.Db.Create(partUpload).Error; err != nil {