From 607ce17a1966eb9cc3be7306d61a9595e51bb26c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=90=D0=BB=D0=B5=D0=BA=D1=81=D0=B5=D0=B9=20=D0=9A=D1=80?=
 =?UTF-8?q?=D1=8F=D0=B6=D0=B5=D0=B2?= <ubivaza@gmail.com>
Date: Tue, 11 Jul 2023 20:44:45 +0300
Subject: [PATCH] support different length for enum

---
 packages/imask/example.html        | 17 +++-----------
 packages/imask/src/masked/enum.ts  | 14 +++++++++--
 packages/imask/test/masked/enum.ts | 37 ++++++++++++++++++++++++++++++
 3 files changed, 52 insertions(+), 16 deletions(-)
 create mode 100644 packages/imask/test/masked/enum.ts

diff --git a/packages/imask/example.html b/packages/imask/example.html
index a57c1c1d..606afa5f 100644
--- a/packages/imask/example.html
+++ b/packages/imask/example.html
@@ -12,20 +12,9 @@ <h1>IMask Core Demo</h1>
   <!-- <script src="https://unpkg.com/imask"></script> -->
   <script type="text/javascript">
     const opts = {
-      mask: [
-        { mask: '' },
-        {
-          mask: 'd % aa',
-          lazy: false,
-          blocks: {
-            d: {
-              mask: Number,
-              expose: true,
-            },
-          },
-          expose: true,
-        },
-      ],
+      mask: IMask.MaskedEnum,
+      enum: ['a', 'bc', 'def'],
+      lazy: false,
     };
 
     var result = document.getElementById('value');
diff --git a/packages/imask/src/masked/enum.ts b/packages/imask/src/masked/enum.ts
index 99fe6afc..7f1ad684 100644
--- a/packages/imask/src/masked/enum.ts
+++ b/packages/imask/src/masked/enum.ts
@@ -15,6 +15,10 @@ export default
 class MaskedEnum extends MaskedPattern {
   declare enum: Array<string>;
 
+  constructor (opts?: MaskedEnumOptions) {
+    super(opts as MaskedPatternOptions); // mask will be created in _update
+  }
+
   override updateOptions (opts: Partial<MaskedEnumOptions>) {
     super.updateOptions(opts);
   }
@@ -23,7 +27,13 @@ class MaskedEnum extends MaskedPattern {
     const { enum: _enum, ...eopts }: MaskedEnumPatternOptions = opts;
 
     if (_enum) {
-      eopts.mask = '*'.repeat(_enum[0].length);
+      const lengths = _enum.map(e => e.length);
+      const requiredLength = Math.min(...lengths);
+      const optionalLength = Math.max(...lengths) - requiredLength;
+
+      eopts.mask = '*'.repeat(requiredLength);
+      if (optionalLength) eopts.mask += '[' + '*'.repeat(optionalLength) + ']';
+
       this.enum = _enum;
     }
 
@@ -31,7 +41,7 @@ class MaskedEnum extends MaskedPattern {
   }
 
   override doValidate (flags: AppendFlags): boolean {
-    return this.enum.some(e => e.indexOf(this.unmaskedValue) >= 0) &&
+    return this.enum.some(e => e.indexOf(this.unmaskedValue) === 0) &&
       super.doValidate(flags);
   }
 }
diff --git a/packages/imask/test/masked/enum.ts b/packages/imask/test/masked/enum.ts
new file mode 100644
index 00000000..c9cf9b00
--- /dev/null
+++ b/packages/imask/test/masked/enum.ts
@@ -0,0 +1,37 @@
+import assert from 'assert';
+import { describe, it } from 'node:test';
+
+import MaskedEnum from '../../src/masked/enum';
+
+
+describe('MaskedEnum', function () {
+  it('should work with different length', function () {
+    const _enum = ['a', 'bc', 'def'];
+    const masked = new MaskedEnum({ enum: _enum });
+
+    _enum.forEach(e => {
+      masked.value = e;
+      assert.equal(masked.value, e);
+    });
+  });
+
+  it('should set required and optional chars', function () {
+    const masked = new MaskedEnum({ enum: ['a', 'bc', 'def'], lazy: false });
+    masked.value = '';
+
+    assert.equal(masked.value, '_');
+  });
+
+  it('should update options', function () {
+    const masked = new MaskedEnum({ enum: ['A', 'B'] });
+
+    masked.value = 'A';
+    assert.equal(masked.value, 'A');
+
+    masked.updateOptions({ enum: ['C', 'D'] });
+    assert.equal(masked.value, '');
+
+    masked.value = 'C';
+    assert.equal(masked.value, 'C');
+  });
+});