Added missing general functions for enum to array

Author: lumburovskalina

case-types-parser.ts

@@ -31,4 +31,11 @@ export function snakeCaseToCamelCase(str: string) {
             .toUpperCase()
             .replace('_', '')
     );
+}
+
+export enum caseType {
+    LOWER,
+    UPPER,
+    CAPITALIZE_FIRST,
+    CAPITALIZE_FIRST_PER_WORD
 }

general.ts

@@ -1,3 +1,5 @@
+import { capitalizeFirst, capitalizeFirstPerWord, caseType } from "./case-types-parser";
+
 export function jsonCopy(src: any): any {
     return JSON.parse(JSON.stringify(src));
 }
@@ -107,4 +109,37 @@ export function formatBytes(bytes, decimals = 2) {
     const sizes = ['Bytes', 'KB', 'MB', 'GB', 'TB', 'PB', 'EB', 'ZB', 'YB'];
     const i = Math.floor(Math.log(bytes) / Math.log(1024));
     return parseFloat((bytes / Math.pow(1024, i)).toFixed(dm)) + ' ' + sizes[i];
-}
+}
+
+
+export type enumToArrayOptions = {
+    caseType?: caseType;
+    prefix?: string;
+    nameFunction?: (name: string) => string;
+}
+
+export function enumToArray(e: Object, options: enumToArrayOptions = null): any[] {
+    const arr = Object.values(e);
+    if (!options) return sortByEnumPos(e, arr);
+    let func;
+    if (options.caseType == caseType.LOWER) func = (x) => x.toLowerCase();
+    else if (options.caseType == caseType.UPPER) func = (x) => x.toUpperCase();
+    else if (options.caseType == caseType.CAPITALIZE_FIRST) func = capitalizeFirst;
+    else if (options.caseType == caseType.CAPITALIZE_FIRST_PER_WORD) func = capitalizeFirstPerWord;
+
+    if (func) return enumToArray(e, { prefix: options.prefix, nameFunction: func });
+    if (!options.nameFunction) return sortByEnumPos(e, arr.map(x => ({ name: options.prefix + x, value: x })));
+    return sortByEnumPos(e, arr.map(x => ({ name: (options.prefix ? options.prefix : "") + options.nameFunction(x), value: x })));
+}
+
+function sortByEnumPos(e: Object, arr: any[]) {
+    const order = [];
+    for (let key in e) {
+        order.push(key);
+    }
+    return arr.sort((a, b) => {
+        const index1 = order.findIndex(key => e[key] === a.code);
+        const index2 = order.findIndex(key => e[key] === b.code);
+        return index1 - index2;
+    });
+}