From e38af0bebe012bd2c8d55c30db854906d057fb8e Mon Sep 17 00:00:00 2001
From: Andrew Bastin <andrewbastin.k@gmail.com>
Date: Tue, 17 Aug 2021 14:28:48 +0530
Subject: [PATCH] feat: firestore sync for global environment

---
 helpers/fb/environments.ts | 83 +++++++++++++++++++++++++++++++++++---
 newstore/environments.ts   |  2 +-
 2 files changed, 79 insertions(+), 6 deletions(-)

diff --git a/helpers/fb/environments.ts b/helpers/fb/environments.ts
index 8d44b3a4b..b0af79e60 100644
--- a/helpers/fb/environments.ts
+++ b/helpers/fb/environments.ts
@@ -4,7 +4,9 @@ import { currentUser$ } from "./auth"
 import {
   Environment,
   environments$,
+  globalEnv$,
   replaceEnvironments,
+  setGlobalEnvVariables,
 } from "~/newstore/environments"
 import { settingsStore } from "~/newstore/settings"
 
@@ -16,6 +18,14 @@ import { settingsStore } from "~/newstore/settings"
  */
 let loadedEnvironments = false
 
+/**
+ * Used locally to prevent infinite loop when global env sync update
+ * is applied to the store which then fires the store sync listener.
+ * When you want to update global env and not want to fire the update listener,
+ * set this to true and then set it back to false once it is done
+ */
+let loadedGlobals = true
+
 async function writeEnvironments(environment: Environment[]) {
   if (currentUser$.value == null)
     throw new Error("Cannot write environments when signed out")
@@ -42,6 +52,32 @@ async function writeEnvironments(environment: Environment[]) {
   }
 }
 
+async function writeGlobalEnvironment(variables: Environment["variables"]) {
+  if (currentUser$.value == null)
+    throw new Error("Cannot write global environment when signed out")
+
+  const ev = {
+    updatedOn: new Date(),
+    author: currentUser$.value.uid,
+    author_name: currentUser$.value.displayName,
+    author_image: currentUser$.value.photoURL,
+    variables,
+  }
+
+  try {
+    await firebase
+      .firestore()
+      .collection("users")
+      .doc(currentUser$.value.uid)
+      .collection("globalEnv")
+      .doc("sync")
+      .set(ev)
+  } catch (e) {
+    console.error("error updating", ev, e)
+    throw e
+  }
+}
+
 export function initEnvironments() {
   environments$.subscribe((envs) => {
     if (
@@ -53,15 +89,33 @@ export function initEnvironments() {
     }
   })
 
-  let snapshotStop: (() => void) | null = null
+  globalEnv$.subscribe((vars) => {
+    if (
+      currentUser$.value &&
+      settingsStore.value.syncEnvironments &&
+      loadedGlobals
+    ) {
+      writeGlobalEnvironment(vars)
+    }
+  })
+
+  let envSnapshotStop: (() => void) | null = null
+  let globalsSnapshotStop: (() => void) | null = null
 
   currentUser$.subscribe((user) => {
-    if (!user && snapshotStop) {
+    if (!user) {
       // User logged out, clean up snapshot listener
-      snapshotStop()
-      snapshotStop = null
+      if (envSnapshotStop) {
+        envSnapshotStop()
+        envSnapshotStop = null
+      }
+
+      if (globalsSnapshotStop) {
+        globalsSnapshotStop()
+        globalsSnapshotStop = null
+      }
     } else if (user) {
-      snapshotStop = firebase
+      envSnapshotStop = firebase
         .firestore()
         .collection("users")
         .doc(user.uid)
@@ -79,6 +133,25 @@ export function initEnvironments() {
           replaceEnvironments(environments[0].environment)
           loadedEnvironments = true
         })
+      globalsSnapshotStop = firebase
+        .firestore()
+        .collection("users")
+        .doc(user.uid)
+        .collection("globalEnv")
+        .onSnapshot((globalsRef) => {
+          const variables: any[] = []
+
+          globalsRef.forEach((doc) => {
+            const variable = doc.data()
+            variable.id = doc.id
+
+            variables.push(variable)
+          })
+
+          loadedGlobals = false
+          setGlobalEnvVariables(variables)
+          loadedGlobals = true
+        })
     }
   })
 }
diff --git a/newstore/environments.ts b/newstore/environments.ts
index 26770307b..0004147fe 100644
--- a/newstore/environments.ts
+++ b/newstore/environments.ts
@@ -235,7 +235,7 @@ export const environments$ = environmentsStore.subject$.pipe(
   pluck("environments")
 )
 
-export const globalEnv$ = environmentsStore.subject$.pipe(pluck("globalEnv"))
+export const globalEnv$ = environmentsStore.subject$.pipe(pluck("globals"))
 
 export const selectedEnvIndex$ = environmentsStore.subject$.pipe(
   pluck("currentEnvironmentIndex")