79023ee190
Reviewed By: bhamodi Differential Revision: D33331422 fbshipit-source-id: 016e8dcc0c0c7f1fc353a348b54fda0d5e2ddc01
67 lines
1.4 KiB
TypeScript
67 lines
1.4 KiB
TypeScript
/**
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*
|
|
* @format
|
|
*/
|
|
|
|
export type GKID = string;
|
|
|
|
export const TEST_PASSING_GK = 'TEST_PASSING_GK';
|
|
export const TEST_FAILING_GK = 'TEST_FAILING_GK';
|
|
export type GKMap = {[key: string]: boolean};
|
|
|
|
const whitelistedGKs: Array<GKID> = [];
|
|
|
|
export function loadGKs(_username: string, _gks: Array<GKID>): Promise<GKMap> {
|
|
return Promise.reject(
|
|
new Error('Implement your custom logic for loading GK'),
|
|
);
|
|
}
|
|
|
|
export function loadDistilleryGK(
|
|
_gk: GKID,
|
|
): Promise<{[key: string]: {result: boolean}}> {
|
|
return Promise.reject(
|
|
new Error('Implement your custom logic for loading GK'),
|
|
);
|
|
}
|
|
|
|
export default class GK {
|
|
static init(_username: string) {}
|
|
|
|
static get(id: GKID): boolean {
|
|
if (process.env.NODE_ENV === 'test' && id === TEST_PASSING_GK) {
|
|
return true;
|
|
}
|
|
if (whitelistedGKs.includes(id)) {
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static async withWhitelistedGK(
|
|
id: GKID,
|
|
callback: () => Promise<void> | void,
|
|
) {
|
|
whitelistedGKs.push(id);
|
|
try {
|
|
const p = callback();
|
|
if (p) {
|
|
await p;
|
|
}
|
|
} finally {
|
|
const idx = whitelistedGKs.indexOf(id);
|
|
if (idx !== -1) {
|
|
whitelistedGKs.splice(idx, 1);
|
|
}
|
|
}
|
|
}
|
|
|
|
static allGKs(): GKMap {
|
|
return {};
|
|
}
|
|
}
|