/**
 * 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 {};
  }
}