From fa954c3bbd9001ec19b602a77547d40ab50f67bc Mon Sep 17 00:00:00 2001
From: Mauricio Siu <47042324+Siumauricio@users.noreply.github.com>
Date: Sat, 15 Mar 2025 18:36:40 -0600
Subject: [PATCH] refactor: remove console log statements on WebSocket
 connection close and adjust compose file handling based on source type

---
 apps/dokploy/server/wss/listen-deployment.ts |  2 --
 packages/server/src/services/compose.ts      | 12 ++++++++----
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/apps/dokploy/server/wss/listen-deployment.ts b/apps/dokploy/server/wss/listen-deployment.ts
index 4a25c6f0e..581c04d7b 100644
--- a/apps/dokploy/server/wss/listen-deployment.ts
+++ b/apps/dokploy/server/wss/listen-deployment.ts
@@ -61,7 +61,6 @@ export const setupDeploymentLogsWebSocketServer = (
 							}
 							stream
 								.on("close", () => {
-									console.log("Connection closed ✅");
 									client.end();
 									ws.close();
 								})
@@ -86,7 +85,6 @@ export const setupDeploymentLogsWebSocketServer = (
 					});
 
 				ws.on("close", () => {
-					console.log("Connection closed ✅, From WS");
 					client.end();
 				});
 			} else {
diff --git a/packages/server/src/services/compose.ts b/packages/server/src/services/compose.ts
index a3ebc26ce..54aa5f1e0 100644
--- a/packages/server/src/services/compose.ts
+++ b/packages/server/src/services/compose.ts
@@ -289,11 +289,11 @@ export const rebuildCompose = async ({
 		// if (admin.cleanupCacheOnCompose) {
 		// 	await cleanupFullDocker(compose?.serverId);
 		// }
-		if (compose.serverId) {
-			await getBuildComposeCommand(compose, deployment.logPath);
-		} else {
-			await buildCompose(compose, deployment.logPath);
+
+		if (compose.sourceType === "raw") {
+			await createComposeFile(compose, deployment.logPath);
 		}
+		await buildCompose(compose, deployment.logPath);
 
 		await updateDeploymentStatus(deployment.deploymentId, "done");
 		await updateCompose(composeId, {
@@ -433,6 +433,10 @@ export const rebuildRemoteCompose = async ({
 		// if (admin.cleanupCacheOnCompose) {
 		// 	await cleanupFullDocker(compose?.serverId);
 		// }
+		if (compose.sourceType === "raw") {
+			const command = getCreateComposeFileCommand(compose, deployment.logPath);
+			await execAsyncRemote(compose.serverId, command);
+		}
 		if (compose.serverId) {
 			await getBuildComposeCommand(compose, deployment.logPath);
 		}