happy-stacks 0.3.0 → 0.5.0

package/scripts/install.mjs

@@ -1,8 +1,8 @@
 import './utils/env/env.mjs';
 import { parseArgs } from './utils/cli/args.mjs';
 import { pathExists } from './utils/fs/fs.mjs';
-import { run } from './utils/proc/proc.mjs';
-import { getComponentDir, getRootDir } from './utils/paths/paths.mjs';
+import { run, runCapture } from './utils/proc/proc.mjs';
+import { getComponentDir, getComponentRepoDir, getRootDir, isHappyMonorepoRoot } from './utils/paths/paths.mjs';
 import { getServerComponentName } from './utils/server/server.mjs';
 import { ensureCliBuilt, ensureDepsInstalled, ensureHappyCliLocalNpmLinked } from './utils/proc/pm.mjs';
 import { dirname, join } from 'node:path';
@@ -12,6 +12,7 @@ import { printResult, wantsHelp, wantsJson } from './utils/cli/cli.mjs';
 import { ensureEnvLocalUpdated } from './utils/env/env_local.mjs';
 import { isTty, prompt, promptSelect, withRl } from './utils/cli/wizard.mjs';
 import { isSandboxed, sandboxAllowsGlobalSideEffects } from './utils/env/sandbox.mjs';
+import { bold, cyan, dim } from './utils/ui/ansi.mjs';
 
 /**
  * Install/setup the local stack:
@@ -24,8 +25,10 @@ import { isSandboxed, sandboxAllowsGlobalSideEffects } from './utils/env/sandbox
 
 const DEFAULT_FORK_REPOS = {
   serverLight: 'https://github.com/leeroybrun/happy-server-light.git',
-  // We don't currently maintain a separate fork for full happy-server; default to upstream.
-  serverFull: 'https://github.com/slopus/happy-server.git',
+  // Both server flavors live as branches in the same fork repo:
+  // - happy-server-light (sqlite)
+  // - happy-server (full)
+  serverFull: 'https://github.com/leeroybrun/happy-server-light.git',
   cli: 'https://github.com/leeroybrun/happy-cli.git',
   ui: 'https://github.com/leeroybrun/happy.git',
 };
@@ -33,8 +36,12 @@ const DEFAULT_FORK_REPOS = {
 const DEFAULT_UPSTREAM_REPOS = {
   // Upstream for server-light lives in the main happy-server repo.
   serverLight: 'https://github.com/slopus/happy-server.git',
-  serverFull: 'https://github.com/slopus/happy-server.git',
-  cli: 'https://github.com/slopus/happy-cli.git',
+  serverFull: 'https://github.com/slopus/happy.git',
+  // slopus/happy is now a monorepo that contains:
+  // - expo-app/ (UI)
+  // - cli/      (happy-cli)
+  // - server/   (happy-server)
+  cli: 'https://github.com/slopus/happy.git',
   ui: 'https://github.com/slopus/happy.git',
 };
 
@@ -44,15 +51,16 @@ function repoUrlsFromOwners({ forkOwner, upstreamOwner }) {
   return {
     forks: {
       serverLight: fork('happy-server-light'),
-      serverFull: fork('happy-server') /* best-effort; user can override */,
+      // Fork convention: server full is a branch in happy-server-light repo (not a separate repo).
+      serverFull: fork('happy-server-light'),
       cli: fork('happy-cli'),
       ui: fork('happy'),
     },
     upstream: {
       // server-light upstream lives in happy-server
       serverLight: up('happy-server'),
-      serverFull: up('happy-server'),
-      cli: up('happy-cli'),
+      serverFull: up('happy'),
+      cli: up('happy'),
       ui: up('happy'),
     },
   };
