[Init] Initial commit - NetMesh terminal manager
Some checks failed
build-packages / resolve bundled mosh-client (push) Has been cancelled
build-packages / resolve bundled et-client (push) Has been cancelled
build-packages / build-macos (push) Has been cancelled
build-packages / build-windows (push) Has been cancelled
build-packages / build-linux-x64 (push) Has been cancelled
build-packages / build-linux-arm64 (push) Has been cancelled
build-packages / release (push) Has been cancelled
build-packages / update Nix release metadata (push) Has been cancelled
build-packages / bump homebrew tap (push) Has been cancelled
test / lint-and-test (push) Has been cancelled
AI automation / Route event (push) Has been cancelled
AI automation / Hand reopened issue to maintainers (push) Has been cancelled
AI automation / Clean source issue state (push) Has been cancelled
AI automation / Reconcile handoffs (push) Has been cancelled
AI automation / Classify issue (push) Has been cancelled
AI automation / Claude Code smoke (push) Has been cancelled
AI automation / Review issue follow-up (push) Has been cancelled
AI automation / Publish issue follow-up (push) Has been cancelled
AI automation / Implement with Claude Code (push) Has been cancelled
AI automation / Publish implement PR (push) Has been cancelled
AI automation / Continue queued issue comments (push) Has been cancelled
AI automation / Codex review loop (push) Has been cancelled
AI automation / Publish Codex fix (push) Has been cancelled
AI automation / Clear Codex dispatch marker (push) Has been cancelled
AI automation / Own PR re-request Codex (push) Has been cancelled
AI automation / External PR re-request Codex (push) Has been cancelled
AI automation / Poll Codex reaction / retry (push) Has been cancelled
build-et-binaries / build-linux-x64 (push) Has been cancelled
build-et-binaries / build-linux-arm64 (push) Has been cancelled
build-et-binaries / build-macos-universal (push) Has been cancelled
build-et-binaries / build-windows-x64 (push) Has been cancelled
build-et-binaries / release (push) Has been cancelled
Some checks failed
build-packages / resolve bundled mosh-client (push) Has been cancelled
build-packages / resolve bundled et-client (push) Has been cancelled
build-packages / build-macos (push) Has been cancelled
build-packages / build-windows (push) Has been cancelled
build-packages / build-linux-x64 (push) Has been cancelled
build-packages / build-linux-arm64 (push) Has been cancelled
build-packages / release (push) Has been cancelled
build-packages / update Nix release metadata (push) Has been cancelled
build-packages / bump homebrew tap (push) Has been cancelled
test / lint-and-test (push) Has been cancelled
AI automation / Route event (push) Has been cancelled
AI automation / Hand reopened issue to maintainers (push) Has been cancelled
AI automation / Clean source issue state (push) Has been cancelled
AI automation / Reconcile handoffs (push) Has been cancelled
AI automation / Classify issue (push) Has been cancelled
AI automation / Claude Code smoke (push) Has been cancelled
AI automation / Review issue follow-up (push) Has been cancelled
AI automation / Publish issue follow-up (push) Has been cancelled
AI automation / Implement with Claude Code (push) Has been cancelled
AI automation / Publish implement PR (push) Has been cancelled
AI automation / Continue queued issue comments (push) Has been cancelled
AI automation / Codex review loop (push) Has been cancelled
AI automation / Publish Codex fix (push) Has been cancelled
AI automation / Clear Codex dispatch marker (push) Has been cancelled
AI automation / Own PR re-request Codex (push) Has been cancelled
AI automation / External PR re-request Codex (push) Has been cancelled
AI automation / Poll Codex reaction / retry (push) Has been cancelled
build-et-binaries / build-linux-x64 (push) Has been cancelled
build-et-binaries / build-linux-arm64 (push) Has been cancelled
build-et-binaries / build-macos-universal (push) Has been cancelled
build-et-binaries / build-windows-x64 (push) Has been cancelled
build-et-binaries / release (push) Has been cancelled
This commit is contained in:
579
application/app/useAppStartupEffects.ts
Normal file
579
application/app/useAppStartupEffects.ts
Normal file
@@ -0,0 +1,579 @@
|
||||
/* eslint-disable @typescript-eslint/no-explicit-any */
|
||||
import { useEffect, useRef } from 'react';
|
||||
import { usePortForwardingAutoStart } from '../state/usePortForwardingAutoStart';
|
||||
import { editorTabStore } from '../state/editorTabStore';
|
||||
import { netcattyBridge } from '../../infrastructure/services/netcattyBridge';
|
||||
import { localStorageAdapter } from '../../infrastructure/persistence/localStorageAdapter';
|
||||
import { toast } from '../../components/ui/toast';
|
||||
import { sftpTransferCenterStore } from '../state/sftpTransferCenterStore';
|
||||
import { resumeTransferWithDedicatedSession } from '../state/sftp/dedicatedTransferResume';
|
||||
import { getSftpTransferResourceKeys, globalSftpTransferScheduler } from '../state/sftp/globalTransferScheduler';
|
||||
import { hasNewSourceFingerprint } from '../state/sftp/transferProgressMetadata';
|
||||
import { STORAGE_KEY_SFTP_TRANSFER_CONCURRENCY } from '../../infrastructure/config/storageKeys';
|
||||
import type { TransferTask } from '../../domain/models';
|
||||
import { isTerminalBootEpochCurrent } from '../../domain/terminalBootEpoch';
|
||||
import {
|
||||
canApplyDedicatedResumeProgress,
|
||||
createDedicatedResumeChildUpdateBatcher,
|
||||
createDedicatedResumeProgressBatcher,
|
||||
} from './dedicatedResumeProgress';
|
||||
|
||||
type StartupEffectsContext = Record<string, any>;
|
||||
|
||||
type KeyboardInteractiveScope = "terminal" | "external";
|
||||
type KeyboardInteractiveRequestLike = {
|
||||
scope?: KeyboardInteractiveScope;
|
||||
sessionId?: string;
|
||||
hostId?: string;
|
||||
requestId?: string;
|
||||
bootEpoch?: number;
|
||||
};
|
||||
type SessionIdLike = { id: string; hostId?: string; hostname?: string; status?: string };
|
||||
type KeyboardInteractiveQueueItem = { requestId: string };
|
||||
|
||||
export function shouldQueueKeyboardInteractiveRequest(
|
||||
request: KeyboardInteractiveRequestLike,
|
||||
sessions: SessionIdLike[],
|
||||
): boolean {
|
||||
if (request.scope !== "terminal") return true;
|
||||
if (!request.sessionId) return false;
|
||||
const session = sessions.find((entry) => entry.id === request.sessionId);
|
||||
if (!session) return false;
|
||||
// Status-bar disconnect keeps the tab; do not queue MFA for aborted panes.
|
||||
if (session.status === "disconnected") return false;
|
||||
// After disconnect → reconnect the tab is connecting again; reject MFA from
|
||||
// a superseded SSH start that still shares this sessionId.
|
||||
if (!isTerminalBootEpochCurrent(request.sessionId, request.bootEpoch)) return false;
|
||||
return true;
|
||||
}
|
||||
|
||||
export function removeKeyboardInteractiveRequest<T extends KeyboardInteractiveQueueItem>(
|
||||
queue: T[],
|
||||
requestId: string,
|
||||
): T[] {
|
||||
return queue.filter(request => request.requestId !== requestId);
|
||||
}
|
||||
|
||||
export function useAppStartupEffects(ctx: StartupEffectsContext) {
|
||||
const {dismissUpdate, enabled = true, groupConfigs, hosts, resumeHosts, identities,
|
||||
hasRuntimeTunnel, installUpdate, isVaultInitialized, keys, knownHosts, openSettingsWindow, portForwardingRules, proxyProfiles, sessions, setKeyboardInteractiveQueue,
|
||||
t, terminalSettings, updateState, workspaces,
|
||||
} = ctx;
|
||||
// Vault hosts for tray/menu; resumeHosts may include ephemeral quick-connect rows.
|
||||
const dedicatedResumeHosts = resumeHosts ?? hosts;
|
||||
const sessionsRef = useRef(sessions);
|
||||
|
||||
useEffect(() => {
|
||||
sessionsRef.current = sessions;
|
||||
}, [sessions]);
|
||||
|
||||
// After app restart (or soft-resume miss), unfinished transfers reconnect via
|
||||
// a dedicated SFTP session. Prefer resumeHosts (vault + ephemeral) so
|
||||
// quick-connect transfers can re-auth without "Cannot find host in your vault".
|
||||
useEffect(() => {
|
||||
if (!enabled || !isVaultInitialized) {
|
||||
sftpTransferCenterStore.setDedicatedResumeHandler(null);
|
||||
return;
|
||||
}
|
||||
sftpTransferCenterStore.setDedicatedResumeHandler(async (task) => {
|
||||
// Keep reconnectRequired true until the first progress/completion so the
|
||||
// play control stays a spinner during auth + session setup.
|
||||
sftpTransferCenterStore.patchTask(task.id, {
|
||||
status: "pending",
|
||||
error: undefined,
|
||||
reconnectRequired: true,
|
||||
speed: 0,
|
||||
phase: undefined,
|
||||
});
|
||||
const children = sftpTransferCenterStore.getSnapshot().tasks.filter(
|
||||
(row) => row.parentTaskId === task.id,
|
||||
);
|
||||
// rAF-coalesce progress so dedicated resume does not flood the global center.
|
||||
type ProgressSample = {
|
||||
transferred: number;
|
||||
total: number;
|
||||
speed: number;
|
||||
checkpointBytes?: number;
|
||||
resumeStage?: TransferTask["resumeStage"];
|
||||
downloadCheckpointBytes?: number;
|
||||
uploadCheckpointBytes?: number;
|
||||
sourceFingerprint?: string;
|
||||
};
|
||||
// One rAF coalesce only — main process already time-throttles IPC.
|
||||
// A second 500ms timer here made dedicated-resume bars jump.
|
||||
const applyProgress = (progress: ProgressSample) => {
|
||||
const current = sftpTransferCenterStore.getSnapshot().tasks.find((row) => row.id === task.id);
|
||||
if (!current || current.status === "cancelled") return;
|
||||
if (current.status === "pausing" || current.status === "paused") {
|
||||
if (hasNewSourceFingerprint(current.sourceFingerprint, progress.sourceFingerprint)) {
|
||||
sftpTransferCenterStore.patchTask(task.id, { sourceFingerprint: progress.sourceFingerprint });
|
||||
}
|
||||
return;
|
||||
}
|
||||
// The final sample can still be queued in requestAnimationFrame after
|
||||
// the resume promise settles. Never let it turn a completed/failed row
|
||||
// back into a permanently "transferring" task.
|
||||
if (!canApplyDedicatedResumeProgress(current.status)) return;
|
||||
// Directory parents use file-count progress; single files use bytes.
|
||||
// Prefer durable contiguous checkpoint when the bridge supplies it.
|
||||
const durableCheckpoint = task.isDirectory
|
||||
? progress.transferred
|
||||
: (progress.checkpointBytes ?? progress.transferred);
|
||||
// Keep progress monotonic so a late force-checkpoint paint cannot hide
|
||||
// later bytes, and the bar never freezes at the pre-quit offset.
|
||||
const nextTransferred = Math.max(current.transferredBytes ?? 0, progress.transferred);
|
||||
const nextCheckpoint = task.isDirectory
|
||||
? Math.max(current.checkpointBytes ?? 0, progress.transferred)
|
||||
: Math.max(current.checkpointBytes ?? 0, durableCheckpoint);
|
||||
sftpTransferCenterStore.patchTask(task.id, {
|
||||
status: "transferring",
|
||||
transferredBytes: nextTransferred,
|
||||
...(progress.total > 0 ? { totalBytes: progress.total } : {}),
|
||||
speed: progress.speed,
|
||||
...(task.isDirectory
|
||||
? { checkpointBytes: nextCheckpoint, progressMode: "files" as const }
|
||||
: {
|
||||
checkpointBytes: nextCheckpoint,
|
||||
resumeStage: progress.resumeStage,
|
||||
downloadCheckpointBytes: progress.downloadCheckpointBytes,
|
||||
uploadCheckpointBytes: progress.uploadCheckpointBytes,
|
||||
sourceFingerprint: progress.sourceFingerprint,
|
||||
}),
|
||||
reconnectRequired: false,
|
||||
error: undefined,
|
||||
phase: "transferring",
|
||||
ownerId: "dedicated-resume",
|
||||
});
|
||||
};
|
||||
const progressBatcher = createDedicatedResumeProgressBatcher<ProgressSample>({
|
||||
requestFrame: (callback) => window.requestAnimationFrame(callback),
|
||||
cancelFrame: (handle) => window.cancelAnimationFrame(handle),
|
||||
canApply: () => {
|
||||
const current = sftpTransferCenterStore.getSnapshot().tasks.find((row) => row.id === task.id);
|
||||
return !!current && canApplyDedicatedResumeProgress(current.status);
|
||||
},
|
||||
apply: applyProgress,
|
||||
});
|
||||
const childUpdateBatcher = createDedicatedResumeChildUpdateBatcher({
|
||||
// Use the restart snapshot, not repeated linear store lookups. Completed
|
||||
// rows disappear as batches compact, but later updates for those ids can
|
||||
// still stay in the same bounded batching path safely.
|
||||
getTaskCount: () => children.length + 1,
|
||||
hasTask: (() => {
|
||||
const retainedChildIds = new Set(children.map((child) => child.id));
|
||||
return (taskId: string) => retainedChildIds.has(taskId);
|
||||
})(),
|
||||
upsertTasks: (updates) => sftpTransferCenterStore.upsertTasks(updates),
|
||||
});
|
||||
let acceptsResumeCallbacks = true;
|
||||
try {
|
||||
return await resumeTransferWithDedicatedSession(
|
||||
task,
|
||||
{
|
||||
hosts: dedicatedResumeHosts,
|
||||
keys,
|
||||
identities,
|
||||
knownHosts,
|
||||
terminalSettings,
|
||||
},
|
||||
(progress) => {
|
||||
if (acceptsResumeCallbacks) progressBatcher.push(progress);
|
||||
},
|
||||
{
|
||||
children,
|
||||
onChildUpdate: (child) => {
|
||||
if (acceptsResumeCallbacks) {
|
||||
childUpdateBatcher.push({ ...child, ownerId: "dedicated-resume" });
|
||||
}
|
||||
},
|
||||
onDirectoryCheckpointUpdate: (checkpoint) => {
|
||||
if (acceptsResumeCallbacks) {
|
||||
sftpTransferCenterStore.patchTask(task.id, {
|
||||
directoryResumeCheckpoint: checkpoint,
|
||||
});
|
||||
}
|
||||
},
|
||||
shouldAbort: () => {
|
||||
const current = sftpTransferCenterStore.getSnapshot().tasks.find((row) => row.id === task.id);
|
||||
// interrupted is the pre-reconnect persisted state — do not abort a
|
||||
// live dedicated walk just because children/parent still show it.
|
||||
return !current
|
||||
|| current.status === "cancelled"
|
||||
|| current.status === "paused";
|
||||
},
|
||||
},
|
||||
);
|
||||
} finally {
|
||||
acceptsResumeCallbacks = false;
|
||||
progressBatcher.finish();
|
||||
childUpdateBatcher.flush();
|
||||
}
|
||||
});
|
||||
return () => sftpTransferCenterStore.setDedicatedResumeHandler(null);
|
||||
}, [dedicatedResumeHosts, enabled, identities, isVaultInitialized, keys, knownHosts, terminalSettings]);
|
||||
|
||||
// Show toast notification when update is available (only when auto-download is idle)
|
||||
const toastedUpdateVersionRef = useRef<string | null>(null);
|
||||
useEffect(() => {
|
||||
if (!enabled) return;
|
||||
// Skip "update available" toast if auto-download has already started or completed
|
||||
if (updateState.autoDownloadStatus !== 'idle') return;
|
||||
// Don't show automatic notification when auto-update is disabled
|
||||
if (localStorageAdapter.readString('netcatty_auto_update_enabled_v1') === 'false') return;
|
||||
if (updateState.hasUpdate && updateState.latestRelease) {
|
||||
const version = updateState.latestRelease.version;
|
||||
if (toastedUpdateVersionRef.current === version) return;
|
||||
toastedUpdateVersionRef.current = version;
|
||||
toast.info(
|
||||
t('update.available.message', { version }),
|
||||
{
|
||||
title: t('update.available.title'),
|
||||
duration: 8000, // Show longer for update notifications
|
||||
onClick: () => {
|
||||
void openSettingsWindow();
|
||||
// Dismiss the update so the toast doesn't re-fire on every render.
|
||||
// On unsupported platforms (where autoDownloadStatus stays 'idle')
|
||||
// this is the only way to suppress the notification for this version.
|
||||
// On supported platforms this toast only shows before auto-download
|
||||
// starts, and the Settings window's own useUpdateCheck will pick up
|
||||
// the download state via IPC events independently of the dismiss.
|
||||
dismissUpdate();
|
||||
},
|
||||
actionLabel: t('update.viewInSettings'),
|
||||
}
|
||||
);
|
||||
}
|
||||
}, [enabled, updateState.hasUpdate, updateState.latestRelease, updateState.autoDownloadStatus, t, openSettingsWindow, dismissUpdate]);
|
||||
|
||||
// Track previous autoDownloadStatus so toast effects fire only on actual transitions,
|
||||
// not when unrelated deps (installUpdate, openSettingsWindow) change their reference.
|
||||
const prevAutoDownloadStatusRef = useRef(updateState.autoDownloadStatus);
|
||||
useEffect(() => {
|
||||
if (!enabled) return;
|
||||
const prev = prevAutoDownloadStatusRef.current;
|
||||
prevAutoDownloadStatusRef.current = updateState.autoDownloadStatus;
|
||||
if (prev === updateState.autoDownloadStatus) return;
|
||||
|
||||
if (updateState.autoDownloadStatus === 'ready') {
|
||||
const version = updateState.latestRelease?.version ?? '';
|
||||
toast.info(
|
||||
t('update.readyToInstall.message', { version }),
|
||||
{
|
||||
title: t('update.readyToInstall.title'),
|
||||
duration: 0,
|
||||
actionLabel: t('update.restartNow'),
|
||||
onClick: () => installUpdate(),
|
||||
}
|
||||
);
|
||||
} else if (updateState.autoDownloadStatus === 'error') {
|
||||
toast.error(
|
||||
t('update.downloadFailed.message'),
|
||||
{
|
||||
title: t('update.downloadFailed.title'),
|
||||
actionLabel: t('update.viewInSettings'),
|
||||
onClick: () => void openSettingsWindow(),
|
||||
}
|
||||
);
|
||||
}
|
||||
}, [enabled, updateState.autoDownloadStatus, updateState.latestRelease?.version, t, installUpdate, openSettingsWindow]);
|
||||
|
||||
// Auto-start port forwarding rules on app launch
|
||||
usePortForwardingAutoStart({
|
||||
enabled,
|
||||
isVaultInitialized,
|
||||
hosts,
|
||||
keys,
|
||||
identities,
|
||||
knownHosts,
|
||||
proxyProfiles,
|
||||
groupConfigs,
|
||||
terminalSettings,
|
||||
});
|
||||
|
||||
// Sync tray menu data + handle tray actions
|
||||
useEffect(() => {
|
||||
if (!enabled) return;
|
||||
const bridge = netcattyBridge.get();
|
||||
if (!bridge?.updateTrayMenuData) return;
|
||||
|
||||
let cancelled = false;
|
||||
const timer = setTimeout(() => {
|
||||
if (cancelled) return;
|
||||
|
||||
const sessionsForTray = sessions.map((s) => {
|
||||
const ws = s.workspaceId ? workspaces.find((w) => w.id === s.workspaceId) : undefined;
|
||||
return {
|
||||
id: s.id,
|
||||
label: s.hostname,
|
||||
hostLabel: s.hostLabel,
|
||||
status: s.status,
|
||||
workspaceId: s.workspaceId,
|
||||
workspaceTitle: ws?.title,
|
||||
aiHidden: s.hiddenFromTabs === true,
|
||||
};
|
||||
});
|
||||
|
||||
const hostsForSystemMenu = hosts
|
||||
.filter((host: any) => typeof host?.id === "string" && host.id.length > 0)
|
||||
.map((host: any) => ({
|
||||
id: host.id,
|
||||
label: host.label,
|
||||
hostname: host.hostname,
|
||||
group: host.group,
|
||||
pinned: host.pinned,
|
||||
lastConnectedAt: host.lastConnectedAt,
|
||||
protocol: host.protocol,
|
||||
}));
|
||||
|
||||
void bridge.updateTrayMenuData({
|
||||
sessions: sessionsForTray,
|
||||
portForwardRules: portForwardingRules.map((rule: any) => ({
|
||||
...rule,
|
||||
canStop: hasRuntimeTunnel(rule.id),
|
||||
})),
|
||||
hosts: hostsForSystemMenu,
|
||||
});
|
||||
}, 250);
|
||||
|
||||
return () => {
|
||||
cancelled = true;
|
||||
clearTimeout(timer);
|
||||
};
|
||||
}, [enabled, hasRuntimeTunnel, hosts, sessions, portForwardingRules, workspaces]);
|
||||
|
||||
// Quit guard: block app exit while any editor tab has unsaved changes.
|
||||
// Main process sends "app:query-dirty-editors"; we respond with the result.
|
||||
useEffect(() => {
|
||||
const bridge = netcattyBridge.get();
|
||||
if (!bridge?.onCheckDirtyEditors) return;
|
||||
const unsub = bridge.onCheckDirtyEditors(async () => {
|
||||
// Always report SOMETHING so the main process doesn't time out for
|
||||
// 5 s on an unhandled exception. If we can't determine the state,
|
||||
// fail open — losing unsaved work is bad, but stranding the user
|
||||
// on a slow quit and then quitting anyway after the timeout is
|
||||
// exactly the same outcome.
|
||||
let hasDirty = false;
|
||||
try {
|
||||
hasDirty = editorTabStore.getTabs().some((tab) => tab.content !== tab.baselineContent);
|
||||
if (hasDirty) toast.warning(t('sftp.editor.quitBlockedByDirty'), 'SFTP');
|
||||
if (!hasDirty) {
|
||||
const unfinishedTasks = sftpTransferCenterStore.getSnapshot().tasks.filter((task) => (
|
||||
!task.parentTaskId && !["completed", "failed", "cancelled"].includes(task.status)
|
||||
));
|
||||
if (unfinishedTasks.length > 0) {
|
||||
await Promise.allSettled(unfinishedTasks.map((task) => sftpTransferCenterStore.pause(task.id)));
|
||||
hasDirty = !window.confirm(t('sftp.transferCenter.quitConfirm', { count: unfinishedTasks.length }));
|
||||
}
|
||||
}
|
||||
} catch (err) {
|
||||
console.error('[App] dirty-editors check failed:', err);
|
||||
}
|
||||
try {
|
||||
bridge.reportDirtyEditorsResult?.(hasDirty);
|
||||
} catch (err) {
|
||||
// Reporting itself shouldn't throw, but if the IPC bridge is in a
|
||||
// bad state we'd rather log than bubble out of the listener and
|
||||
// disable the quit guard for the rest of the session.
|
||||
console.error('[App] reportDirtyEditorsResult failed:', err);
|
||||
}
|
||||
});
|
||||
return unsub;
|
||||
}, [enabled, t]);
|
||||
|
||||
useEffect(() => {
|
||||
const bridge = netcattyBridge.get();
|
||||
const unsubscribeEvents = bridge?.onGlobalSftpTransferEvent?.((event) => {
|
||||
sftpTransferCenterStore.ingestBackgroundEvent(event);
|
||||
});
|
||||
const restartBackgroundTransfer = async (taskId: string, fromBeginning: boolean) => {
|
||||
const task = sftpTransferCenterStore.getSnapshot().tasks.find((candidate) => candidate.id === taskId);
|
||||
if (!task || !bridge?.openSftpForSession || !bridge.startStreamTransfer) return;
|
||||
const sessionId = task.direction === "upload" ? task.targetConnectionId : task.sourceConnectionId;
|
||||
if (!sessionId || sessionId === "agent" || sessionId === "local") {
|
||||
sftpTransferCenterStore.ingestBackgroundEvent({
|
||||
type: "failed",
|
||||
transferId: taskId,
|
||||
error: "The original server session is unavailable",
|
||||
endedAt: Date.now(),
|
||||
});
|
||||
return;
|
||||
}
|
||||
let sftpId: string | undefined;
|
||||
try {
|
||||
const checkpointBytes = fromBeginning ? 0 : (task.checkpointBytes ?? task.transferredBytes ?? 0);
|
||||
sftpTransferCenterStore.ingestBackgroundEvent({ type: "queued", transferId: taskId });
|
||||
// Admit first so agent resume does not pin session-backed SFTP handles
|
||||
// while waiting for main-process concurrency.
|
||||
const result = await globalSftpTransferScheduler.run(
|
||||
"background-agent",
|
||||
task.id,
|
||||
getSftpTransferResourceKeys({
|
||||
sourceHostId: task.sourceHostId,
|
||||
targetHostId: task.targetHostId,
|
||||
}),
|
||||
() => localStorageAdapter.readNumber(STORAGE_KEY_SFTP_TRANSFER_CONCURRENCY),
|
||||
async () => {
|
||||
sftpId = await bridge.openSftpForSession!(sessionId);
|
||||
return bridge.startStreamTransfer!({
|
||||
transferId: task.id,
|
||||
sourcePath: task.sourcePath,
|
||||
targetPath: task.targetPath,
|
||||
sourceType: task.direction === "upload" ? "local" : "sftp",
|
||||
targetType: task.direction === "download" ? "local" : "sftp",
|
||||
sourceSftpId: task.direction === "download" ? sftpId : undefined,
|
||||
targetSftpId: task.direction === "upload" ? sftpId : undefined,
|
||||
// Keep host-scoped path gates across session reopen (Codex P1).
|
||||
sourceHostId: task.sourceHostId,
|
||||
targetHostId: task.targetHostId,
|
||||
totalBytes: task.totalBytes,
|
||||
resumable: task.resumable !== false,
|
||||
checkpointBytes,
|
||||
resumeStage: fromBeginning ? undefined : task.resumeStage,
|
||||
downloadCheckpointBytes: fromBeginning ? 0 : task.downloadCheckpointBytes,
|
||||
uploadCheckpointBytes: fromBeginning ? 0 : task.uploadCheckpointBytes,
|
||||
sourceFingerprint: fromBeginning ? undefined : task.sourceFingerprint,
|
||||
skipAdmission: true,
|
||||
});
|
||||
},
|
||||
);
|
||||
// Same-id retry stole ownership; wait for the live owner's terminal
|
||||
// status instead of treating this invoke as completed (Codex P2).
|
||||
if (result?.superseded === true) {
|
||||
// Wait for live owner terminal status only (no fixed deadline).
|
||||
for (;;) {
|
||||
const latest = sftpTransferCenterStore.getSnapshot().tasks.find((candidate) => candidate.id === task.id);
|
||||
const status = latest?.status;
|
||||
if (status === "completed" || status === "cancelled" || status === "failed") {
|
||||
if (status === "failed") {
|
||||
throw new Error(latest?.error || "Transfer failed");
|
||||
}
|
||||
if (status === "cancelled") {
|
||||
sftpTransferCenterStore.ingestBackgroundEvent({
|
||||
type: "cancelled",
|
||||
transferId: task.id,
|
||||
endedAt: Date.now(),
|
||||
});
|
||||
}
|
||||
// completed: events already applied; cancelled handled above.
|
||||
break;
|
||||
}
|
||||
await new Promise((resolve) => setTimeout(resolve, 200));
|
||||
}
|
||||
} else if (result?.cancelled || result?.error === "Transfer cancelled") {
|
||||
sftpTransferCenterStore.ingestBackgroundEvent({ type: "cancelled", transferId: task.id, endedAt: Date.now() });
|
||||
} else if (result?.error) {
|
||||
throw new Error(result.error);
|
||||
} else {
|
||||
sftpTransferCenterStore.ingestBackgroundEvent({ type: "completed", transferId: task.id, endedAt: Date.now() });
|
||||
}
|
||||
} catch (error) {
|
||||
sftpTransferCenterStore.ingestBackgroundEvent({
|
||||
type: "failed",
|
||||
transferId: task.id,
|
||||
error: error instanceof Error ? error.message : String(error),
|
||||
endedAt: Date.now(),
|
||||
});
|
||||
} finally {
|
||||
if (sftpId) await bridge.closeSftp?.(sftpId).catch(() => {});
|
||||
}
|
||||
};
|
||||
const unregisterOwner = sftpTransferCenterStore.registerOwner("background-agent", {
|
||||
pause: async (taskId) => {
|
||||
const result = await bridge?.pauseTransfer?.(taskId);
|
||||
if (result?.success) sftpTransferCenterStore.ingestBackgroundEvent({
|
||||
type: "paused",
|
||||
transferId: taskId,
|
||||
checkpointBytes: result.checkpointBytes,
|
||||
resumeStage: result.resumeStage,
|
||||
downloadCheckpointBytes: result.downloadCheckpointBytes,
|
||||
uploadCheckpointBytes: result.uploadCheckpointBytes,
|
||||
sourceFingerprint: result.sourceFingerprint,
|
||||
});
|
||||
},
|
||||
resume: async (taskId) => {
|
||||
const result = await bridge?.resumeTransfer?.(taskId);
|
||||
if (result?.success) {
|
||||
sftpTransferCenterStore.ingestBackgroundEvent({ type: "resumed", transferId: taskId });
|
||||
} else {
|
||||
sftpTransferCenterStore.markReconnectRequired(
|
||||
taskId,
|
||||
result?.reason ?? "The original server connection is unavailable",
|
||||
);
|
||||
setTimeout(() => { void sftpTransferCenterStore.resume(taskId); }, 0);
|
||||
}
|
||||
},
|
||||
cancel: async (taskId) => {
|
||||
await bridge?.cancelTransfer?.(taskId);
|
||||
sftpTransferCenterStore.ingestBackgroundEvent({ type: "cancelled", transferId: taskId, endedAt: Date.now() });
|
||||
},
|
||||
retry: async (taskId) => { await restartBackgroundTransfer(taskId, true); },
|
||||
prioritize: async (taskId) => { await bridge?.prioritizeTransfer?.(taskId); },
|
||||
dismiss: (taskId, prunedTask) => {
|
||||
const task = prunedTask
|
||||
?? sftpTransferCenterStore.getSnapshot().tasks.find((candidate) => candidate.id === taskId);
|
||||
if (!task) return;
|
||||
void bridge?.cleanupTransferArtifacts?.({
|
||||
transferId: task.id,
|
||||
sourcePath: task.sourcePath,
|
||||
targetPath: task.targetPath,
|
||||
stagedTargetPath: task.stagedTargetPath,
|
||||
});
|
||||
},
|
||||
});
|
||||
return () => {
|
||||
unsubscribeEvents?.();
|
||||
unregisterOwner();
|
||||
};
|
||||
}, [enabled]);
|
||||
|
||||
// Keyboard-interactive authentication (2FA/MFA) event listener
|
||||
useEffect(() => {
|
||||
const bridge = netcattyBridge.get();
|
||||
if (!bridge?.onKeyboardInteractive) return;
|
||||
|
||||
const unsubscribe = bridge.onKeyboardInteractive((request) => {
|
||||
if (!shouldQueueKeyboardInteractiveRequest(request, sessionsRef.current)) {
|
||||
if (request.scope === "terminal" && request.requestId) {
|
||||
void bridge.respondKeyboardInteractive?.(request.requestId, [], true);
|
||||
}
|
||||
return;
|
||||
}
|
||||
console.log('[App] Keyboard-interactive request received:', request);
|
||||
// Add to queue instead of replacing - supports multiple concurrent sessions
|
||||
setKeyboardInteractiveQueue(prev => [...prev, {
|
||||
requestId: request.requestId,
|
||||
sessionId: request.sessionId,
|
||||
hostId: request.hostId,
|
||||
name: request.name,
|
||||
instructions: request.instructions,
|
||||
prompts: request.prompts,
|
||||
hostname: request.hostname,
|
||||
savedPassword: request.savedPassword,
|
||||
allowSavePassword: request.allowSavePassword !== false,
|
||||
}]);
|
||||
});
|
||||
const unsubscribeCancelled = bridge.onKeyboardInteractiveCancelled?.((event) => {
|
||||
setKeyboardInteractiveQueue(prev => removeKeyboardInteractiveRequest(prev, event.requestId));
|
||||
});
|
||||
const onTerminalDisconnected = (event: Event) => {
|
||||
const sessionId = (event as CustomEvent<{ sessionId?: string }>).detail?.sessionId;
|
||||
if (!sessionId) return;
|
||||
setKeyboardInteractiveQueue((prev) => {
|
||||
const doomed = prev.filter((request) => request.sessionId === sessionId);
|
||||
for (const request of doomed) {
|
||||
void bridge.respondKeyboardInteractive?.(request.requestId, [], true);
|
||||
}
|
||||
return prev.filter((request) => request.sessionId !== sessionId);
|
||||
});
|
||||
};
|
||||
window.addEventListener("netcatty:terminal-session-disconnected", onTerminalDisconnected);
|
||||
|
||||
return () => {
|
||||
unsubscribe?.();
|
||||
unsubscribeCancelled?.();
|
||||
window.removeEventListener("netcatty:terminal-session-disconnected", onTerminalDisconnected);
|
||||
};
|
||||
}, [enabled, setKeyboardInteractiveQueue]);
|
||||
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user