import { io } from "socket.io-client";

type Promisify<T extends Record<string | symbol | number, any>> = {
  [K in keyof T]: (...params: Parameters<T[K]>) => Promise<ReturnType<T[K]>>;
};

export default function Client<API>(
  endpoint: string = "http://localhost:8080"
) {
  const socket = io(endpoint);

  const queue: { [key: string]: (value: unknown) => void } = {};

  socket.on("function-response", (msg) => {
    const { result, id, status, error } = msg;

    if (status > 200) {
      throw new Error(`ServerError: ${error}`);
    }

    queue[id](result);

    delete queue[id];
  });

  const waitForResult = (id: string, resolve: (value: unknown) => void) => {
    queue[id] = resolve;
  };

  return new Proxy(
    {},
    {
      get(_, procedureName) {
        return function (...params: any[]) {
          const id = `${new Date().valueOf()}-${procedureName.toString()}-${JSON.stringify(
            params
          )}`;

          socket.emit("function-call", {
            procedureName,
            params,
            id,
          });

          return new Promise((resolve) => waitForResult(id, resolve));
        };
      },
    }
  ) as Promisify<API>;
}