@@ -77,15 +85,62 @@ function resolveRepoSource({ flags }) {
 
 function getRepoUrls({ repoSource }) {
   const defaults = repoSource === 'upstream' ? DEFAULT_UPSTREAM_REPOS : DEFAULT_FORK_REPOS;
+  const ui = process.env.HAPPY_LOCAL_UI_REPO_URL?.trim() || defaults.ui;
   return {
     // Backwards compatible: HAPPY_LOCAL_SERVER_REPO_URL historically referred to the server-light component.
     serverLight: process.env.HAPPY_LOCAL_SERVER_LIGHT_REPO_URL?.trim() || process.env.HAPPY_LOCAL_SERVER_REPO_URL?.trim() || defaults.serverLight,
-    serverFull: process.env.HAPPY_LOCAL_SERVER_FULL_REPO_URL?.trim() || defaults.serverFull,
-    cli: process.env.HAPPY_LOCAL_CLI_REPO_URL?.trim() || defaults.cli,
-    ui: process.env.HAPPY_LOCAL_UI_REPO_URL?.trim() || defaults.ui,
+    // Default to the UI repo when using a monorepo (override to keep split repos).
+    serverFull: process.env.HAPPY_LOCAL_SERVER_FULL_REPO_URL?.trim() || defaults.serverFull || ui,
+    cli: process.env.HAPPY_LOCAL_CLI_REPO_URL?.trim() || defaults.cli || ui,
+    ui,
   };
 }
 
+async function ensureGitBranchCheckedOut({ repoDir, branch, label }) {
+  if (!(await pathExists(join(repoDir, '.git')))) return;
+  const b = String(branch ?? '').trim();
+  if (!b) return;
+
+  try {
+    const head = (await runCapture('git', ['rev-parse', '--abbrev-ref', 'HEAD'], { cwd: repoDir })).trim();
+    if (head && head === b) return;
+  } catch {
+    // ignore
+  }
+
+  // Ensure branch exists locally, otherwise fetch it from origin.
+  let hasLocal = true;
+  try {
+    await run('git', ['show-ref', '--verify', '--quiet', `refs/heads/${b}`], { cwd: repoDir });
+  } catch {
+    hasLocal = false;
+  }
+  if (!hasLocal) {
+    try {
+      await run('git', ['fetch', '--quiet', 'origin', b], { cwd: repoDir });
+    } catch {
+      throw new Error(
+        `[local] ${label}: expected branch "${b}" to exist in ${repoDir}.\n` +
+          `[local] Fix: use --forks for happy-server-light (sqlite), or use --server=happy-server with --upstream.`
+      );
+    }
+  }
+
+  try {
+    await run('git', ['checkout', '-q', b], { cwd: repoDir });
+  } catch {
+    // If remote-tracking branch exists but local doesn't, create it.
+    try {
+      await run('git', ['checkout', '-q', '-B', b, `origin/${b}`], { cwd: repoDir });
+    } catch {
+      throw new Error(
+        `[local] ${label}: failed to checkout branch "${b}" in ${repoDir}.\n` +
+          `[local] Fix: re-run with --force in worktree flows, or delete the checkout and re-run install/bootstrap.`
+      );
+    }
+  }
+}
+
 async function ensureComponentPresent({ dir, label, repoUrl, allowClone }) {
   if (await pathExists(dir)) {
     return;
@@ -120,31 +175,33 @@ async function ensureUpstreamRemote({ repoDir, upstreamUrl }) {
 async function interactiveWizard({ rootDir, defaults }) {
   return await withRl(async (rl) => {
     const repoSource = await promptSelect(rl, {
-      title: 'Select repo source:',
+      title: `${bold('Repo source')}\n${dim('Where should Happy Stacks clone the component repos from?')}`,
       options: [
-        { label: `forks (default, recommended)`, value: 'forks' },
-        { label: `upstream (slopus/*)`, value: 'upstream' },
+        { label: `${cyan('forks')} (default, recommended)`, value: 'forks' },
+        { label: `${cyan('upstream')} (slopus/*)`, value: 'upstream' },
       ],
       defaultIndex: defaults.repoSource === 'upstream' ? 1 : 0,
     });
 
+    // eslint-disable-next-line no-console
+    console.log(dim('Tip: keep the defaults unless you maintain your own forks.'));
     const forkOwner = await prompt(rl, `GitHub fork owner (default: ${defaults.forkOwner}): `, { defaultValue: defaults.forkOwner });
     const upstreamOwner = await prompt(rl, `GitHub upstream owner (default: ${defaults.upstreamOwner}): `, {
       defaultValue: defaults.upstreamOwner,
     });
 
     const serverMode = await promptSelect(rl, {
-      title: 'Which server components should be set up?',
+      title: `${bold('Server components')}\n${dim('Choose which server repo(s) to clone and install deps for.')}`,
       options: [
-        { label: 'happy-server-light only (default)', value: 'happy-server-light' },
-        { label: 'happy-server only (full server)', value: 'happy-server' },
-        { label: 'both (server-light + full server)', value: 'both' },
+        { label: `${cyan('happy-server-light')} only (default)`, value: 'happy-server-light' },
+        { label: `${cyan('happy-server')} only (full server)`, value: 'happy-server' },
+        { label: `both (${cyan('server-light')} + ${cyan('full server')})`, value: 'both' },
       ],
       defaultIndex: defaults.serverComponentName === 'both' ? 2 : defaults.serverComponentName === 'happy-server' ? 1 : 0,
     });
 
     const allowClone = await promptSelect(rl, {
-      title: 'Clone missing component repos?',
+      title: `${bold('Cloning')}\n${dim('If repos are missing under components/, should we clone them automatically?')}`,
       options: [
         { label: 'yes (default)', value: true },
         { label: 'no', value: false },
@@ -154,8 +211,8 @@ async function interactiveWizard({ rootDir, defaults }) {
 
     const enableAutostart = await promptSelect(rl, {
       title: isSandboxed()
-        ? 'Enable macOS autostart (LaunchAgent)? (NOTE: sandbox mode; this is global OS state)'
-        : 'Enable macOS autostart (LaunchAgent)?',
+        ? `${bold('Autostart (macOS)')}\n${dim('Sandbox mode: this is global OS state; normally disabled in sandbox.')}`
+        : `${bold('Autostart (macOS)')}\n${dim('Install a LaunchAgent so Happy starts at login?')}`,
       options: [
         { label: 'no (default)', value: false },
         { label: 'yes', value: true },
@@ -164,7 +221,7 @@ async function interactiveWizard({ rootDir, defaults }) {
     });
 
     const buildTauri = await promptSelect(rl, {
-      title: 'Build Tauri desktop app as part of setup?',
+      title: `${bold('Desktop app (optional)')}\n${dim('Build the Tauri desktop app as part of setup? (slow; requires extra toolchain)')}`,
       options: [
         { label: 'no (default)', value: false },
         { label: 'yes', value: true },
@@ -173,7 +230,7 @@ async function interactiveWizard({ rootDir, defaults }) {
     });
 
     const configureGit = await promptSelect(rl, {
-      title: 'Configure upstream Git remotes and create mirror branches (slopus/main)?',
+      title: `${bold('Git remotes')}\n${dim('Configure upstream remotes and create/update mirror branches (e.g. slopus/main)?')}`,
       options: [
         { label: 'yes (default)', value: true },
         { label: 'no', value: false },
@@ -201,7 +258,22 @@ async function main() {
   if (wantsHelp(argv, { flags })) {
     printResult({
       json,
-      data: { flags: ['--forks', '--upstream', '--clone', '--no-clone', '--autostart', '--no-autostart', '--server=...'], json: true },
+      data: {
+        flags: [
+          '--forks',
+          '--upstream',
+          '--clone',
+          '--no-clone',
+          '--autostart',
+          '--no-autostart',
+          '--server=...',
+          '--no-ui-build',
+          '--no-ui-deps',
+          '--no-cli-deps',
+          '--no-cli-build',
+        ],
+        json: true,
+      },
       text: [
         '[bootstrap] usage:',
         '  happys bootstrap [--forks|--upstream] [--server=happy-server|happy-server-light|both] [--json]',
@@ -270,70 +342,125 @@ async function main() {
   const disableAutostart = flags.has('--no-autostart');
 
   const serverComponentName = (wizard?.serverComponentName ?? getServerComponentName({ kv })).trim();
-  const serverLightDir = getComponentDir(rootDir, 'happy-server-light');
-  const serverFullDir = getComponentDir(rootDir, 'happy-server');
-  const cliDir = getComponentDir(rootDir, 'happy-cli');
+  // Safety: upstream server-light is not a separate upstream repo/branch today.
+  // Upstream slopus/happy-server is Postgres-only, while happy-server-light requires sqlite.
+  if (repoSource === 'upstream' && (serverComponentName === 'happy-server-light' || serverComponentName === 'both')) {
+    throw new Error(
+      `[bootstrap] --upstream is not supported for happy-server-light (sqlite).\n` +
+        `Reason: upstream ${DEFAULT_UPSTREAM_REPOS.serverLight} does not provide a happy-server-light branch.\n` +
+        `Fix:\n` +
+        `- use --forks (recommended), OR\n` +
+        `- use --server=happy-server with --upstream`
+    );
+  }
+  // Repo roots (clone locations)
+  const uiRepoDir = getComponentRepoDir(rootDir, 'happy');
+  const serverLightRepoDir = getComponentRepoDir(rootDir, 'happy-server-light');
+
+  // Ensure UI exists first (monorepo anchor in slopus/happy).
+  await ensureComponentPresent({
+    dir: uiRepoDir,
+    label: 'UI',
+    repoUrl: repos.ui,
+    allowClone,
+  });
+
+  // Package dirs (where we run installs/builds). Recompute after cloning UI.
   const uiDir = getComponentDir(rootDir, 'happy');
+  const cliDir = getComponentDir(rootDir, 'happy-cli');
+  const serverFullDir = getComponentDir(rootDir, 'happy-server');
+
+  const cliRepoDir = getComponentRepoDir(rootDir, 'happy-cli');
+  const serverFullRepoDir = getComponentRepoDir(rootDir, 'happy-server');
+  const hasMonorepo = isHappyMonorepoRoot(uiRepoDir);
 
-  // Ensure components exist (embedded layout)
+  // Ensure other components exist.
+  // - server-light stays separate for now.
+  // - full server + cli may be embedded in the monorepo.
   if (serverComponentName === 'both' || serverComponentName === 'happy-server-light') {
-  await ensureComponentPresent({
-      dir: serverLightDir,
-    label: 'SERVER',
+    await ensureComponentPresent({
+      dir: serverLightRepoDir,
+      label: 'SERVER',
       repoUrl: repos.serverLight,
       allowClone,
     });
   }
-  if (serverComponentName === 'both' || serverComponentName === 'happy-server') {
+  if (!hasMonorepo) {
+    if (serverComponentName === 'both' || serverComponentName === 'happy-server') {
+      await ensureComponentPresent({
+        dir: serverFullRepoDir,
+        label: 'SERVER_FULL',
+        repoUrl: repos.serverFull,
+        allowClone,
+      });
+    }
     await ensureComponentPresent({
-      dir: serverFullDir,
-      label: 'SERVER_FULL',
-      repoUrl: repos.serverFull,
-    allowClone,
-  });
+      dir: cliRepoDir,
+      label: 'CLI',
+      repoUrl: repos.cli,
+      allowClone,
+    });
+  } else {
+    if ((serverComponentName === 'both' || serverComponentName === 'happy-server') && !(await pathExists(serverFullDir))) {
+      throw new Error(`[bootstrap] expected monorepo server package at ${serverFullDir} (missing).`);
+    }
+    if (!(await pathExists(cliDir))) {
+      throw new Error(`[bootstrap] expected monorepo cli package at ${cliDir} (missing).`);
+    }
+  }
+
+  // Ensure expected branches are checked out for server flavors (avoids "server-light directory contains full server" mistakes).
+  if (serverComponentName === 'both' || serverComponentName === 'happy-server-light') {
+    await ensureGitBranchCheckedOut({ repoDir: serverLightRepoDir, branch: 'happy-server-light', label: 'SERVER' });
+  }
+  if (serverComponentName === 'both' || serverComponentName === 'happy-server') {
+    // In fork mode (split repos), full server is a branch in the fork server repo.
+    // In upstream mode and in monorepo mode, use main.
+    const serverFullBranch = isHappyMonorepoRoot(serverFullRepoDir) ? 'main' : repoSource === 'upstream' ? 'main' : 'happy-server';
+    await ensureGitBranchCheckedOut({ repoDir: serverFullRepoDir, branch: serverFullBranch, label: 'SERVER_FULL' });
   }
-  await ensureComponentPresent({
-    dir: cliDir,
-    label: 'CLI',
-    repoUrl: repos.cli,
-    allowClone,
-  });
-  await ensureComponentPresent({
-    dir: uiDir,
-    label: 'UI',
-    repoUrl: repos.ui,
-    allowClone,
-  });
 
   const cliDirFinal = cliDir;
   const uiDirFinal = uiDir;
 
   // Install deps
+  const skipUiDeps = flags.has('--no-ui-deps') || (process.env.HAPPY_STACKS_INSTALL_NO_UI_DEPS ?? '').trim() === '1';
+  const skipCliDeps = flags.has('--no-cli-deps') || (process.env.HAPPY_STACKS_INSTALL_NO_CLI_DEPS ?? '').trim() === '1';
   if (serverComponentName === 'both' || serverComponentName === 'happy-server-light') {
-    await ensureDepsInstalled(serverLightDir, 'happy-server-light');
+    await ensureDepsInstalled(getComponentDir(rootDir, 'happy-server-light'), 'happy-server-light');
   }
   if (serverComponentName === 'both' || serverComponentName === 'happy-server') {
     await ensureDepsInstalled(serverFullDir, 'happy-server');
   }
-  await ensureDepsInstalled(uiDirFinal, 'happy');
-  await ensureDepsInstalled(cliDirFinal, 'happy-cli');
+  if (!skipUiDeps) {
+    await ensureDepsInstalled(uiDirFinal, 'happy');
+  }
+  if (!skipCliDeps) {
+    await ensureDepsInstalled(cliDirFinal, 'happy-cli');
+  }
 
   // CLI build + link
-  const buildCli = (process.env.HAPPY_LOCAL_CLI_BUILD ?? '1') !== '0';
-  const npmLinkCli = (process.env.HAPPY_LOCAL_NPM_LINK ?? '1') !== '0';
-  await ensureCliBuilt(cliDirFinal, { buildCli });
-  await ensureHappyCliLocalNpmLinked(rootDir, { npmLinkCli });
+  const skipCliBuild = flags.has('--no-cli-build') || (process.env.HAPPY_STACKS_INSTALL_NO_CLI_BUILD ?? '').trim() === '1';
+  if (!skipCliBuild) {
+    const buildCli = (process.env.HAPPY_LOCAL_CLI_BUILD ?? '1') !== '0';
+    const npmLinkCli = (process.env.HAPPY_LOCAL_NPM_LINK ?? '1') !== '0';
+    await ensureCliBuilt(cliDirFinal, { buildCli });
+    await ensureHappyCliLocalNpmLinked(rootDir, { npmLinkCli });
+  }
 
   // Build UI (so run works without expo dev server)
+  const skipUiBuild = flags.has('--no-ui-build') || (process.env.HAPPY_STACKS_INSTALL_NO_UI_BUILD ?? '').trim() === '1';
   const buildArgs = [join(rootDir, 'scripts', 'build.mjs')];
   // Tauri builds are opt-in (slow + requires additional toolchain).
   const buildTauri = wizard?.buildTauri ?? (flags.has('--tauri') && !flags.has('--no-tauri'));
-  if (buildTauri) {
-    buildArgs.push('--tauri');
-  } else if (flags.has('--no-tauri')) {
-    buildArgs.push('--no-tauri');
+  if (!skipUiBuild) {
+    if (buildTauri) {
+      buildArgs.push('--tauri');
+    } else if (flags.has('--no-tauri')) {
+      buildArgs.push('--no-tauri');
+    }
+    await run(process.execPath, buildArgs, { cwd: rootDir });
   }
-  await run(process.execPath, buildArgs, { cwd: rootDir });
 
   // Optional autostart (macOS)
   if (disableAutostart) {
@@ -346,14 +473,16 @@ async function main() {
   if (wizard?.configureGit) {
     // Ensure upstream remotes exist so `happys wt sync-all` works consistently.
     const upstreamRepos = getRepoUrls({ repoSource: 'upstream' });
-    await ensureUpstreamRemote({ repoDir: uiDir, upstreamUrl: upstreamRepos.ui });
-    await ensureUpstreamRemote({ repoDir: cliDir, upstreamUrl: upstreamRepos.cli });
+    await ensureUpstreamRemote({ repoDir: uiRepoDir, upstreamUrl: upstreamRepos.ui });
+    if (cliRepoDir !== uiRepoDir) {
+      await ensureUpstreamRemote({ repoDir: cliRepoDir, upstreamUrl: upstreamRepos.cli });
+    }
     // server-light and server-full both track upstream happy-server
-    if (await pathExists(serverLightDir)) {
-      await ensureUpstreamRemote({ repoDir: serverLightDir, upstreamUrl: upstreamRepos.serverLight });
+    if (await pathExists(serverLightRepoDir)) {
+      await ensureUpstreamRemote({ repoDir: serverLightRepoDir, upstreamUrl: upstreamRepos.serverLight });
     }
-    if (await pathExists(serverFullDir)) {
-      await ensureUpstreamRemote({ repoDir: serverFullDir, upstreamUrl: upstreamRepos.serverFull });
+    if (serverFullRepoDir !== uiRepoDir && (await pathExists(serverFullRepoDir))) {
+      await ensureUpstreamRemote({ repoDir: serverFullRepoDir, upstreamUrl: upstreamRepos.serverFull });
     }
 
     // Create/update mirror branches like slopus/main for each repo (best-effort).
@@ -370,7 +499,16 @@ async function main() {
       ok: true,
       repoSource,
       serverComponentName,
-      dirs: { serverLightDir, serverFullDir, cliDir: cliDirFinal, uiDir: uiDirFinal },
+      dirs: {
+        uiRepoDir,
+        uiDir: uiDirFinal,
+        cliRepoDir,
+        cliDir: cliDirFinal,
+        serverLightRepoDir,
+        serverLightDir: getComponentDir(rootDir, 'happy-server-light'),
+        serverFullRepoDir,
+        serverFullDir,
+      },
       cloned: allowClone,
       autostart: enableAutostart ? 'enabled' : sandboxed && enableAutostartRaw && !allowGlobal ? 'skipped (sandbox)' : disableAutostart ? 'disabled' : 'unchanged',
       interactive: Boolean(wizard),

package/scripts/lint.mjs

@@ -1,11 +1,12 @@
 import './utils/env/env.mjs';
 import { parseArgs } from './utils/cli/args.mjs';
 import { printResult, wantsHelp, wantsJson } from './utils/cli/cli.mjs';
-import { getComponentDir, getRootDir } from './utils/paths/paths.mjs';
+import { componentDirEnvKey, getComponentDir, getRootDir } from './utils/paths/paths.mjs';
 import { ensureDepsInstalled } from './utils/proc/pm.mjs';
 import { pathExists } from './utils/fs/fs.mjs';
 import { run } from './utils/proc/proc.mjs';
 import { detectPackageManagerCmd, pickFirstScript, readPackageJsonScripts } from './utils/proc/package_scripts.mjs';
+import { getInvokedCwd, inferComponentFromCwd } from './utils/cli/cwd_scope.mjs';
 
 const DEFAULT_COMPONENTS = ['happy', 'happy-cli', 'happy-server-light', 'happy-server'];
 
@@ -40,18 +41,37 @@ async function main() {
         'examples:',
         '  happys lint',
         '  happys lint happy happy-cli',
+        '',
+        'note:',
+        '  If run from inside a component checkout/worktree and no components are provided, defaults to that component.',
       ].join('\n'),
     });
     return;
   }
 
+  const rootDir = getRootDir(import.meta.url);
+
   const positionals = argv.filter((a) => !a.startsWith('--'));
-  const requested = positionals.length ? positionals : ['all'];
+  const inferred =
+    positionals.length === 0
+      ? inferComponentFromCwd({
+          rootDir,
+          invokedCwd: getInvokedCwd(process.env),
+          components: DEFAULT_COMPONENTS,
+        })
+      : null;
+  if (inferred) {
+    const stacksKey = componentDirEnvKey(inferred.component);
+    const legacyKey = stacksKey.replace(/^HAPPY_STACKS_/, 'HAPPY_LOCAL_');
+    if (!(process.env[stacksKey] ?? '').toString().trim() && !(process.env[legacyKey] ?? '').toString().trim()) {
+      process.env[stacksKey] = inferred.repoDir;
+    }
+  }
+
+  const requested = positionals.length ? positionals : inferred ? [inferred.component] : ['all'];
   const wantAll = requested.includes('all');
   const components = wantAll ? DEFAULT_COMPONENTS : requested;
 
-  const rootDir = getRootDir(import.meta.url);
-
   const results = [];
   for (const component of components) {
     if (!DEFAULT_COMPONENTS.includes(component)) {

package/scripts/migrate.mjs

@@ -1,7 +1,6 @@
 import './utils/env/env.mjs';
 import { copyFile, mkdir, readFile } from 'node:fs/promises';
 import { basename, join } from 'node:path';
-import { createRequire } from 'node:module';
 
 import { parseArgs } from './utils/cli/args.mjs';
 import { printResult, wantsHelp, wantsJson } from './utils/cli/cli.mjs';
@@ -13,6 +12,7 @@ import { ensureHappyServerManagedInfra, applyHappyServerMigrations } from './uti
 import { runCapture } from './utils/proc/proc.mjs';
 import { pickNextFreeTcpPort } from './utils/net/ports.mjs';
 import { getEnvValue } from './utils/env/values.mjs';
+import { importPrismaClientForHappyServerLight, importPrismaClientFromNodeModules } from './utils/server/prisma_import.mjs';
 
 function usage() {
   return [
@@ -54,14 +54,6 @@ async function ensureTargetSecretMatchesSource({ sourceSecretPath, targetSecretP
   }
 }
 
-async function importPrismaClientFrom(dir) {
-  const req = createRequire(import.meta.url);
-  const resolved = req.resolve('@prisma/client', { paths: [dir] });
-  // eslint-disable-next-line import/no-dynamic-require
-  const mod = req(resolved);
-  return mod.PrismaClient;
-}
-
 async function migrateLightToServer({ rootDir, fromStack, toStack, includeFiles, force, json }) {
   const from = resolveStackEnvPath(fromStack);
   const to = resolveStackEnvPath(toStack);
@@ -139,8 +131,8 @@ async function migrateLightToServer({ rootDir, fromStack, toStack, includeFiles,
   await copyFile(fromParsed.path, snapshotPath);
   const snapshotDbUrl = `file:${snapshotPath}`;
 
-  const SourcePrismaClient = await importPrismaClientFrom(lightDir);
-  const TargetPrismaClient = await importPrismaClientFrom(fullDir);
+  const SourcePrismaClient = await importPrismaClientForHappyServerLight({ serverDir: lightDir });
+  const TargetPrismaClient = await importPrismaClientFromNodeModules({ dir: fullDir });
 
   const sourceDb = new SourcePrismaClient({ datasources: { db: { url: snapshotDbUrl } } });
   const targetDb = new TargetPrismaClient({ datasources: { db: { url: infra.env.DATABASE_URL } } });
@@ -289,4 +281,3 @@ main().catch((err) => {
   console.error(err);
   process.exit(1);
 });
-