posthog-js-lite 2.0.0-alpha4 → 2.0.0-alpha5

package/lib/posthog-web/src/types.d.ts

@@ -0,0 +1,6 @@
+import { PosthogCoreOptions } from '../../posthog-core/src';
+export declare type PostHogOptions = {
+    autocapture?: boolean;
+    persistence?: 'localStorage' | 'sessionStorage' | 'cookie' | 'memory';
+    persistence_name?: string;
+} & PosthogCoreOptions;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "posthog-js-lite",
-  "version": "2.0.0-alpha4",
+  "version": "2.0.0-alpha5",
   "main": "lib/index.cjs.js",
   "module": "lib/index.esm.js",
   "types": "lib/index.d.ts",

package/src/posthog-web.ts

@@ -1,21 +1,16 @@
 import {
   PostHogCore,
-  PosthogCoreOptions,
   PostHogFetchOptions,
   PostHogFetchResponse,
   PostHogPersistedProperty,
 } from '../../posthog-core/src'
 import { getContext } from './context'
-import { localStore, cookieStore, sessionStorage } from './storage'
+import { PostHogStorage, getStorage } from './storage'
 import { version } from '../package.json'
-
-export interface PostHogOptions extends PosthogCoreOptions {
-  autocapture?: boolean
-  persistence_name?: string
-}
+import { PostHogOptions } from './types'
 
 export class PostHog extends PostHogCore {
-  private _storage = localStore || sessionStorage || cookieStore
+  private _storage: PostHogStorage
   private _storageCache: any
   private _storageKey: string
 
@@ -24,6 +19,7 @@ export class PostHog extends PostHogCore {
 
     // posthog-js stores options in one object on
     this._storageKey = options?.persistence_name ? `ph_${options.persistence_name}` : `ph_${apiKey}_posthog`
+    this._storage = getStorage(options?.persistence || 'localStorage')
   }
 
   getPersistedProperty<T>(key: PostHogPersistedProperty): T | undefined {

package/src/storage.ts

@@ -1,3 +1,5 @@
+import { PostHogOptions } from './types'
+
 export type PostHogStorage = {
   getItem: (key: string) => string | null | undefined
   setItem: (key: string, value: string) => void
@@ -112,3 +114,49 @@ const checkStoreIsSupported = (storage: PostHogStorage, key = '__mplssupport__')
 
 export const localStore = checkStoreIsSupported(_localStore) ? _localStore : undefined
 export const sessionStorage = checkStoreIsSupported(_sessionStore) ? _sessionStore : undefined
+
+const createMemoryStorage = (): PostHogStorage => {
+  const _cache: { [key: string]: any | undefined } = {}
+
+  const store: PostHogStorage = {
+    getItem(key) {
+      return _cache[key]
+    },
+
+    setItem(key, value) {
+      _cache[key] = value !== null ? value : undefined
+    },
+
+    removeItem(key) {
+      delete _cache[key]
+    },
+    clear() {
+      for (const key in _cache) {
+        delete _cache[key]
+      }
+    },
+    getAllKeys() {
+      const keys = []
+      for (const key in _cache) {
+        keys.push(key)
+      }
+      return keys
+    },
+  }
+  return store
+}
+
+export const getStorage = (type: PostHogOptions['persistence']): PostHogStorage => {
+  switch (type) {
+    case 'cookie':
+      return cookieStore || localStore || sessionStorage || createMemoryStorage()
+    case 'localStorage':
+      return localStore || sessionStorage || createMemoryStorage()
+    case 'sessionStorage':
+      return sessionStorage || createMemoryStorage()
+    case 'memory':
+      return createMemoryStorage()
+    default:
+      return createMemoryStorage()
+  }
+}

package/src/types.ts

@@ -0,0 +1,7 @@
+import { PosthogCoreOptions } from '../../posthog-core/src'
+
+export type PostHogOptions = {
+  autocapture?: boolean
+  persistence?: 'localStorage' | 'sessionStorage' | 'cookie' | 'memory'
+  persistence_name?: string
+} & PosthogCoreOptions