Merge pull request #78 from trustpilot/add-session-persister-to-config

Add session persister to config

src/inMemoryPersister.ts

@@ -0,0 +1,13 @@
+import { UserSessionPersister } from './userSessionPersister';
+
+export class InMemoryPersister implements UserSessionPersister {
+  private endOfLife: number = 0;
+  private storage: string = '';
+  public loadUserSession() {
+      return !this.endOfLife || (Date.now() > this.endOfLife) ? '' : this.storage;
+  }
+  public saveUserSession(userSession: string, daysToLive: number) {
+      this.storage = userSession;
+      this.endOfLife = Date.now() + (daysToLive * 24 * 60 * 60 * 1000);
+  }
+}

src/index.ts

@@ -1,3 +1,4 @@
+import { InMemoryPersister } from './inMemoryPersister';
 import {
     config,
     create,
@@ -12,6 +13,7 @@ import {
 } from './main';
 import { SplitTest } from './splitTest';
 import { uiFactory } from './ui';
+import { CookiePersister, UserSessionPersister } from './userSessionPersister';
 
 const ui = uiFactory(
     testsObservable,
@@ -47,6 +49,9 @@ export {
     reset,
     ui,
     SplitTest,
+    UserSessionPersister,
+    InMemoryPersister,
+    CookiePersister,
 };
 
 export default {

src/main.ts

@@ -30,6 +30,11 @@ export function config(userConfig: Partial<Config> = {}) {
     if (userConfig.onVariationChange) {
         _config.onVariationChange = userConfig.onVariationChange;
     }
+    if (userConfig.sessionPersister) {
+        const session = _config.sessionPersister.loadUserSession() || '';
+        _config.sessionPersister = userConfig.sessionPersister;
+        _config.sessionPersister.saveUserSession(session, _config.userSessionDaysToLive);
+    }
 }
 
 /**

src/userSessionPersister.ts

@@ -1,40 +1,35 @@
 import config from './config';
-
 export interface UserSessionPersister {
     loadUserSession(): string | null;
     saveUserSession(userSession: string, daysToLive: number): void;
 }
-
-function createCookie(name: string, value: string, days: number): void {
-    let expires = '';
-    if (days) {
-        const date = new Date();
-        date.setTime(date.getTime() + (days * 24 * 60 * 60 * 1000));
-        expires = '; expires=' + date.toUTCString();
-    }
-    document.cookie = name + '=' + value + expires + '; path=/';
-}
-
-function readCookie(name: string): string | null {
-    const nameEq = name + '=';
-    const ca = document.cookie.split(';');
-    for (let c of ca) {
-        while (c.charAt(0) === ' ') {
-            c = c.substring(1, c.length);
-        }
-        if (c.indexOf(nameEq) === 0) {
-            return c.substring(nameEq.length, c.length);
-        }
-    }
-    return null;
-}
-
 export class CookiePersister implements UserSessionPersister {
-    public loadUserSession() {
-        return readCookie(config.cookieName);
-    }
-
-    public saveUserSession(userSession: string, daysToLive: number) {
-        createCookie(config.cookieName, userSession, daysToLive);
-    }
+  private static createCookie(name: string, value: string, days: number): void {
+      let expires = '';
+      if (days) {
+          const date = new Date();
+          date.setTime(date.getTime() + (days * 24 * 60 * 60 * 1000));
+          expires = '; expires=' + date.toUTCString();
+      }
+      document.cookie = name + '=' + value + expires + '; path=/';
+  }
+  private static readCookie(name: string): string | null {
+      const nameEq = name + '=';
+      const ca = document.cookie.split(';');
+      for (let c of ca) {
+          while (c.charAt(0) === ' ') {
+              c = c.substring(1, c.length);
+          }
+          if (c.indexOf(nameEq) === 0) {
+              return c.substring(nameEq.length, c.length);
+          }
+      }
+      return null;
+  }
+  public loadUserSession() {
+      return CookiePersister.readCookie(config.cookieName);
+  }
+  public saveUserSession(userSession: string, daysToLive: number) {
+    CookiePersister.createCookie(config.cookieName, userSession, daysToLive);
+  }
 }