Restart workspace

presleyp · presleyp · commit 09d2214c41a6 · 2022-12-07T22:59:08.000Z
diff --git a/site/src/pages/WorkspacePage/WorkspacePage.tsx b/site/src/pages/WorkspacePage/WorkspacePage.tsx
@@ -3,7 +3,7 @@ import { useMachine } from "@xstate/react"
 import { AlertBanner } from "components/AlertBanner/AlertBanner"
 import { ChooseOne, Cond } from "components/Conditionals/ChooseOne"
 import { FC, useEffect } from "react"
-import { useParams } from "react-router-dom"
+import { useLocation, useParams } from "react-router-dom"
 import { Loader } from "components/Loader/Loader"
 import { firstOrItem } from "util/array"
 import { workspaceMachine } from "xServices/workspace/workspaceXService"
@@ -15,6 +15,7 @@ export const WorkspacePage: FC = () => {
     useParams()
   const username = firstOrItem(usernameQueryParam, null)
   const workspaceName = firstOrItem(workspaceQueryParam, null)
+  const shouldRestartWorkspace = useLocation().state?.shouldRestartWorkspace ?? false
   const [workspaceState, workspaceSend] = useMachine(workspaceMachine)
   const {
     workspace,
@@ -69,6 +70,7 @@ export const WorkspacePage: FC = () => {
           workspaceState={workspaceState}
           quotaState={quotaState}
           workspaceSend={workspaceSend}
+          shouldRestartWorkspace={shouldRestartWorkspace}
         />
       </Cond>
       <Cond>
diff --git a/site/src/pages/WorkspacePage/WorkspaceReadyPage.tsx b/site/src/pages/WorkspacePage/WorkspaceReadyPage.tsx
@@ -30,12 +30,14 @@ interface WorkspaceReadyPageProps {
   workspaceState: StateFrom<typeof workspaceMachine>
   quotaState: StateFrom<typeof quotaMachine>
   workspaceSend: (event: WorkspaceEvent) => void
+  shouldRestartWorkspace: boolean
 }
 
 export const WorkspaceReadyPage = ({
   workspaceState,
   quotaState,
   workspaceSend,
+  shouldRestartWorkspace
 }: WorkspaceReadyPageProps): JSX.Element => {
   const [bannerState, bannerSend] = useActor(
     workspaceState.children["scheduleBannerMachine"],
@@ -71,6 +73,13 @@ export const WorkspaceReadyPage = ({
     bannerSend({ type: "REFRESH_WORKSPACE", workspace })
   }, [bannerSend, workspace])
 
+  useEffect(() => {
+    // workspaces can be started without being stopped first
+    shouldRestartWorkspace && workspaceSend({ type: "START" })
+    // clear location state
+    window.history.replaceState({}, document.title)
+  }, [shouldRestartWorkspace, workspaceSend])
+
   return (
     <>
       <Helmet>
