WIP add store migration

PostHog · Nov 3, 2023 · 29257fb · 29257fb
1 parent 9cf3648
commit 29257fb
Show file tree

Hide file tree

Showing 2 changed files with 69 additions and 5 deletions.
diff --git a/src/posthog-persistence.ts b/src/posthog-persistence.ts
@@ -1,12 +1,19 @@
 /* eslint camelcase: "off" */
 
 import { _each, _extend, _include, _strip_empty_properties } from './utils'
-import { cookieStore, localStore, localPlusCookieStore, memoryStore, sessionStore } from './storage'
+import {
+    COOKIE_PERSISTED_PROPERTIES,
+    cookieStore,
+    localPlusCookieStore,
+    localStore,
+    memoryStore,
+    sessionStore,
+} from './storage'
 import { PersistentStore, PostHogConfig, Properties } from './types'
 import {
-    PERSISTENCE_RESERVED_PROPERTIES,
-    EVENT_TIMERS_KEY,
     ENABLED_FEATURE_FLAGS,
+    EVENT_TIMERS_KEY,
+    PERSISTENCE_RESERVED_PROPERTIES,
     POSTHOG_QUOTA_LIMITED,
     USER_STATE,
 } from './constants'
@@ -78,7 +85,8 @@ export class PostHogPersistence {
         } else if (storage_type === 'memory') {
             this.storage = memoryStore
         } else {
-            this.storage = cookieStore
+            const { storage } = this._getAndMigrateToDefaultStore()
+            this.storage = storage
         }
 
         this.user_state = 'anonymous'
@@ -314,4 +322,60 @@ export class PostHogPersistence {
         this.props[POSTHOG_QUOTA_LIMITED] = state
         this.save()
     }
+
+    _getAndMigrateToDefaultStore = (): { storage: PersistentStore; wasPreviouslyDifferentStore: boolean } => {
+        // Check if there's any data in local/cookie/session storage, and if needed,
+        // migrate to cookie+localstorage
+        const defaultStore = localPlusCookieStore
+        const currentStore = this._getCurrentStore()
+
+        if (currentStore && currentStore !== defaultStore) {
+            this._migrateStoreToDefaultStore(currentStore, defaultStore)
+            return { storage: defaultStore, wasPreviouslyDifferentStore: !!currentStore }
+        } else {
+            return { storage: defaultStore, wasPreviouslyDifferentStore: false }
+        }
+    }
+
+    _getCurrentStore = () => {
+        // Test by picking a property that is stored in the cookie for local+cookie
+        const testKey = COOKIE_PERSISTED_PROPERTIES[0]
+
+        // If that key exists in the local store, then we're definitely using
+        // local storage
+        const isLocalStore = localStore.is_supported() && localStore.get(this.name)?.[testKey]
+        if (isLocalStore) {
+            return localStore
+        }
+        // If not using pure localStore, we must be using localPlusCookie if
+        // there's data in both
+        const isLocalPlusCookie =
+            localPlusCookieStore.is_supported() &&
+            localStore.is_supported() &&
+            localStore.get(this.name) &&
+            cookieStore.is_supported() &&
+            cookieStore.get(this.name)
+        if (isLocalPlusCookie) {
+            return localPlusCookieStore
+        }
+        // If there's any data in cookieStore at this point then we're on cookieStore
+        const isCookie = cookieStore.is_supported() && cookieStore.get(this.name)
+        if (isCookie) {
+            return cookieStore
+        }
+
+        const isSession = sessionStore.is_supported() && sessionStore.get(this.name)
+        if (isSession) {
+            return sessionStore
+        }
+
+        // memory store, no existing store, or existing store no longer supported
+        return undefined
+    }
+
+    _migrateStoreToDefaultStore(store: PersistentStore, defaultStore: PersistentStore) {
+        const values = store.get(this.name)
+        store.remove(this.name, this.cross_subdomain)
+        defaultStore.set(this.name, values, this.expire_days, this.cross_subdomain, this.secure)
+    }
 }
diff --git a/src/storage.ts b/src/storage.ts
@@ -170,7 +170,7 @@ export const localStore: PersistentStore = {
 // Use localstorage for most data but still use cookie for COOKIE_PERSISTED_PROPERTIES
 // This solves issues with cookies having too much data in them causing headers too large
 // Also makes sure we don't have to send a ton of data to the server
-const COOKIE_PERSISTED_PROPERTIES = [DISTINCT_ID, SESSION_ID, SESSION_RECORDING_IS_SAMPLED]
+export const COOKIE_PERSISTED_PROPERTIES = [DISTINCT_ID, SESSION_ID, SESSION_RECORDING_IS_SAMPLED]
 
 export const localPlusCookieStore: PersistentStore = {
     ...localStore,