Update script-terminal-modal.tsx

This commit is contained in:
MacRimi
2025-12-06 22:26:42 +01:00
parent f90f6f364a
commit ea58b70435

View File

@@ -29,13 +29,6 @@ interface ScriptTerminalModalProps {
description: string description: string
} }
interface TerminalInstance {
term: any | null
ws: WebSocket | null
fitAddon: any | null
isConnected: boolean
}
export function ScriptTerminalModal({ export function ScriptTerminalModal({
open, open,
onClose, onClose,
@@ -45,14 +38,12 @@ export function ScriptTerminalModal({
title, title,
description, description,
}: ScriptTerminalModalProps) { }: ScriptTerminalModalProps) {
const [terminal, setTerminal] = useState<TerminalInstance>({ const termRef = useRef<any>(null)
term: null, const wsRef = useRef<WebSocket | null>(null)
ws: null, const fitAddonRef = useRef<any>(null)
fitAddon: null,
isConnected: false,
})
const sessionIdRef = useRef<string>(Math.random().toString(36).substring(2, 8)) const sessionIdRef = useRef<string>(Math.random().toString(36).substring(2, 8))
const [isConnected, setIsConnected] = useState(false)
const [isComplete, setIsComplete] = useState(false) const [isComplete, setIsComplete] = useState(false)
const [exitCode, setExitCode] = useState<number | null>(null) const [exitCode, setExitCode] = useState<number | null>(null)
const [currentInteraction, setCurrentInteraction] = useState<WebInteraction | null>(null) const [currentInteraction, setCurrentInteraction] = useState<WebInteraction | null>(null)
@@ -78,25 +69,27 @@ export function ScriptTerminalModal({
if (waitingTimeoutRef.current) { if (waitingTimeoutRef.current) {
clearTimeout(waitingTimeoutRef.current) clearTimeout(waitingTimeoutRef.current)
} }
if (terminal.ws) { if (wsRef.current) {
terminal.ws.close() wsRef.current.close()
wsRef.current = null
} }
if (terminal.term) { if (termRef.current) {
terminal.term.dispose() termRef.current.dispose()
termRef.current = null
} }
setTerminal({ term: null, ws: null, fitAddon: null, isConnected: false })
sessionIdRef.current = Math.random().toString(36).substring(2, 8) sessionIdRef.current = Math.random().toString(36).substring(2, 8)
setIsComplete(false) setIsComplete(false)
setExitCode(null) setExitCode(null)
setInteractionInput("") setInteractionInput("")
setCurrentInteraction(null) setCurrentInteraction(null)
setIsWaitingNextInteraction(false) setIsWaitingNextInteraction(false)
setIsConnected(false)
} }
}, [open]) }, [open])
useEffect(() => { useEffect(() => {
const container = terminalContainerRef.current const container = terminalContainerRef.current
if (!open || !container || terminal.term) { if (!open || !container || termRef.current) {
return return
} }
@@ -148,6 +141,9 @@ export function ScriptTerminalModal({
term.loadAddon(fitAddon) term.loadAddon(fitAddon)
term.open(container) term.open(container)
termRef.current = term
fitAddonRef.current = fitAddon
setTimeout(() => { setTimeout(() => {
try { try {
fitAddon.fit() fitAddon.fit()
@@ -158,9 +154,10 @@ export function ScriptTerminalModal({
const wsUrl = getScriptWebSocketUrl(sessionIdRef.current) const wsUrl = getScriptWebSocketUrl(sessionIdRef.current)
const ws = new WebSocket(wsUrl) const ws = new WebSocket(wsUrl)
wsRef.current = ws
ws.onopen = () => { ws.onopen = () => {
setTerminal((prev) => ({ ...prev, isConnected: true, term, ws, fitAddon })) setIsConnected(true)
const initMessage = { const initMessage = {
script_path: scriptPath, script_path: scriptPath,
@@ -207,18 +204,18 @@ export function ScriptTerminalModal({
options: msg.interaction.options, options: msg.interaction.options,
default: msg.interaction.default, default: msg.interaction.default,
}) })
return // Don't write JSON to terminal return
} }
if (msg.type === "error") { if (msg.type === "error") {
terminal.term.writeln(`\x1b[31m${msg.message}\x1b[0m`) term.writeln(`\x1b[31m${msg.message}\x1b[0m`)
return return
} }
} catch { } catch {
// Not JSON, it's regular terminal output // Not JSON, es output normal de terminal
} }
terminal.term.write(event.data) term.write(event.data)
setIsWaitingNextInteraction(false) setIsWaitingNextInteraction(false)
if (waitingTimeoutRef.current) { if (waitingTimeoutRef.current) {
@@ -227,13 +224,13 @@ export function ScriptTerminalModal({
} }
ws.onerror = (error) => { ws.onerror = (error) => {
setTerminal((prev) => ({ ...prev, isConnected: false })) setIsConnected(false)
terminal.term.writeln("\x1b[31mWebSocket error occurred\x1b[0m") term.writeln("\x1b[31mWebSocket error occurred\x1b[0m")
} }
ws.onclose = (event) => { ws.onclose = (event) => {
setTerminal((prev) => ({ ...prev, isConnected: false })) setIsConnected(false)
terminal.term.writeln("\x1b[33mConnection closed\x1b[0m") term.writeln("\x1b[33mConnection closed\x1b[0m")
if (!isComplete) { if (!isComplete) {
setIsComplete(true) setIsComplete(true)
@@ -249,7 +246,7 @@ export function ScriptTerminalModal({
checkConnectionInterval.current = setInterval(() => { checkConnectionInterval.current = setInterval(() => {
if (ws) { if (ws) {
setTerminal((prev) => ({ ...prev, isConnected: ws.readyState === WebSocket.OPEN })) setIsConnected(ws.readyState === WebSocket.OPEN)
} }
}, 500) }, 500)
@@ -279,7 +276,7 @@ export function ScriptTerminalModal({
} }
initializeTerminal() initializeTerminal()
}, [open, terminal.term]) }, [open])
const getScriptWebSocketUrl = (sid: string): string => { const getScriptWebSocketUrl = (sid: string): string => {
if (typeof window === "undefined") { if (typeof window === "undefined") {
@@ -292,7 +289,7 @@ export function ScriptTerminalModal({
} }
const handleInteractionResponse = (value: string) => { const handleInteractionResponse = (value: string) => {
if (!terminal.ws || !currentInteraction) { if (!wsRef.current || !currentInteraction) {
return return
} }
@@ -309,8 +306,8 @@ export function ScriptTerminalModal({
value: value, value: value,
}) })
if (terminal.ws.readyState === WebSocket.OPEN) { if (wsRef.current.readyState === WebSocket.OPEN) {
terminal.ws.send(response) wsRef.current.send(response)
} }
setCurrentInteraction(null) setCurrentInteraction(null)
@@ -322,14 +319,14 @@ export function ScriptTerminalModal({
} }
const handleCloseModal = () => { const handleCloseModal = () => {
if (terminal.ws && terminal.ws.readyState === WebSocket.OPEN) { if (wsRef.current && wsRef.current.readyState === WebSocket.OPEN) {
terminal.ws.close() wsRef.current.close()
} }
if (checkConnectionInterval.current) { if (checkConnectionInterval.current) {
clearInterval(checkConnectionInterval.current) clearInterval(checkConnectionInterval.current)
} }
if (terminal.term) { if (termRef.current) {
terminal.term.dispose() termRef.current.dispose()
} }
onClose() onClose()
} }
@@ -413,10 +410,10 @@ export function ScriptTerminalModal({
<div className="flex items-center gap-3"> <div className="flex items-center gap-3">
<Activity className="h-5 w-5 text-blue-500" /> <Activity className="h-5 w-5 text-blue-500" />
<div <div
className={`w-2 h-2 rounded-full ${terminal.isConnected ? "bg-green-500" : "bg-red-500"}`} className={`w-2 h-2 rounded-full ${isConnected ? "bg-green-500" : "bg-red-500"}`}
title={terminal.isConnected ? "Connected" : "Disconnected"} title={isConnected ? "Connected" : "Disconnected"}
></div> ></div>
<span className="text-xs text-muted-foreground">{terminal.isConnected ? "Online" : "Offline"}</span> <span className="text-xs text-muted-foreground">{isConnected ? "Online" : "Offline"}</span>
</div> </div>
<Button <Button