import { afterEach, describe, expect, test } from 'vitest';
import { copyFileSync, mkdtempSync, mkdirSync, rmSync, writeFileSync } from 'node:fs';
import { execFileSync, spawnSync } from 'node:child_process';
import { tmpdir } from 'node:os';
import path from 'node:path';
import { fileURLToPath } from 'node:url';

const packageRoot = fileURLToPath(new URL('../', import.meta.url));
const fixtureRoots: string[] = [];

function createFixture () {
  const fixtureRoot = mkdtempSync(path.join(tmpdir(), 'napcat-loader-'));
  fixtureRoots.push(fixtureRoot);
  const installDir = path.join(fixtureRoot, '中文 space # percent%');
  mkdirSync(installDir);
  return { fixtureRoot, installDir };
}

afterEach(() => {
  for (const fixtureRoot of fixtureRoots.splice(0)) {
    rmSync(fixtureRoot, { recursive: true, force: true });
  }
});

describe('loader paths', () => {
  test('Shell loads the flat package from an unrelated working directory', () => {
    const { fixtureRoot, installDir } = createFixture();
    const loaderPath = path.join(installDir, 'loadNapCat.cjs');
    copyFileSync(path.join(packageRoot, 'napcat-shell-loader/loadNapCat.js'), loaderPath);
    writeFileSync(path.join(installDir, 'napcat.mjs'), 'console.log("SHELL_LOADED");');
    const environment = { ...process.env };
    delete environment['NAPCAT_MAIN_PATH'];
    const output = execFileSync(process.execPath, [loaderPath], { cwd: fixtureRoot, env: environment, encoding: 'utf8' });
    expect(output).toContain('SHELL_LOADED');
  });

  test('Shell uses the actual entry path when the loader is redirected into QQ', () => {
    const { fixtureRoot, installDir } = createFixture();
    const loaderPath = path.join(fixtureRoot, 'loadNapCat.cjs');
    const entryPath = path.join(installDir, 'napcat.mjs');
    copyFileSync(path.join(packageRoot, 'napcat-shell-loader/loadNapCat.js'), loaderPath);
    writeFileSync(entryPath, 'console.log("REDIRECTED_SHELL_LOADED");');
    const output = execFileSync(process.execPath, [loaderPath], {
      cwd: fixtureRoot,
      env: { ...process.env, NAPCAT_MAIN_PATH: entryPath },
      encoding: 'utf8',
    });
    expect(output).toContain('REDIRECTED_SHELL_LOADED');
  });

  test('Framework native loader preserves URL-sensitive path characters', () => {
    const { fixtureRoot, installDir } = createFixture();
    const loaderPath = path.join(installDir, 'nativeLoader.cjs');
    copyFileSync(path.join(packageRoot, 'napcat-framework/nativeLoader.cjs'), loaderPath);
    writeFileSync(path.join(installDir, 'napcat.mjs'), 'export async function NCoreInitFramework () { console.log("FRAMEWORK_LOADED"); }');
    const output = execFileSync(process.execPath, ['-e', 'require(process.argv[1]).initializeNapCat({}, {}, () => {});', loaderPath], {
      cwd: fixtureRoot,
      encoding: 'utf8',
    });
    expect(output).toContain('FRAMEWORK_LOADED');
    expect(output).not.toContain('[Error]');
  });

  test('Framework dlopen loader preserves URL-sensitive path characters', () => {
    const { fixtureRoot, installDir } = createFixture();
    const loaderPath = path.join(installDir, 'napcat.cjs');
    copyFileSync(path.join(packageRoot, 'napcat-framework/napcat.cjs'), loaderPath);
    writeFileSync(path.join(installDir, 'napcat.mjs'), 'export async function NCoreInitFramework () { console.log("DLOPEN_FRAMEWORK_LOADED"); } export function getWebUiUrl () { return ""; }');
    const script = 'process.dlopen = (module) => { module.exports = { NodeIKernelLoginService: { get: () => ({}) }, NodeIQQNTWrapperSession: { create: () => ({}) } }; }; require(process.argv[1]); process.dlopen({ exports: {} }, "wrapper.node");';
    const output = execFileSync(process.execPath, ['-e', script, loaderPath], { cwd: fixtureRoot, encoding: 'utf8' });
    expect(output).toContain('DLOPEN_FRAMEWORK_LOADED');
    expect(output).not.toContain('[Error]');
  });

  test.skipIf(process.platform !== 'win32')('Node batch uses the bundled runtime, forwards arguments and returns the exit code', () => {
    const { fixtureRoot, installDir } = createFixture();
    copyFileSync(process.execPath, path.join(installDir, 'node.exe'));
    copyFileSync(path.join(packageRoot, 'napcat-develop/napcat.bat'), path.join(installDir, 'napcat.bat'));
    writeFileSync(path.join(installDir, 'index.js'), 'console.log(JSON.stringify({ cwd: process.cwd(), args: process.argv.slice(2) })); process.exitCode = 7;');
    const result = spawnSync('cmd.exe', ['/d', '/s', '/c', `""${path.join(installDir, 'napcat.bat')}" -q 123456"`], {
      cwd: fixtureRoot,
      encoding: 'utf8',
      windowsVerbatimArguments: true,
    });
    expect(result.error).toBeUndefined();
    expect(result.status).toBe(7);
    expect(JSON.parse(result.stdout)).toEqual({ cwd: installDir, args: ['-q', '123456'] });
  });
});