- Added constructor attribute to record interface;

- Added ability to override default constructor on item set;
- Added ability to use Array constructor for a key inside brackets [].

Author: Valerii Radchenko

package.json

@@ -1,6 +1,6 @@
 {
   "name": "ambary",
-  "version": "1.1.0",
+  "version": "1.2.0",
   "description": "Plain object in memory database",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",

src/base-ambary.ts

@@ -1,4 +1,9 @@
-import { IAmbaryRecord, IAmbaryOptions, IAmbarySetOptions } from './interfaces';
+import {
+  IAmbaryRecord,
+  IAmbaryOptions,
+  IAmbarySetOptions,
+  Constructor,
+} from './interfaces';
 
 export class BaseAmbary {
   private index = new Map<string, IAmbaryRecord>();
@@ -23,15 +28,17 @@ export class BaseAmbary {
     return typeof this.get(fullPath) !== 'undefined';
   }
 
-  protected set({ key, value, path }: IAmbarySetOptions) {
+  protected set({ key, value, path, Constructor = Object }: IAmbarySetOptions) {
     let parent = this.model;
 
     for (const pathKey of path) {
       const level = parent[pathKey];
 
       if (!level) {
-        parent[pathKey] = {};
-        parent = parent[pathKey];
+        const result = this.overrideKeyConstructor(pathKey, Constructor);
+
+        parent[result.key] = new result.Constructor();
+        parent = parent[result.key];
         continue;
       }
 
@@ -68,6 +75,7 @@ export class BaseAmbary {
       path,
       type,
       group,
+      constructor: value?.constructor || null,
       fullPath: [...path, key].join('.'),
     };
   }
@@ -95,7 +103,7 @@ export class BaseAmbary {
     return result;
   }
 
-  protected createRecordIterator = function*(
+  protected createRecordIterator = function* (
     model: Record<string, any> = this.model,
     path = [],
   ) {
@@ -126,6 +134,13 @@ export class BaseAmbary {
     return ['Object', 'Array'].includes(type) ? 'model' : 'value';
   }
 
+  private overrideKeyConstructor(key: string, DefaultConstructor: Constructor) {
+    const isArray = key.charAt(0) + key.charAt(key.length - 1) === '[]';
+    return isArray
+      ? { Constructor: Array, key: key.substr(1, key.length - 2) }
+      : { Constructor: DefaultConstructor, key };
+  }
+
   toJSON() {
     return this.model;
   }

src/interfaces.ts

@@ -1,9 +1,16 @@
+export type Constructor<
+  T = object,
+  A extends any[] = any[],
+  Static = {}
+> = (new (...a: A) => T) & Static;
+
 export interface IAmbaryRecord {
   key: string;
   value: any;
   path: string[];
   type: string;
   group: 'model' | 'value';
+  constructor: Constructor;
   fullPath: string;
 }
 
@@ -31,4 +38,5 @@ export interface IAmbarySetOptions {
   key: string;
   value: any;
   path: string[];
+  Constructor?: Constructor;
 }

src/specs/ambary.spec.ts

@@ -85,6 +85,7 @@ describe('Ambary', () => {
       path: ['level1_2', 'level2_1'],
       type: 'Number',
       group: 'value',
+      constructor: Number,
       fullPath: 'level1_2.level2_1.level3_2',
     });
   });
@@ -98,6 +99,7 @@ describe('Ambary', () => {
       path: ['level1_2', 'level2_1'],
       type: 'String',
       group: 'value',
+      constructor: String,
       fullPath: 'level1_2.level2_1.level3_1',
     });
   });
@@ -113,6 +115,7 @@ describe('Ambary', () => {
       path: ['level1_2', 'level2_1'],
       type: 'Number',
       group: 'value',
+      constructor: Number,
       fullPath: 'level1_2.level2_1.level3_2',
     });
   });

src/specs/base-ambary.spec.ts

@@ -84,6 +84,19 @@ describe('BaseAmbary', () => {
     expect(result).toEqual('test set');
   });
 
+  it('should set ambary item to non-existing parent(Array)', () => {
+    ambary.set({
+      key: 'level4_1',
+      value: 'test set',
+      path: ['[level1_3]', '0', 'level3_1'],
+    });
+
+    const result = ambary.get('level1_3.0.level3_1.level4_1');
+
+    expect(result).toEqual('test set');
+    expect(Array.isArray(ambary.get('level1_3'))).toBeTruthy();
+  });
+
   it('should return index iterator', () => {
     ambary.makeIndex();
     const iterator = ambary.getIndexIterator();