import { getErrorMessage } from "@/lib/error-message.js";
import { useState, useEffect, useCallback, useMemo, useRef } from "react";
import { useTranslation } from "react-i18next";
import { toast } from "sonner";
import {
Boxes,
Check,
ChevronLeft,
Download,
ExternalLink,
Loader2,
Package,
Play,
Plus,
RefreshCw,
Search,
Server,
Settings2,
Share2,
Shield,
Trash2,
Upload,
User,
X,
} from "lucide-react";
import { Button } from "@/components/button";
import { Input } from "@/components/input";
import { Textarea } from "@/components/textarea";
import { Checkbox } from "@/components/checkbox";
import { Badge } from "@/components/badge";
import {
Dialog,
DialogContent,
DialogFooter,
DialogHeader,
DialogTitle,
DialogDescription,
} from "@/components/dialog";
import { FOLDER_COLORS } from "@/lib/theme";
import { getSSHHosts } from "@/api/ssh-host-management-api";
import { getUserList } from "@/api/user-management-api";
import {
getRoles,
type SharePermissionLevel,
type ShareTarget,
} from "@/api/rbac-api";
import { Select2 } from "@/components/select2";
import {
DropdownMenu,
DropdownMenuContent,
DropdownMenuItem,
DropdownMenuTrigger,
} from "@/components/dropdown-menu";
import {
extractSnippetInputs,
type SnippetInput,
} from "@/lib/snippet-variables";
import {
listFleets,
createFleet,
updateFleet,
deleteFleet,
getFleetMembers,
addFleetMember,
removeFleetMember,
runFleetCommand,
pushFleetFile,
pullFleetFile,
getFleetInventory,
refreshFleetInventory,
runFleetPackageAction,
shareFleet,
type FleetRow,
type FleetMemberRow,
type FleetHostResult,
type FleetInventoryEntry,
type FleetPackageAction,
type FleetShareHostResult,
} from "@/api/fleets-api";
import type { SSHHost } from "@/types/index";
const SHARE_PERMISSION_LEVELS: SharePermissionLevel[] = [
"connect",
"view",
"edit",
"manage",
];
const SHARE_EXPIRY_PRESETS = [
{ key: "never", hours: undefined },
{ key: "oneHour", hours: 1 },
{ key: "oneDay", hours: 24 },
{ key: "sevenDays", hours: 24 * 7 },
{ key: "thirtyDays", hours: 24 * 30 },
{ key: "custom", hours: undefined },
] as const;
type ShareExpiryPresetKey = (typeof SHARE_EXPIRY_PRESETS)[number]["key"];
function formatUptime(seconds: number | null): string {
if (seconds === null) return "-";
const d = Math.floor(seconds / 86400);
const h = Math.floor((seconds % 86400) / 3600);
const m = Math.floor((seconds % 3600) / 60);
if (d > 0) return `${d}d ${h}h`;
if (h > 0) return `${h}h ${m}m`;
return `${m}m`;
}
function ResultsList({ results }: { results: FleetHostResult[] }) {
const { t } = useTranslation();
if (results.length === 0) return null;
return (
{results.map((r) => (
{r.hostName}
{r.success
? t("newUi.sidebar.fleets.resultSuccess")
: t("newUi.sidebar.fleets.resultFailed")}
{(r.output || r.error) && (
{r.error ? r.error : r.output}
)}
))}
);
}
function FleetFormDialog({
open,
onClose,
fleet,
onSaved,
}: {
open: boolean;
onClose: () => void;
fleet: FleetRow | null;
onSaved: () => void;
}) {
const { t } = useTranslation();
const [name, setName] = useState("");
const [description, setDescription] = useState("");
const [color, setColor] = useState(FOLDER_COLORS[0]);
const [tagRulesText, setTagRulesText] = useState("");
const [saving, setSaving] = useState(false);
useEffect(() => {
if (!open) return;
setName(fleet?.name ?? "");
setDescription(fleet?.description ?? "");
setColor(fleet?.color ?? FOLDER_COLORS[0]);
setTagRulesText((fleet?.tagRules ?? []).join(", "));
}, [open, fleet]);
async function handleSave() {
if (!name.trim()) {
toast.error(t("newUi.sidebar.fleets.nameRequired"));
return;
}
const tagRules = tagRulesText
.split(",")
.map((tag) => tag.trim())
.filter(Boolean);
setSaving(true);
try {
if (fleet) {
await updateFleet(fleet.id, {
name: name.trim(),
description: description.trim() || null,
color,
tagRules,
});
toast.success(t("newUi.sidebar.fleets.fleetUpdated"));
} else {
await createFleet({
name: name.trim(),
description: description.trim() || null,
color,
tagRules,
});
toast.success(t("newUi.sidebar.fleets.fleetCreated"));
}
onSaved();
onClose();
} catch (error) {
const message = getErrorMessage(error, "");
toast.error(message || t("newUi.sidebar.fleets.saveFailed"));
} finally {
setSaving(false);
}
}
return (
);
}
function MemberPickerDialog({
open,
onClose,
fleetId,
allHosts,
memberIds,
onChanged,
}: {
open: boolean;
onClose: () => void;
fleetId: number;
allHosts: SSHHost[];
memberIds: Set;
onChanged: () => void;
}) {
const { t } = useTranslation();
const [search, setSearch] = useState("");
const [busyId, setBusyId] = useState(null);
const visibleHosts = useMemo(
() =>
allHosts.filter((h) =>
h.name.toLowerCase().includes(search.trim().toLowerCase()),
),
[allHosts, search],
);
async function toggleHost(host: SSHHost, isMember: boolean) {
setBusyId(host.id);
try {
if (isMember) {
await removeFleetMember(fleetId, host.id);
} else {
await addFleetMember(fleetId, host.id);
}
onChanged();
} catch (error) {
const message = getErrorMessage(error, "");
toast.error(message || t("newUi.sidebar.fleets.memberUpdateFailed"));
} finally {
setBusyId(null);
}
}
return (
);
}
function FleetShareDialog({
open,
onClose,
fleet,
}: {
open: boolean;
onClose: () => void;
fleet: FleetRow | null;
}) {
const { t } = useTranslation();
const [targetTab, setTargetTab] = useState<"user" | "role">("user");
const [search, setSearch] = useState("");
const [shareUsers, setShareUsers] = useState<
{ id: string; username: string }[]
>([]);
const [shareRoles, setShareRoles] = useState<
{ id: number; name: string; displayName?: string }[]
>([]);
const [selectedUserIds, setSelectedUserIds] = useState>(
new Set(),
);
const [selectedRoleIds, setSelectedRoleIds] = useState>(
new Set(),
);
const [permissionLevel, setPermissionLevel] =
useState("connect");
const [expiryPreset, setExpiryPreset] =
useState("never");
const [customHours, setCustomHours] = useState("");
const [submitting, setSubmitting] = useState(false);
const [summary, setSummary] = useState<{
hostsShared: number;
hostsTotal: number;
hostResults: FleetShareHostResult[];
} | null>(null);
useEffect(() => {
if (!open) return;
setSearch("");
setSelectedUserIds(new Set());
setSelectedRoleIds(new Set());
setPermissionLevel("connect");
setExpiryPreset("never");
setCustomHours("");
setTargetTab("user");
setSummary(null);
Promise.all([
getUserList().catch(() => ({ users: [] })),
getRoles().catch(() => ({ roles: [] })),
]).then(([usersRes, rolesRes]) => {
setShareUsers(
(usersRes.users ?? []).map((u) => ({
id: String(u.userId),
username: u.username,
})),
);
setShareRoles(
(rolesRes.roles ?? [])
.filter((r) => !r.isSystem)
.map((r) => ({
id: Number(r.id),
name: r.name,
displayName: r.displayName,
})),
);
});
}, [open]);
const filteredUsers = useMemo(() => {
const q = search.trim().toLowerCase();
return q
? shareUsers.filter((u) => u.username.toLowerCase().includes(q))
: shareUsers;
}, [shareUsers, search]);
const filteredRoles = useMemo(() => {
const q = search.trim().toLowerCase();
return q
? shareRoles.filter(
(r) =>
r.name.toLowerCase().includes(q) ||
(r.displayName ?? "").toLowerCase().includes(q),
)
: shareRoles;
}, [shareRoles, search]);
const selectedCount = selectedUserIds.size + selectedRoleIds.size;
const durationHours = (() => {
if (expiryPreset === "never") return undefined;
if (expiryPreset === "custom") {
const hours = Number(customHours);
return Number.isFinite(hours) && hours > 0 ? hours : undefined;
}
return (
SHARE_EXPIRY_PRESETS.find((p) => p.key === expiryPreset)?.hours ??
undefined
);
})();
async function handleShare() {
if (!fleet || selectedCount === 0) return;
const targets: ShareTarget[] = [
...[...selectedUserIds].map(
(id) => ({ type: "user", id }) as ShareTarget,
),
...[...selectedRoleIds].map(
(id) => ({ type: "role", id }) as ShareTarget,
),
];
setSubmitting(true);
try {
const result = await shareFleet(fleet.id, {
targets,
permissionLevel,
...(durationHours ? { durationHours } : {}),
});
setSummary({
hostsShared: result.hostsShared,
hostsTotal: result.hostsTotal,
hostResults: result.hostResults,
});
setSelectedUserIds(new Set());
setSelectedRoleIds(new Set());
toast.success(
t("newUi.sidebar.fleets.fleetSharedSuccessfully", {
shared: result.hostsShared,
total: result.hostsTotal,
}),
);
} catch (error) {
const message = getErrorMessage(error, "");
toast.error(message || t("newUi.sidebar.fleets.failedToShareFleet"));
} finally {
setSubmitting(false);
}
}
return (
);
}
function RunCommandTab({ fleetId }: { fleetId: number }) {
const { t } = useTranslation();
const [command, setCommand] = useState("");
const [inputs, setInputs] = useState([]);
const [inputValues, setInputValues] = useState>({});
const [running, setRunning] = useState(false);
const [results, setResults] = useState(null);
useEffect(() => {
setInputs(extractSnippetInputs(command));
}, [command]);
async function handleRun() {
if (!command.trim()) {
toast.error(t("newUi.sidebar.fleets.commandRequired"));
return;
}
setRunning(true);
setResults(null);
try {
const { results: r } = await runFleetCommand(
fleetId,
command,
inputs.length > 0 ? inputValues : undefined,
);
setResults(r);
const failed = r.filter((x) => !x.success).length;
if (failed === 0) {
toast.success(t("newUi.sidebar.fleets.commandSucceeded"));
} else {
toast.warning(
t("newUi.sidebar.fleets.commandPartialFailure", {
failed,
total: r.length,
}),
);
}
} catch (error) {
const message = getErrorMessage(error, "");
toast.error(message || t("newUi.sidebar.fleets.commandFailed"));
} finally {
setRunning(false);
}
}
return (
);
}
function TransferTab({ fleetId }: { fleetId: number }) {
const { t } = useTranslation();
const [direction, setDirection] = useState<"push" | "pull">("push");
const [remotePath, setRemotePath] = useState("");
const [file, setFile] = useState(null);
const [busy, setBusy] = useState(false);
const [results, setResults] = useState(null);
async function handleTransfer() {
if (!remotePath.trim()) {
toast.error(t("newUi.sidebar.fleets.remotePathRequired"));
return;
}
if (direction === "push" && !file) {
toast.error(t("newUi.sidebar.fleets.fileRequired"));
return;
}
setBusy(true);
setResults(null);
try {
if (direction === "push" && file) {
const { results: r } = await pushFleetFile(fleetId, file, remotePath);
setResults(r);
} else {
const {
results: r,
blob,
fileName,
} = await pullFleetFile(fleetId, remotePath);
setResults(r);
const url = URL.createObjectURL(blob);
const a = document.createElement("a");
a.href = url;
a.download = fileName;
document.body.appendChild(a);
a.click();
document.body.removeChild(a);
URL.revokeObjectURL(url);
}
toast.success(t("newUi.sidebar.fleets.transferComplete"));
} catch (error) {
const message = getErrorMessage(error, "");
toast.error(message || t("newUi.sidebar.fleets.transferFailed"));
} finally {
setBusy(false);
}
}
return (
setRemotePath(e.target.value)}
placeholder="/etc/hosts"
className="font-mono text-xs"
/>
{direction === "push" && (
setFile(e.target.files?.[0] ?? null)}
/>
)}
{direction === "push"
? t("newUi.sidebar.fleets.pushHint")
: t("newUi.sidebar.fleets.pullHint")}
{results &&
}
);
}
function InventoryTab({ fleetId }: { fleetId: number }) {
const { t } = useTranslation();
const [entries, setEntries] = useState([]);
const [loading, setLoading] = useState(true);
const [refreshing, setRefreshing] = useState(false);
const load = useCallback(async () => {
setLoading(true);
try {
const data = await getFleetInventory(fleetId);
setEntries(data);
} catch (error) {
const message = getErrorMessage(error, "");
toast.error(message || t("newUi.sidebar.fleets.inventoryLoadFailed"));
} finally {
setLoading(false);
}
}, [fleetId, t]);
useEffect(() => {
load();
}, [load]);
async function handleRefresh() {
setRefreshing(true);
try {
const { results } = await refreshFleetInventory(fleetId);
const failed = results.filter((r) => !r.success).length;
if (failed > 0) {
toast.warning(
t("newUi.sidebar.fleets.commandPartialFailure", {
failed,
total: results.length,
}),
);
} else {
toast.success(t("newUi.sidebar.fleets.inventoryRefreshed"));
}
await load();
} catch (error) {
const message = getErrorMessage(error, "");
toast.error(message || t("newUi.sidebar.fleets.inventoryRefreshFailed"));
} finally {
setRefreshing(false);
}
}
return (
{loading ? (
) : (
{entries.map((entry) => (
{entry.hostName}
{entry.inventory ? (
{t("newUi.sidebar.fleets.inventoryOs")}:{" "}
{entry.inventory.osPrettyName ?? "-"}
{t("newUi.sidebar.fleets.inventoryKernel")}:{" "}
{entry.inventory.kernel ?? "-"}
{t("newUi.sidebar.fleets.inventoryArch")}:{" "}
{entry.inventory.architecture ?? "-"}
{t("newUi.sidebar.fleets.inventoryUptime")}:{" "}
{formatUptime(entry.inventory.uptimeSeconds)}
{t("newUi.sidebar.fleets.inventoryPackageManager")}:{" "}
{entry.inventory.packageManager ?? "-"}
{t("newUi.sidebar.fleets.inventoryCollectedAt")}:{" "}
{new Date(entry.inventory.collectedAt).toLocaleString()}
) : (
{t("newUi.sidebar.fleets.inventoryNoData")}
)}
))}
)}
);
}
function PackagesTab({ fleetId }: { fleetId: number }) {
const { t } = useTranslation();
const [action, setAction] = useState("install");
const [packageName, setPackageName] = useState("");
const [running, setRunning] = useState(false);
const [results, setResults] = useState(null);
async function handleRun() {
if (action !== "upgrade-all" && !packageName.trim()) {
toast.error(t("newUi.sidebar.fleets.packageNameRequired"));
return;
}
setRunning(true);
setResults(null);
try {
const { results: r } = await runFleetPackageAction(
fleetId,
action,
action === "upgrade-all" ? undefined : packageName.trim(),
);
setResults(r);
const failed = r.filter((x) => !x.success).length;
if (failed === 0) {
toast.success(t("newUi.sidebar.fleets.commandSucceeded"));
} else {
toast.warning(
t("newUi.sidebar.fleets.commandPartialFailure", {
failed,
total: r.length,
}),
);
}
} catch (error) {
const message = getErrorMessage(error, "");
toast.error(message || t("newUi.sidebar.fleets.commandFailed"));
} finally {
setRunning(false);
}
}
return (
{action !== "upgrade-all" && (
setPackageName(e.target.value)}
placeholder="curl"
className="font-mono text-xs"
/>
)}
{t("newUi.sidebar.fleets.packageHint")}
{results &&
}
);
}
type FleetActionView = "run" | "transfer" | "inventory" | "packages";
function FleetDetail({
fleet,
allHosts,
onBack,
onFleetChanged,
onOpenFleetInventory,
}: {
fleet: FleetRow;
allHosts: SSHHost[];
onBack: () => void;
onFleetChanged: () => void;
onOpenFleetInventory?: (fleetId: number) => void;
}) {
const { t } = useTranslation();
const [members, setMembers] = useState([]);
const [loadingMembers, setLoadingMembers] = useState(true);
const [pickerOpen, setPickerOpen] = useState(false);
const [shareOpen, setShareOpen] = useState(false);
const [actionView, setActionView] = useState("run");
const loadMembers = useCallback(async () => {
setLoadingMembers(true);
try {
const data = await getFleetMembers(fleet.id);
setMembers(data);
} catch (error) {
const message = getErrorMessage(error, "");
toast.error(message || t("newUi.sidebar.fleets.membersLoadFailed"));
} finally {
setLoadingMembers(false);
}
}, [fleet.id, t]);
useEffect(() => {
loadMembers();
}, [loadMembers]);
const memberIds = useMemo(() => new Set(members.map((m) => m.id)), [members]);
function handleMembersChanged() {
loadMembers();
onFleetChanged();
}
return (
{fleet.name}
{t("newUi.sidebar.fleets.memberCount", { count: members.length })}
{loadingMembers ? (
) : members.length === 0 ? (
{t("newUi.sidebar.fleets.noMembers")}
) : (
{actionView === "inventory" && onOpenFleetInventory && (
)}
{actionView === "run" &&
}
{actionView === "transfer" &&
}
{actionView === "inventory" &&
}
{actionView === "packages" &&
}
)}
setPickerOpen(false)}
fleetId={fleet.id}
allHosts={allHosts}
memberIds={memberIds}
onChanged={handleMembersChanged}
/>
setShareOpen(false)}
fleet={fleet}
/>
);
}
export function FleetsPanel({
active,
onOpenFleetInventory,
}: {
active?: boolean;
onOpenFleetInventory?: (fleetId: number) => void;
}) {
const { t } = useTranslation();
const [fleets, setFleets] = useState([]);
const [allHosts, setAllHosts] = useState([]);
const [loading, setLoading] = useState(true);
const [search, setSearch] = useState("");
const [selectedFleet, setSelectedFleet] = useState(null);
const [formOpen, setFormOpen] = useState(false);
const [editingFleet, setEditingFleet] = useState(null);
const [deleteTarget, setDeleteTarget] = useState(null);
const [shareTarget, setShareTarget] = useState(null);
const hasLoadedRef = useRef(false);
const loadFleets = useCallback(async () => {
try {
const data = await listFleets();
setFleets(data);
setSelectedFleet((prev) =>
prev ? (data.find((f) => f.id === prev.id) ?? null) : null,
);
} catch (error) {
const message = getErrorMessage(error, "");
toast.error(message || t("newUi.sidebar.fleets.loadFailed"));
}
}, [t]);
useEffect(() => {
if (!active || hasLoadedRef.current) return;
hasLoadedRef.current = true;
setLoading(true);
Promise.all([listFleets(), getSSHHosts({ includeStatus: false })])
.then(([fleetData, hosts]) => {
setFleets(fleetData);
setAllHosts(hosts as unknown as SSHHost[]);
})
.catch((error: unknown) => {
const message = getErrorMessage(error, "");
toast.error(message || t("newUi.sidebar.fleets.loadFailed"));
})
.finally(() => setLoading(false));
}, [active, hasLoadedRef, t]);
async function handleDelete() {
if (!deleteTarget) return;
try {
await deleteFleet(deleteTarget.id);
toast.success(t("newUi.sidebar.fleets.fleetDeleted"));
if (selectedFleet?.id === deleteTarget.id) setSelectedFleet(null);
setDeleteTarget(null);
loadFleets();
} catch (error) {
const message = getErrorMessage(error, "");
toast.error(message || t("newUi.sidebar.fleets.deleteFailed"));
}
}
const visibleFleets = fleets.filter((f) =>
f.name.toLowerCase().includes(search.trim().toLowerCase()),
);
if (selectedFleet) {
return (
setSelectedFleet(null)}
onFleetChanged={loadFleets}
onOpenFleetInventory={onOpenFleetInventory}
/>
);
}
return (
{loading ? (
) : visibleFleets.length === 0 ? (
{t("newUi.sidebar.fleets.noFleets")}
) : (
{visibleFleets.map((fleet) => (
setSelectedFleet(fleet)}
>
{fleet.name}
{fleet.description && (
{fleet.description}
)}
{t("newUi.sidebar.fleets.memberCount", {
count: fleet.memberCount,
})}
))}
)}
setFormOpen(false)}
fleet={editingFleet}
onSaved={loadFleets}
/>
setShareTarget(null)}
fleet={shareTarget}
/>
);
}