import { AnyRouter } from "@trpc/server";
import { createRecursiveProxy } from "@trpc/server/shared";
import { getQueryKey } from "../../internals/getQueryKey";
import { type CreateSolidQueryHooks } from "../hooks/createHooksInternal";

/**
 * Create proxy for decorating procedures
 * @internal
 */
export function createSolidProxyDecoration<
  TRouter extends AnyRouter,
  TSSRContext = unknown
>(name: string, hooks: CreateSolidQueryHooks<TRouter, TSSRContext>) {
  return createRecursiveProxy((opts) => {
    const args = opts.args;

    const pathCopy = [name, ...opts.path];

    // The last arg is for instance `.useMutation` or `.useQuery()`
    // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
    const lastArg = pathCopy.pop()!;

    // The `path` ends up being something like `post.byId`
    const path = pathCopy.join(".");
    if (lastArg === "useMutation") {
      return (hooks as any)[lastArg](path, ...args);
    }
    const [input, ...rest] = args;
    return (hooks as any)[lastArg](
      () => getQueryKey(path, typeof input === "function" ? input() : input),
      () => rest.map((arg) => (typeof arg === "function" ? arg() : arg))
    );
  });
}