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;

    const queryKey = getQueryKey(path, input);
    return (hooks as any)[lastArg](queryKey, ...rest);
  });
}