yaak/apps/yaak-proxy/components/ProxyLayout.tsx

import { type } from '@tauri-apps/plugin-os';
import type { ProxyHeader } from '@yaakapp-internal/proxy-lib';
import {
  HeaderSize,
  IconButton,
  Table,
  TableBody,
  TableCell,
  TableHead,
  TableHeaderCell,
  TableRow,
  TruncatedWideTableCell,
} from '@yaakapp-internal/ui';
import classNames from 'classnames';
import { useAtomValue } from 'jotai';
import { useRpcQueryWithEvent } from '../hooks/useRpcQueryWithEvent';
import { ActionButton } from './ActionButton';
import { filteredExchangesAtom, Sidebar } from './Sidebar';

export function ProxyLayout() {
  const osType = type();
  const exchanges = useAtomValue(filteredExchangesAtom);
  const { data: proxyState } = useRpcQueryWithEvent('get_proxy_state', {}, 'proxy_state_changed');
  const isRunning = proxyState?.state === 'running';

  return (
    <div
      className={classNames(
        'h-full w-full grid grid-rows-[auto_1fr]',
        osType === 'linux' && 'border border-border-subtle',
      )}
    >
      <HeaderSize
        size="lg"
        osType={osType}
        hideWindowControls={false}
        useNativeTitlebar={false}
        interfaceScale={1}
        className="x-theme-appHeader bg-surface"
      >
        <div className="grid grid-cols-[minmax(0,1fr)_auto]">
          <div data-tauri-drag-region className="flex items-center text-sm px-2">
            Yaak Proxy
          </div>
          <div>
            <IconButton icon="alarm_clock" title="Yo" />
          </div>
        </div>
      </HeaderSize>
      <div className="grid grid-cols-[auto_1fr] min-h-0">
        <Sidebar />
        <main className="overflow-auto p-4">
          <div className="flex items-center gap-3 mb-4">
            <ActionButton
              action={{ scope: 'global', action: 'proxy_start' }}
              size="sm"
              tone="primary"
              disabled={isRunning}
            />
            <ActionButton
              action={{ scope: 'global', action: 'proxy_stop' }}
              size="sm"
              variant="border"
              disabled={!isRunning}
            />
            <span
              className={classNames(
                'text-xs font-medium',
                isRunning ? 'text-success' : 'text-text-subtlest',
              )}
            >
              {isRunning ? 'Running on :9090' : 'Stopped'}
            </span>
          </div>

          {exchanges.length === 0 ? (
            <p className="text-text-subtlest text-sm">No traffic yet</p>
          ) : (
            <Table scrollable>
              <TableHead>
                <TableRow>
                  <TableHeaderCell>Method</TableHeaderCell>
                  <TableHeaderCell>URL</TableHeaderCell>
                  <TableHeaderCell>Status</TableHeaderCell>
                  <TableHeaderCell>Type</TableHeaderCell>
                </TableRow>
              </TableHead>
              <TableBody>
                {exchanges.map((ex) => (
                  <TableRow key={ex.id}>
                    <TableCell className="font-mono text-2xs">{ex.method}</TableCell>
                    <TruncatedWideTableCell className="font-mono text-2xs">
                      {ex.url}
                    </TruncatedWideTableCell>
                    <TableCell>
                      <StatusBadge status={ex.resStatus} error={ex.error} />
                    </TableCell>
                    <TableCell className="text-text-subtle text-xs">
                      {getContentType(ex.resHeaders)}
                    </TableCell>
                  </TableRow>
                ))}
              </TableBody>
            </Table>
          )}
        </main>
      </div>
    </div>
  );
}

function StatusBadge({ status, error }: { status: number | null; error: string | null }) {
  if (error) return <span className="text-xs text-danger">Error</span>;
  if (status == null) return <span className="text-xs text-text-subtlest">—</span>;

  const color =
    status >= 500
      ? 'text-danger'
      : status >= 400
        ? 'text-warning'
        : status >= 300
          ? 'text-notice'
          : 'text-success';

  return <span className={classNames('text-xs font-mono', color)}>{status}</span>;
}

function getContentType(headers: ProxyHeader[]): string {
  const ct = headers.find((h) => h.name.toLowerCase() === 'content-type')?.value;
  if (ct == null) return '—';
  // Strip parameters (e.g. "; charset=utf-8")
  return ct.split(';')[0]?.trim() ?? ct;
}