From 285d0e86bf7b0022a1d5f524998c47cd313435ad Mon Sep 17 00:00:00 2001
From: Shreerang Kale <shreerangkale@gmail.com>
Date: Wed, 5 Feb 2025 16:02:47 +0530
Subject: [PATCH] Add custom domain support for auctions flow

---
 packages/backend/src/resolvers.ts                            | 5 +++--
 packages/backend/src/service.ts                              | 4 ++++
 .../projects/project/deployments/DeploymentMenu.tsx          | 1 +
 packages/gql-client/src/client.ts                            | 4 +++-
 4 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/packages/backend/src/resolvers.ts b/packages/backend/src/resolvers.ts
index d9f9128d..401f3c67 100644
--- a/packages/backend/src/resolvers.ts
+++ b/packages/backend/src/resolvers.ts
@@ -345,10 +345,11 @@ export const createResolvers = async (service: Service): Promise<any> => {
         {
           projectId,
           deploymentId,
-        }: { deploymentId: string; projectId: string },
+          deployerId,
+        }: { deploymentId: string; projectId: string, deployerId: string },
       ) => {
         try {
-          return await service.rollbackDeployment(projectId, deploymentId);
+          return await service.rollbackDeployment(projectId, deploymentId, deployerId);
         } catch (err) {
           log(err);
           return false;
diff --git a/packages/backend/src/service.ts b/packages/backend/src/service.ts
index 207aabfe..4ae6a5b8 100644
--- a/packages/backend/src/service.ts
+++ b/packages/backend/src/service.ts
@@ -1193,6 +1193,7 @@ export class Service {
   async rollbackDeployment(
     projectId: string,
     deploymentId: string,
+    deployerId: string,
   ): Promise<boolean> {
     // TODO: Implement transactions
     const oldCurrentDeployment = await this.db.getDeployment({
@@ -1204,6 +1205,9 @@ export class Service {
         project: {
           id: projectId,
         },
+        deployer: {
+          deployerId
+        },
         isCurrent: true,
         isCanonical: false,
       },
diff --git a/packages/frontend/src/components/projects/project/deployments/DeploymentMenu.tsx b/packages/frontend/src/components/projects/project/deployments/DeploymentMenu.tsx
index cb87aa98..fad462ee 100644
--- a/packages/frontend/src/components/projects/project/deployments/DeploymentMenu.tsx
+++ b/packages/frontend/src/components/projects/project/deployments/DeploymentMenu.tsx
@@ -99,6 +99,7 @@ export const DeploymentMenu = ({
     const isRollbacked = await client.rollbackDeployment(
       project.id,
       deployment.id,
+      deployment.deployer.deployerId
     );
     if (isRollbacked.rollbackDeployment) {
       await onUpdate();
diff --git a/packages/gql-client/src/client.ts b/packages/gql-client/src/client.ts
index 49ddbb60..7d272670 100644
--- a/packages/gql-client/src/client.ts
+++ b/packages/gql-client/src/client.ts
@@ -337,13 +337,15 @@ export class GQLClient {
 
   async rollbackDeployment(
     projectId: string,
-    deploymentId: string
+    deploymentId: string,
+    deployerId: string,
   ): Promise<types.RollbackDeploymentResponse> {
     const { data } = await this.client.mutate({
       mutation: mutations.rollbackDeployment,
       variables: {
         projectId,
         deploymentId,
+        deployerId,
       },
     });