initialization

coder · greyscaled · May 26, 2022 · May 24, 2022 · May 24, 2022 · May 24, 2022
commit ebc49651cca65431d7bb4799f63140076eb2d95a
diff --git a/site/src/pages/WorkspaceSchedulePage/WorkspaceSchedulePage.test.tsx b/site/src/pages/WorkspaceSchedulePage/WorkspaceSchedulePage.test.tsx
@@ -1,6 +1,7 @@
 import * as TypesGen from "../../api/typesGenerated"
 import { WorkspaceScheduleFormValues } from "../../components/WorkspaceStats/WorkspaceScheduleForm"
-import { formValuesToAutoStartRequest, formValuesToTTLRequest } from "./WorkspaceSchedulePage"
+import * as Mocks from "../../testHelpers/entities"
+import { formValuesToAutoStartRequest, formValuesToTTLRequest, workspaceToInitialValues } from "./WorkspaceSchedulePage"
 
 const validValues: WorkspaceScheduleFormValues = {
   sunday: false,
@@ -149,4 +150,73 @@ describe("WorkspaceSchedulePage", () => {
       expect(formValuesToTTLRequest(values)).toEqual(request)
     })
   })
+
+  describe("workspaceToInitialValues", () => {
+    it.each<[TypesGen.Workspace, WorkspaceScheduleFormValues]>([
+      // Empty case
+      [
+        {
+          ...Mocks.MockWorkspace,
+          autostart_schedule: "",
+          ttl: undefined,
+        },
+        {
+          sunday: false,
+          monday: false,
+          tuesday: false,
+          wednesday: false,
+          thursday: false,
+          friday: false,
+          saturday: false,
+          startTime: "",
+          timezone: "",
+          ttl: 0,
+        },
+      ],
+
+      // Basic case: 9:30 1-5 UTC running for 2 hours
+      [
+        {
+          ...Mocks.MockWorkspace,
+          autostart_schedule: "30 9 * * 1-5",
+          ttl: 7_200_000_000_000,
+        },
+        {
+          sunday: false,
+          monday: true,
+          tuesday: true,
+          wednesday: true,
+          thursday: true,
+          friday: true,
+          saturday: false,
+          startTime: "09:30",
+          timezone: "",
+          ttl: 2,
+        },
+      ],
+
+      // Complex case: 4:20 1 3-4 6 Canada/Eastern for 8 hours
+      [
+        {
+          ...Mocks.MockWorkspace,
+          autostart_schedule: "CRON_TZ=Canada/Eastern 20 16 * * 1,3-4,6",
+          ttl: 28_800_000_000_000,
+        },
+        {
+          sunday: false,
+          monday: true,
+          tuesday: false,
+          wednesday: true,
+          thursday: true,
+          friday: false,
+          saturday: true,
+          startTime: "16:20",
+          timezone: "Canada/Eastern",
+          ttl: 8,
+        },
+      ],
+    ])(`workspaceToInitialValues(%p) returns %p`, (workspace, formValues) => {
+      expect(workspaceToInitialValues(workspace)).toEqual(formValues)
+    })
+  })
 })
diff --git a/site/src/pages/WorkspaceSchedulePage/WorkspaceSchedulePage.tsx b/site/src/pages/WorkspaceSchedulePage/WorkspaceSchedulePage.tsx
@@ -9,6 +9,7 @@ import {
   WorkspaceScheduleFormValues,
 } from "../../components/WorkspaceStats/WorkspaceScheduleForm"
 import { firstOrItem } from "../../util/array"
+import { dowToWeeklyFlag, extractTimezone, stripTimezone } from "../../util/schedule"
 import { workspaceSchedule } from "../../xServices/workspaceSchedule/workspaceScheduleXService"
 
 export const formValuesToAutoStartRequest = (
@@ -81,6 +82,51 @@ export const formValuesToTTLRequest = (values: WorkspaceScheduleFormValues): Typ
   }
 }
 
