blitzkit/packages/core/src/blitz/fetchBlitz.ts

import { assertSecret, EventManager } from "../blitzkit";
import { patientFetch } from "../blitzkit/patientFetch";
import type { Region } from "./regions";

const RETRY_ERRORS = ["REQUEST_LIMIT_EXCEEDED", "SOURCE_NOT_AVAILABLE"];

export const retryAbleBlitzFetchEvent = new EventManager();
export const blitzFetchQueueAvailableEvent = new EventManager();

type BlitzResponse<Data extends object> =
  | {
      status: "error";
      error: { field: string; message: string; code: 402; value: null };
    }
  | {
      status: "ok";
      data: Data;
    };

const MAX_CALLS_PER_SECOND = 10;

const queue: { url: string; resolve: (data: object) => void }[] = [];
let inProgress = 0;

function decrement() {
  inProgress--;
  blitzFetchQueueAvailableEvent.emit(undefined);
  manageQueue();
}

async function manageQueue() {
  if (queue.length > 0 && inProgress < MAX_CALLS_PER_SECOND) {
    inProgress++;

    const timeout = setTimeout(decrement, 1000);
    const request = queue.shift()!;
    const data = (await patientFetch(request.url)
      .then((response) => response.json())
      .catch((error) => ({ status: "error", error }))) as BlitzResponse<
      typeof request.resolve
    >;

    if (data.status === "ok") {
      request.resolve(data.data);
    } else {
      if (RETRY_ERRORS.includes(data.error.message)) {
        retryAbleBlitzFetchEvent.emit(undefined);
        clearTimeout(timeout);
        setTimeout(() => {
          queue.push(request);
          decrement();
        }, 1000 / MAX_CALLS_PER_SECOND);
      } else {
        clearTimeout(timeout);
        decrement();

        throw new Error(`Wargaming response error status: "${data.status}"`, {
          cause: `Message: "${data.error.message}"\nURL: "${request.url}"`,
        });
      }
    }
  }
}

export type FetchBlitzParams = Record<string, string | number | undefined>;

export function fetchBlitz<Data extends object>(
  region: Region,
  path: string,
  params: FetchBlitzParams = {}
) {
  return new Promise<Data>((resolve) => {
    queue.push({
      url: `https://api.wotblitz.${region}/wotb/${path}/?${Object.entries({
        application_id: assertSecret(
          import.meta.env.PUBLIC_WARGAMING_APPLICATION_ID
        ),
        ...params,
      })
        .filter(([_, value]) => value !== undefined)
        .map(([key, value]) => `${key}=${encodeURIComponent(value!)}`)
        .join("&")}`,
      resolve(data) {
        resolve(data as Data);
      },
    });
    manageQueue();
  });
}