feat(site): add reset function and throw error when initializing twice

Author: bartlangelaan

src/HnServerResponse.ts

@@ -0,0 +1,28 @@
+export interface Entity {
+  __hn?: {
+    view_modes: {
+      [viewMode: string]: boolean,
+    };
+    entity: {
+      type: string,
+      bundle: string,
+    };
+    url: string;
+    status?: number;
+  };
+}
+
+export default interface HnServerResponse {
+  data?: {
+    [uuid: string]: Entity | any,
+  };
+  paths?: {
+    [path: string]: string,
+  };
+  __hn?: {
+    request?: {
+      user?: string,
+      token?: string,
+    };
+  };
+}

src/Site.ts

@@ -4,38 +4,50 @@ import * as deepmerge from 'deepmerge';
 import * as getNested from 'get-nested';
 import { stringify } from 'query-string';
 import SiteInitializeParams from './SiteInitializeParams';
+import HnServerResponse from './HnServerResponse';
 
 polyfill();
 
 const propertiesToHydrate = ['tokensToVerify', 'user', 'data'];
 
 class Site {
 
-  private initialized: boolean = false;
+  private initialized: boolean;
 
   // Created when initializing
   private url: string;
 
   // Can be hydrated and dehydrated
-  private tokensToVerify: string[] = [];
+  private tokensToVerify: string[];
   private user: string;
-  private data = {
-    data: {},
-    paths: {},
-  };
+  private data: HnServerResponse;
 
   // Not hydrated
-  private pagesLoading = {};
+  private pagesLoading: {[s: string]: Promise<string>};
 
   constructor(initParams?: SiteInitializeParams) {
+    this.reset();
     if (initParams) this.initialize(initParams);
   }
 
   initialize({ url }: SiteInitializeParams) {
+    if (this.initialized) throw Error('The site is already initialized.');
     this.initialized = true;
     this.url = url;
   }
 
+  reset() {
+    this.initialized = false;
+    this.url = null;
+    this.tokensToVerify = [];
+    this.user = null;
+    this.data = {
+      data: {},
+      paths: {},
+    };
+    this.pagesLoading = {};
+  }
+
   /**
    * Creates an object that can be hydrated by the hydrate function.
    */
@@ -56,7 +68,7 @@ class Site {
     });
   }
 
-  private fetch(path, options = {}): Promise<object> {
+  private fetch(path, options = {}): Promise<HnServerResponse> {
     if (!this.initialized) {
       throw Error('Site is not intitialized. Pass an object when creating a site, or use the ' +
         'initialize method.');
@@ -77,7 +89,7 @@ class Site {
       });
   }
 
-  public getPage(path, loadFromServer = false): Promise<void> {
+  public getPage(path, loadFromServer = false): Promise<string> {
     if (!this.pagesLoading[ path ]) {
       const dataMaybeAlreadyLoaded = getNested(() => this.data.data[this.data.paths[path]]);
       if (getNested(() => dataMaybeAlreadyLoaded.__hn.view_modes.includes('default'))) {
@@ -96,16 +108,16 @@ class Site {
         _hn_user: this.user ? this.user : undefined,
         _hn_verify: tokensToVerify,
       }))
-        .then((page: Response) => {
+        .then((page: HnServerResponse) => {
 
           // Get the user id, to pass to all new requests.
-          this.user = getNested(() => page['__hn'].request.user, this.user);
+          this.user = getNested(() => page.__hn.request.user, this.user);
 
           // Remove all sent tokens from the tokensToVerify.
           this.tokensToVerify = this.tokensToVerify.filter(t => tokensToVerify.indexOf(t) === -1);
 
           // Add new token to tokensToVerify.
-          const newToken = getNested(() => page['__hn'].request.token);
+          const newToken = getNested(() => page.__hn.request.token);
           if (newToken) this.tokensToVerify.push(newToken);
 
           // Add all data to the global data storage.
@@ -119,12 +131,13 @@ class Site {
               [path]: '500',
             },
           });
-        });
+        })
+        .then(() => this.data.paths[ path ]);
     }
-    return this.pagesLoading[ path ].then(() => this.data.paths[ path ]);
+    return this.pagesLoading[ path ];
   }
 
-  private addData(data: object) {
+  private addData(data: HnServerResponse) {
     this.data = deepmerge(this.data, data, { arrayMerge: (a, b) => b });
   }