import { spawn } from 'node:child_process';
import { chmod, copyFile, mkdir } from 'node:fs/promises';
import { dirname, join, resolve } from 'node:path';
import { fileURLToPath } from 'node:url';
import { Arch } from 'builder-util';

const root = resolve(dirname(fileURLToPath(import.meta.url)), '../native/desktop-bridge');
const targets = {
  win32: { x64: 'x86_64-pc-windows-msvc', arm64: 'aarch64-pc-windows-msvc' },
  darwin: { x64: 'x86_64-apple-darwin', arm64: 'aarch64-apple-darwin' },
  linux: { x64: 'x86_64-unknown-linux-gnu', arm64: 'aarch64-unknown-linux-gnu' },
};

export function nativeTarget(platform, arch) {
  const target = targets[platform]?.[arch];
  if (!target) throw new Error(`Unsupported native helper target: ${platform}/${arch}`);
  return target;
}

/** Build for the actual installer architecture, then copy before signing. */
export async function packageNativeHelper(context) {
  const platform = context.electronPlatformName;
  const target = nativeTarget(platform, Arch[context.arch]);
  const targetDir = join(root, 'target');
  await new Promise((resolve, reject) => {
    const child = spawn('cargo', ['build', '--locked', '--release', '--target', target], {
      cwd: root,
      env: { ...process.env, CARGO_TARGET_DIR: targetDir },
      stdio: 'inherit',
      windowsHide: true,
    });
    child.once('error', reject);
    child.once('close', (code) =>
      code === 0 ? resolve() : reject(new Error(`Native helper build failed (${code})`)),
    );
  });
  const name = 'voicestudio-desktop-bridge' + (platform === 'win32' ? '.exe' : '');
  const destination = join(context.packager.getResourcesDir(context.appOutDir), 'native');
  await mkdir(destination, { recursive: true });
  const binary = join(destination, name);
  await copyFile(join(targetDir, target, 'release', name), binary);
  if (platform === 'win32') await context.packager.signIf(binary);
  else await chmod(binary, 0o755);
}