feat(edit-docs): read from input directory instead of the zip

This commit is contained in:
Elian Doran 2025-05-27 19:05:13 +03:00
parent 0bb294753f
commit ad0c73d210
No known key found for this signature in database
3 changed files with 66 additions and 53 deletions

View File

@ -1,12 +1,13 @@
import { extractZip, initializeDatabase, startElectron } from "./utils.js"; import { extractZip, importData, initializeDatabase, startElectron } from "./utils.js";
import { initializeTranslations } from "@triliumnext/server/src/services/i18n.js"; import { initializeTranslations } from "@triliumnext/server/src/services/i18n.js";
import debounce from "@triliumnext/client/src/services/debounce.js"; import debounce from "@triliumnext/client/src/services/debounce.js";
import fs from "fs/promises"; import fs from "fs/promises";
import { join } from "path"; import { join } from "path";
import cls from "@triliumnext/server/src/services/cls.js";
// Paths are relative to apps/edit-docs/dist. // Paths are relative to apps/edit-docs/dist.
const DEMO_ZIP_PATH = join(__dirname, "../../server/src/assets/db/demo.zip"); const DEMO_ZIP_PATH = join(__dirname, "../../server/src/assets/db/demo.zip");
const OUTPUT_DIR = join(__dirname, "../demo"); const DEMO_ZIP_DIR_PATH = join(__dirname, "../demo");
async function main() { async function main() {
const initializedPromise = startElectron(() => { const initializedPromise = startElectron(() => {
@ -15,10 +16,23 @@ async function main() {
}); });
await initializeTranslations(); await initializeTranslations();
await initializeDatabase(false); await initializeDatabase(true);
cls.init(async () => {
await importData(DEMO_ZIP_DIR_PATH);
setOptions();
initializedPromise.resolve();
});
initializedPromise.resolve(); initializedPromise.resolve();
} }
async function setOptions() {
const optionsService = (await import("@triliumnext/server/src/services/options.js")).default;
optionsService.setOption("eraseUnusedAttachmentsAfterSeconds", 10);
optionsService.setOption("eraseUnusedAttachmentsAfterTimeScale", 60);
optionsService.setOption("compressImages", "false");
}
async function registerHandlers() { async function registerHandlers() {
const events = (await import("@triliumnext/server/src/services/events.js")).default; const events = (await import("@triliumnext/server/src/services/events.js")).default;
const eraseService = (await import("@triliumnext/server/src/services/erase.js")).default; const eraseService = (await import("@triliumnext/server/src/services/erase.js")).default;
@ -27,8 +41,8 @@ async function registerHandlers() {
eraseService.eraseUnusedAttachmentsNow(); eraseService.eraseUnusedAttachmentsNow();
await exportData(); await exportData();
await fs.rmdir(OUTPUT_DIR, { recursive: true }).catch(() => {}); await fs.rmdir(DEMO_ZIP_DIR_PATH, { recursive: true }).catch(() => {});
await extractZip(DEMO_ZIP_PATH, OUTPUT_DIR); await extractZip(DEMO_ZIP_PATH, DEMO_ZIP_DIR_PATH);
}, 10_000); }, 10_000);
events.subscribe(events.ENTITY_CHANGED, async (e) => { events.subscribe(events.ENTITY_CHANGED, async (e) => {
if (e.entityName === "options") { if (e.entityName === "options") {

View File

@ -3,15 +3,11 @@ import fsExtra from "fs-extra";
import path from "path"; import path from "path";
import type { NoteMetaFile } from "@triliumnext/server/src/services/meta/note_meta.js"; import type { NoteMetaFile } from "@triliumnext/server/src/services/meta/note_meta.js";
import { initializeTranslations } from "@triliumnext/server/src/services/i18n.js"; import { initializeTranslations } from "@triliumnext/server/src/services/i18n.js";
import archiver, { type Archiver } from "archiver";
import type { WriteStream } from "fs";
import debounce from "@triliumnext/client/src/services/debounce.js"; import debounce from "@triliumnext/client/src/services/debounce.js";
import { extractZip, initializeDatabase, startElectron } from "./utils.js"; import { extractZip, importData, initializeDatabase, startElectron } from "./utils.js";
import cls from "@triliumnext/server/src/services/cls.js"; import cls from "@triliumnext/server/src/services/cls.js";
import type { AdvancedExportOptions } from "@triliumnext/server/src/services/export/zip.js"; import type { AdvancedExportOptions } from "@triliumnext/server/src/services/export/zip.js";
import TaskContext from "@triliumnext/server/src/services/task_context.js";
import { parseNoteMetaFile } from "@triliumnext/server/src/services/in_app_help.js"; import { parseNoteMetaFile } from "@triliumnext/server/src/services/in_app_help.js";
import { resolve } from "path";
import type NoteMeta from "@triliumnext/server/src/services/meta/note_meta.js"; import type NoteMeta from "@triliumnext/server/src/services/meta/note_meta.js";
interface NoteMapping { interface NoteMapping {
@ -79,49 +75,6 @@ async function setOptions() {
optionsService.setOption("compressImages", "false"); optionsService.setOption("compressImages", "false");
} }
async function importData(path: string) {
const buffer = await createImportZip(path);
const importService = (await import("@triliumnext/server/src/services/import/zip.js")).default;
const context = new TaskContext("no-progress-reporting", "import", false);
const becca = (await import("@triliumnext/server/src/becca/becca.js")).default;
const rootNote = becca.getRoot();
if (!rootNote) {
throw new Error("Missing root note for import.");
}
await importService.importZip(context, buffer, rootNote, {
preserveIds: true
});
}
async function createImportZip(path: string) {
const inputFile = "input.zip";
const archive = archiver("zip", {
zlib: { level: 0 }
});
console.log("Archive path is ", resolve(path))
archive.directory(path, "/");
const outputStream = fsExtra.createWriteStream(inputFile);
archive.pipe(outputStream);
await waitForEnd(archive, outputStream);
try {
return await fsExtra.readFile(inputFile);
} finally {
await fsExtra.rm(inputFile);
}
}
function waitForEnd(archive: Archiver, stream: WriteStream) {
return new Promise<void>(async (res, rej) => {
stream.on("finish", () => res());
await archive.finalize();
});
}
async function exportData(noteId: string, format: "html" | "markdown", outputPath: string, ignoredFiles?: Set<string>) { async function exportData(noteId: string, format: "html" | "markdown", outputPath: string, ignoredFiles?: Set<string>) {
const zipFilePath = "output.zip"; const zipFilePath = "output.zip";

View File

@ -5,6 +5,10 @@ import path from "path";
import electron from "electron"; import electron from "electron";
import { deferred, type DeferredPromise } from "@triliumnext/server/src/services/utils.js"; import { deferred, type DeferredPromise } from "@triliumnext/server/src/services/utils.js";
import windowService from "@triliumnext/server/src/services/window.js"; import windowService from "@triliumnext/server/src/services/window.js";
import archiver, { type Archiver } from "archiver";
import type { WriteStream } from "fs";
import TaskContext from "@triliumnext/server/src/services/task_context.js";
import { resolve } from "path";
export function initializeDatabase(skipDemoDb: boolean) { export function initializeDatabase(skipDemoDb: boolean) {
return new Promise<void>(async (resolve) => { return new Promise<void>(async (resolve) => {
@ -44,6 +48,48 @@ export function startElectron(callback: () => void): DeferredPromise<void> {
return initializedPromise; return initializedPromise;
} }
export async function importData(path: string) {
const buffer = await createImportZip(path);
const importService = (await import("@triliumnext/server/src/services/import/zip.js")).default;
const context = new TaskContext("no-progress-reporting", "import", false);
const becca = (await import("@triliumnext/server/src/becca/becca.js")).default;
const rootNote = becca.getRoot();
if (!rootNote) {
throw new Error("Missing root note for import.");
}
await importService.importZip(context, buffer, rootNote, {
preserveIds: true
});
}
async function createImportZip(path: string) {
const inputFile = "input.zip";
const archive = archiver("zip", {
zlib: { level: 0 }
});
console.log("Archive path is ", resolve(path))
archive.directory(path, "/");
const outputStream = fsExtra.createWriteStream(inputFile);
archive.pipe(outputStream);
await waitForEnd(archive, outputStream);
try {
return await fsExtra.readFile(inputFile);
} finally {
await fsExtra.rm(inputFile);
}
}
function waitForEnd(archive: Archiver, stream: WriteStream) {
return new Promise<void>(async (res, rej) => {
stream.on("finish", () => res());
await archive.finalize();
});
}
export async function extractZip(zipFilePath: string, outputPath: string, ignoredFiles?: Set<string>) { export async function extractZip(zipFilePath: string, outputPath: string, ignoredFiles?: Set<string>) {
const deferred = (await import("@triliumnext/server/src/services/utils.js")).deferred; const deferred = (await import("@triliumnext/server/src/services/utils.js")).deferred;