Update xstate machine

presleyp · presleyp · commit e3fb8cc9fddf · 2022-12-07T21:27:33.000Z
diff --git a/site/src/xServices/workspaceSchedule/workspaceScheduleXService.ts b/site/src/xServices/workspaceSchedule/workspaceScheduleXService.ts
@@ -5,7 +5,6 @@
 import { assign, createMachine } from "xstate"
 import * as API from "../../api/api"
 import * as TypesGen from "../../api/typesGenerated"
-import { displaySuccess } from "../../components/GlobalSnackbar/utils"
 
 export const Language = {
   successMessage: "Successfully updated workspace schedule.",
@@ -24,6 +23,8 @@ export interface WorkspaceScheduleContext {
   permissions?: Permissions
   checkPermissionsError?: Error | unknown
   submitScheduleError?: Error | unknown
+  autoStopChanged?: boolean
+  shouldRestartWorkspace?: boolean
 }
 
 export const checks = {
@@ -50,8 +51,12 @@ export type WorkspaceScheduleEvent =
       ttl: TypesGen.UpdateWorkspaceTTLRequest
       autoStopChanged: boolean
     }
+  | { type: "RESTART_WORKSPACE" }
+  | { type: "APPLY_LATER" }
 
-export const workspaceSchedule = createMachine(
+export const workspaceSchedule =
+/** @xstate-layout N4IgpgJg5mDOIC5QHcD2AnA1rADgQwGMwBlAgC0gFcAbEgFzzrAGIBxAUQBUB9AdQHkASgGliABQCCAYXYBtAAwBdRKBypYASzobUAOxUgAHogCMAJgB0AVgAcAThsAWeXZMA2O2YDsdq1YA0IACeiGbyFl428o4AzFYmMT4xbl5mAL5pgWhYuIQk5FS0xAxMFjB02rpQvBjY+ETMEHpgFhq6AG6omC3lNTn1YArKSCBqmtp6BsYI7tbybmZ+9mYO8u6OgSEIMY6OFo5eC34mdqkLNhlZtblEpBQQNPSMPWAVbdXXA8xg6OgYFjhqIwAGYYAC2ZVefTqeSGBjGWh0+hG02SNmsXhMVhWdkcbkcNm8Xk2oRcFgW7hi8h8djcVkuIGyMNuBQeRRKLzeVTEPzBGlgmj0sEazVaHS6LQKBEwPPQfIFSNgcJGCImyNA0wAtOEYrjsa4vD5bMkEiSEG44uTFosCa41lY3AymTd8vdHsVnpCuVBZfLBbphT8-ugAUC6KC5RYpTLefz-UqlPD1IjJijEMkrBE3G4bO5cfN5GYzGadnsDgt5rmnFZHL4nZ88ndCk9Sjh0HAwLo6AAxcHMYgAVQAQgBZACSPGIUgAEuwACIDgAyckTKuTaqmiEzqTMdOzJixJi82OLwUQNic5MxrjxusS5nr-UbrPdHIssEoACM+d6m2yWE0ugtG0nTdO+X4-n+jzKqo65IpuMx7CYF7ZjYMSJLihpeBsZ4zO4Jj7HahomPINaEo4j7Mq6zYeqUH7flolRQFBtAikBYqgS09GQS+tCyCYwyweM8Fpggjg1hYu5UrqZiOCsjjmGaB5uARtYkfq7jUjYjqZIyDYsm67KetxjHvCxLBBv8gIguC4EMXQ5kwaMcGphq6YpBEB4JFYuqxN4Jhmq4bj7CkiReNEpykSYlEuuZtFcWQqDIO8ghwAw6B0HOGh4NQqBQMwgjsMQnASIIPACCI4jSCugnOcJrlGOeMT7LYyExMhClybESmJMFrU2OFDo+OYOlXE+Bk0W+sCJclVSpbA6WZdluX5RIYhiIuACa3CLhInDsIITmqiJbnbG4Oq1gpaxOHmMRKWY2kWPYObnehOwxDFAxxW+lnoGwXB8EIoiSDIR0ueqjXbFYdgWKcVZmCN5x+EpBJPa4ngmLE8jtV4GS6boqAQHABjOl9vEtmASb1RDWqo75GmGr4aFuGampYpYTgI-IUS+Nzhy47ppPPoZFOtBAtBUymNOmOEUQ1nidJWMe1IPazdIWDsXMHseXgxFEAtjVR32euUTHQi6ksbqJXgWPI8y1rJix23ESso3sdvRLrKxmDEjvpIL+nUf+8VekxvpxoqlsnZD6K5nYLhHnJNjYmsZoEjbuu+ASJqdbrn3C5Nnpth2Xa9nKUcNdMuxPdjaFOFS2ErGakTNQjDu3qkJF2PnE3B1NEGmVU5kV9LMzhNDSvzBa1IqZ4OFbLSMMWj7OZoXrhIfQH41B6+xkzSlaV4BlWU5XlI8ISReyYbJ5h2A4cnI7hI2ZtmcSRNpOZxP7huxeTIe-efUSthMyyU8PHO+R4LRKQks4HMmMDhfzWNFLeRs-5vkApTNc1MEK6hAe4c6pE6QDTCCzJ+tZY4nD1qkdCqQBYZCAA */
+createMachine(
   {
     id: "workspaceScheduleState",
     predictableActionArguments: true,
@@ -110,37 +115,45 @@ export const workspaceSchedule = createMachine(
       },
       presentForm: {
         on: {
-          SUBMIT_SCHEDULE: "submittingSchedule",
+          SUBMIT_SCHEDULE: { target: "submittingSchedule", actions: "assignAutoStopChanged" },
         },
       },
       submittingSchedule: {
         invoke: {
           src: "submitSchedule",
           id: "submitSchedule",
-          onDone: {
-            target: "submitSuccess",
-            actions: "displaySuccess",
-          },
+          onDone: [
+            {
+              cond: "autoStartChanged",
+              target: "showingRestartDialog",
+            },
+            { target: "done" }
+          ],
           onError: {
             target: "presentForm",
             actions: ["assignSubmissionError"],
           },
         },
         tags: "loading",
       },
-      submitSuccess: {
+      showingRestartDialog: {
         on: {
-          SUBMIT_SCHEDULE: "submittingSchedule",
+          RESTART_WORKSPACE: { target: "done", actions: "assignRestartWorkspace" },
+          APPLY_LATER: "done"
         },
       },
       error: {
         on: {
           GET_WORKSPACE: "gettingWorkspace",
         },
       },
+      done: {}
     },
   },
   {
+    guards: {
+      autoStartChanged: (context) => Boolean(context.autoStopChanged)
+    },
     actions: {
       assignSubmissionError: assign({
         submitScheduleError: (_, event) => event.data,
@@ -159,6 +172,12 @@ export const workspaceSchedule = createMachine(
       assignGetPermissionsError: assign({
         checkPermissionsError: (_, event) => event.data,
       }),
+      assignAutoStopChanged: assign({
+        autoStopChanged: (_) => true
+      }),
+      assignRestartWorkspace: assign({
+        shouldRestartWorkspace: (_) => true
+      }),
       clearGetPermissionsError: assign({
         checkPermissionsError: (_) => undefined,
       }),
@@ -168,9 +187,6 @@ export const workspaceSchedule = createMachine(
       clearGetWorkspaceError: (context) => {
         assign({ ...context, getWorkspaceError: undefined })
       },
-      displaySuccess: () => {
-        displaySuccess(Language.successMessage)
-      },
     },
 
     services: {
