-
Notifications
You must be signed in to change notification settings - Fork 0
/
helpers.ts
46 lines (38 loc) · 1017 Bytes
/
helpers.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
const isPowerOfTwo = (x: number): boolean => {
return x != 0 && (x & (x - 1)) == 0;
};
export function getEnumFlags<
O extends object,
K extends O[keyof O] = O[keyof O]
>(obj: O): K[] {
const isFlag = (arg: string | number | K): arg is K => {
const nArg = Number(arg);
const isNumber = !Number.isNaN(nArg);
return isNumber && isPowerOfTwo(nArg);
};
const enumFlags: K[] = [];
Object.keys(obj).forEach(key => {
const nKey = Number(key);
if (isFlag(nKey)) {
enumFlags.push(nKey);
}
});
return enumFlags;
}
export function getEnumValues<
O extends object,
K extends O[keyof O] = O[keyof O]
>(obj: O): K[] {
const isFlag = (arg: string | number | K): arg is K => {
const nArg = Number(arg);
return !Number.isNaN(nArg);
};
const enumFlags: K[] = [];
Object.keys(obj).forEach(key => {
const nKey = Number(key);
if (isFlag(nKey)) {
enumFlags.push(nKey);
}
});
return enumFlags;
}