feat(mongodb): bson dumps and database backup/restore
This commit is contained in:
1 parent
ecc1135de3
commit
f5673ca0a1
54 files changed
+6355
-106
No files matched your search
@@ -0,0 +1,438 @@
|
||||
<script setup lang="ts">
|
||||
import { computed, nextTick, onBeforeUnmount, ref, watch } from "vue";
|
||||
import { useI18n } from "vue-i18n";
|
||||
import { ArrowLeft, ArrowRight, DatabaseBackup, FolderOpen, Loader2, Play, Square, X } from "@lucide/vue";
|
||||
import { Dialog, DialogFooter, DialogHeader, DialogScrollContent, DialogTitle } from "@/components/ui/dialog";
|
||||
import { Button } from "@/components/ui/button";
|
||||
import { Input } from "@/components/ui/input";
|
||||
import { Label } from "@/components/ui/label";
|
||||
import { Select, SelectContent, SelectItem, SelectTrigger, SelectValue } from "@/components/ui/select";
|
||||
import { useConnectionStore } from "@/stores/connectionStore";
|
||||
import { connectionIsEffectivelyReadOnly } from "@/lib/database/readOnlyWriteAccess";
|
||||
import { executeWithProductionContextGuard } from "@/lib/database/productionExecutionGuard";
|
||||
import { isTauriRuntime } from "@/lib/backend/tauriRuntime";
|
||||
import { uuid } from "@/lib/common/utils";
|
||||
import * as api from "@/lib/backend/api";
|
||||
|
||||
const props = defineProps<{ connectionId: string; database: string; mode: "dump" | "restore" }>();
|
||||
const open = defineModel<boolean>("open", { default: false });
|
||||
const { t } = useI18n();
|
||||
const store = useConnectionStore();
|
||||
const desktop = isTauriRuntime();
|
||||
const restoring = computed(() => props.mode === "restore");
|
||||
const connection = computed(() => store.getConfig(props.connectionId));
|
||||
const readonly = computed(() => restoring.value && connectionIsEffectivelyReadOnly(connection.value));
|
||||
const format = ref<api.MongoDumpFormat>("archive");
|
||||
const gzip = ref(false);
|
||||
const catalog = ref<api.MongoDumpCatalog | null>(null);
|
||||
const sourceRef = ref<string | null>(null);
|
||||
const source = ref<api.MongoDumpSourceInput | null>(null);
|
||||
const sourceName = ref("");
|
||||
const sourceDatabase = ref("");
|
||||
const targetDatabase = ref(props.database);
|
||||
const outputPath = ref(desktop ? "" : `${props.database}.archive`);
|
||||
const selected = ref<string[]>([]);
|
||||
const dropExisting = ref(false);
|
||||
const restoreOptions = ref(true);
|
||||
const restoreIndexes = ref(true);
|
||||
const stopOnError = ref(true);
|
||||
const objcheck = ref(false);
|
||||
const batchSize = ref(500);
|
||||
const loading = ref(false);
|
||||
const confirming = ref(false);
|
||||
const running = ref(false);
|
||||
const cancelling = ref(false);
|
||||
const error = ref("");
|
||||
const step = ref<"configure" | "confirm" | "result">("configure");
|
||||
const progress = ref<api.MongoDatabaseDumpProgress | null>(null);
|
||||
const taskId = ref("");
|
||||
const fileInput = ref<HTMLInputElement | null>(null);
|
||||
let generation = 0;
|
||||
let readController: AbortController | null = null;
|
||||
let uploadController: AbortController | null = null;
|
||||
const readUpload = ref<{ loaded: number; total: number } | null>(null);
|
||||
function gib(value: number) {
|
||||
return (value / 1024 ** 3).toFixed(2);
|
||||
}
|
||||
|
||||
const entries = computed(() => catalog.value?.collections.filter((entry) => !restoring.value || entry.database === sourceDatabase.value) ?? []);
|
||||
const allSelected = computed(() => entries.value.length > 0 && selected.value.length === entries.value.length);
|
||||
const canReview = computed(
|
||||
() =>
|
||||
!!catalog.value &&
|
||||
!loading.value &&
|
||||
!confirming.value &&
|
||||
!running.value &&
|
||||
!readonly.value &&
|
||||
(entries.value.length === 0 || selected.value.length > 0) &&
|
||||
(restoring.value ? !!sourceRef.value && !!targetDatabase.value.trim() && (!catalog.value.databases.length || catalog.value.databases.includes(sourceDatabase.value)) : !!outputPath.value.trim()),
|
||||
);
|
||||
|
||||
function releaseSource() {
|
||||
const reference = sourceRef.value;
|
||||
sourceRef.value = null;
|
||||
if (reference) void api.releaseMongodbRestoreSource(reference);
|
||||
}
|
||||
function invalidate() {
|
||||
generation += 1;
|
||||
releaseSource();
|
||||
catalog.value = null;
|
||||
selected.value = [];
|
||||
error.value = "";
|
||||
step.value = "configure";
|
||||
}
|
||||
function cleanup() {
|
||||
generation += 1;
|
||||
readController?.abort();
|
||||
if (!running.value) releaseSource();
|
||||
}
|
||||
onBeforeUnmount(cleanup);
|
||||
watch(
|
||||
open,
|
||||
async (value) => {
|
||||
if (!value) {
|
||||
cleanup();
|
||||
return;
|
||||
}
|
||||
if (restoring.value) return;
|
||||
const current = ++generation;
|
||||
loading.value = true;
|
||||
try {
|
||||
await store.ensureConnected(props.connectionId);
|
||||
const result = await api.inspectMongodbDatabaseDump(props.connectionId, props.database);
|
||||
if (current !== generation) return;
|
||||
catalog.value = result;
|
||||
selected.value = result.collections.map((entry) => entry.name);
|
||||
} catch (cause) {
|
||||
if (current === generation) error.value = String(cause instanceof Error ? cause.message : cause);
|
||||
} finally {
|
||||
if (current === generation) loading.value = false;
|
||||
}
|
||||
},
|
||||
{ immediate: true },
|
||||
);
|
||||
watch([format, gzip], () => {
|
||||
if (restoring.value) invalidate();
|
||||
else outputPath.value = desktop ? "" : `${props.database}.archive${gzip.value ? ".gz" : ""}`;
|
||||
});
|
||||
watch(format, () => {
|
||||
if (restoring.value) {
|
||||
source.value = null;
|
||||
sourceName.value = "";
|
||||
}
|
||||
});
|
||||
|
||||
function setFormat(value: unknown) {
|
||||
if (value === "directory" || value === "archive") format.value = value;
|
||||
}
|
||||
function setSourceDatabase(value: unknown) {
|
||||
if (typeof value !== "string") return;
|
||||
sourceDatabase.value = value;
|
||||
selected.value = entries.value.map((entry) => entry.name);
|
||||
}
|
||||
function selectAll(event: Event) {
|
||||
selected.value = (event.target as HTMLInputElement).checked ? entries.value.map((entry) => entry.name) : [];
|
||||
}
|
||||
function selectCollection(name: string, event: Event) {
|
||||
selected.value = (event.target as HTMLInputElement).checked ? [...selected.value, name] : selected.value.filter((value) => value !== name);
|
||||
}
|
||||
|
||||
async function chooseSource() {
|
||||
if (!desktop) {
|
||||
fileInput.value?.click();
|
||||
return;
|
||||
}
|
||||
const { open: choose } = await import("@tauri-apps/plugin-dialog");
|
||||
const path = await choose({ directory: format.value === "directory", multiple: false });
|
||||
if (typeof path === "string") {
|
||||
invalidate();
|
||||
source.value = path;
|
||||
sourceName.value = path;
|
||||
}
|
||||
}
|
||||
function chooseFiles(event: Event) {
|
||||
const input = event.target as HTMLInputElement;
|
||||
const files = Array.from(input.files ?? []);
|
||||
input.value = "";
|
||||
if (!files.length) return;
|
||||
invalidate();
|
||||
source.value = format.value === "directory" ? files : files[0]!;
|
||||
sourceName.value = format.value === "directory" ? `${(files[0]!.webkitRelativePath || files[0]!.name).split("/")[0]} (${files.length})` : files[0]!.name;
|
||||
}
|
||||
async function readSource() {
|
||||
if (!source.value) return;
|
||||
releaseSource();
|
||||
const current = ++generation;
|
||||
loading.value = true;
|
||||
error.value = "";
|
||||
catalog.value = null;
|
||||
readUpload.value = null;
|
||||
readController = new AbortController();
|
||||
try {
|
||||
const result = await api.prepareMongodbRestoreSource(source.value, format.value, gzip.value, {
|
||||
signal: readController.signal,
|
||||
onUploadProgress: (loaded, total) => {
|
||||
readUpload.value = { loaded, total };
|
||||
},
|
||||
});
|
||||
if (current !== generation) {
|
||||
void api.releaseMongodbRestoreSource(result.sourceRef);
|
||||
return;
|
||||
}
|
||||
sourceRef.value = result.sourceRef;
|
||||
catalog.value = result;
|
||||
setSourceDatabase(result.databases.length === 1 ? result.databases[0] : "");
|
||||
} catch (cause) {
|
||||
if (current === generation) error.value = String(cause instanceof Error ? cause.message : cause);
|
||||
} finally {
|
||||
if (current === generation) loading.value = false;
|
||||
readController = null;
|
||||
}
|
||||
}
|
||||
async function chooseDestination() {
|
||||
const { open: choose, save } = await import("@tauri-apps/plugin-dialog");
|
||||
if (format.value === "directory") {
|
||||
const parent = await choose({ directory: true, multiple: false });
|
||||
if (typeof parent === "string") outputPath.value = `${parent.replace(/[\\/]$/, "")}/${props.database}-dump`;
|
||||
} else {
|
||||
const path = await save({ defaultPath: `${props.database}.archive${gzip.value ? ".gz" : ""}`, filters: [{ name: "MongoDB archive", extensions: [gzip.value ? "gz" : "archive"] }] });
|
||||
if (path) outputPath.value = path;
|
||||
}
|
||||
}
|
||||
|
||||
async function refreshDatabase() {
|
||||
await store.loadMongoDatabases(props.connectionId);
|
||||
await store.loadMongoCollections(props.connectionId, targetDatabase.value);
|
||||
for (const collection of selected.value) {
|
||||
store.mongoImportCompleted = { connectionId: props.connectionId, database: targetDatabase.value, collection, at: Date.now() };
|
||||
await nextTick();
|
||||
}
|
||||
}
|
||||
async function execute() {
|
||||
if (!canReview.value) return;
|
||||
confirming.value = true;
|
||||
const current = generation;
|
||||
try {
|
||||
targetDatabase.value = targetDatabase.value.trim();
|
||||
if (restoring.value) {
|
||||
const allowed = await executeWithProductionContextGuard({
|
||||
connection: connection.value,
|
||||
database: targetDatabase.value.trim(),
|
||||
source: t("mongoDump.restoreTitle"),
|
||||
reviewText: `${dropExisting.value ? "DROP selected collections; " : ""}RESTORE ${JSON.stringify(sourceDatabase.value)} INTO ${JSON.stringify(targetDatabase.value.trim())}: ${selected.value.map((name) => JSON.stringify(name)).join(", ")}`,
|
||||
execute: async () => true,
|
||||
});
|
||||
if (!allowed) return;
|
||||
}
|
||||
if (current !== generation || !open.value) return;
|
||||
running.value = true;
|
||||
confirming.value = false;
|
||||
step.value = "result";
|
||||
error.value = "";
|
||||
progress.value = null;
|
||||
taskId.value = uuid();
|
||||
const onProgress = (value: api.MongoDatabaseDumpProgress) => {
|
||||
progress.value = value;
|
||||
};
|
||||
await store.ensureConnected(props.connectionId);
|
||||
if (restoring.value) {
|
||||
uploadController = new AbortController();
|
||||
const selectedPaths = new Set(entries.value.filter((entry) => selected.value.includes(entry.name)).flatMap((entry) => entry.sourceFiles ?? []));
|
||||
const upload = !desktop && format.value === "directory" && Array.isArray(source.value) ? { files: source.value.filter((file) => selectedPaths.has(file.webkitRelativePath || file.name)), gzip: gzip.value, signal: uploadController.signal } : undefined;
|
||||
progress.value = await api.restoreMongodbDatabase(
|
||||
{
|
||||
taskId: taskId.value,
|
||||
connectionId: props.connectionId,
|
||||
database: targetDatabase.value.trim(),
|
||||
sourceDatabase: sourceDatabase.value,
|
||||
sourceRef: sourceRef.value!,
|
||||
collections: [...selected.value],
|
||||
dropExisting: dropExisting.value,
|
||||
restoreOptions: restoreOptions.value,
|
||||
restoreIndexes: restoreIndexes.value,
|
||||
stopOnError: stopOnError.value,
|
||||
objcheck: objcheck.value,
|
||||
batchSize: Math.max(100, Math.min(5000, Math.trunc(Number(batchSize.value)) || 500)),
|
||||
},
|
||||
onProgress,
|
||||
upload,
|
||||
);
|
||||
void refreshDatabase().catch((cause) => console.error("Refresh after MongoDB restore failed", cause));
|
||||
} else {
|
||||
progress.value = await api.dumpMongodbDatabase({ taskId: taskId.value, connectionId: props.connectionId, database: props.database, filePath: outputPath.value.trim(), format: format.value, gzip: gzip.value, collections: [...selected.value] }, onProgress);
|
||||
}
|
||||
} catch (cause) {
|
||||
error.value = String(cause instanceof Error ? cause.message : cause);
|
||||
if (cause instanceof DOMException && cause.name === "AbortError") {
|
||||
progress.value = { taskId: taskId.value, status: "cancelled", phase: "done", collection: null, collectionsDone: 0, collectionsTotal: selected.value.length, documentsRead: 0, documentsWritten: 0, documentsFailed: 0, indexesCreated: 0, elapsedMs: 0, errorMessage: null, filePath: null };
|
||||
error.value = "";
|
||||
}
|
||||
} finally {
|
||||
confirming.value = false;
|
||||
running.value = false;
|
||||
cancelling.value = false;
|
||||
uploadController = null;
|
||||
}
|
||||
}
|
||||
async function cancel() {
|
||||
cancelling.value = true;
|
||||
uploadController?.abort();
|
||||
try {
|
||||
await api.cancelMongodbDatabaseDump(taskId.value);
|
||||
} catch (cause) {
|
||||
cancelling.value = false;
|
||||
error.value = String(cause instanceof Error ? cause.message : cause);
|
||||
}
|
||||
}
|
||||
function close() {
|
||||
if (!running.value && !confirming.value) open.value = false;
|
||||
}
|
||||
</script>
|
||||
|
||||
<template>
|
||||
<Dialog
|
||||
:open="open"
|
||||
@update:open="
|
||||
(value) => {
|
||||
if (!value) close();
|
||||
}
|
||||
"
|
||||
>
|
||||
<DialogScrollContent class="flex max-h-[calc(var(--dbx-viewport-height)-4rem)] min-h-0 flex-col overflow-hidden sm:max-w-[860px]">
|
||||
<DialogHeader class="shrink-0 pr-8"
|
||||
><DialogTitle class="flex items-center gap-2 text-base"><DatabaseBackup class="h-4 w-4" />{{ t(restoring ? "mongoDump.restoreTitle" : "mongoDump.dumpTitle") }}</DialogTitle></DialogHeader
|
||||
>
|
||||
<div class="min-h-0 flex-1 space-y-4 overflow-y-auto py-3 text-sm">
|
||||
<div class="border-b pb-2 font-medium break-all">{{ connection?.name }} / {{ props.database }}</div>
|
||||
<div v-if="readonly" class="text-xs text-amber-600">{{ t("mongoDump.readonly") }}</div>
|
||||
<template v-if="step === 'configure'">
|
||||
<div class="grid grid-cols-1 gap-3 sm:grid-cols-2">
|
||||
<div class="space-y-1">
|
||||
<Label>{{ t("mongoDump.format") }}</Label
|
||||
><Select :model-value="format" :disabled="loading" @update:model-value="setFormat"
|
||||
><SelectTrigger class="h-8"><SelectValue /></SelectTrigger
|
||||
><SelectContent
|
||||
><SelectItem value="archive">{{ t("mongoDump.archive") }}</SelectItem
|
||||
><SelectItem v-if="desktop || restoring" value="directory">{{ t("mongoDump.directory") }}</SelectItem></SelectContent
|
||||
></Select
|
||||
>
|
||||
</div>
|
||||
<label class="flex min-h-8 items-center gap-2 self-end"><input v-model="gzip" type="checkbox" :disabled="loading" />{{ t("mongoDump.gzip") }}</label>
|
||||
</div>
|
||||
<div v-if="restoring" class="space-y-2">
|
||||
<Label>{{ t("mongoDump.source") }}</Label>
|
||||
<input ref="fileInput" type="file" class="hidden" :multiple="format === 'directory'" :webkitdirectory="format === 'directory' ? '' : undefined" @change="chooseFiles" />
|
||||
<div class="flex flex-wrap items-center gap-2">
|
||||
<div class="min-w-0 flex-1 basis-full break-all text-xs sm:basis-0">{{ sourceName || t("tableImport.noFileSelected") }}</div>
|
||||
<Button variant="outline" size="sm" :disabled="loading" @click="chooseSource"><FolderOpen class="mr-1.5 h-4 w-4" />{{ t("tableImport.selectFile") }}</Button
|
||||
><Button size="sm" :disabled="!source || loading" @click="readSource"><Loader2 v-if="loading" class="mr-1.5 h-4 w-4 animate-spin" /><Play v-else class="mr-1.5 h-4 w-4" />{{ t("mongoDump.readSource") }}</Button>
|
||||
</div>
|
||||
<div class="grid grid-cols-1 gap-3 sm:grid-cols-2">
|
||||
<div class="space-y-1">
|
||||
<Label>{{ t("mongoDump.sourceDatabase") }}</Label
|
||||
><Select :model-value="sourceDatabase" :disabled="!catalog || loading" @update:model-value="setSourceDatabase"
|
||||
><SelectTrigger class="h-8 w-full min-w-0 [&>span]:min-w-0 [&>span]:truncate"><SelectValue /></SelectTrigger
|
||||
><SelectContent
|
||||
><SelectItem v-for="db in catalog?.databases ?? []" :key="db" :value="db">{{ db }}</SelectItem></SelectContent
|
||||
></Select
|
||||
>
|
||||
</div>
|
||||
<div class="space-y-1">
|
||||
<Label>{{ t("mongoDump.targetDatabase") }}</Label
|
||||
><Input v-model="targetDatabase" class="h-8" :disabled="loading" />
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
<div v-else class="space-y-1">
|
||||
<Label>{{ t("mongoDump.destination") }}</Label>
|
||||
<div class="flex gap-2">
|
||||
<Input v-model="outputPath" class="h-8 min-w-0" :readonly="desktop" /><Button v-if="desktop" variant="outline" size="icon" class="h-8 w-8 shrink-0" :title="t('mongoDump.chooseDestination')" :aria-label="t('mongoDump.chooseDestination')" @click="chooseDestination"
|
||||
><FolderOpen class="h-4 w-4"
|
||||
/></Button>
|
||||
</div>
|
||||
</div>
|
||||
<div v-if="loading" class="flex items-center gap-2 text-xs text-muted-foreground"><Loader2 class="h-4 w-4 animate-spin" />{{ t(restoring ? "mongoDump.preparing" : "mongoDump.loading") }}</div>
|
||||
<div v-if="loading && readUpload" class="text-xs tabular-nums">{{ t("mongoDump.uploadBytes", { done: gib(readUpload.loaded), total: gib(readUpload.total) }) }}</div>
|
||||
<div v-if="catalog" class="space-y-2">
|
||||
<div class="text-xs text-muted-foreground">{{ t("mongoDump.selected", { count: selected.length }) }}</div>
|
||||
<div class="max-h-64 overflow-auto border-y">
|
||||
<table class="w-full min-w-[480px] table-fixed text-xs">
|
||||
<thead class="sticky top-0 bg-background">
|
||||
<tr>
|
||||
<th class="w-8 p-2"><input type="checkbox" :checked="allSelected" :aria-label="t('mongoDump.selectAll')" @change="selectAll" /></th>
|
||||
<th class="w-2/5 p-2 text-left">{{ t("mongoDump.collection") }}</th>
|
||||
<th class="p-2 text-left">{{ t("mongoDump.kind") }}</th>
|
||||
<th class="p-2 text-right">{{ t("mongoDump.documents") }}</th>
|
||||
<th class="p-2 text-right">{{ t("mongoDump.indexes") }}</th>
|
||||
</tr>
|
||||
</thead>
|
||||
<tbody>
|
||||
<tr v-for="entry in entries" :key="entry.name" class="border-t">
|
||||
<td class="p-2"><input type="checkbox" :checked="selected.includes(entry.name)" :aria-label="entry.name" @change="selectCollection(entry.name, $event)" /></td>
|
||||
<td class="p-2 break-all">{{ entry.name }}</td>
|
||||
<td class="p-2 break-all">{{ entry.kind }}</td>
|
||||
<td class="p-2 text-right">{{ entry.documents ?? "-" }}</td>
|
||||
<td class="p-2 text-right">{{ entry.indexes }}</td>
|
||||
</tr>
|
||||
<tr v-if="!entries.length">
|
||||
<td colspan="5" class="p-4 text-center text-muted-foreground">{{ t("mongoDump.empty") }}</td>
|
||||
</tr>
|
||||
</tbody>
|
||||
</table>
|
||||
</div>
|
||||
</div>
|
||||
<div v-if="restoring" class="space-y-3 border-t pt-3">
|
||||
<label class="flex items-center gap-2"><input v-model="dropExisting" type="checkbox" />{{ t("mongoDump.dropExisting") }}</label>
|
||||
<div class="grid grid-cols-1 gap-3 sm:grid-cols-2">
|
||||
<label class="flex items-center gap-2"><input v-model="restoreOptions" type="checkbox" />{{ t("mongoDump.restoreOptions") }}</label
|
||||
><label class="flex items-center gap-2"><input v-model="restoreIndexes" type="checkbox" />{{ t("mongoDump.restoreIndexes") }}</label
|
||||
><label class="flex items-center gap-2"><input v-model="stopOnError" type="checkbox" />{{ t("mongoDump.stopOnError") }}</label>
|
||||
<label class="flex items-center gap-2"><input v-model="objcheck" type="checkbox" />{{ t("mongoDump.objcheck") }}</label>
|
||||
<div class="flex items-center gap-2">
|
||||
<Label class="shrink-0">{{ t("mongoDump.batchSize") }}</Label
|
||||
><Input v-model.number="batchSize" type="number" min="100" max="5000" class="h-8 w-24" />
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
</template>
|
||||
<template v-else-if="step === 'confirm'"
|
||||
><p>{{ t(restoring ? "mongoDump.confirmRestore" : "mongoDump.confirmDump", { count: selected.length, database: restoring ? targetDatabase : props.database }) }}</p>
|
||||
<p v-if="restoring" :class="dropExisting ? 'text-destructive' : 'text-muted-foreground'">{{ t(dropExisting ? "mongoDump.confirmDrop" : "mongoDump.confirmAppend") }}</p>
|
||||
<div class="max-h-56 overflow-auto border-y py-2 text-xs">
|
||||
<div v-for="name in selected" :key="name" class="py-1 break-all">{{ name }}</div>
|
||||
</div>
|
||||
<div v-if="!restoring" class="break-all text-xs text-muted-foreground">{{ outputPath }}</div></template
|
||||
>
|
||||
<template v-else
|
||||
><div class="flex items-center gap-2 font-medium">
|
||||
<Loader2 v-if="running" class="h-4 w-4 animate-spin" />{{ running ? t(`mongoDump.phase.${progress?.phase ?? "preparing"}`) : t(progress?.status === "done" ? "mongoDump.done" : progress?.status === "cancelled" ? "mongoDump.cancelled" : "mongoDump.failed") }}
|
||||
</div>
|
||||
<div class="break-all text-xs">{{ progress?.collection }}</div>
|
||||
<div v-if="progress?.phase === 'uploading'" class="text-xs tabular-nums">{{ t("mongoDump.uploadBytes", { done: gib(progress.bytesProcessed ?? 0), total: gib(progress.bytesTotal ?? 0) }) }}</div>
|
||||
<div v-if="restoring && progress?.phase === 'data'" class="text-xs tabular-nums">{{ t("mongoDump.readBytes", { size: gib(progress.bytesProcessed ?? 0) }) }}</div>
|
||||
<div v-if="progress?.phase === 'validating'" class="text-xs tabular-nums">{{ t("mongoDump.validationBytes", { size: gib(progress.bytesProcessed ?? 0), count: progress.documentsValidated ?? 0 }) }}</div>
|
||||
<progress class="h-2 w-full" :value="progress?.collectionsDone ?? 0" :max="Math.max(1, progress?.collectionsTotal ?? selected.length)" />
|
||||
<div class="text-xs">{{ t("mongoDump.progress", { done: progress?.collectionsDone ?? 0, total: progress?.collectionsTotal ?? selected.length }) }}</div>
|
||||
<dl class="grid grid-cols-2 gap-3 border-y py-3 text-xs">
|
||||
<dt>{{ t("mongoDump.read") }}</dt>
|
||||
<dd class="text-right tabular-nums">{{ progress?.documentsRead ?? 0 }}</dd>
|
||||
<dt v-if="restoring">{{ t("mongoDump.written") }}</dt>
|
||||
<dd v-if="restoring" class="text-right tabular-nums">{{ progress?.documentsWritten ?? 0 }}</dd>
|
||||
<dt>{{ t("mongoDump.rejected") }}</dt>
|
||||
<dd class="text-right tabular-nums">{{ progress?.documentsFailed ?? 0 }}</dd>
|
||||
<dt v-if="restoring">{{ t("mongoDump.createdIndexes") }}</dt>
|
||||
<dd v-if="restoring" class="text-right tabular-nums">{{ progress?.indexesCreated ?? 0 }}</dd>
|
||||
</dl></template
|
||||
>
|
||||
<div v-if="error" class="break-words border-l-2 border-destructive pl-3 text-xs text-destructive" role="alert">{{ error }}</div>
|
||||
</div>
|
||||
<DialogFooter class="shrink-0"
|
||||
><Button v-if="running" variant="destructive" :disabled="cancelling" @click="cancel"><Square class="mr-1.5 h-4 w-4" />{{ t("mongoDump.cancel") }}</Button
|
||||
><Button v-else variant="outline" :disabled="confirming" @click="close"><X class="mr-1.5 h-4 w-4" />{{ t("mongoDump.close") }}</Button
|
||||
><Button v-if="step !== 'configure' && !running" variant="outline" :disabled="confirming" @click="step = 'configure'"><ArrowLeft class="mr-1.5 h-4 w-4" />{{ t("mongoDump.back") }}</Button
|
||||
><Button v-if="step === 'configure'" :disabled="!canReview" @click="step = 'confirm'"><ArrowRight class="mr-1.5 h-4 w-4" />{{ t("mongoDump.review") }}</Button
|
||||
><Button v-if="step === 'confirm'" :disabled="!canReview" @click="execute"><Loader2 v-if="confirming" class="mr-1.5 h-4 w-4 animate-spin" /><Play v-else class="mr-1.5 h-4 w-4" />{{ t(restoring ? "mongoDump.restore" : "mongoDump.dump") }}</Button></DialogFooter
|
||||
>
|
||||
</DialogScrollContent>
|
||||
</Dialog>
|
||||
</template>
|
||||
@@ -1,5 +1,5 @@
|
||||
<script setup lang="ts">
|
||||
import { computed, ref, watch } from "vue";
|
||||
import { computed, onBeforeUnmount, ref, watch } from "vue";
|
||||
import { uuid } from "@/lib/common/utils";
|
||||
import { useI18n } from "vue-i18n";
|
||||
import { isTauriRuntime } from "@/lib/backend/tauriRuntime";
|
||||
@@ -14,7 +14,7 @@ import { useConnectionStore } from "@/stores/connectionStore";
|
||||
import { connectionIsEffectivelyReadOnly } from "@/lib/database/readOnlyWriteAccess";
|
||||
import { executeWithProductionContextGuard } from "@/lib/database/productionExecutionGuard";
|
||||
import { TABLE_IMPORT_ENCODING_OPTIONS } from "@/lib/table/tableImport";
|
||||
import { importPreviewInput, importSourceDisplayName, importTextDelimiterForName, uploadedImportSourceFromPreview, type UploadedImportSource } from "@/lib/import/importSource";
|
||||
import { importPreviewInput, importSourceDisplayName, uploadedImportSourceFromPreview, type UploadedImportSource } from "@/lib/import/importSource";
|
||||
import { useToast } from "@/composables/useToast";
|
||||
import { translateBackendError } from "@/i18n/backend-errors";
|
||||
import * as api from "@/lib/backend/api";
|
||||
@@ -93,13 +93,6 @@ const parseOptions = computed((): api.MongoImportParseOptions => {
|
||||
};
|
||||
});
|
||||
|
||||
function formatFromName(name: string): api.MongoImportFormat {
|
||||
const lower = name.toLowerCase();
|
||||
if (lower.endsWith(".ndjson") || lower.endsWith(".jsonl")) return "ndjson";
|
||||
if (lower.endsWith(".json")) return "json";
|
||||
return "csv";
|
||||
}
|
||||
|
||||
function releasePreviewSource() {
|
||||
const sourceRef = uploadedSource?.sourceRef;
|
||||
uploadedSource = null;
|
||||
@@ -111,10 +104,7 @@ function assignSource(source: string | File) {
|
||||
releasePreviewSource();
|
||||
selectedSource.value = source;
|
||||
sourceName.value = importSourceDisplayName(source);
|
||||
format.value = formatFromName(sourceName.value);
|
||||
typeMode.value = format.value === "csv" ? "auto" : "extendedJson";
|
||||
columnTypeOverrides.value = {};
|
||||
delimiter.value = importTextDelimiterForName(sourceName.value);
|
||||
wizardStep.value = "source";
|
||||
queuePreview();
|
||||
}
|
||||
@@ -127,7 +117,7 @@ async function selectFile() {
|
||||
const { open: openDialog } = await import("@tauri-apps/plugin-dialog");
|
||||
const selected = await openDialog({
|
||||
multiple: false,
|
||||
filters: [{ name: t("mongo.import.fileFilter"), extensions: ["csv", "json", "ndjson", "jsonl", "tsv"] }],
|
||||
filters: [{ name: t("mongo.import.fileFilter"), extensions: ["bson", "gz", "csv", "json", "ndjson", "jsonl", "tsv"] }],
|
||||
});
|
||||
if (!selected || Array.isArray(selected)) return;
|
||||
assignSource(selected);
|
||||
@@ -143,15 +133,17 @@ function handleFileInputChange(event: Event) {
|
||||
function queuePreview() {
|
||||
if (!selectedSource.value) return;
|
||||
if (previewReloadTimer) clearTimeout(previewReloadTimer);
|
||||
const requestId = ++previewRequestId;
|
||||
loadingPreview.value = true;
|
||||
previewReloadTimer = setTimeout(() => {
|
||||
void loadPreview();
|
||||
previewReloadTimer = null;
|
||||
void loadPreview(requestId);
|
||||
}, 200);
|
||||
}
|
||||
|
||||
async function loadPreview() {
|
||||
async function loadPreview(requestId: number) {
|
||||
const source = selectedSource.value;
|
||||
if (!source) return;
|
||||
const requestId = ++previewRequestId;
|
||||
const keepExistingPreview = !!preview.value;
|
||||
if (!keepExistingPreview) {
|
||||
loadingPreview.value = true;
|
||||
@@ -165,7 +157,10 @@ async function loadPreview() {
|
||||
previewLimit: previewLimit.value,
|
||||
sourceRef: input.sourceRef,
|
||||
});
|
||||
if (requestId !== previewRequestId) return;
|
||||
if (requestId !== previewRequestId) {
|
||||
if (next.sourceRef && next.sourceRef !== uploadedSource?.sourceRef) void api.releaseMongodbImportSource(next.sourceRef);
|
||||
return;
|
||||
}
|
||||
preview.value = next;
|
||||
previewError.value = "";
|
||||
uploadedSource = uploadedImportSourceFromPreview(next) ?? uploadedSource;
|
||||
@@ -178,13 +173,21 @@ async function loadPreview() {
|
||||
}
|
||||
}
|
||||
|
||||
watch([format, hasHeader, typeMode], () => {
|
||||
if (Object.keys(columnTypeOverrides.value).length) columnTypeOverrides.value = {};
|
||||
});
|
||||
watch(
|
||||
[format, hasHeader, typeMode],
|
||||
() => {
|
||||
if (Object.keys(columnTypeOverrides.value).length) columnTypeOverrides.value = {};
|
||||
},
|
||||
{ flush: "sync" },
|
||||
);
|
||||
|
||||
watch([format, encoding, delimiter, hasHeader, trim, emptyAsNull, typeMode, recognizeObjectIdHex, previewLimit, columnTypeOverrides], () => {
|
||||
if (selectedSource.value) queuePreview();
|
||||
});
|
||||
watch(
|
||||
[format, encoding, delimiter, hasHeader, trim, emptyAsNull, typeMode, recognizeObjectIdHex, previewLimit, columnTypeOverrides],
|
||||
() => {
|
||||
if (selectedSource.value) queuePreview();
|
||||
},
|
||||
{ flush: "sync" },
|
||||
);
|
||||
|
||||
watch(open, (value) => {
|
||||
if (value) return;
|
||||
@@ -197,6 +200,10 @@ watch(open, (value) => {
|
||||
});
|
||||
|
||||
function reset() {
|
||||
previewRequestId += 1;
|
||||
if (previewReloadTimer) clearTimeout(previewReloadTimer);
|
||||
previewReloadTimer = null;
|
||||
loadingPreview.value = false;
|
||||
releasePreviewSource();
|
||||
selectedSource.value = null;
|
||||
sourceName.value = "";
|
||||
@@ -211,8 +218,12 @@ function reset() {
|
||||
importId.value = "";
|
||||
}
|
||||
|
||||
onBeforeUnmount(() => {
|
||||
if (!running.value) reset();
|
||||
});
|
||||
|
||||
function canConfirm() {
|
||||
return !!preview.value && !loadingPreview.value && !previewError.value && !effectivelyReadOnly.value && (!preview.value.errors.length || skipErrorRows.value);
|
||||
return !!preview.value && preview.value.format === format.value && !loadingPreview.value && !previewError.value && !effectivelyReadOnly.value && (!preview.value.errors.length || skipErrorRows.value);
|
||||
}
|
||||
|
||||
async function startImport() {
|
||||
@@ -287,7 +298,7 @@ function formatPreviewCell(value: unknown) {
|
||||
}
|
||||
|
||||
function setFormat(value: unknown) {
|
||||
if (value === "csv" || value === "json" || value === "ndjson") format.value = value;
|
||||
if (value === "csv" || value === "json" || value === "ndjson" || value === "bson") format.value = value;
|
||||
}
|
||||
|
||||
function setEncoding(value: unknown) {
|
||||
@@ -343,7 +354,7 @@ function requestClose() {
|
||||
</DialogHeader>
|
||||
<div class="min-h-0 flex-1 space-y-4 overflow-y-auto py-2 pr-1 text-sm">
|
||||
<div class="grid grid-cols-[minmax(0,1fr)_auto] items-center gap-2">
|
||||
<input ref="fileInput" type="file" class="hidden" accept=".csv,.json,.ndjson,.jsonl,.tsv" @change="handleFileInputChange" />
|
||||
<input ref="fileInput" type="file" class="hidden" accept=".bson,.bson.gz,.csv,.json,.ndjson,.jsonl,.tsv" @change="handleFileInputChange" />
|
||||
<div class="flex h-10 min-w-0 items-center gap-2 rounded-md border bg-muted/20 px-3">
|
||||
<span class="shrink-0 text-xs text-muted-foreground">{{ t("mongo.import.target") }}</span>
|
||||
<span class="min-w-0 truncate text-sm font-medium">{{ targetLabel }}</span>
|
||||
@@ -366,10 +377,11 @@ function requestClose() {
|
||||
<SelectItem value="csv">CSV</SelectItem>
|
||||
<SelectItem value="json">JSON</SelectItem>
|
||||
<SelectItem value="ndjson">NDJSON</SelectItem>
|
||||
<SelectItem value="bson">BSON dump</SelectItem>
|
||||
</SelectContent>
|
||||
</Select>
|
||||
</div>
|
||||
<div class="space-y-1.5">
|
||||
<div v-if="format !== 'bson'" class="space-y-1.5">
|
||||
<Label class="text-xs">{{ t("tableImport.encoding") }}</Label>
|
||||
<Select :model-value="encoding" @update:model-value="setEncoding">
|
||||
<SelectTrigger class="h-8 text-xs"><SelectValue /></SelectTrigger>
|
||||
@@ -382,7 +394,7 @@ function requestClose() {
|
||||
<Label class="text-xs">{{ t("tableImport.delimiter") }}</Label>
|
||||
<Input v-model="delimiter" class="h-8 text-xs font-mono" :aria-label="t('tableImport.delimiter')" />
|
||||
</div>
|
||||
<div class="space-y-1.5">
|
||||
<div v-if="format !== 'bson'" class="space-y-1.5">
|
||||
<Label class="text-xs">{{ t("mongo.import.typeMode") }}</Label>
|
||||
<Select :model-value="typeMode" :aria-label="t('mongo.import.typeMode')" @update:model-value="setTypeMode">
|
||||
<SelectTrigger class="h-8 text-xs"><SelectValue /></SelectTrigger>
|
||||
@@ -405,13 +417,14 @@ function requestClose() {
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div class="grid grid-cols-2 gap-x-4 gap-y-2 rounded-md border p-3">
|
||||
<div v-if="format !== 'bson'" class="grid grid-cols-2 gap-x-4 gap-y-2 rounded-md border p-3">
|
||||
<label class="flex items-center gap-2 text-xs"><input v-model="hasHeader" type="checkbox" class="h-3.5 w-3.5 accent-primary" :aria-label="t('tableImport.hasHeader')" /> {{ t("tableImport.hasHeader") }}</label>
|
||||
<label class="flex items-center gap-2 text-xs"><input v-model="trim" type="checkbox" class="h-3.5 w-3.5 accent-primary" /> {{ t("tableImport.trimValues") }}</label>
|
||||
<label class="flex items-center gap-2 text-xs"><input v-model="emptyAsNull" type="checkbox" class="h-3.5 w-3.5 accent-primary" /> {{ t("mongo.import.emptyAsNull") }}</label>
|
||||
<label class="flex items-center gap-2 text-xs"><input v-model="recognizeObjectIdHex" type="checkbox" class="h-3.5 w-3.5 accent-primary" /> {{ t("mongo.import.recognizeObjectIdHex") }}</label>
|
||||
<label class="flex items-center gap-2 text-xs"><input v-model="skipErrorRows" type="checkbox" class="h-3.5 w-3.5 accent-primary" /> {{ t("mongo.import.skipErrorRows") }}</label>
|
||||
</div>
|
||||
<label v-else class="flex items-center gap-2 text-xs"><input v-model="skipErrorRows" type="checkbox" class="h-3.5 w-3.5 accent-primary" /> {{ t("mongo.import.skipErrorRows") }}</label>
|
||||
|
||||
<div v-if="loadingPreview && !preview" class="flex items-center gap-2 text-xs text-muted-foreground">
|
||||
<Loader2 class="h-3.5 w-3.5 animate-spin" />
|
||||
@@ -438,7 +451,7 @@ function requestClose() {
|
||||
check-position="right"
|
||||
@update:model-value="(value) => setColumnType(column.name, value)"
|
||||
/>
|
||||
<span v-else-if="format !== 'csv'" class="text-muted-foreground">({{ column.inferredType }})</span>
|
||||
<span v-else-if="format !== 'csv' && format !== 'bson'" class="text-muted-foreground">({{ column.inferredType }})</span>
|
||||
</th>
|
||||
</tr>
|
||||
</thead>
|
||||
|
||||
+233
@@ -0,0 +1,233 @@
|
||||
// @vitest-environment happy-dom
|
||||
import { createApp, nextTick, type App } from "vue";
|
||||
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
||||
|
||||
const mocks = vi.hoisted(() => ({
|
||||
api: { inspectMongodbDatabaseDump: vi.fn(), prepareMongodbRestoreSource: vi.fn(), releaseMongodbRestoreSource: vi.fn(), dumpMongodbDatabase: vi.fn(), restoreMongodbDatabase: vi.fn(), cancelMongodbDatabaseDump: vi.fn() },
|
||||
store: { getConfig: vi.fn(), ensureConnected: vi.fn(), loadMongoDatabases: vi.fn(), loadMongoCollections: vi.fn(), mongoImportCompleted: null },
|
||||
guard: vi.fn(),
|
||||
}));
|
||||
vi.mock("@/lib/backend/api", () => mocks.api);
|
||||
vi.mock("@/lib/backend/tauriRuntime", () => ({ isTauriRuntime: () => false }));
|
||||
vi.mock("@/stores/connectionStore", () => ({ useConnectionStore: () => mocks.store }));
|
||||
vi.mock("vue-i18n", async (original) => ({ ...(await original<typeof import("vue-i18n")>()), useI18n: () => ({ t: (key: string) => key }) }));
|
||||
vi.mock("@/lib/database/productionExecutionGuard", () => ({ executeWithProductionContextGuard: mocks.guard }));
|
||||
import MongoDatabaseDumpDialog from "../MongoDatabaseDumpDialog.vue";
|
||||
|
||||
const catalog = {
|
||||
databases: ["source"],
|
||||
collections: [
|
||||
{ database: "source", name: "records", kind: "collection", documents: 2, sizeBytes: 30, indexes: 2 },
|
||||
{ database: "source", name: "empty", kind: "collection", documents: 0, sizeBytes: 0, indexes: 0 },
|
||||
],
|
||||
};
|
||||
const done = { taskId: "task", status: "done", phase: "done", collection: "records", collectionsDone: 2, collectionsTotal: 2, documentsRead: 2, documentsWritten: 2, documentsFailed: 0, indexesCreated: 2, elapsedMs: 10, errorMessage: null, filePath: null };
|
||||
let app: App | null;
|
||||
async function flush() {
|
||||
for (let i = 0; i < 5; i++) {
|
||||
await Promise.resolve();
|
||||
await nextTick();
|
||||
}
|
||||
}
|
||||
function button(key: string): HTMLButtonElement {
|
||||
const button = Array.from(document.querySelectorAll("button")).find((button) => button.textContent?.trim() === key);
|
||||
if (!button) throw new Error(`Missing button ${key}`);
|
||||
return button;
|
||||
}
|
||||
function checkbox(key: string): HTMLInputElement {
|
||||
const label = Array.from(document.querySelectorAll("label")).find((label) => label.textContent?.trim() === key);
|
||||
const checkbox = label?.querySelector<HTMLInputElement>("input[type=checkbox]");
|
||||
if (!checkbox) throw new Error(`Missing checkbox ${key}`);
|
||||
return checkbox;
|
||||
}
|
||||
async function mount(mode: "dump" | "restore") {
|
||||
app = createApp(MongoDatabaseDumpDialog, { open: true, connectionId: "c1", database: "target", mode });
|
||||
app.mount(document.body);
|
||||
await flush();
|
||||
}
|
||||
async function filesSelected(files: File[]) {
|
||||
const input = document.querySelector<HTMLInputElement>("input[type=file]")!;
|
||||
Object.defineProperty(input, "files", { configurable: true, value: files });
|
||||
input.dispatchEvent(new Event("change"));
|
||||
await flush();
|
||||
}
|
||||
async function formatSelected(value: string) {
|
||||
document.querySelector<HTMLElement>("[role=combobox]")!.dispatchEvent(new KeyboardEvent("keydown", { key: "ArrowDown", bubbles: true, cancelable: true }));
|
||||
await vi.advanceTimersByTimeAsync(10);
|
||||
await flush();
|
||||
const option = Array.from(document.querySelectorAll<HTMLElement>("[role=option]")).find((option) => option.textContent?.trim() === value)!;
|
||||
option.focus();
|
||||
option.dispatchEvent(new KeyboardEvent("keydown", { key: "Enter", bubbles: true, cancelable: true }));
|
||||
await vi.advanceTimersByTimeAsync(10);
|
||||
await flush();
|
||||
}
|
||||
|
||||
describe("MongoDatabaseDumpDialog", () => {
|
||||
beforeEach(() => {
|
||||
vi.useFakeTimers();
|
||||
vi.clearAllMocks();
|
||||
mocks.store.getConfig.mockReturnValue({ name: "MongoDB", read_only: false });
|
||||
mocks.store.ensureConnected.mockResolvedValue(undefined);
|
||||
mocks.api.inspectMongodbDatabaseDump.mockResolvedValue(catalog);
|
||||
mocks.api.prepareMongodbRestoreSource.mockResolvedValue({ ...catalog, sourceRef: "prepared-one" });
|
||||
mocks.api.releaseMongodbRestoreSource.mockResolvedValue(true);
|
||||
mocks.api.dumpMongodbDatabase.mockResolvedValue(done);
|
||||
mocks.api.restoreMongodbDatabase.mockResolvedValue(done);
|
||||
mocks.guard.mockImplementation(async ({ execute }) => execute());
|
||||
});
|
||||
afterEach(() => {
|
||||
app?.unmount();
|
||||
app = null;
|
||||
document.body.innerHTML = "";
|
||||
vi.useRealTimers();
|
||||
});
|
||||
|
||||
it("exports selected collections using the explicit archive and gzip options", async () => {
|
||||
await mount("dump");
|
||||
checkbox("mongoDump.gzip").click();
|
||||
document.querySelector<HTMLInputElement>('input[aria-label="empty"]')!.click();
|
||||
await flush();
|
||||
button("mongoDump.review").click();
|
||||
await flush();
|
||||
button("mongoDump.dump").click();
|
||||
await flush();
|
||||
expect(mocks.api.dumpMongodbDatabase).toHaveBeenCalledWith(expect.objectContaining({ connectionId: "c1", database: "target", format: "archive", gzip: true, collections: ["records"], filePath: "target.archive.gz" }), expect.any(Function));
|
||||
expect(mocks.guard).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("restores a directory into the chosen database only after drop confirmation", async () => {
|
||||
await mount("restore");
|
||||
await formatSelected("mongoDump.directory");
|
||||
checkbox("mongoDump.gzip").click();
|
||||
await filesSelected([new File([], "records.bson.gz")]);
|
||||
button("mongoDump.readSource").click();
|
||||
await flush();
|
||||
expect(mocks.api.prepareMongodbRestoreSource).toHaveBeenCalledWith(expect.any(Array), "directory", true, expect.objectContaining({ signal: expect.any(AbortSignal) }));
|
||||
checkbox("mongoDump.dropExisting").click();
|
||||
await flush();
|
||||
button("mongoDump.review").click();
|
||||
await flush();
|
||||
expect(document.body.textContent).toContain("mongoDump.confirmDrop");
|
||||
expect(mocks.api.restoreMongodbDatabase).not.toHaveBeenCalled();
|
||||
button("mongoDump.restore").click();
|
||||
await flush();
|
||||
expect(mocks.api.restoreMongodbDatabase).toHaveBeenCalledWith(
|
||||
expect.objectContaining({ database: "target", sourceDatabase: "source", sourceRef: "prepared-one", dropExisting: true, restoreOptions: true, restoreIndexes: true, objcheck: false, batchSize: 500 }),
|
||||
expect.any(Function),
|
||||
expect.objectContaining({ gzip: true }),
|
||||
);
|
||||
expect(mocks.guard).toHaveBeenCalledWith(expect.objectContaining({ database: "target", reviewText: expect.stringContaining("DROP selected collections") }));
|
||||
});
|
||||
|
||||
it("passes optional objcheck without a separate prevalidation request", async () => {
|
||||
await mount("restore");
|
||||
expect(checkbox("mongoDump.objcheck").checked).toBe(false);
|
||||
await filesSelected([new File([], "backup.archive")]);
|
||||
button("mongoDump.readSource").click();
|
||||
await flush();
|
||||
checkbox("mongoDump.objcheck").click();
|
||||
button("mongoDump.review").click();
|
||||
await flush();
|
||||
button("mongoDump.restore").click();
|
||||
await flush();
|
||||
expect(mocks.api.restoreMongodbDatabase).toHaveBeenCalledWith(expect.objectContaining({ objcheck: true }), expect.any(Function), undefined);
|
||||
expect(mocks.api.prepareMongodbRestoreSource).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
|
||||
it("starts only one restore when confirmation is pending and the button is clicked twice", async () => {
|
||||
let allow!: (value: boolean) => void;
|
||||
mocks.guard.mockReturnValue(
|
||||
new Promise<boolean>((resolve) => {
|
||||
allow = resolve;
|
||||
}),
|
||||
);
|
||||
await mount("restore");
|
||||
await filesSelected([new File([], "backup.archive")]);
|
||||
button("mongoDump.readSource").click();
|
||||
await flush();
|
||||
button("mongoDump.review").click();
|
||||
await flush();
|
||||
const restore = button("mongoDump.restore");
|
||||
restore.click();
|
||||
restore.click();
|
||||
await flush();
|
||||
expect(mocks.guard).toHaveBeenCalledTimes(1);
|
||||
expect(button("mongoDump.back").disabled).toBe(true);
|
||||
expect(button("mongoDump.close").disabled).toBe(true);
|
||||
expect(mocks.api.restoreMongodbDatabase).not.toHaveBeenCalled();
|
||||
allow(true);
|
||||
await flush();
|
||||
expect(mocks.api.restoreMongodbDatabase).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
|
||||
it("does not restore if the dialog is unmounted while confirmation is pending", async () => {
|
||||
let allow!: (value: boolean) => void;
|
||||
mocks.guard.mockReturnValue(
|
||||
new Promise<boolean>((resolve) => {
|
||||
allow = resolve;
|
||||
}),
|
||||
);
|
||||
await mount("restore");
|
||||
await filesSelected([new File([], "backup.archive")]);
|
||||
button("mongoDump.readSource").click();
|
||||
await flush();
|
||||
button("mongoDump.review").click();
|
||||
await flush();
|
||||
button("mongoDump.restore").click();
|
||||
await flush();
|
||||
app!.unmount();
|
||||
app = null;
|
||||
allow(true);
|
||||
await flush();
|
||||
expect(mocks.api.restoreMongodbDatabase).not.toHaveBeenCalled();
|
||||
expect(mocks.store.ensureConnected).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("unlocks the dialog when production confirmation is declined", async () => {
|
||||
mocks.guard.mockResolvedValueOnce(false);
|
||||
await mount("restore");
|
||||
await filesSelected([new File([], "backup.archive")]);
|
||||
button("mongoDump.readSource").click();
|
||||
await flush();
|
||||
button("mongoDump.review").click();
|
||||
await flush();
|
||||
button("mongoDump.restore").click();
|
||||
await flush();
|
||||
expect(mocks.api.restoreMongodbDatabase).not.toHaveBeenCalled();
|
||||
expect(button("mongoDump.restore").disabled).toBe(false);
|
||||
button("mongoDump.restore").click();
|
||||
await flush();
|
||||
expect(mocks.api.restoreMongodbDatabase).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
|
||||
it("invalidates a prepared source when gzip changes without inferring the format from its name", async () => {
|
||||
await mount("restore");
|
||||
await filesSelected([new File([], "misleading.bson.gz")]);
|
||||
button("mongoDump.readSource").click();
|
||||
await flush();
|
||||
expect(mocks.api.prepareMongodbRestoreSource).toHaveBeenCalledWith(expect.any(File), "archive", false, expect.any(Object));
|
||||
checkbox("mongoDump.gzip").click();
|
||||
await flush();
|
||||
expect(mocks.api.releaseMongodbRestoreSource).toHaveBeenCalledWith("prepared-one");
|
||||
expect(button("mongoDump.review").disabled).toBe(true);
|
||||
expect(mocks.api.prepareMongodbRestoreSource).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
|
||||
it("releases a late preview after the dialog is unmounted", async () => {
|
||||
let resolve!: (value: unknown) => void;
|
||||
mocks.api.prepareMongodbRestoreSource.mockReturnValueOnce(
|
||||
new Promise((done) => {
|
||||
resolve = done;
|
||||
}),
|
||||
);
|
||||
await mount("restore");
|
||||
await filesSelected([new File([], "backup.archive")]);
|
||||
button("mongoDump.readSource").click();
|
||||
await flush();
|
||||
app!.unmount();
|
||||
app = null;
|
||||
resolve({ ...catalog, sourceRef: "late-source" });
|
||||
await flush();
|
||||
expect(mocks.api.releaseMongodbRestoreSource).toHaveBeenCalledWith("late-source");
|
||||
});
|
||||
});
|
||||
@@ -143,6 +143,53 @@ describe("MongoImportDialog", () => {
|
||||
app.unmount();
|
||||
});
|
||||
|
||||
it("does not override the selected format from the file extension", async () => {
|
||||
const app = createApp(
|
||||
defineComponent({
|
||||
setup() {
|
||||
return () => h(MongoImportDialog, { open: true, connectionId: "c1", database: "shop", collection: "orders" });
|
||||
},
|
||||
}),
|
||||
);
|
||||
app.mount(document.body);
|
||||
await nextTick();
|
||||
|
||||
const input = document.querySelector("input[type='file']") as HTMLInputElement;
|
||||
const file = new File([new Uint8Array([5, 0, 0, 0, 0])], "orders.bson.gz", { type: "application/gzip" });
|
||||
Object.defineProperty(input, "files", { value: [file] });
|
||||
input.dispatchEvent(new Event("change"));
|
||||
await vi.advanceTimersByTimeAsync(200);
|
||||
await Promise.resolve();
|
||||
await nextTick();
|
||||
|
||||
expect(api.previewMongodbImportFile.mock.calls[0]?.[1]).toMatchObject({
|
||||
format: "csv",
|
||||
parseOptions: { delimiter: ",", typeMode: "auto" },
|
||||
});
|
||||
expect(document.body.textContent).toContain("tableImport.encoding");
|
||||
expect(document.body.textContent).toContain("mongo.import.typeMode");
|
||||
app.unmount();
|
||||
});
|
||||
|
||||
it("keeps an explicitly selected BSON format when changing files", async () => {
|
||||
api.previewMongodbImportFile.mockResolvedValue(previewResult({ format: "bson" }));
|
||||
const app = createApp(MongoImportDialog, { open: true, connectionId: "c1", database: "shop", collection: "orders" });
|
||||
app.mount(document.body);
|
||||
await nextTick();
|
||||
await setLabeledSelect("tableImport.sourceFormat", "bson");
|
||||
expect(document.body.textContent).not.toContain("tableImport.encoding");
|
||||
|
||||
for (const name of ["orders.bson", "renamed.json"]) {
|
||||
const input = document.querySelector("input[type='file']") as HTMLInputElement;
|
||||
Object.defineProperty(input, "files", { configurable: true, value: [new File([new Uint8Array([5, 0, 0, 0, 0])], name)] });
|
||||
input.dispatchEvent(new Event("change"));
|
||||
await vi.advanceTimersByTimeAsync(200);
|
||||
await nextTick();
|
||||
expect(api.previewMongodbImportFile.mock.lastCall?.[1]).toMatchObject({ format: "bson" });
|
||||
}
|
||||
app.unmount();
|
||||
});
|
||||
|
||||
it("reuses the uploaded sourceRef on later previews and releases it on close", async () => {
|
||||
const open = ref(true);
|
||||
const app = createApp(
|
||||
@@ -180,6 +227,9 @@ describe("MongoImportDialog", () => {
|
||||
|
||||
const header = document.querySelector("input[type='checkbox']") as HTMLInputElement;
|
||||
header.click();
|
||||
await nextTick();
|
||||
const next = Array.from(document.querySelectorAll("button")).find((button) => button.textContent?.includes("tableImport.next"));
|
||||
expect(next?.disabled).toBe(true);
|
||||
await vi.advanceTimersByTimeAsync(200);
|
||||
await Promise.resolve();
|
||||
await nextTick();
|
||||
@@ -194,7 +244,7 @@ describe("MongoImportDialog", () => {
|
||||
app.unmount();
|
||||
});
|
||||
|
||||
async function mountWithFile(fileName: string, preview: MongoImportPreview) {
|
||||
async function mountWithFile(fileName: string, preview: MongoImportPreview, format: MongoImportPreview["format"] = "csv") {
|
||||
api.previewMongodbImportFile.mockResolvedValue(preview);
|
||||
const app = createApp(
|
||||
defineComponent({
|
||||
@@ -211,6 +261,7 @@ describe("MongoImportDialog", () => {
|
||||
);
|
||||
app.mount(document.body);
|
||||
await nextTick();
|
||||
if (format !== "csv") await setLabeledSelect("tableImport.sourceFormat", format);
|
||||
const input = document.querySelector("input[type='file']") as HTMLInputElement;
|
||||
const file = new File(["id\n1"], fileName, { type: "text/csv" });
|
||||
Object.defineProperty(input, "files", { configurable: true, value: [file] });
|
||||
@@ -317,6 +368,7 @@ describe("MongoImportDialog", () => {
|
||||
fileName: "orders.json",
|
||||
columns: [{ name: "id", inferredType: "string" }],
|
||||
}),
|
||||
"json",
|
||||
);
|
||||
expect(columnTypeTrigger()).toBeNull();
|
||||
expect(document.body.textContent).toContain("(string)");
|
||||
|
||||
@@ -14,6 +14,7 @@ const SchemaDiagramDialog = defineAsyncComponent(() => import("@/components/diag
|
||||
const DatabaseDocsDialog = defineAsyncComponent(() => import("@/components/docs/DatabaseDocsDialog.vue"));
|
||||
const TableImportDialog = defineAsyncComponent(() => import("@/components/import/TableImportDialog.vue"));
|
||||
const MongoImportDialog = defineAsyncComponent(() => import("@/components/document/MongoImportDialog.vue"));
|
||||
const MongoDatabaseDumpDialog = defineAsyncComponent(() => import("@/components/document/MongoDatabaseDumpDialog.vue"));
|
||||
const FieldLineageDialog = defineAsyncComponent(() => import("@/components/lineage/FieldLineageDialog.vue"));
|
||||
const ConfigPassphraseDialog = defineAsyncComponent(() => import("@/components/config/ConfigPassphraseDialog.vue"));
|
||||
const ConfigConnectionSelectDialog = defineAsyncComponent(() => import("@/components/config/ConfigConnectionSelectDialog.vue"));
|
||||
@@ -301,6 +302,13 @@ watch(
|
||||
:prefill-table="dialogs.tableImportPrefillTable.value"
|
||||
/>
|
||||
<MongoImportDialog v-model:open="dialogs.showMongoImportDialog.value" :connection-id="dialogs.mongoImportPrefillConnectionId.value" :database="dialogs.mongoImportPrefillDatabase.value" :collection="dialogs.mongoImportPrefillCollection.value" />
|
||||
<MongoDatabaseDumpDialog
|
||||
v-if="dialogs.showMongoDatabaseDumpDialog.value"
|
||||
v-model:open="dialogs.showMongoDatabaseDumpDialog.value"
|
||||
:connection-id="dialogs.mongoDatabaseDumpPrefillConnectionId.value"
|
||||
:database="dialogs.mongoDatabaseDumpPrefillDatabase.value"
|
||||
:mode="dialogs.mongoDatabaseDumpMode.value"
|
||||
/>
|
||||
<DataGenerateDialog
|
||||
v-if="dialogs.showTableDataGenerateDialog.value"
|
||||
v-model:open="dialogs.showTableDataGenerateDialog.value"
|
||||
|
||||
@@ -410,15 +410,31 @@ const { copyStructureAs, copyStructureDocText, copyStructurePreview, exportData,
|
||||
acceptedSelectionIds: () => acceptedSelectionIds,
|
||||
});
|
||||
|
||||
const { openAllDatabasesExport, openDataCompare, openDatabaseExport, openDatabaseSearch, openDiagram, openDocs, openFieldLineage, openMongoImport, openScheduledBackups, openSchemaDiff, openSchemaDiffForRoutine, openSqlFileExecution, openStructureEditor, openTableImport, openTransfer } =
|
||||
useSidebarTreeToolRuntime({
|
||||
activeNode,
|
||||
connectionStore,
|
||||
queryStore,
|
||||
settingsStore,
|
||||
tableChildObjectName: tableChildDropObjectName,
|
||||
acceptedSelectionIds: () => acceptedSelectionIds,
|
||||
});
|
||||
const {
|
||||
openAllDatabasesExport,
|
||||
openDataCompare,
|
||||
openDatabaseExport,
|
||||
openDatabaseSearch,
|
||||
openDiagram,
|
||||
openDocs,
|
||||
openFieldLineage,
|
||||
openMongoImport,
|
||||
openMongoDatabaseDump,
|
||||
openScheduledBackups,
|
||||
openSchemaDiff,
|
||||
openSchemaDiffForRoutine,
|
||||
openSqlFileExecution,
|
||||
openStructureEditor,
|
||||
openTableImport,
|
||||
openTransfer,
|
||||
} = useSidebarTreeToolRuntime({
|
||||
activeNode,
|
||||
connectionStore,
|
||||
queryStore,
|
||||
settingsStore,
|
||||
tableChildObjectName: tableChildDropObjectName,
|
||||
acceptedSelectionIds: () => acceptedSelectionIds,
|
||||
});
|
||||
|
||||
const emit = defineEmits<{
|
||||
"rename-started": [];
|
||||
@@ -5870,6 +5886,8 @@ function buildSpecialSidebarMenu(context: SidebarMenuFactoryContext): boolean {
|
||||
if (node.type === "mongo-db") {
|
||||
items.push({ label: "", separator: true });
|
||||
items.push({ label: t("transfer.dataTransfer"), action: openTransfer, icon: ArrowRightLeft });
|
||||
items.push({ label: t("mongoDump.menuDump"), action: () => openMongoDatabaseDump("dump"), icon: Upload });
|
||||
items.push({ label: t("mongoDump.menuRestore"), action: () => openMongoDatabaseDump("restore"), icon: Download });
|
||||
}
|
||||
if (node.type === "redis-db") {
|
||||
items.push({ label: "", separator: true });
|
||||
@@ -5950,6 +5968,8 @@ function buildSpecialSidebarMenu(context: SidebarMenuFactoryContext): boolean {
|
||||
children: [
|
||||
{ label: "CSV", action: () => void exportMongoCollection("csv") },
|
||||
{ label: "NDJSON", action: () => void exportMongoCollection("ndjson") },
|
||||
{ label: "BSON dump", action: () => void exportMongoCollection("bson") },
|
||||
{ label: "BSON dump (gzip)", action: () => void exportMongoCollection("bsonGzip") },
|
||||
],
|
||||
});
|
||||
if (canDropMongoCollection.value) {
|
||||
|
||||
Reference in new issue
Block a user