/**
 * @license
 * Copyright 2026 Qwen Team
 * SPDX-License-Identifier: Apache-2.0
 */

import { spawnSync } from 'node:child_process';
import {
  mkdirSync,
  mkdtempSync,
  readFileSync,
  rmSync,
  writeFileSync,
} from 'node:fs';
import { tmpdir } from 'node:os';
import path from 'node:path';
import { fileURLToPath } from 'node:url';
import { afterEach, describe, expect, it } from 'vitest';
import {
  getTestCiWorkspacePackageJsonPaths,
  getWorkspacePackageJsonPaths,
} from '../workspaces.js';

describe('workspace helpers', () => {
  const tempDirs = [];

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

  it('honors negated workspace patterns', () => {
    const root = mkdtempSync(path.join(tmpdir(), 'qwen-workspaces-'));
    tempDirs.push(root);

    for (const packagePath of [
      'packages/cli/package.json',
      'packages/core/package.json',
      'packages/desktop/package.json',
      'packages/channels/base/package.json',
    ]) {
      writeFile(root, packagePath, '{}\n');
    }

    expect(
      getWorkspacePackageJsonPaths(root, [
        'packages/*',
        'packages/channels/base',
        '!packages/desktop',
      ]),
    ).toEqual([
      'packages/channels/base/package.json',
      'packages/cli/package.json',
      'packages/core/package.json',
    ]);
  });

  it('normalizes Windows-style workspace patterns', () => {
    const root = mkdtempSync(path.join(tmpdir(), 'qwen-workspaces-'));
    tempDirs.push(root);

    for (const packagePath of [
      'packages/cli/package.json',
      'packages/core/package.json',
      'packages/desktop/package.json',
      'packages/channels/base/package.json',
    ]) {
      writeFile(root, packagePath, '{}\n');
    }

    expect(
      getWorkspacePackageJsonPaths(root, [
        'packages\\*',
        'packages\\channels\\base',
        '!packages\\desktop',
      ]),
    ).toEqual([
      'packages/channels/base/package.json',
      'packages/cli/package.json',
      'packages/core/package.json',
    ]);
  });

  it('returns exactly the package.json paths of workspaces with a test:ci script', () => {
    const root = mkdtempSync(path.join(tmpdir(), 'qwen-workspaces-'));
    tempDirs.push(root);

    writeFile(
      root,
      'package.json',
      '{"workspaces": ["integrations/*", "packages/*", ' +
        '"packages/channels/nested-ci", "packages/channels/nested-no-ci"]}\n',
    );
    // Both repo layouts are witness-bearing: the flat integrations/* layout
    // and the nested packages/channels/* layout. The nested members need
    // explicit root-list entries because packages/* does not match nested
    // directories — exactly why a filter mutation that drops the subtree
    // (say `!path.startsWith('packages/channels/')`) would stay invisible
    // to every consumer that derives its set from the selector. This
    // exact-set pin is what catches it.
    writeFile(
      root,
      'integrations/with-ci/package.json',
      '{"scripts": {"test:ci": "vitest run"}}\n',
    );
    writeFile(
      root,
      'packages/without-ci/package.json',
      '{"scripts": {"test": "vitest run"}}\n',
    );
    writeFile(
      root,
      'packages/channels/nested-ci/package.json',
      '{"scripts": {"test:ci": "vitest run"}}\n',
    );
    writeFile(
      root,
      'packages/channels/nested-no-ci/package.json',
      '{"scripts": {"test": "vitest run"}}\n',
    );

    expect(getTestCiWorkspacePackageJsonPaths(root)).toEqual([
      'integrations/with-ci/package.json',
      'packages/channels/nested-ci/package.json',
    ]);
  });

  it('selects the workspaces npm runs test:ci in', () => {
    // npm resolves the workspaces itself for `npm run test:ci:workspaces`
    // and for the Linux test job's lane script
    // (.github/scripts/ci/run-unit-test-lanes.sh), which lists them with
    // `npm pkg get`. The pins that gate on this selector's set must name the
    // suites those runs start.
    const root = path.resolve(
      path.dirname(fileURLToPath(import.meta.url)),
      '../..',
    );
    const result = spawnSync(
      'npm',
      ['pkg', 'get', 'scripts.test:ci', '--workspaces', '--json'],
      { cwd: root, encoding: 'utf8', shell: process.platform === 'win32' },
    );
    expect(result.status, result.stderr).toBe(0);
    const npmNames = Object.entries(JSON.parse(result.stdout))
      .filter(([, script]) => typeof script === 'string')
      .map(([name]) => name)
      .sort();
    const selectorNames = getTestCiWorkspacePackageJsonPaths(root)
      .map(
        (packageJsonPath) =>
          JSON.parse(readFileSync(path.join(root, packageJsonPath), 'utf8'))
            .name,
      )
      .sort();
    expect(npmNames).toEqual(selectorNames);
    expect(npmNames).toContain('@qwen-code/web-shell');
  });

  function writeFile(root, relativePath, content) {
    const filePath = path.join(root, relativePath);
    mkdirSync(path.dirname(filePath), { recursive: true });
    writeFileSync(filePath, content);
  }
});