From 11da11c326133b0bd59ba833213adbbaaeafc2eb Mon Sep 17 00:00:00 2001
From: karamvir <karamvir.singh98@gmail.com>
Date: Sat, 5 Aug 2023 23:22:56 -0700
Subject: [PATCH] fix deployment action button show states

---
 .../src/resources/deployment/components/actions.tsx   | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/frontend/src/resources/deployment/components/actions.tsx b/frontend/src/resources/deployment/components/actions.tsx
index 81b54f676..a6b3b5f15 100644
--- a/frontend/src/resources/deployment/components/actions.tsx
+++ b/frontend/src/resources/deployment/components/actions.tsx
@@ -1,5 +1,5 @@
 import { ActionWithDialog, ConfirmButton } from "@components/util";
-import { RefreshCw, Play, Trash, Pause } from "lucide-react";
+import { Play, Trash, Pause, Rocket } from "lucide-react";
 import { useExecute, useRead } from "@hooks";
 import { DockerContainerState } from "@monitor/client/dist/types";
 import { useNavigate } from "react-router-dom";
@@ -16,7 +16,7 @@ export const RedeployContainer = ({ deployment_id }: DeploymentId) => {
     <ConfirmButton
       title={deployment?.status ? "Redeploy" : "Deploy"}
       intent="success"
-      icon={<RefreshCw className="h-4 w-4" />}
+      icon={<Rocket className="h-4 w-4" />}
       onClick={() => mutate({ deployment_id })}
       disabled={isLoading}
     />
@@ -60,6 +60,9 @@ const StopContainer = ({ deployment_id }: DeploymentId) => {
 export const StartOrStopContainer = ({ deployment_id }: DeploymentId) => {
   const deployments = useRead("ListDeployments", {}).data;
   const deployment = deployments?.find((d) => d.id === deployment_id);
+
+  if (deployment?.state === DockerContainerState.NotDeployed) return null;
+
   if (deployment?.state === DockerContainerState.Running)
     return <StopContainer deployment_id={deployment_id} />;
   return <StartContainer deployment_id={deployment_id} />;
@@ -69,6 +72,10 @@ export const RemoveContainer = ({ deployment_id }: DeploymentId) => {
   const { data: d } = useRead("GetDeployment", { id: deployment_id });
   const { mutate, isLoading } = useExecute("RemoveContainer");
 
+  const deployments = useRead("ListDeployments", {}).data;
+  const deployment = deployments?.find((d) => d.id === deployment_id);
+  if (deployment?.state === DockerContainerState.NotDeployed) return null;
+
   if (!d) return null;
   return (
     <ActionWithDialog