trilium/apps/desktop/scripts/rebuild.mts

import { join } from "path";
import { cpSync, existsSync, mkdirSync, readFileSync, rmSync } from "fs";
import { execSync } from "child_process";
import { rebuild } from "@electron/rebuild"
import { isNixOS, resetPath } from "../../../scripts/utils.mjs";
import packageJson from "../package.json" with { type: "json" };

const desktopProjectRoot = join(import.meta.dirname, "..");
const workspaceRoot = join(desktopProjectRoot, "../..");

function copyNativeDependencies() {
    const destPath = join(desktopProjectRoot, "node_modules/better-sqlite3");
    
    if (existsSync(destPath)) {
        rmSync(destPath, { recursive: true });
    }
    mkdirSync(destPath);
    cpSync(join(workspaceRoot, "node_modules/better-sqlite3"), destPath, { recursive: true, dereference: true });
}

function rebuildNativeDependencies() {
    const electronVersion = determineElectronVersion();

    if (!electronVersion) {
        console.error("Unable to determine Electron version.");
        process.exit(1);
    }

    console.log(`Rebuilding ${desktopProjectRoot} with ${electronVersion}...`);

    rebuild({
        projectRootPath: desktopProjectRoot,
        buildPath: desktopProjectRoot,
        electronVersion
    });

    if (isNixOS()) {
        console.log("Patching ELF...");

        return execSync(`nix-shell -p auto-patchelf gcc.cc.lib --run "auto-patchelf --paths node_modules/better-sqlite3/build/Release/better_sqlite3.node --libs ${libStdPath}"`, {
            cwd: desktopProjectRoot,
            stdio: "inherit"
        });
    }
}

function determineElectronVersion() {
    if (isNixOS()) {
        console.log("Detected NixOS, reading Electron version from PATH");
        resetPath();

        try {
            return execSync("electron --version", { }).toString("utf-8");
        } catch (e) {
            console.error("Got error while trying to read the Electron version from shell. Make sure that an Electron version is in the PATH (e.g. `nix-shell -p electron`)");
            process.exit(1);
        }
    } else {
        console.log("Using Electron version from package.json");
        return packageJson.devDependencies.electron;
    }
}

copyNativeDependencies();
rebuildNativeDependencies();