repomix-mirror/website/server/scripts/bundle.mjs

/**
 * Bundle script for website server
 *
 * Creates production-ready bundles using Rolldown and collects WASM files.
 * Generates two separate bundles:
 * - server.mjs: Full server bundle with all dependencies
 * - worker.mjs: Minimal worker bundle for tinypool workers
 *
 * Usage: node scripts/bundle.mjs
 */

import { execSync } from 'node:child_process';
import { cpSync, existsSync, mkdirSync, readdirSync, rmSync, statSync } from 'node:fs';
import { dirname, join } from 'node:path';
import { fileURLToPath } from 'node:url';
import { rolldown } from 'rolldown';

const __dirname = dirname(fileURLToPath(import.meta.url));
const rootDir = join(__dirname, '..');
const distBundledDir = join(rootDir, 'dist-bundled');
const wasmDir = join(distBundledDir, 'wasm');

/**
 * Clean dist-bundled directory
 */
function cleanDistBundled() {
  console.log('Cleaning dist-bundled...');
  if (existsSync(distBundledDir)) {
    rmSync(distBundledDir, { recursive: true });
  }
  mkdirSync(distBundledDir, { recursive: true });
}

/**
 * Build TypeScript to JavaScript
 */
function buildTypeScript() {
  console.log('Building TypeScript...');
  execSync('npm run build', { cwd: rootDir, stdio: 'inherit' });
}

/**
 * Bundle server with Rolldown (full bundle with all server dependencies)
 */
async function bundleAll() {
  console.log('Bundling with code splitting...');

  const build = await rolldown({
    input: {
      server: join(rootDir, 'dist/index.js'),
      worker: join(rootDir, 'dist/worker-entry.js'),
    },
    platform: 'node',
    external: ['tinypool', 'tiktoken'],
  });

  await build.write({
    dir: distBundledDir,
    format: 'esm',
    entryFileNames: '[name].mjs',
    chunkFileNames: '[name]-[hash].mjs',
    // Note: No banner - Rolldown generates necessary shims via rolldown-runtime chunk
    minify: true,
    legalComments: 'inline',
    // Force code splitting with advancedChunks
    advancedChunks: {
      groups: [
        {
          name: 'shared',
          minSize: 100_000, // 100KB minimum for shared chunks
          minShareCount: 2, // Module must be used by at least 2 entry points
        },
      ],
    },
  });

  // Report bundle sizes
  const files = readdirSync(distBundledDir).filter((f) => f.endsWith('.mjs'));
  for (const file of files) {
    const size = getFileSizeMB(join(distBundledDir, file));
    console.log(`Bundle created: dist-bundled/${file} (${size} MB)`);
  }
}

/**
 * Get file size in MB
 */
function getFileSizeMB(filePath) {
  const stats = statSync(filePath);
  return (stats.size / 1024 / 1024).toFixed(2);
}

/**
 * Collect WASM files from node_modules
 */
function collectWasmFiles() {
  console.log('Collecting WASM files...');

  // Create wasm directory
  if (!existsSync(wasmDir)) {
    mkdirSync(wasmDir, { recursive: true });
  }

  // Copy web-tree-sitter.wasm to dist-bundled root
  // (web-tree-sitter looks for WASM file in the same directory as the JS file)
  const webTreeSitterWasm = join(rootDir, 'node_modules/web-tree-sitter/web-tree-sitter.wasm');
  if (existsSync(webTreeSitterWasm)) {
    cpSync(webTreeSitterWasm, join(distBundledDir, 'web-tree-sitter.wasm'));
    console.log('Copied web-tree-sitter.wasm to dist-bundled/');
  } else {
    console.warn('Warning: web-tree-sitter.wasm not found');
  }

  // Find and copy tree-sitter language WASM files
  const treeSitterWasmsDir = join(rootDir, 'node_modules/@repomix/tree-sitter-wasms/out');

  if (existsSync(treeSitterWasmsDir)) {
    const wasmFiles = readdirSync(treeSitterWasmsDir).filter((f) => f.endsWith('.wasm'));

    for (const file of wasmFiles) {
      cpSync(join(treeSitterWasmsDir, file), join(wasmDir, file));
    }

    console.log(`Copied ${wasmFiles.length} language WASM files to dist-bundled/wasm/`);
  } else {
    console.warn('Warning: tree-sitter-wasms not found');
  }
}

/**
 * Main bundle process
 */
async function main() {
  console.log('Starting bundle process...\n');

  cleanDistBundled();
  buildTypeScript();
  await bundleAll();
  collectWasmFiles();

  console.log('\nBundle complete!');
}

main().catch((err) => {
  console.error('Bundle failed:', err);
  process.exit(1);
});