import type {
  Folder,
  GrpcRequest,
  HttpRequest,
  HttpRequestHeader,
  WebsocketRequest,
  Workspace,
} from '@yaakapp-internal/models';
import { foldersAtom, workspacesAtom } from '@yaakapp-internal/models';
import { atom, useAtomValue } from 'jotai';
import { defaultHeaders } from '../lib/defaultHeaders';

const ancestorsAtom = atom((get) => [...get(foldersAtom), ...get(workspacesAtom)]);

export type HeaderModel = HttpRequest | GrpcRequest | WebsocketRequest | Folder | Workspace;

export function useInheritedHeaders(baseModel: HeaderModel | null) {
  const parents = useAtomValue(ancestorsAtom);

  if (baseModel == null) return [];
  if (baseModel.model === 'workspace') return defaultHeaders;

  const next = (child: HeaderModel): HttpRequestHeader[] => {
    // Short-circuit at workspace level - return global defaults + workspace headers
    if (child.model === 'workspace') {
      return [...defaultHeaders, ...child.headers];
    }

    // Recurse up the tree
    const parent = parents.find((p) => {
      if (child.folderId) return p.id === child.folderId;
      return p.id === child.workspaceId;
    });

    // Failed to find parent (should never happen)
    if (parent == null) {
      return [];
    }

    const headers = next(parent);
    return [...headers, ...parent.headers];
  };

  const allHeaders = next(baseModel);

  // Deduplicate by header name (case-insensitive), keeping the latest (most specific) value
  const headersByName = new Map<string, HttpRequestHeader>();
  for (const header of allHeaders) {
    headersByName.set(header.name.toLowerCase(), header);
  }

  return Array.from(headersByName.values());
}