posthog-node 2.0.0-alpha3 → 2.0.0-alpha6

package/src/posthog-node.ts

@@ -7,12 +7,15 @@ import {
   PostHogFetchResponse,
   PostHogPersistedProperty,
 } from '../../posthog-core/src'
+import { PostHogMemoryStorage } from '../../posthog-core/src/storage-memory'
 import { EventMessageV1, GroupIdentifyMessage, IdentifyMessageV1, PostHogNodeV1 } from './types'
 
-export type PostHogOptions = PosthogCoreOptions
+export type PostHogOptions = PosthogCoreOptions & {
+  persistence?: 'memory'
+}
 
 class PostHog extends PostHogCore {
-  private _memoryStorage: { [key: string]: any | undefined } = {}
+  private _memoryStorage = new PostHogMemoryStorage()
 
   constructor(apiKey: string, options: PostHogOptions = {}) {
     options.captureMode = options?.captureMode || 'json'
@@ -22,11 +25,11 @@ class PostHog extends PostHogCore {
   }
 
   getPersistedProperty(key: PostHogPersistedProperty): any | undefined {
-    return this._memoryStorage[key]
+    return this._memoryStorage.getProperty(key)
   }
 
   setPersistedProperty(key: PostHogPersistedProperty, value: any | null): void {
-    this._memoryStorage[key] = value !== null ? value : undefined
+    return this._memoryStorage.setProperty(key, value)
   }
 
   getSessionId(): string | undefined {
@@ -59,10 +62,14 @@ export class PostHogGlobal implements PostHogNodeV1 {
   }
 
   private reInit(distinctId: string): void {
-    // Remove all state except the queue
-    const queue = this._sharedClient.getPersistedProperty(PostHogPersistedProperty.Queue)
-    this._sharedClient.reset()
-    this._sharedClient.setPersistedProperty(PostHogPersistedProperty.Queue, queue)
+    // Certain properties we want to persist
+    const propertiesToKeep = [PostHogPersistedProperty.Queue, PostHogPersistedProperty.OptedOut]
+
+    for (const key in PostHogPersistedProperty) {
+      if (!propertiesToKeep.includes(key as any)) {
+        this._sharedClient.setPersistedProperty((PostHogPersistedProperty as any)[key], null)
+      }
+    }
     this._sharedClient.setPersistedProperty(PostHogPersistedProperty.DistinctId, distinctId)
   }