Merge pull request #2 from jeanlescure/add-length

Add length option

mod.d.ts

@@ -8,4 +8,5 @@ type Options = {
   dictionary: string[],
   skipShuffle: boolean,
   debug: boolean,
+  length: number,
 };

mod.ts

@@ -2,7 +2,7 @@
 // @deno-types="./mod.d.ts"
 import { version } from './version.json';
 
-const DEFAULT_RANDOM_ID_LEN: number = 6;
+const DEFAULT_ID_LENGTH: number = 6;
 
 const DIGIT_FIRST_ASCII: number = 48;
 const DIGIT_LAST_ASCII: number = 58;
@@ -21,6 +21,7 @@ const DEFAULT_OPTIONS: Options = {
   dictionary: [],
   skipShuffle: false,
   debug: false,
+  length: DEFAULT_ID_LENGTH,
 };
 
 class ShortUniqueId extends Function {
@@ -42,6 +43,8 @@ class ShortUniqueId extends Function {
 
   dictLength: number = 0;
 
+  uuidLength: number;
+
   /* tslint:disable consistent-return */
   log(...args: any[]) {
     const finalArgs = [...args];
@@ -72,8 +75,11 @@ class ShortUniqueId extends Function {
     const {
       dictionary: userDict,
       skipShuffle,
+      length,
     } = options;
 
+    this.uuidLength = length;
+
     if (userDict && userDict.length > 1) {
       this.setDictionary(userDict);
     } else {
@@ -158,7 +164,7 @@ class ShortUniqueId extends Function {
   }
 
   /*  Generates UUID by creating each part randomly. */
-  randomUUID(uuidLength: number = DEFAULT_RANDOM_ID_LEN) {
+  randomUUID(uuidLength: number = this.uuidLength) {
     let id;
     let randomPartIdx;
     let j;

test.ts

@@ -49,9 +49,10 @@ test({
     const uid: ShortUniqueId = new ShortUniqueId({
       dictionary: ['a', '1'],
       skipShuffle: true,
+      length: 2,
     });
     /* tslint:disable no-magic-numbers */
-    assert((/^[a1][a1]$/).test(uid(2)));
+    assert((/^[a1][a1]$/).test(uid()));
     /* tslint:enable no-magic-numbers */
     assertEquals(
       [uid.sequentialUUID(), uid.sequentialUUID()].join(''),