import { useState, useEffect, useCallback, useRef } from "react"; import { Download, FolderOpen, HardDrive } from "lucide-react"; import { save } from "@tauri-apps/plugin-dialog"; import { Button } from "../ui/Button"; import { BackupProgress } from "./BackupProgress"; import { useBackupStore } from "../../stores/backupStore"; import { useNotificationStore } from "../../stores/notificationStore"; import { detectPgTools, pgDump, getSchemas } from "../../lib/commands"; import type { PgToolStatus } from "../../lib/types"; interface BackupPageProps { connectionId: string; } type BackupFormat = "plain" | "custom" | "tar" | "directory"; const PLATFORM_INSTALL_INSTRUCTIONS: Record = { darwin: "brew install libpq", linux: "sudo apt install postgresql-client # Debian/Ubuntu\nsudo dnf install postgresql # Fedora\nsudo pacman -S postgresql # Arch", win32: "Download PostgreSQL installer from https://www.postgresql.org/download/windows/ and ensure pg_dump is in your PATH.", }; function getPlatformInstructions(): string { const platform = typeof navigator !== "undefined" ? navigator.platform.toLowerCase() : ""; if (platform.includes("mac") || platform.includes("darwin")) return PLATFORM_INSTALL_INSTRUCTIONS.darwin; if (platform.includes("linux")) return PLATFORM_INSTALL_INSTRUCTIONS.linux; if (platform.includes("win")) return PLATFORM_INSTALL_INSTRUCTIONS.win32; return PLATFORM_INSTALL_INSTRUCTIONS.linux; } export function BackupPage({ connectionId }: BackupPageProps) { const [format, setFormat] = useState("custom"); const [filePath, setFilePath] = useState(""); const [schema, setSchema] = useState(""); const [noOwner, setNoOwner] = useState(true); const [toolStatus, setToolStatus] = useState(null); const [checkingTools, setCheckingTools] = useState(true); const [availableSchemas, setAvailableSchemas] = useState([]); const activeJobId = useBackupStore((s) => s.activeJobId); const jobs = useBackupStore((s) => s.jobs); const startJob = useBackupStore((s) => s.startJob); const notify = useNotificationStore((s) => s.notify); const activeJob = jobs.find((j) => j.id === activeJobId); const isRunning = activeJob?.status === "running"; // Track our job ID so we only react to jobs we started const pendingJobRef = useRef(null); // React to job completion/failure via store events useEffect(() => { if (!pendingJobRef.current || !activeJob) return; if (activeJob.id !== pendingJobRef.current) return; if (activeJob.status === "completed") { notify("Backup completed successfully", "success"); pendingJobRef.current = null; } else if (activeJob.status === "failed") { notify( `Backup failed: ${activeJob.error_message || "Unknown error"}`, "error", ); pendingJobRef.current = null; } }, [activeJob, notify]); useEffect(() => { setCheckingTools(true); detectPgTools() .then((status) => setToolStatus(status)) .catch(() => setToolStatus({ pg_dump_found: false, pg_restore_found: false, pg_dump_version: null, pg_restore_version: null, pg_dump_source: null, pg_restore_source: null, }), ) .finally(() => setCheckingTools(false)); getSchemas(connectionId) .then((schemas) => setAvailableSchemas(schemas)) .catch(() => setAvailableSchemas([])); }, [connectionId]); const handlePickFile = useCallback(async () => { const extensions: Record = { plain: ["sql"], custom: ["dump", "custom"], tar: ["tar"], directory: [], }; const picked = await save({ defaultPath: `backup.${ format === "custom" ? "dump" : format === "plain" ? "sql" : "tar" }`, filters: [{ name: "Backup", extensions: extensions[format] }], }); if (picked) setFilePath(picked); }, [format]); const handleStartBackup = useCallback(async () => { if (!filePath) { notify("Please select a file path", "error"); return; } const jobId = `dump-${Date.now()}`; startJob(jobId, "dump"); pendingJobRef.current = jobId; try { // pgDump returns the job ID immediately — completion // comes via Tauri events handled by the backupStore await pgDump(connectionId, { format, filePath, schema: schema || undefined, tables: undefined, noOwner, }); } catch (e) { // If the command itself fails (e.g. connection not found), // the event won't fire — handle here const msg = e instanceof Error ? e.message : String(e); useBackupStore.getState().failJob(jobId, msg); } }, [filePath, format, schema, noOwner, connectionId, startJob, notify]); const toolsMissing = toolStatus && !toolStatus.pg_dump_found; const toolsBundled = toolStatus?.pg_dump_source === "bundled"; return (
{/* Toolbar header */}
Backup Create a database backup via pg_dump
{/* Content */}
{/* Tool check */} {checkingTools && (

Checking for pg_dump...

)} {toolsMissing && !toolsBundled && (

pg_dump not found

The PostgreSQL client tools are required for backup/restore operations. Install them using:

                                {getPlatformInstructions()}
                            
)} {!checkingTools && !toolsMissing && ( <> {/* Configuration card */}
{/* Format */}
{/* Output file */}
setFilePath(e.target.value) } placeholder="/path/to/backup.dump" className="flex-1 px-4 py-2 text-sm text-text placeholder-text-muted/50 border-b border-border focus:border-accent focus:outline-none transition-colors" />
{/* Schema (optional) */}
{/* No-owner toggle */}
{/* Progress */} {activeJob && (
)} {/* Actions */}
)}
); }