fixup! Use new context

Emyrk · Emyrk · commit 69c573443265 · 2023-04-26T09:59:35.000-05:00
diff --git a/site/src/contexts/ProxyContext.tsx b/site/src/contexts/ProxyContext.tsx
@@ -0,0 +1,159 @@
+import { Region } from "api/typesGenerated"
+import { useDashboard } from "components/Dashboard/DashboardProvider"
+import { createContext, FC, PropsWithChildren, useContext, useState } from "react"
+
+interface ProxyContextValue {
+  value: PreferredProxy
+  setValue: (regions: Region[], selectedRegion: Region | undefined) => void
+}
+
+interface PreferredProxy {
+  // Regions is a list of all the regions returned by coderd.
+  regions: Region[]
+  // SelectedRegion is the region the user has selected.
+  selectedRegion: Region | undefined
+  // PreferredPathAppURL is the URL of the proxy or it is the empty string
+  // to indicte using relative paths. To add a path to this:
+  //  PreferredPathAppURL + "/path/to/app"
+  preferredPathAppURL: string
+  // PreferredWildcardHostname is a hostname that includes a wildcard.
+  // TODO: If the preferred proxy does not have this set, should we default to'
+  //    the primary's?
+  //  Example: "*.example.com"
+  preferredWildcardHostname: string
+}
+
+const ProxyContext = createContext<ProxyContextValue | undefined>(undefined)
+
+/**
+ * ProxyProvider interacts with local storage to indicate the preferred workspace proxy.
+ */
+export const ProxyProvider: FC<PropsWithChildren> = ({ children }) => {
+  // Try to load the preferred proxy from local storage.
+  let savedProxy = loadPreferredProxy()
+  if (!savedProxy) {
+    savedProxy = getURLs([], undefined)
+  }
+
+  // The initial state is no regions and no selected region.
+  const [state, setState] = useState<PreferredProxy>(savedProxy)
+
+  // ******************************* //
+  // ** This code can be removed  **
+  // ** when the experimental is  ** 
+  // **       dropped             ** //
+  const dashboard = useDashboard()
+  // If the experiment is disabled, then make the setState do a noop.
+  // This preserves an empty state, which is the default behavior.
+  if (!dashboard?.experiments.includes("moons")) {
+    return (
+      <ProxyContext.Provider value={{
+        value: getURLs([], undefined),
+        setValue: () => {
+          // Does a noop
+        },
+      }}>
+        {children}
+      </ProxyContext.Provider >
+    )
+  }
+  // ******************************* //
+
+  // TODO: @emyrk Should make an api call to /regions endpoint to update the
+  // regions list.
+
+  return (
+    <ProxyContext.Provider value={{
+      value: state,
+      // A function that takes the new regions and selected region and updates
+      // the state with the appropriate urls.
+      setValue: (regions, selectedRegion) => {
+        const preferred = getURLs(regions, selectedRegion)
+        // Save to local storage to persist the user's preference across reloads
+        // and other tabs.
+        savePreferredProxy(preferred)
+        // Set the state for the current context.
+        setState(preferred)
+      },
+    }}>
+      {children}
+    </ProxyContext.Provider >
+  )
+}
+
+export const useProxy = (): ProxyContextValue => {
+  const context = useContext(ProxyContext)
+
+  if (!context) {
+    throw new Error("useProxy should be used inside of <ProxyProvider />")
+  }
+
+  return context
+}
+
+
+/**
+ * getURLs is a helper function to calculate the urls to use for a given proxy configuration. By default, it is
+ * assumed no proxy is configured and relative paths should be used.
+ * 
+ * @param regions Is the list of regions returned by coderd. If this is empty, default behavior is used.
+ * @param selectedRegion Is the region the user has selected. If this is undefined, default behavior is used.
+ */
+const getURLs = (regions: Region[], selectedRegion: Region | undefined): PreferredProxy => {
+  // By default we set the path app to relative and disable wilcard hostnames.
+  // We will set these values if we find a proxy we can use that supports them.
+  let pathAppURL = ""
+  let wildcardHostname = ""
+
+  if (selectedRegion === undefined) {
+    // If no region is selected, default to the primary region.
+    selectedRegion = regions.find((region) => region.name === "primary")
+  } else {
+    // If a region is selected, make sure it is in the list of regions. If it is not
+    // we should default to the primary.
+    selectedRegion = regions.find((region) => region.id === selectedRegion?.id)
+  }
+
+  // Only use healthy regions.
+  if (selectedRegion && selectedRegion.healthy) {
+    // By default use relative links for the primary region.
+    // This is the default, and we should not change it.
+    if (selectedRegion.name !== "primary") {
+      pathAppURL = selectedRegion.path_app_url
+    }
+    wildcardHostname = selectedRegion.wildcard_hostname
+  }
+
+  // TODO: @emyrk Should we notify the user if they had an unhealthy region selected?
+
+
+  return {
+    regions: regions,
+    selectedRegion: selectedRegion,
+    // Trim trailing slashes to be consistent
+    preferredPathAppURL: pathAppURL.replace(/\/$/, ""),
+    preferredWildcardHostname: wildcardHostname,
+  }
+}
+
+// Local storage functions
+
+export const savePreferredProxy = (saved: PreferredProxy): void => {
+  window.localStorage.setItem("preferred-proxy", JSON.stringify(saved))
+}
+
+export const loadPreferredProxy = (): PreferredProxy | undefined => {
+  const str = localStorage.getItem("preferred-proxy")
+  if (str === undefined || str === null) {
+    return undefined
+  }
+  const proxy = JSON.parse(str)
+  if (proxy.id === undefined || proxy.id === null) {
+    return undefined
+  }
+  return proxy
+}
+
+export const clearPreferredProxy = (): void => {
+  localStorage.removeItem("preferred-proxy")
+}
