feat: save consent in localStorage for browsers

src/BrowserMetrics.ts

@@ -1,10 +1,15 @@
 import Countly from 'countly-sdk-web'
 
 import MetricsProvider, { MetricsProviderConstructorOptions } from './MetricsProvider.js'
+import { BrowserStorageProvider } from './BrowserStorageProvider.js'
 
 export class BrowserMetricsProvider extends MetricsProvider<typeof Countly> {
   constructor (args: Omit<MetricsProviderConstructorOptions<typeof Countly>, 'metricsService'>) {
-    super({ ...args, metricsService: Countly })
+    super({
+      metricsService: Countly,
+      storageProvider: BrowserStorageProvider,
+      ...args
+    })
   }
 }
 

src/BrowserStorageProvider.ts

@@ -0,0 +1,25 @@
+import type { StorageProvider } from './types/index.js'
+
+export const BrowserStorageProvider: StorageProvider = {
+  setStore: (consentArray) => {
+    try {
+      const jsonString = JSON.stringify(consentArray)
+      window.localStorage.setItem('@ipfs-shipyard/ignite-metrics:consent', jsonString)
+    } catch (err) {
+      // eslint-disable-next-line no-console
+      console.error(err)
+    }
+  },
+  getStore: () => {
+    try {
+      const jsonString = window.localStorage.getItem('@ipfs-shipyard/ignite-metrics:consent')
+      if (jsonString != null) {
+        return JSON.parse(jsonString)
+      }
+    } catch (err) {
+      // eslint-disable-next-line no-console
+      console.error(err)
+    }
+    return []
+  }
+}

src/MetricsProvider.ts

@@ -1,16 +1,17 @@
 import { COUNTLY_API_URL } from './config.js'
 import type { metricFeatures, CountlyWebSdk } from 'countly-sdk-web'
 import type { CountlyNodeSdk } from 'countly-sdk-nodejs'
-import type { consentTypes, consentTypesExceptAll } from './types/index.js'
+import type { consentTypes, consentTypesExceptAll, StorageProvider } from './types/index.js'
 
 export interface MetricsProviderConstructorOptions<T> {
   appKey: string
   url?: string
   autoTrack?: boolean
   metricsService: T
+  storageProvider?: StorageProvider | null
 }
 
-export default class MetricsProvider<T extends CountlyWebSdk & CountlyNodeSdk> {
+export default class MetricsProvider<T extends CountlyWebSdk | CountlyNodeSdk> {
   private readonly groupedFeatures: Record<consentTypes, metricFeatures[]> = this.mapAllEvents({
     minimal: ['sessions', 'views', 'events'],
     performance: ['crashes', 'apm'],
@@ -22,9 +23,17 @@ export default class MetricsProvider<T extends CountlyWebSdk & CountlyNodeSdk> {
   private sessionStarted: boolean = false
   private readonly _consentGranted: Set<consentTypes> = new Set()
   private readonly metricsService: T
-
-  constructor ({ autoTrack = true, url = COUNTLY_API_URL, appKey, metricsService }: MetricsProviderConstructorOptions<T>) {
+  private readonly storageProvider: StorageProvider | null
+
+  constructor ({
+    autoTrack = true,
+    url = COUNTLY_API_URL,
+    appKey,
+    metricsService,
+    storageProvider
+  }: MetricsProviderConstructorOptions<T>) {
     this.metricsService = metricsService
+    this.storageProvider = storageProvider ?? null
     this.metricsService.init({
       app_key: appKey,
       url,
@@ -36,6 +45,8 @@ export default class MetricsProvider<T extends CountlyWebSdk & CountlyNodeSdk> {
     if (autoTrack) {
       this.setupAutoTrack()
     }
+
+    this.getConsentStore().forEach(this.addConsent.bind(this))
   }
 
   mapAllEvents (eventMap: Record<consentTypesExceptAll, metricFeatures[]>): Record<consentTypes, metricFeatures[]> {
@@ -66,6 +77,7 @@ export default class MetricsProvider<T extends CountlyWebSdk & CountlyNodeSdk> {
     }
     consent.forEach(c => this._consentGranted.add(c))
     this.metricsService.add_consent(consent)
+    this.setConsentStore()
   }
 
   removeConsent (consent: consentTypes | consentTypes[]): void {
@@ -74,6 +86,20 @@ export default class MetricsProvider<T extends CountlyWebSdk & CountlyNodeSdk> {
     }
     consent.forEach(c => this._consentGranted.delete(c))
     this.metricsService.remove_consent(consent, true)
+    this.setConsentStore()
+  }
+
+  private setConsentStore (): void {
+    if (this.storageProvider != null) {
+      this.storageProvider.setStore(Array.from(this._consentGranted))
+    }
+  }
+
+  private getConsentStore (): consentTypes[] {
+    if (this.storageProvider != null) {
+      return this.storageProvider.getStore()
+    }
+    return []
   }
 
   checkConsent (consent: consentTypes | metricFeatures): boolean {

src/NodeMetrics.ts

@@ -4,7 +4,10 @@ import MetricsProvider, { MetricsProviderConstructorOptions } from './MetricsPro
 
 export class NodeMetricsProvider extends MetricsProvider<typeof Countly> {
   constructor (args: Omit<MetricsProviderConstructorOptions<typeof Countly>, 'metricsService'>) {
-    super({ ...args, metricsService: Countly })
+    super({
+      metricsService: Countly,
+      ...args
+    })
   }
 }
 

src/types/index.d.ts

@@ -1,3 +1,8 @@
 
 export type consentTypesExceptAll = 'minimal' | 'performance' | 'ux' | 'feedback' | 'location'
 export type consentTypes = 'all' | consentTypesExceptAll
+
+export interface StorageProvider {
+  setStore: (values: consentTypes[]) => void
+  getStore: () => consentTypes[]
+}