From 39b784aa60fdbfe133180182b363b8f33e8b3950 Mon Sep 17 00:00:00 2001
From: alex <alex@dytry.ch>
Date: Tue, 14 Mar 2023 16:08:12 +0000
Subject: [PATCH] alex/maintenance-mode: Add maintenance mode (#1164)

---
 apps/www/components/MaintenanceMode.tsx | 108 ++++++++++++++++++++++++
 apps/www/pages/_app.tsx                 |   5 +-
 2 files changed, 112 insertions(+), 1 deletion(-)
 create mode 100644 apps/www/components/MaintenanceMode.tsx

diff --git a/apps/www/components/MaintenanceMode.tsx b/apps/www/components/MaintenanceMode.tsx
new file mode 100644
index 000000000..d9655a880
--- /dev/null
+++ b/apps/www/components/MaintenanceMode.tsx
@@ -0,0 +1,108 @@
+import { ReactNode, useEffect, useRef, useState } from 'react'
+import { light, styled } from '~styles'
+
+const CONTROL_SERVER = process.env.NEXT_PUBLIC_CONTROL_SERVER
+  ? process.env.NEXT_PUBLIC_CONTROL_SERVER
+  : process.env.NEXT_PUBLIC_ENABLE_DEV_CONTROL_SERVER
+  ? 'http://localhost:3001'
+  : null
+
+type MaintenanceModeConfig =
+  | {
+      enabled: false
+      ttlSeconds: number
+    }
+  | {
+      enabled: true
+      ttlSeconds: number
+      titleMessage: string
+      bodyMessageHtml: string
+    }
+
+type ControlResponse = {
+  maintenance: MaintenanceModeConfig
+}
+
+export function MaintenanceMode({ children }: { children: ReactNode }) {
+  const [maintenanceMode, setMaintenanceMode] = useState<MaintenanceModeConfig | null>(null)
+
+  useEffect(() => {
+    if (!CONTROL_SERVER) return
+    let isCancelled = false
+    const updateMaintenanceMode = async () => {
+      try {
+        const response = await fetch(`${CONTROL_SERVER}/control.json`)
+        const data: ControlResponse = await response.json()
+        if (isCancelled) return
+        setMaintenanceMode(data.maintenance)
+      } catch {
+        setMaintenanceMode({
+          enabled: false,
+          ttlSeconds: 60,
+        })
+      }
+    }
+
+    if (maintenanceMode) {
+      const refreshTimeout = setTimeout(updateMaintenanceMode, maintenanceMode.ttlSeconds * 1000)
+      return () => {
+        isCancelled = true
+        clearTimeout(refreshTimeout)
+      }
+    } else {
+      updateMaintenanceMode()
+      return () => {
+        isCancelled = true
+      }
+    }
+  }, [maintenanceMode])
+
+  const isMaintenanceModeEnabled = maintenanceMode?.enabled ?? false
+  const wasInMaintenanceModeRef = useRef(isMaintenanceModeEnabled)
+  useEffect(() => {
+    // If we were in maintenance mode and now we're not, reload the page
+    if (wasInMaintenanceModeRef.current && !isMaintenanceModeEnabled) {
+      window.location.reload()
+    }
+    wasInMaintenanceModeRef.current = isMaintenanceModeEnabled
+  }, [isMaintenanceModeEnabled])
+
+  if (maintenanceMode?.enabled) {
+    return (
+      <div className={`tldraw ${light}`}>
+        <Container>
+          <Modal>
+            <Heading>{maintenanceMode.titleMessage}</Heading>
+            <div dangerouslySetInnerHTML={{ __html: maintenanceMode.bodyMessageHtml }} />
+          </Modal>
+        </Container>
+      </div>
+    )
+  }
+
+  return <>{children}</>
+}
+
+const Container = styled('div', {
+  backgroundColor: '$canvas',
+  position: 'absolute',
+  inset: 0,
+  padding: '1rem',
+  display: 'flex',
+  alignItems: 'center',
+  justifyContent: 'center',
+})
+
+const Modal = styled('div', {
+  backgroundColor: '$panel',
+  padding: '1rem 1.5rem',
+  borderRadius: '$2',
+  boxShadow: '$8',
+  maxWidth: '400px',
+  width: '100%',
+})
+
+const Heading = styled('h2', {
+  margin: 0,
+  marginTop: '1rem',
+})
diff --git a/apps/www/pages/_app.tsx b/apps/www/pages/_app.tsx
index 9afe0b9bc..4510a30e6 100644
--- a/apps/www/pages/_app.tsx
+++ b/apps/www/pages/_app.tsx
@@ -1,6 +1,7 @@
 import '@fontsource/recursive'
 import Head from 'next/head'
 import type React from 'react'
+import { MaintenanceMode } from '~components/MaintenanceMode'
 import '~styles/globals.css'
 import useGtag from '~utils/useGtag'
 
@@ -50,7 +51,9 @@ function MyApp({ Component, pageProps }: any) {
 
         <title>tldraw</title>
       </Head>
-      <Component {...pageProps} />
+      <MaintenanceMode>
+        <Component {...pageProps} />
+      </MaintenanceMode>
     </>
   )
 }