import {
  AdapterResponse,
  DataAdapterKeyPath,
  IDataAdapter,
} from '../interfaces/IDataAdapter';
import StatsigServer from '../StatsigServer';
import {
  GateForConfigSpecTest,
  GatesForIdListTest,
} from './BootstrapWithDataAdapter.data';
let hitNetwork = false;
jest.mock('node-fetch', () =>
  jest.fn().mockImplementation(() => {
    hitNetwork = true;
    return Promise.resolve();
  }),
);

function makeDownloadConfigSpecsResponse(gates?: any[]) {
  return {
    time: Date.now(),
    feature_gates: gates ?? [],
    dynamic_configs: [],
    layer_configs: [],
    has_updates: true,
  };
}

class BootstrapDataAdapter implements IDataAdapter {
  private specs: string;
  private idLists: Record<string, string>;
  private idListLookup: string;

  constructor(specs: Record<string, unknown>, idLists: Record<string, string>) {
    this.specs = JSON.stringify(specs);
    this.idLists = idLists;
    this.idListLookup = JSON.stringify(Object.keys(idLists));
  }

  get(key: string): Promise<AdapterResponse> {
    if (key.includes(DataAdapterKeyPath.V1Rulesets)) {
      return Promise.resolve({ result: this.specs });
    } else if (key.includes(DataAdapterKeyPath.V1IDLists)) {
      return Promise.resolve({ result: this.idListLookup });
    } else {
      const second_part = key.split('|')[1]
      const idListName = second_part.split('::')[1];
      return Promise.resolve({ result: this.idLists[idListName] });
    }
  }

  set = () => Promise.resolve();
  initialize = () => Promise.resolve();
  shutdown = () => Promise.resolve();
}

describe('Bootstrap with DataAdapter', () => {
  beforeEach(async () => {
    hitNetwork = false;
  });

  it('makes no network calls when bootstrapping', async () => {
    const adapter = new BootstrapDataAdapter(
      makeDownloadConfigSpecsResponse(),
      {},
    );

    const statsig = new StatsigServer('secret-key', {
      dataAdapter: adapter,
    });
    await statsig.initializeAsync();

    expect(hitNetwork).toBe(false);
  });

  it('bootstraps id lists', async () => {
    const adapter = new BootstrapDataAdapter(
      makeDownloadConfigSpecsResponse(GatesForIdListTest),
      {
        /* a-user + b-user */
        user_id_list: ['+Z/hEKLio', '+M5m6a10x'].join('\n'),
      },
    );

    const statsig = new StatsigServer('secret-key', {
      dataAdapter: adapter,
    });
    await statsig.initializeAsync();

    await Promise.all(
      [
        {
          user: { userID: 'a-user' },
          expectedValue: true,
        },
        {
          user: { userID: 'b-user' },
          expectedValue: true,
        },
        {
          user: { userID: 'c-user' },
          expectedValue: false,
        },
      ].map(
        async ({ user, expectedValue }) => {
          const gateResult = statsig.checkGate(user, 'test_id_list');
          expect(gateResult).toBe(expectedValue)
        }
      ),
    );
  });

  it('bootstraps config specs', async () => {
    const adapter = new BootstrapDataAdapter(
      makeDownloadConfigSpecsResponse(GateForConfigSpecTest),
      {},
    );

    const statsig = new StatsigServer('secret-key', {
      dataAdapter: adapter,
    });
    await statsig.initializeAsync();
    const gateResult = statsig.checkGate({ userID: 'a-user' }, 'test_public');
    expect(gateResult).toBeTruthy();
  });
});