\#83: Added albumID as a arg while uploading (#87)

prabhuomkar · Nov 16, 2021 · a41ac4b · a41ac4b
1 parent 394d7e0
commit a41ac4b
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 7 deletions.
diff --git a/api/internal/graph/generated/generated.go b/api/internal/graph/generated/generated.go
diff --git a/api/internal/graph/resolvers/schema.resolvers.go b/api/internal/graph/resolvers/schema.resolvers.go
@@ -28,7 +28,7 @@ const (
 	actionTypePermanent = "permanent"
 )
 
-func (r *mutationResolver) Upload(ctx context.Context, file graphql.Upload) (bool, error) {
+func (r *mutationResolver) Upload(ctx context.Context, file graphql.Upload, albumID *string) (bool, error) {
 	result, err := r.CDN.Upload(file.File, file.Filename, file.Size, "", "")
 	if err != nil {
 		return false, err
@@ -64,6 +64,22 @@ func (r *mutationResolver) Upload(ctx context.Context, file graphql.Upload) (boo
 		return false, nil
 	}
 
+	if albumID != nil {
+		albumOID, err := primitive.ObjectIDFromHex(*albumID)
+		if err != nil {
+			return false, err
+		}
+
+		_, err = r.DB.Collection(models.ColAlbums).UpdateByID(ctx, albumOID, bson.D{
+			{Key: "$set", Value: bson.D{
+				{Key: "$push", Value: bson.D{{Key: "mediaItems", Value: insertedID}}},
+			}},
+		})
+		if err != nil {
+			return false, err
+		}
+	}
+
 	return true, nil
 }
 

diff --git a/api/schema/schema.graphql b/api/schema/schema.graphql
@@ -13,7 +13,7 @@ extend type Query {
 }
 
 extend type Mutation {
-  upload(file: Upload!): Boolean!
+  upload(file: Upload!, albumId: String): Boolean!
   updateFavourite(id: String!, type: String!): Boolean!
   delete(id: String!, type: String!): Boolean!
 }