yaak-mountain-loop/apps/yaak-client/lib/requestBodyConversion.ts

import type { HttpRequest } from "@yaakapp-internal/models";
import {
  BODY_TYPE_BINARY,
  BODY_TYPE_FORM_MULTIPART,
  BODY_TYPE_FORM_URLENCODED,
  BODY_TYPE_GRAPHQL,
  BODY_TYPE_JSON,
  BODY_TYPE_NONE,
} from "./model_util";

type Body = HttpRequest["body"];
type BodyType = HttpRequest["bodyType"];
type GraphQLBody = {
  query: string;
  variables: string | undefined;
};

export function convertRequestBody({
  body,
  fromBodyType,
  toBodyType,
}: {
  body: Body;
  fromBodyType: BodyType;
  toBodyType: BodyType;
}): Body {
  if (toBodyType === BODY_TYPE_NONE) {
    return {};
  }

  if (toBodyType === BODY_TYPE_GRAPHQL) {
    return toGraphQLBody(body);
  }

  if (toBodyType === BODY_TYPE_FORM_URLENCODED || toBodyType === BODY_TYPE_FORM_MULTIPART) {
    return toFormBody(body);
  }

  if (toBodyType === BODY_TYPE_BINARY) {
    return typeof body.filePath === "string" ? { filePath: body.filePath } : {};
  }

  return toTextBody(body, fromBodyType, toBodyType);
}

export function normalizeGraphQLBody(body: Body): GraphQLBody {
  return toGraphQLBody(body);
}

function toGraphQLBody(body: Body): GraphQLBody {
  if (typeof body.query === "string") {
    return {
      query: body.query,
      variables: typeof body.variables === "string" ? body.variables : undefined,
    };
  }

  if (typeof body.text === "string") {
    const parsed = parseJsonObject(body.text);
    if (parsed != null) {
      const query = typeof parsed.query === "string" ? parsed.query : "";
      const variables =
        parsed.variables == null ? undefined : JSON.stringify(parsed.variables, null, 2);

      return { query, variables };
    }

    return { query: body.text, variables: undefined };
  }

  return { query: "", variables: undefined };
}

function toFormBody(body: Body): Body {
  if (Array.isArray(body.form)) {
    return {
      form: body.form.map((p) => ({
        enabled: p.enabled !== false,
        name: typeof p.name === "string" ? p.name : "",
        value: stringifyFormValue(p.value ?? p.file),
        contentType: typeof p.contentType === "string" ? p.contentType : undefined,
        filename: typeof p.filename === "string" ? p.filename : undefined,
        file: typeof p.file === "string" ? p.file : undefined,
        id: typeof p.id === "string" ? p.id : undefined,
      })),
    };
  }

  if (typeof body.text === "string") {
    const parsed = parseJsonObject(body.text);
    if (parsed != null) {
      return { form: objectToFormPairs(parsed) };
    }

    return { form: urlEncodedTextToFormPairs(body.text) };
  }

  if (typeof body.query === "string") {
    return {
      form: [
        { enabled: true, name: "query", value: body.query },
        ...(typeof body.variables === "string"
          ? [{ enabled: true, name: "variables", value: body.variables }]
          : []),
      ],
    };
  }

  return { form: [] };
}

function toTextBody(body: Body, fromBodyType: BodyType, toBodyType: BodyType): Body {
  const sendJsonComments =
    typeof body.sendJsonComments === "boolean" ? { sendJsonComments: body.sendJsonComments } : {};

  if (typeof body.text === "string") {
    return { text: body.text, ...sendJsonComments };
  }

  if (Array.isArray(body.form)) {
    if (toBodyType === BODY_TYPE_JSON) {
      return { text: JSON.stringify(formBodyToObject(body.form), null, 2) };
    }

    return { text: formBodyToUrlEncodedText(body.form) };
  }

  if (typeof body.query === "string") {
    if (toBodyType === BODY_TYPE_JSON || fromBodyType === BODY_TYPE_GRAPHQL) {
      const value: Record<string, unknown> = { query: body.query };
      if (typeof body.variables === "string" && body.variables.trim() !== "") {
        value.variables = parseJson(body.variables) ?? body.variables;
      }

      return { text: JSON.stringify(value, null, 2) };
    }

    return { text: body.query };
  }

  if (typeof body.filePath === "string") {
    return { text: body.filePath };
  }

  return { text: "" };
}

function formBodyToUrlEncodedText(form: unknown[]): string {
  const params = new URLSearchParams();

  for (const pair of form) {
    if (!isRecord(pair)) continue;
    if (pair.enabled === false) continue;
    if (typeof pair.name !== "string" || pair.name === "") continue;
    params.append(pair.name, stringifyFormValue(pair.value));
  }

  return params.toString();
}

function urlEncodedTextToFormPairs(text: string) {
  if (!text.includes("=") && !text.includes("&")) {
    return [];
  }

  return Array.from(new URLSearchParams(text).entries()).map(([name, value]) => ({
    enabled: true,
    name,
    value,
  }));
}

function formBodyToObject(form: unknown[]) {
  const result: Record<string, unknown> = {};

  for (const pair of form) {
    if (!isRecord(pair)) continue;
    if (pair.enabled === false) continue;
    if (typeof pair.name !== "string" || pair.name === "") continue;

    const value = stringifyFormValue(pair.value);
    if (pair.name in result) {
      const existing = result[pair.name];
      result[pair.name] = Array.isArray(existing) ? [...existing, value] : [existing, value];
    } else {
      result[pair.name] = value;
    }
  }

  return result;
}

function objectToFormPairs(value: Record<string, unknown>) {
  return Object.entries(value).flatMap(([name, rawValue]) => {
    if (Array.isArray(rawValue)) {
      return rawValue.map((item) => ({ enabled: true, name, value: stringifyFormValue(item) }));
    }

    return [{ enabled: true, name, value: stringifyFormValue(rawValue) }];
  });
}

function stringifyFormValue(value: unknown): string {
  if (value == null) return "";
  if (typeof value === "string") return value;
  return JSON.stringify(value);
}

function parseJsonObject(text: string): Record<string, unknown> | null {
  const parsed = parseJson(text);
  return isRecord(parsed) ? parsed : null;
}

function parseJson(text: string): unknown | null {
  try {
    return JSON.parse(text);
  } catch {
    return null;
  }
}

function isRecord(value: unknown): value is Record<string, unknown> {
  return value != null && typeof value === "object" && !Array.isArray(value);
}