feat(type): support lower/mixed case as identifier for enum values

```typescript enum Units { MILLIGRAM = 'm', GRAM = 'g', } expect(cast<Units>('MILLIGRAM')).toBe(Units.MILLIGRAM); expect(cast<Units>('MilLiGrAm')).toBe(Units.MILLIGRAM); ```
deepkit · May 4, 2024 · ce0166e · ce0166e
1 parent 0216af4
commit ce0166e
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/packages/type/src/serializer.ts b/packages/type/src/serializer.ts
@@ -1977,7 +1977,15 @@ export class Serializer {
         this.serializeRegistry.register(ReflectionKind.enum, (type, state) => state.addSetter(state.accessor));
         this.deserializeRegistry.register(ReflectionKind.enum, (type, state) => {
             const valuesVar = state.setVariable('values', type.values);
+            const lowercaseNames = state.setVariable('lowercaseNames', Object.keys(type.enum).map(v => v.toLowerCase()));
+            const allLowercased = Object.keys(type.enum).every(v => v.toLowerCase() === v);
+            const enumValues = state.setVariable('enumValues', type.values);
+            const allowLowercase = allLowercased ? '' : `
+                ${state.accessor} = ${enumValues}[${lowercaseNames}.indexOf(String(${state.accessor}).toLowerCase())] ?? ${state.accessor};
+            `;
+
             state.addCodeForSetter(`
+                ${allowLowercase}
                 ${state.setter} = ${state.accessor};
                 if (${valuesVar}.indexOf(${state.accessor}) === -1) ${state.throwCode('enum', `'No valid value of ' + ${valuesVar}.join(', ')`)}
             `);

diff --git a/packages/type/tests/serializer.spec.ts b/packages/type/tests/serializer.spec.ts
@@ -1019,6 +1019,22 @@ test('naming strategy camel case', () => {
     }
 });
 
+test('enum mixed case', () => {
+    enum Units {
+        MILLIGRAM = 'm',
+        GRAM = 'g',
+        KILOGRAM = 'k',
+    }
+
+    expect(cast<Units>('milligram')).toBe('m');
+    expect(cast<Units>('milligram')).toBe(Units.MILLIGRAM);
+
+    expect(cast<Units>('MilliGRAM')).toBe(Units.MILLIGRAM);
+
+    expect(cast<Units>('gram')).toBe('g');
+    expect(cast<Units>('gram')).toBe(Units.GRAM);
+});
+
 test('enum union', () => {
     enum StatEnginePowerUnit {
         Hp = 'hp',