import { initClient, tsRestFetchApi } from "@ts-rest/core";
import { contract } from "./contract";
const { version: SDK_VERSION } = require("../package.json");

/**
 * Provides raw API access to the Inferable API.
 */
export const createApiClient = ({
  baseUrl,
  machineId,
  clientAbortController,
  apiSecret,
}: {
  baseUrl?: string;
  machineId?: string;
  clientAbortController?: AbortController;
  apiSecret?: string;
}) =>
  initClient(contract, {
    baseUrl: baseUrl ?? "https://api.inferable.ai",
    baseHeaders: {
      "x-machine-sdk-version": SDK_VERSION,
      "x-machine-sdk-language": "typescript",
      ...(apiSecret ? { authorization: apiSecret } : {}),
      ...(machineId ? { "x-machine-id": machineId } : {}),
    },
    api: async (args) => {
      try {
        return await tsRestFetchApi({
          ...args,
          ...(clientAbortController
            ? { signal: clientAbortController.signal }
            : {}),
        });
      } catch (e) {
        return {
          status: -1,
          headers: new Headers(),
          body: e,
        };
      }
    },
  });