Merge pull request #7 from suzanne-lee/get-key-count

Added getKeyCount()
AnyhowStep · Jun 22, 2019 · 62ad529 · 62ad529
2 parents ccd20e8 + 9bde92d
commit 62ad529
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 1 deletion.
diff --git a/dist/src/main/util.d.ts b/dist/src/main/util.d.ts
@@ -16,6 +16,7 @@ export declare function toKey<E extends typeof Enum, V extends number | string>(
 export declare function toKey<E extends typeof Enum, K extends StringKeyOf<E>>(e: E, value: E[K]): K;
 export declare function toKey<E extends typeof Enum>(e: E, value: E[StringKeyOf<E>]): StringKeyOf<E>;
 export declare function toKey<E extends typeof Enum>(e: E, mixed: any): (StringKeyOf<E>) | undefined;
+export declare function getKeyCount<E extends typeof Enum>(e: E): number;
 export declare class WrappedEnum<E extends typeof Enum> {
     private readonly e;
     private readonly keys;
@@ -34,4 +35,5 @@ export declare class WrappedEnum<E extends typeof Enum> {
     toKey<K extends StringKeyOf<E>>(mixed: E[K]): K;
     toKey(mixed: E[StringKeyOf<E>]): StringKeyOf<E>;
     toKey(mixed: any): (StringKeyOf<E>) | undefined;
+    getKeyCount(): number;
 }
diff --git a/dist/src/main/util.js b/dist/src/main/util.js
diff --git a/dist/src/main/util.js.map b/dist/src/main/util.js.map
diff --git a/src/main/util.ts b/src/main/util.ts
@@ -90,6 +90,9 @@ export function toKey<E extends typeof Enum> (e : E, mixed : any) : (StringKeyOf
 export function toKey<E extends typeof Enum> (e : E, mixed : any) : (StringKeyOf<E>)|undefined {
     return toKeyInternal(e, getKeys(e), mixed);
 }
+export function getKeyCount<E extends typeof Enum> (e : E) : number {
+    return getKeys(e).length;
+}
 
 export class WrappedEnum<E extends typeof Enum> {
     private readonly e : E;
@@ -136,4 +139,7 @@ export class WrappedEnum<E extends typeof Enum> {
     public toKey (mixed : any) : (StringKeyOf<E>)|undefined {
         return toKeyInternal(this.e, this.keys, mixed);
     }
+    public getKeyCount () : number {
+        return this.keys.length;
+    }
 }