+export const workspaceToInitialValues = (workspace: TypesGen.Workspace): WorkspaceScheduleFormValues => {
+  const schedule = workspace.autostart_schedule
+
+  if (!schedule) {
+    return {
+      sunday: false,
+      monday: false,
+      tuesday: false,
+      wednesday: false,
+      thursday: false,
+      friday: false,
+      saturday: false,
+      startTime: "",
+      timezone: "",
+      ttl: 0,
+    }
+  }
+
+  const timezone = extractTimezone(schedule, "")
+  const cronString = stripTimezone(schedule)
+
+  // parts has the following format: "mm HH * * dow"
+  const parts = cronString.split(" ")
+
+  // -> we skip month and day-of-month
+  const mm = parts[0]
+  const HH = parts[1]
+  const dow = parts[4]
+
+  const weeklyFlags = dowToWeeklyFlag(dow)
+
+  return {
+    sunday: weeklyFlags[0],
+    monday: weeklyFlags[1],
+    tuesday: weeklyFlags[2],
+    wednesday: weeklyFlags[3],
+    thursday: weeklyFlags[4],
+    friday: weeklyFlags[5],
+    saturday: weeklyFlags[6],
+    startTime: `${HH.padStart(2, "0")}:${mm.padStart(2, "0")}`,
+    timezone,
+    ttl: workspace.ttl ? workspace.ttl / (1_000_000 * 1000 * 60 * 60) : 0,
+  }
+}
+
 export const WorkspaceSchedulePage: React.FC = () => {
   const navigate = useNavigate()
   const { workspace: workspaceQueryParam } = useParams()

diff --git a/site/src/util/schedule.test.ts b/site/src/util/schedule.test.ts
@@ -1,4 +1,4 @@
-import { extractTimezone, stripTimezone } from "./schedule"
+import { dowToWeeklyFlag, extractTimezone, stripTimezone, WeeklyFlag } from "./schedule"
 
 describe("util/schedule", () => {
   describe("stripTimezone", () => {
@@ -20,4 +20,25 @@ describe("util/schedule", () => {
       expect(extractTimezone(input)).toBe(expected)
     })
   })
+
+  describe("dowToWeeklyFlag", () => {
+    it.each<[string, WeeklyFlag]>([
+      // All days
+      ["*", [true, true, true, true, true, true, true]],
+      ["1-7", [true, true, true, true, true, true, true]],
+
+      // Single number modulo 7
+      ["3", [false, false, false, true, false, false, false]],
+      ["0", [true, false, false, false, false, false, false]],
+      ["7", [true, false, false, false, false, false, false]],
+      ["8", [false, true, false, false, false, false, false]],
+
+      // Comma-separated Numbers, Ranges and Mixes
+      ["1,3,5", [false, true, false, true, false, true, false]],
+      ["1-2,4-5", [false, true, true, false, true, true, false]],
+      ["1,3-4,6", [false, true, false, true, true, false, true]],
+    ])(`dowToWeeklyFlag(%p) returns %p`, (dow, weeklyFlag) => {
+      expect(dowToWeeklyFlag(dow)).toEqual(weeklyFlag)
+    })
+  })
 })
diff --git a/site/src/util/schedule.ts b/site/src/util/schedule.ts
@@ -19,14 +19,83 @@ export const stripTimezone = (raw: string): string => {
 
 /**
  * extractTimezone returns a leading timezone from a schedule string if one is
- * specified; otherwise DEFAULT_TIMEZONE
+ * specified; otherwise the specified defaultTZ
  */
-export const extractTimezone = (raw: string): string => {
+export const extractTimezone = (raw: string, defaultTZ = DEFAULT_TIMEZONE): string => {
   const matches = raw.match(/CRON_TZ=\S*\s/g)
 
   if (matches && matches.length) {
     return matches[0].replace(/CRON_TZ=/, "").trim()
   } else {
-    return DEFAULT_TIMEZONE
+    return defaultTZ
   }
 }
+
+/**
+ * WeeklyFlag is an array represnting which days of the week are set or flagged
+ *
+ * @remarks
+ *
+ * A WeeklyFlag has an array size of 7 and should never have its size modified.
+ * The 0th index is Sunday
+ * The 6th index is Saturday
+ */
+export type WeeklyFlag = [boolean, boolean, boolean, boolean, boolean, boolean, boolean]
+
+/**
+ * dowToWeeklyFlag converts a dow cron string to a WeeklyFlag array.
+ *
+ * @example
+ *
+ * dowToWeeklyFlag("1") // [false, true, false, false, false, false, false]
+ * dowToWeeklyFlag("1-5") // [false, true, true, true, true, true, false]
+ * dowToWeeklyFlag("1,3-4,6") // [false, true, false, true, true, false, true]
+ */
+export const dowToWeeklyFlag = (dow: string): WeeklyFlag => {
+  if (dow === "*") {
+    return [true, true, true, true, true, true, true]
+  }
+
+  const results: WeeklyFlag = [false, false, false, false, false, false, false]
+
+  const commaSeparatedRangeOrNum = dow.split(",")
+
+  for (const rangeOrNum of commaSeparatedRangeOrNum) {
+    const flags = processRangeOrNum(rangeOrNum)
+
+    flags.forEach((value, idx) => {
+      if (value) {
+        results[idx] = true
+      }
+    })
+  }
+
+  return results
+}
+
+/**
+ * processRangeOrNum is a helper for dowToWeeklyFlag. It processes a range or
+ * number (modulo 7) into a Weeklyflag boolean array.
+ *
+ * @example
+ *
+ * processRangeOrNum("1") // [false, true, false, false, false, false, false]
+ * processRangeOrNum("1-5") // [false, true, true, true, true, true, false]
+ */
+const processRangeOrNum = (rangeOrNum: string): WeeklyFlag => {
+  const result: WeeklyFlag = [false, false, false, false, false, false, false]
+
+  const isRange = /^[0-9]-[0-9]$/.test(rangeOrNum)
+
+  if (isRange) {
+    const [first, last] = rangeOrNum.split("-")
+
+    for (let i = Number(first); i <= Number(last); i++) {
+      result[i % 7] = true
+    }
+  } else {
+    result[Number(rangeOrNum) % 7] = true
+  }
+
+  return result
+}