aiops/frontend/src/components/deploy/WorkspaceTermDialog.tsx

import { useRef, useEffect, useState } from 'react';
import {
  Dialog,
  DialogContent,
  DialogDescription,
  DialogTitle,
} from '@/components/ui/dialog';
import { Terminal as XTerm } from 'xterm';
import { FitAddon } from 'xterm-addon-fit';
import 'xterm/css/xterm.css';
import { Maximize2, Minimize2, X, FolderOpen } from 'lucide-react';
import { cn } from '@/lib/utils';
import { useUserStore } from '@/store';
import { pipelineApi } from '@/api';
import type { JsonData } from '@/types/common';

interface WorkspaceTermDialogProps {
  open: boolean;
  onOpenChange: (open: boolean) => void;
  runId: number | null;
  buildNumber: number;
}

/**
 * Workspace 终端 Dialog — 进入流水线构建目录查看文件
 */
export function WorkspaceTermDialog({
  open,
  onOpenChange,
  runId,
  buildNumber,
}: WorkspaceTermDialogProps) {
  const containerRef = useRef<HTMLDivElement>(null);
  const termRef = useRef<XTerm | null>(null);
  const wsRef = useRef<WebSocket | null>(null);
  const fitRef = useRef<FitAddon | null>(null);
  const [connected, setConnected] = useState(false);
  const [fullscreen, setFullscreen] = useState(false);
  const [workspace, setWorkspace] = useState('');

  // 获取 workspace 路径
  useEffect(() => {
    if (!open || !runId) return;
    pipelineApi.getWorkspace(runId)
      .then((data: JsonData) => setWorkspace(data?.workspace_path || ''))
      .catch(() => {});
  }, [open, runId]);

  // 连接终端
  useEffect(() => {
    if (!open || !runId) return;

    const term = new XTerm({
      cursorBlink: true,
      fontSize: 13,
      fontFamily: 'Menlo, Monaco, "Courier New", monospace',
      allowProposedApi: true,
      theme: {
        background: '#0a0e14',
        foreground: '#d6d6d6',
        cursor: '#d6d6d6',
        selectionBackground: 'rgba(255,255,255,0.2)',
      },
    });
    const fit = new FitAddon();
    term.loadAddon(fit);
    termRef.current = term;
    fitRef.current = fit;

    const proto = location.protocol === 'https:' ? 'wss' : 'ws';
    const token = useUserStore.getState().token || '';
    const wsUrl = `${proto}://${location.host}/ws/pipeline/${runId}/workspace-shell?token=${encodeURIComponent(token)}`;
    const ws = new WebSocket(wsUrl);
    wsRef.current = ws;

    term.onData((data) => {
      if (ws.readyState === WebSocket.OPEN) {
        ws.send(JSON.stringify({ status: 0, data }));
      }
    });

    ws.onmessage = (event) => {
      try {
        const msg = JSON.parse(event.data);
        if (msg.status === 0 && typeof msg.data === 'string') {
          term.write(msg.data);
        }
      } catch {
        term.write(typeof event.data === 'string' ? event.data : '');
      }
    };

    ws.onopen = () => {
      setConnected(true);
      if (containerRef.current) {
        term.open(containerRef.current);
        requestAnimationFrame(() => {
          try { fit.fit(); term.focus(); } catch { /* noop */ }
        });
      }
    };

    ws.onclose = () => {
      setConnected(false);
      term.write('\r\n\x1b[31m[连接已关闭]\x1b[0m\r\n');
    };

    ws.onerror = () => {
      term.write('\r\n\x1b[31m[连接发生错误]\x1b[0m\r\n');
    };

    return () => {
      try { ws.close(); term.dispose(); } catch { /* noop */ }
      termRef.current = null;
      wsRef.current = null;
      setConnected(false);
    };
  }, [open, runId]);

  // ResizeObserver for fit
  useEffect(() => {
    if (!open) return;
    const container = containerRef.current;
    if (!container) return;

    let timer: ReturnType<typeof setTimeout>;
    const ro = new ResizeObserver(() => {
      clearTimeout(timer);
      timer = setTimeout(() => {
        if (fitRef.current && termRef.current) {
          try {
            fitRef.current.fit();
            termRef.current.focus();
            if (wsRef.current?.readyState === WebSocket.OPEN) {
              const { cols, rows } = termRef.current;
              wsRef.current.send(JSON.stringify({ status: 1, cols, rows }));
            }
          } catch { /* noop */ }
        }
      }, 60);
    });
    ro.observe(container);
    return () => { clearTimeout(timer); ro.disconnect(); };
  }, [open]);

  return (
    <Dialog open={open} onOpenChange={onOpenChange}>
      <DialogContent
        showCloseButton={false}
        // 禁用 Radix Dialog 默认的 ESC 关闭：工作区终端会用到 ESC，关闭只能点头部 X 按钮 // Logan.Li
        onEscapeKeyDown={(e) => e.preventDefault()}
        className={cn(
          'bg-[var(--dg-bg-secondary)] border-[var(--dg-border)] p-0 gap-0',
          '!duration-0 !transition-none',
          fullscreen
            ? '!max-w-none !w-screen !h-screen !rounded-none'
            : ''
        )}
        style={fullscreen ? { maxWidth: '100vw', width: '100vw', height: '100vh' } : { maxWidth: '900px', width: '90vw' }}
      >
        <DialogTitle className="sr-only">Workspace 终端 #{buildNumber}</DialogTitle>
        <DialogDescription className="sr-only">流水线构建目录终端</DialogDescription>
        <div className="flex items-center justify-between px-4 py-2 border-b border-[var(--dg-border)]">
          <div className="flex items-center gap-2 text-sm min-w-0">
            <div className={cn('w-2 h-2 rounded-full shrink-0', connected ? 'bg-emerald-400' : 'bg-red-400')} />
            <FolderOpen className="h-4 w-4 text-[var(--dg-text-secondary)] shrink-0" />
            <span className="text-[var(--dg-text-primary)] font-medium truncate">
              #{buildNumber} 构建目录
            </span>
            {workspace && (
              <span className="text-[var(--dg-text-muted)] text-xs truncate hidden sm:inline">
                {workspace}
              </span>
            )}
          </div>
          <div className="flex items-center gap-1">
            <button
              onClick={() => setFullscreen(!fullscreen)}
              className="p-1.5 rounded hover:bg-[var(--dg-bg-hover)] text-[var(--dg-text-secondary)]"
            >
              {fullscreen ? <Minimize2 className="h-4 w-4" /> : <Maximize2 className="h-4 w-4" />}
            </button>
            <button
              onClick={() => onOpenChange(false)}
              className="p-1.5 rounded hover:bg-[var(--dg-bg-hover)] text-[var(--dg-text-secondary)]"
            >
              <X className="h-4 w-4" />
            </button>
          </div>
        </div>
        <div
          ref={containerRef}
          className="bg-[#0a0e14]"
          style={{ height: fullscreen ? 'calc(100vh - 49px)' : '480px' }}
        />
      </DialogContent>
    </Dialog>
  );
}