yaak/apps/yaak-client/lib/model_util.ts

import type {
  AnyModel,
  Cookie,
  Environment,
  HttpResponseEvent,
  HttpResponseHeader,
} from "@yaakapp-internal/models";
import { getMimeTypeFromContentType } from "./contentType";

export const BODY_TYPE_NONE = null;
export const BODY_TYPE_GRAPHQL = "graphql";
export const BODY_TYPE_JSON = "application/json";
export const BODY_TYPE_BINARY = "binary";
export const BODY_TYPE_OTHER = "other";
export const BODY_TYPE_FORM_URLENCODED = "application/x-www-form-urlencoded";
export const BODY_TYPE_FORM_MULTIPART = "multipart/form-data";
export const BODY_TYPE_XML = "text/xml";

export function cookieDomain(cookie: Cookie): string {
  if (cookie.domain === "NotPresent" || cookie.domain === "Empty") {
    return "n/a";
  }
  if ("HostOnly" in cookie.domain) {
    return cookie.domain.HostOnly;
  }
  if ("Suffix" in cookie.domain) {
    return cookie.domain.Suffix;
  }
  return "unknown";
}

export function modelsEq(a: AnyModel, b: AnyModel) {
  if (a.model !== b.model) {
    return false;
  }
  if (a.model === "key_value" && b.model === "key_value") {
    return a.key === b.key && a.namespace === b.namespace;
  }
  if ("id" in a && "id" in b) {
    return a.id === b.id;
  }
  return false;
}

export function getContentTypeFromHeaders(headers: HttpResponseHeader[] | null): string | null {
  return headers?.find((h) => h.name.toLowerCase() === "content-type")?.value ?? null;
}

export function getCharsetFromContentType(headers: HttpResponseHeader[]): string | null {
  const contentType = getContentTypeFromHeaders(headers);
  if (contentType == null) return null;

  const mimeType = getMimeTypeFromContentType(contentType);
  return mimeType.parameters.get("charset") ?? null;
}

export function isBaseEnvironment(environment: Environment): boolean {
  return environment.parentModel === "workspace";
}

export function isSubEnvironment(environment: Environment): boolean {
  return environment.parentModel === "environment";
}

export function isFolderEnvironment(environment: Environment): boolean {
  return environment.parentModel === "folder";
}

export function getCookieCounts(events: HttpResponseEvent[] | undefined): {
  sent: number;
  received: number;
} {
  if (!events) return { sent: 0, received: 0 };

  // Use Sets to deduplicate by cookie name
  const sentNames = new Set<string>();
  const receivedNames = new Set<string>();

  for (const event of events) {
    const e = event.event;
    if (e.type === "header_up" && e.name.toLowerCase() === "cookie") {
      // Parse "Cookie: name=value; name2=value2" format
      for (const pair of e.value.split(";")) {
        const name = pair.split("=")[0]?.trim();
        if (name) sentNames.add(name);
      }
    } else if (e.type === "header_down" && e.name.toLowerCase() === "set-cookie") {
      // Parse "Set-Cookie: name=value; ..." - first part before ; is name=value
      const name = e.value.split(";")[0]?.split("=")[0]?.trim();
      if (name) receivedNames.add(name);
    }
  }

  return { sent: sentNames.size, received: receivedNames.size };
}