fallback to file

ShMcK · ShMcK · commit bf9ed01adf41 · 2021-11-20T19:52:28.000-08:00
Signed-off-by: shmck &lt;shawn.j.mckay@gmail.com&gt;
diff --git a/src/environment.ts b/src/environment.ts
@@ -46,3 +46,6 @@ export const CONTENT_SECURITY_POLICY_EXEMPTIONS: string | null =
 
 // optional token for authorization/authentication of webhook calls
 export const WEBHOOK_TOKEN = process.env.CODEROAD_WEBHOOK_TOKEN || null
+
+// a path to write session state to a file. Useful for maintaining session across containers
+export const SESSION_FILE_PATH = process.env.CODEROAD_SESSION_FILE_PATH || null
diff --git a/src/services/node/index.ts b/src/services/node/index.ts
@@ -7,6 +7,7 @@ import { WORKSPACE_ROOT } from '../../environment'
 const asyncExec = promisify(cpExec)
 const asyncRemoveFile = promisify(fs.unlink)
 const asyncReadFile = promisify(fs.readFile)
+const asyncWriteFile = promisify(fs.writeFile)
 
 interface ExecParams {
   command: string
@@ -27,5 +28,12 @@ export const removeFile = (...paths: string[]) => {
 }
 
 export const readFile = (...paths: string[]) => {
-  return asyncReadFile(join(...paths))
+  return asyncReadFile(join(...paths), 'utf8')
+}
+
+export const writeFile = (data: any, ...paths: string[]) => {
+  const filePath = join(...paths)
+  return asyncWriteFile(filePath, JSON.stringify(data)).catch((err) => {
+    console.error(`Failed to write to ${filePath}`)
+  })
 }
diff --git a/src/services/storage/index.ts b/src/services/storage/index.ts
@@ -1,4 +1,6 @@
 import * as vscode from 'vscode'
+import { readFile, writeFile } from '../node'
+import { SESSION_FILE_PATH } from '../../environment'
 
 // NOTE: localStorage is not available on client
 // and must be stored in editor
@@ -19,6 +21,18 @@ class Storage<T> {
     const value: string | undefined = await this.storage.get(this.key)
     if (value) {
       return JSON.parse(value)
+    } else if (SESSION_FILE_PATH) {
+      // optionally read from file as a fallback to localstorage
+      const sessionFile = await readFile(SESSION_FILE_PATH)
+      try {
+        const session = JSON.parse(sessionFile)
+        if (session && session[this.key]) {
+          // TODO: validate session
+          return session[this.key]
+        }
+      } catch (err) {
+        console.error(`Failed to parse session file: ${SESSION_FILE_PATH}`)
+      }
     }
     return this.defaultValue
   }
@@ -32,7 +46,12 @@ class Storage<T> {
       ...current,
       ...value,
     })
-    this.storage.update(this.key, next)
+    this.storage.update(this.key, next).then(() => {
+      // optionally write to file
+      if (SESSION_FILE_PATH) {
+        writeFile(this.storage, SESSION_FILE_PATH)
+      }
+    })
   }
   public reset = () => {
     this.set(this.defaultValue)
