@nsxbet/playwright-orchestrator 0.2.0

package/dist/commands/extract-timing.js

@@ -0,0 +1,196 @@
+import * as fs from 'node:fs';
+import * as path from 'node:path';
+import { Command, Flags } from '@oclif/core';
+import { buildTestId } from '../core/index.js';
+export default class ExtractTiming extends Command {
+    static description = 'Extract timing data from Playwright JSON report (file-level or test-level)';
+    static examples = [
+        '<%= config.bin %> extract-timing --report-file ./playwright-report/results.json --output-file ./timing.json',
+        '<%= config.bin %> extract-timing --report-file ./results.json --shard 1 --project "Mobile Chrome" --level test',
+    ];
+    static flags = {
+        'report-file': Flags.string({
+            char: 'r',
+            description: 'Path to Playwright JSON report file',
+            required: true,
+        }),
+        'output-file': Flags.string({
+            char: 'o',
+            description: 'Path to write timing JSON output',
+        }),
+        shard: Flags.integer({
+            char: 's',
+            description: 'Shard index for the artifact',
+            default: 1,
+        }),
+        project: Flags.string({
+            char: 'p',
+            description: 'Playwright project name',
+            default: 'default',
+        }),
+        level: Flags.string({
+            char: 'l',
+            description: 'Extraction level: file or test',
+            default: 'test',
+            options: ['file', 'test'],
+        }),
+        verbose: Flags.boolean({
+            char: 'v',
+            description: 'Show verbose output',
+            default: false,
+        }),
+    };
+    async run() {
+        const { flags } = await this.parse(ExtractTiming);
+        // Read Playwright report
+        const reportPath = path.resolve(flags['report-file']);
+        let report;
+        try {
+            const content = fs.readFileSync(reportPath, 'utf-8');
+            report = JSON.parse(content);
+        }
+        catch {
+            this.error(`Failed to read Playwright report: ${reportPath}`);
+        }
+        let output;
+        if (flags.level === 'test') {
+            // Extract test-level durations
+            const testDurations = this.extractTestDurations(report);
+            if (flags.verbose) {
+                this.log(`Extracted timing for ${Object.keys(testDurations).length} tests`);
+            }
+            // Create test-level artifact
+            const artifact = {
+                shard: flags.shard,
+                project: flags.project,
+                tests: testDurations,
+            };
+            output = JSON.stringify(artifact, null, 2);
+        }
+        else {
+            // Extract file-level durations (legacy)
+            const fileDurations = this.extractFileDurations(report);
+            if (flags.verbose) {
+                this.log(`Extracted timing for ${Object.keys(fileDurations).length} files`);
+            }
+            // Create file-level artifact
+            const artifact = {
+                shard: flags.shard,
+                project: flags.project,
+                files: fileDurations,
+            };
+            output = JSON.stringify(artifact, null, 2);
+        }
+        // Output
+        if (flags['output-file']) {
+            fs.writeFileSync(flags['output-file'], output, 'utf-8');
+            if (flags.verbose) {
+                this.log(`Wrote timing data to ${flags['output-file']}`);
+            }
+        }
+        else {
+            this.log(output);
+        }
+    }
+    /**
+     * Extract test-level durations from Playwright report
+     *
+     * Each test is identified by: file::describe::testTitle
+     */
+    extractTestDurations(report) {
+        const testDurations = {};
+        for (const suite of report.suites) {
+            this.extractTestsFromSuite(suite, [], testDurations);
+        }
+        return testDurations;
+    }
+    /**
+     * Recursively extract test durations from a suite
+     */
+    extractTestsFromSuite(suite, parentTitles, testDurations) {
+        const file = this.normalizeFilePath(suite.file);
+        const currentTitles = suite.title && suite.title !== ''
+            ? [...parentTitles, suite.title]
+            : parentTitles;
+        // Process specs (actual tests)
+        if (suite.specs) {
+            for (const spec of suite.specs) {
+                const titlePath = [...currentTitles, spec.title];
+                const testId = buildTestId(file, titlePath);
+                // Sum all result durations (including retries)
+                let totalDuration = 0;
+                for (const test of spec.tests) {
+                    for (const result of test.results) {
+                        totalDuration += result.duration;
+                    }
+                }
+                testDurations[testId] = totalDuration;
+            }
+        }
+        // Process nested suites
+        if (suite.suites) {
+            for (const nestedSuite of suite.suites) {
+                // Pass the file from parent if nested suite doesn't have one
+                const nestedWithFile = {
+                    ...nestedSuite,
+                    file: nestedSuite.file || suite.file,
+                };
+                this.extractTestsFromSuite(nestedWithFile, currentTitles, testDurations);
+            }
+        }
+    }
+    /**
+     * Extract file durations from Playwright report (legacy file-level)
+     *
+     * The report structure has nested suites where the top-level suite
+     * represents the file. We sum up all test durations within each file.
+     */
+    extractFileDurations(report) {
+        const fileDurations = {};
+        for (const suite of report.suites) {
+            const file = this.normalizeFilePath(suite.file);
+            const duration = this.calculateSuiteDuration(suite);
+            fileDurations[file] = duration;
+        }
+        return fileDurations;
+    }
+    /**
+     * Normalize file path to be relative and consistent
+     */
+    normalizeFilePath(filePath) {
+        // Extract just the filename or relative path
+        // Remove any absolute path prefix
+        const parts = filePath.split('/');
+        // Find the index of common test directories
+        const testDirIndex = parts.findIndex((p) => p === 'e2e' || p === 'test' || p === '__tests__');
+        if (testDirIndex !== -1) {
+            return parts.slice(testDirIndex + 1).join('/');
+        }
+        // Just return the filename
+        return parts[parts.length - 1] ?? filePath;
+    }
+    /**
+     * Calculate total duration for a suite (recursively including nested suites)
+     */
+    calculateSuiteDuration(suite) {
+        let total = 0;
+        // Sum durations from specs
+        if (suite.specs) {
+            for (const spec of suite.specs) {
+                for (const test of spec.tests) {
+                    for (const result of test.results) {
+                        total += result.duration;
+                    }
+                }
+            }
+        }
+        // Sum durations from nested suites
+        if (suite.suites) {
+            for (const nestedSuite of suite.suites) {
+                total += this.calculateSuiteDuration(nestedSuite);
+            }
+        }
+        return total;
+    }
+}
+//# sourceMappingURL=extract-timing.js.map

package/dist/commands/extract-timing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"extract-timing.js","sourceRoot":"","sources":["../../src/commands/extract-timing.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAM7C,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAE/C,MAAM,CAAC,OAAO,OAAO,aAAc,SAAQ,OAAO;IAChD,MAAM,CAAU,WAAW,GACzB,4EAA4E,CAAC;IAE/E,MAAM,CAAU,QAAQ,GAAG;QACzB,6GAA6G;QAC7G,gHAAgH;KACjH,CAAC;IAEF,MAAM,CAAU,KAAK,GAAG;QACtB,aAAa,EAAE,KAAK,CAAC,MAAM,CAAC;YAC1B,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,qCAAqC;YAClD,QAAQ,EAAE,IAAI;SACf,CAAC;QACF,aAAa,EAAE,KAAK,CAAC,MAAM,CAAC;YAC1B,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,kCAAkC;SAChD,CAAC;QACF,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC;YACnB,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,8BAA8B;YAC3C,OAAO,EAAE,CAAC;SACX,CAAC;QACF,OAAO,EAAE,KAAK,CAAC,MAAM,CAAC;YACpB,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,yBAAyB;YACtC,OAAO,EAAE,SAAS;SACnB,CAAC;QACF,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC;YAClB,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,gCAAgC;YAC7C,OAAO,EAAE,MAAM;YACf,OAAO,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC;SAC1B,CAAC;QACF,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC;YACrB,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,qBAAqB;YAClC,OAAO,EAAE,KAAK;SACf,CAAC;KACH,CAAC;IAEF,KAAK,CAAC,GAAG;QACP,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;QAElD,yBAAyB;QACzB,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC;QACtD,IAAI,MAAwB,CAAC;QAE7B,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YACrD,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAqB,CAAC;QACnD,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,CAAC,KAAK,CAAC,qCAAqC,UAAU,EAAE,CAAC,CAAC;QAChE,CAAC;QAED,IAAI,MAAc,CAAC;QAEnB,IAAI,KAAK,CAAC,KAAK,KAAK,MAAM,EAAE,CAAC;YAC3B,+BAA+B;YAC/B,MAAM,aAAa,GAAG,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAExD,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBAClB,IAAI,CAAC,GAAG,CACN,wBAAwB,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,MAAM,QAAQ,CAClE,CAAC;YACJ,CAAC;YAED,6BAA6B;YAC7B,MAAM,QAAQ,GAA4B;gBACxC,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,KAAK,EAAE,aAAa;aACrB,CAAC;YAEF,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC7C,CAAC;aAAM,CAAC;YACN,wCAAwC;YACxC,MAAM,aAAa,GAAG,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAExD,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBAClB,IAAI,CAAC,GAAG,CACN,wBAAwB,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,MAAM,QAAQ,CAClE,CAAC;YACJ,CAAC;YAED,6BAA6B;YAC7B,MAAM,QAAQ,GAAwB;gBACpC,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,KAAK,EAAE,aAAa;aACrB,CAAC;YAEF,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC7C,CAAC;QAED,SAAS;QACT,IAAI,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC;YACzB,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YACxD,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBAClB,IAAI,CAAC,GAAG,CAAC,wBAAwB,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACnB,CAAC;IACH,CAAC;IAED;;;;OAIG;IACK,oBAAoB,CAC1B,MAAwB;QAExB,MAAM,aAAa,GAA2B,EAAE,CAAC;QAEjD,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClC,IAAI,CAAC,qBAAqB,CAAC,KAAK,EAAE,EAAE,EAAE,aAAa,CAAC,CAAC;QACvD,CAAC;QAED,OAAO,aAAa,CAAC;IACvB,CAAC;IAED;;OAEG;IACK,qBAAqB,CAC3B,KAAoC,EACpC,YAAsB,EACtB,aAAqC;QAErC,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,aAAa,GACjB,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK,KAAK,EAAE;YAC/B,CAAC,CAAC,CAAC,GAAG,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC;YAChC,CAAC,CAAC,YAAY,CAAC;QAEnB,+BAA+B;QAC/B,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YAChB,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;gBAC/B,MAAM,SAAS,GAAG,CAAC,GAAG,aAAa,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;gBACjD,MAAM,MAAM,GAAG,WAAW,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;gBAE5C,+CAA+C;gBAC/C,IAAI,aAAa,GAAG,CAAC,CAAC;gBACtB,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBAC9B,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;wBAClC,aAAa,IAAI,MAAM,CAAC,QAAQ,CAAC;oBACnC,CAAC;gBACH,CAAC;gBAED,aAAa,CAAC,MAAM,CAAC,GAAG,aAAa,CAAC;YACxC,CAAC;QACH,CAAC;QAED,wBAAwB;QACxB,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACjB,KAAK,MAAM,WAAW,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;gBACvC,6DAA6D;gBAC7D,MAAM,cAAc,GAAG;oBACrB,GAAG,WAAW;oBACd,IAAI,EAAE,WAAW,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI;iBACrC,CAAC;gBACF,IAAI,CAAC,qBAAqB,CACxB,cAAc,EACd,aAAa,EACb,aAAa,CACd,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACK,oBAAoB,CAC1B,MAAwB;QAExB,MAAM,aAAa,GAA2B,EAAE,CAAC;QAEjD,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YAClC,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;YACpD,aAAa,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC;QACjC,CAAC;QAED,OAAO,aAAa,CAAC;IACvB,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,QAAgB;QACxC,6CAA6C;QAC7C,kCAAkC;QAClC,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAElC,4CAA4C;QAC5C,MAAM,YAAY,GAAG,KAAK,CAAC,SAAS,CAClC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,KAAK,IAAI,CAAC,KAAK,MAAM,IAAI,CAAC,KAAK,WAAW,CACxD,CAAC;QAEF,IAAI,YAAY,KAAK,CAAC,CAAC,EAAE,CAAC;YACxB,OAAO,KAAK,CAAC,KAAK,CAAC,YAAY,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACjD,CAAC;QAED,2BAA2B;QAC3B,OAAO,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,QAAQ,CAAC;IAC7C,CAAC;IAED;;OAEG;IACK,sBAAsB,CAAC,KAAoC;QACjE,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,2BAA2B;QAC3B,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YAChB,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;gBAC/B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBAC9B,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;wBAClC,KAAK,IAAI,MAAM,CAAC,QAAQ,CAAC;oBAC3B,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,mCAAmC;QACnC,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACjB,KAAK,MAAM,WAAW,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;gBACvC,KAAK,IAAI,IAAI,CAAC,sBAAsB,CAAC,WAAW,CAAC,CAAC;YACpD,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC"}

package/dist/commands/list-tests.d.ts

@@ -0,0 +1,16 @@
+import { Command } from '@oclif/core';
+export default class ListTests extends Command {
+    static description: string;
+    static examples: string[];
+    static flags: {
+        'test-dir': import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        project: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        'output-format': import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        'glob-pattern': import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        'use-fallback': import("@oclif/core/interfaces").BooleanFlag<boolean>;
+        verbose: import("@oclif/core/interfaces").BooleanFlag<boolean>;
+    };
+    run(): Promise<void>;
+    private outputResult;
+}
+//# sourceMappingURL=list-tests.d.ts.map

package/dist/commands/list-tests.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"list-tests.d.ts","sourceRoot":"","sources":["../../src/commands/list-tests.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,EAAS,MAAM,aAAa,CAAC;AAQ7C,MAAM,CAAC,OAAO,OAAO,SAAU,SAAQ,OAAO;IAC5C,OAAgB,WAAW,SACiD;IAE5E,OAAgB,QAAQ,WAGtB;IAEF,OAAgB,KAAK;;;;;;;MA6BnB;IAEI,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;IAwC1B,OAAO,CAAC,YAAY;CA4BrB"}

package/dist/commands/list-tests.js

@@ -0,0 +1,98 @@
+import * as path from 'node:path';
+import { Command, Flags } from '@oclif/core';
+import { discoverTests, discoverTestsFromFiles, groupTestsByFile, } from '../core/index.js';
+export default class ListTests extends Command {
+    static description = 'Discover all tests in a Playwright project using --list or file parsing';
+    static examples = [
+        '<%= config.bin %> list-tests --test-dir ./src/test/e2e',
+        '<%= config.bin %> list-tests --test-dir ./e2e --project "Mobile Chrome" --output-format json',
+    ];
+    static flags = {
+        'test-dir': Flags.string({
+            char: 'd',
+            description: 'Path to test directory',
+            required: true,
+        }),
+        project: Flags.string({
+            char: 'p',
+            description: 'Playwright project name',
+        }),
+        'output-format': Flags.string({
+            char: 'f',
+            description: 'Output format',
+            default: 'json',
+            options: ['json', 'text'],
+        }),
+        'glob-pattern': Flags.string({
+            description: 'Glob pattern for test files (used for fallback discovery)',
+            default: '**/*.spec.ts',
+        }),
+        'use-fallback': Flags.boolean({
+            description: 'Use file parsing instead of Playwright --list',
+            default: false,
+        }),
+        verbose: Flags.boolean({
+            char: 'v',
+            description: 'Show verbose output',
+            default: false,
+        }),
+    };
+    async run() {
+        const { flags } = await this.parse(ListTests);
+        const testDir = path.resolve(flags['test-dir']);
+        if (flags.verbose) {
+            this.log(`Discovering tests in ${testDir}...`);
+        }
+        let tests;
+        try {
+            if (flags['use-fallback']) {
+                // Use file parsing
+                tests = discoverTestsFromFiles(testDir, flags['glob-pattern']);
+                if (flags.verbose) {
+                    this.log('Using file parsing for test discovery');
+                }
+            }
+            else {
+                // Use Playwright --list
+                tests = discoverTests(testDir, flags.project);
+                if (flags.verbose) {
+                    this.log('Using Playwright --list for test discovery');
+                }
+            }
+        }
+        catch {
+            // Fallback to file parsing if Playwright --list fails
+            if (flags.verbose) {
+                this.warn('Playwright --list failed, falling back to file parsing');
+            }
+            tests = discoverTestsFromFiles(testDir, flags['glob-pattern']);
+        }
+        if (flags.verbose) {
+            this.log(`Discovered ${tests.length} tests`);
+        }
+        this.outputResult(tests, flags['output-format'], flags.verbose);
+    }
+    outputResult(tests, format, verbose = false) {
+        if (format === 'json') {
+            this.log(JSON.stringify(tests));
+        }
+        else {
+            // Text format - group by file
+            const grouped = groupTestsByFile(tests);
+            this.log('\n=== Discovered Tests ===\n');
+            for (const [file, fileTests] of grouped) {
+                this.log(`${file}:`);
+                for (const test of fileTests) {
+                    const titlePath = test.titlePath.join(' > ');
+                    this.log(`  - ${titlePath}`);
+                    if (verbose) {
+                        this.log(`    ID: ${test.testId}`);
+                    }
+                }
+                this.log('');
+            }
+            this.log(`Total: ${tests.length} tests in ${grouped.size} files`);
+        }
+    }
+}
+//# sourceMappingURL=list-tests.js.map

package/dist/commands/list-tests.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"list-tests.js","sourceRoot":"","sources":["../../src/commands/list-tests.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAEL,aAAa,EACb,sBAAsB,EACtB,gBAAgB,GACjB,MAAM,kBAAkB,CAAC;AAE1B,MAAM,CAAC,OAAO,OAAO,SAAU,SAAQ,OAAO;IAC5C,MAAM,CAAU,WAAW,GACzB,yEAAyE,CAAC;IAE5E,MAAM,CAAU,QAAQ,GAAG;QACzB,wDAAwD;QACxD,8FAA8F;KAC/F,CAAC;IAEF,MAAM,CAAU,KAAK,GAAG;QACtB,UAAU,EAAE,KAAK,CAAC,MAAM,CAAC;YACvB,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,wBAAwB;YACrC,QAAQ,EAAE,IAAI;SACf,CAAC;QACF,OAAO,EAAE,KAAK,CAAC,MAAM,CAAC;YACpB,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,yBAAyB;SACvC,CAAC;QACF,eAAe,EAAE,KAAK,CAAC,MAAM,CAAC;YAC5B,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,eAAe;YAC5B,OAAO,EAAE,MAAM;YACf,OAAO,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC;SAC1B,CAAC;QACF,cAAc,EAAE,KAAK,CAAC,MAAM,CAAC;YAC3B,WAAW,EAAE,2DAA2D;YACxE,OAAO,EAAE,cAAc;SACxB,CAAC;QACF,cAAc,EAAE,KAAK,CAAC,OAAO,CAAC;YAC5B,WAAW,EAAE,+CAA+C;YAC5D,OAAO,EAAE,KAAK;SACf,CAAC;QACF,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC;YACrB,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,qBAAqB;YAClC,OAAO,EAAE,KAAK;SACf,CAAC;KACH,CAAC;IAEF,KAAK,CAAC,GAAG;QACP,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAE9C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC;QAEhD,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,GAAG,CAAC,wBAAwB,OAAO,KAAK,CAAC,CAAC;QACjD,CAAC;QAED,IAAI,KAAuB,CAAC;QAE5B,IAAI,CAAC;YACH,IAAI,KAAK,CAAC,cAAc,CAAC,EAAE,CAAC;gBAC1B,mBAAmB;gBACnB,KAAK,GAAG,sBAAsB,CAAC,OAAO,EAAE,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;gBAC/D,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;oBAClB,IAAI,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC;gBACpD,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,wBAAwB;gBACxB,KAAK,GAAG,aAAa,CAAC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;gBAC9C,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;oBAClB,IAAI,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;gBACzD,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,sDAAsD;YACtD,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBAClB,IAAI,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;YACtE,CAAC;YACD,KAAK,GAAG,sBAAsB,CAAC,OAAO,EAAE,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;QACjE,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,GAAG,CAAC,cAAc,KAAK,CAAC,MAAM,QAAQ,CAAC,CAAC;QAC/C,CAAC;QAED,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,eAAe,CAAC,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IAClE,CAAC;IAEO,YAAY,CAClB,KAAuB,EACvB,MAAc,EACd,OAAO,GAAG,KAAK;QAEf,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;YACtB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;QAClC,CAAC;aAAM,CAAC;YACN,8BAA8B;YAC9B,MAAM,OAAO,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;YAExC,IAAI,CAAC,GAAG,CAAC,8BAA8B,CAAC,CAAC;YAEzC,KAAK,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,OAAO,EAAE,CAAC;gBACxC,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC;gBACrB,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;oBAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBAC7C,IAAI,CAAC,GAAG,CAAC,OAAO,SAAS,EAAE,CAAC,CAAC;oBAC7B,IAAI,OAAO,EAAE,CAAC;wBACZ,IAAI,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;oBACrC,CAAC;gBACH,CAAC;gBACD,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACf,CAAC;YAED,IAAI,CAAC,GAAG,CAAC,UAAU,KAAK,CAAC,MAAM,aAAa,OAAO,CAAC,IAAI,QAAQ,CAAC,CAAC;QACpE,CAAC;IACH,CAAC"}

package/dist/commands/merge-timing.d.ts

@@ -0,0 +1,19 @@
+import { Command } from '@oclif/core';
+export default class MergeTiming extends Command {
+    static description: string;
+    static examples: string[];
+    static flags: {
+        existing: import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        new: import("@oclif/core/interfaces").OptionFlag<string[], import("@oclif/core/interfaces").CustomOptions>;
+        output: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        alpha: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        'prune-days': import("@oclif/core/interfaces").OptionFlag<number, import("@oclif/core/interfaces").CustomOptions>;
+        'current-files': import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        level: import("@oclif/core/interfaces").OptionFlag<string, import("@oclif/core/interfaces").CustomOptions>;
+        verbose: import("@oclif/core/interfaces").BooleanFlag<boolean>;
+    };
+    run(): Promise<void>;
+    private runTestLevel;
+    private runFileLevel;
+}
+//# sourceMappingURL=merge-timing.d.ts.map

package/dist/commands/merge-timing.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"merge-timing.d.ts","sourceRoot":"","sources":["../../src/commands/merge-timing.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,OAAO,EAAS,MAAM,aAAa,CAAC;AAgB7C,MAAM,CAAC,OAAO,OAAO,WAAY,SAAQ,OAAO;IAC9C,OAAgB,WAAW,SACiD;IAE5E,OAAgB,QAAQ,WAItB;IAEF,OAAgB,KAAK;;;;;;;;;MAwCnB;IAEI,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;YAeZ,YAAY;YAkHZ,YAAY;CA4G3B"}

package/dist/commands/merge-timing.js

@@ -0,0 +1,217 @@
+import * as fs from 'node:fs';
+import * as path from 'node:path';
+import { Command, Flags } from '@oclif/core';
+import { DEFAULT_EMA_ALPHA, DEFAULT_PRUNE_DAYS, isTimingDataV2, loadTimingData, mergeTestTimingData, mergeTimingData, pruneTestTimingData, pruneTimingData, saveTimingData, } from '../core/index.js';
+export default class MergeTiming extends Command {
+    static description = 'Merge timing data from multiple shards using Exponential Moving Average';
+    static examples = [
+        '<%= config.bin %> merge-timing --existing ./timing.json --new ./shard-1.json ./shard-2.json --output ./timing.json',
+        '<%= config.bin %> merge-timing --new ./shard-*.json --output ./timing.json --alpha 0.3 --prune-days 30',
+        '<%= config.bin %> merge-timing --new ./shard-*.json --output ./timing.json --level test',
+    ];
+    static flags = {
+        existing: Flags.string({
+            char: 'e',
+            description: 'Path to existing timing data JSON (optional)',
+        }),
+        new: Flags.string({
+            char: 'n',
+            description: 'Paths to new timing artifact files',
+            multiple: true,
+            required: true,
+        }),
+        output: Flags.string({
+            char: 'o',
+            description: 'Path to write merged timing data',
+            required: true,
+        }),
+        alpha: Flags.string({
+            char: 'a',
+            description: 'EMA smoothing factor (0-1)',
+            default: String(DEFAULT_EMA_ALPHA),
+        }),
+        'prune-days': Flags.integer({
+            description: 'Remove entries older than N days',
+            default: DEFAULT_PRUNE_DAYS,
+        }),
+        'current-files': Flags.string({
+            description: 'Path to file listing current test files/IDs (for pruning deleted tests)',
+        }),
+        level: Flags.string({
+            char: 'l',
+            description: 'Data level: file or test',
+            default: 'test',
+            options: ['file', 'test'],
+        }),
+        verbose: Flags.boolean({
+            char: 'v',
+            description: 'Show verbose output',
+            default: false,
+        }),
+    };
+    async run() {
+        const { flags } = await this.parse(MergeTiming);
+        const alpha = Number.parseFloat(flags.alpha);
+        if (Number.isNaN(alpha) || alpha < 0 || alpha > 1) {
+            this.error('Alpha must be a number between 0 and 1');
+        }
+        if (flags.level === 'test') {
+            await this.runTestLevel(flags, alpha);
+        }
+        else {
+            await this.runFileLevel(flags, alpha);
+        }
+    }
+    async runTestLevel(flags, alpha) {
+        // Load existing timing data
+        let existingData = null;
+        if (flags.existing) {
+            const loaded = loadTimingData(flags.existing);
+            if (isTimingDataV2(loaded)) {
+                existingData = loaded;
+            }
+            else if (flags.verbose) {
+                this.warn('Existing timing data is v1 (file-level), starting fresh for test-level');
+            }
+        }
+        if (flags.verbose) {
+            const testCount = existingData
+                ? Object.keys(existingData.tests).length
+                : 0;
+            this.log(`Loaded existing timing data with ${testCount} tests`);
+        }
+        // Load new timing artifacts
+        const newArtifacts = [];
+        for (const artifactPath of flags.new) {
+            try {
+                const content = fs.readFileSync(path.resolve(artifactPath), 'utf-8');
+                const artifact = JSON.parse(content);
+                // Check if it has tests (v2) or files (v1)
+                if (artifact.tests) {
+                    newArtifacts.push(artifact);
+                    if (flags.verbose) {
+                        this.log(`Loaded timing artifact from shard ${artifact.shard} with ${Object.keys(artifact.tests).length} tests`);
+                    }
+                }
+                else if (flags.verbose) {
+                    this.warn(`Artifact ${artifactPath} is file-level, skipping for test-level merge`);
+                }
+            }
+            catch {
+                this.warn(`Failed to load timing artifact: ${artifactPath}`);
+            }
+        }
+        if (newArtifacts.length === 0) {
+            this.warn('No valid test-level timing artifacts found');
+            return;
+        }
+        // Merge timing data using EMA
+        let mergedData = mergeTestTimingData(existingData, newArtifacts, alpha);
+        if (flags.verbose) {
+            this.log(`Merged timing data now has ${Object.keys(mergedData.tests).length} tests`);
+        }
+        // Load current test IDs for pruning if provided
+        let currentTestIds;
+        if (flags['current-files']) {
+            try {
+                const content = fs.readFileSync(flags['current-files'], 'utf-8');
+                currentTestIds = content.split('\n').filter((line) => line.trim());
+            }
+            catch {
+                this.warn(`Failed to load current test IDs list: ${flags['current-files']}`);
+            }
+        }
+        // Prune old entries
+        const beforePrune = Object.keys(mergedData.tests).length;
+        mergedData = pruneTestTimingData(mergedData, flags['prune-days'], currentTestIds);
+        const afterPrune = Object.keys(mergedData.tests).length;
+        if (flags.verbose && beforePrune !== afterPrune) {
+            this.log(`Pruned ${beforePrune - afterPrune} old entries`);
+        }
+        // Save merged data
+        saveTimingData(flags.output, mergedData);
+        if (flags.verbose) {
+            this.log(`Saved merged timing data to ${flags.output}`);
+        }
+        // Output summary
+        this.log(JSON.stringify({
+            tests: Object.keys(mergedData.tests).length,
+            updatedAt: mergedData.updatedAt,
+            version: mergedData.version,
+        }));
+    }
+    async runFileLevel(flags, alpha) {
+        // Load existing timing data
+        const existingData = flags.existing
+            ? loadTimingData(flags.existing)
+            : { version: 1, updatedAt: new Date().toISOString(), files: {} };
+        if (flags.verbose) {
+            const fileCount = 'files' in existingData ? Object.keys(existingData.files).length : 0;
+            this.log(`Loaded existing timing data with ${fileCount} files`);
+        }
+        // Load new timing artifacts
+        const newArtifacts = [];
+        for (const artifactPath of flags.new) {
+            try {
+                const content = fs.readFileSync(path.resolve(artifactPath), 'utf-8');
+                const artifact = JSON.parse(content);
+                // Check if it has files (v1)
+                if (artifact.files) {
+                    newArtifacts.push(artifact);
+                    if (flags.verbose) {
+                        this.log(`Loaded timing artifact from shard ${artifact.shard} with ${Object.keys(artifact.files).length} files`);
+                    }
+                }
+                else if (flags.verbose) {
+                    this.warn(`Artifact ${artifactPath} is test-level, skipping for file-level merge`);
+                }
+            }
+            catch {
+                this.warn(`Failed to load timing artifact: ${artifactPath}`);
+            }
+        }
+        if (newArtifacts.length === 0) {
+            this.warn('No valid file-level timing artifacts found');
+            return;
+        }
+        // Merge timing data using EMA
+        let mergedData = mergeTimingData(existingData, newArtifacts, alpha);
+        if (flags.verbose) {
+            this.log(`Merged timing data now has ${Object.keys(mergedData.files).length} files`);
+        }
+        // Load current files for pruning if provided
+        let currentFiles;
+        if (flags['current-files']) {
+            try {
+                const content = fs.readFileSync(flags['current-files'], 'utf-8');
+                currentFiles = content.split('\n').filter((line) => line.trim());
+            }
+            catch {
+                this.warn(`Failed to load current files list: ${flags['current-files']}`);
+            }
+        }
+        // Prune old entries
+        const beforePrune = Object.keys(mergedData.files).length;
+        const prunedData = pruneTimingData(mergedData, flags['prune-days'], currentFiles);
+        // pruneTimingData can return v1 or v2, but for file-level we expect v1
+        if ('files' in prunedData) {
+            mergedData = prunedData;
+        }
+        const afterPrune = Object.keys(mergedData.files).length;
+        if (flags.verbose && beforePrune !== afterPrune) {
+            this.log(`Pruned ${beforePrune - afterPrune} old entries`);
+        }
+        // Save merged data
+        saveTimingData(flags.output, mergedData);
+        if (flags.verbose) {
+            this.log(`Saved merged timing data to ${flags.output}`);
+        }
+        // Output summary
+        this.log(JSON.stringify({
+            files: Object.keys(mergedData.files).length,
+            updatedAt: mergedData.updatedAt,
+            version: mergedData.version,
+        }));
+    }
+}
+//# sourceMappingURL=merge-timing.js.map

package/dist/commands/merge-timing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"merge-timing.js","sourceRoot":"","sources":["../../src/commands/merge-timing.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EACL,iBAAiB,EACjB,kBAAkB,EAClB,cAAc,EACd,cAAc,EACd,mBAAmB,EACnB,eAAe,EACf,mBAAmB,EACnB,eAAe,EAEf,cAAc,GAGf,MAAM,kBAAkB,CAAC;AAE1B,MAAM,CAAC,OAAO,OAAO,WAAY,SAAQ,OAAO;IAC9C,MAAM,CAAU,WAAW,GACzB,yEAAyE,CAAC;IAE5E,MAAM,CAAU,QAAQ,GAAG;QACzB,oHAAoH;QACpH,wGAAwG;QACxG,yFAAyF;KAC1F,CAAC;IAEF,MAAM,CAAU,KAAK,GAAG;QACtB,QAAQ,EAAE,KAAK,CAAC,MAAM,CAAC;YACrB,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,8CAA8C;SAC5D,CAAC;QACF,GAAG,EAAE,KAAK,CAAC,MAAM,CAAC;YAChB,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,oCAAoC;YACjD,QAAQ,EAAE,IAAI;YACd,QAAQ,EAAE,IAAI;SACf,CAAC;QACF,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC;YACnB,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,kCAAkC;YAC/C,QAAQ,EAAE,IAAI;SACf,CAAC;QACF,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC;YAClB,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,4BAA4B;YACzC,OAAO,EAAE,MAAM,CAAC,iBAAiB,CAAC;SACnC,CAAC;QACF,YAAY,EAAE,KAAK,CAAC,OAAO,CAAC;YAC1B,WAAW,EAAE,kCAAkC;YAC/C,OAAO,EAAE,kBAAkB;SAC5B,CAAC;QACF,eAAe,EAAE,KAAK,CAAC,MAAM,CAAC;YAC5B,WAAW,EACT,yEAAyE;SAC5E,CAAC;QACF,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC;YAClB,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,0BAA0B;YACvC,OAAO,EAAE,MAAM;YACf,OAAO,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC;SAC1B,CAAC;QACF,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC;YACrB,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,qBAAqB;YAClC,OAAO,EAAE,KAAK;SACf,CAAC;KACH,CAAC;IAEF,KAAK,CAAC,GAAG;QACP,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAEhD,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC7C,IAAI,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YAClD,IAAI,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,KAAK,CAAC,KAAK,KAAK,MAAM,EAAE,CAAC;YAC3B,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,KAOC,EACD,KAAa;QAEb,4BAA4B;QAC5B,IAAI,YAAY,GAAwB,IAAI,CAAC;QAC7C,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YACnB,MAAM,MAAM,GAAG,cAAc,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC9C,IAAI,cAAc,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC3B,YAAY,GAAG,MAAM,CAAC;YACxB,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBACzB,IAAI,CAAC,IAAI,CACP,wEAAwE,CACzE,CAAC;YACJ,CAAC;QACH,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,MAAM,SAAS,GAAG,YAAY;gBAC5B,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,MAAM;gBACxC,CAAC,CAAC,CAAC,CAAC;YACN,IAAI,CAAC,GAAG,CAAC,oCAAoC,SAAS,QAAQ,CAAC,CAAC;QAClE,CAAC;QAED,4BAA4B;QAC5B,MAAM,YAAY,GAA8B,EAAE,CAAC;QAEnD,KAAK,MAAM,YAAY,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;YACrC,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,OAAO,CAAC,CAAC;gBACrE,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAA4B,CAAC;gBAEhE,2CAA2C;gBAC3C,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;oBACnB,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;oBAC5B,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;wBAClB,IAAI,CAAC,GAAG,CACN,qCAAqC,QAAQ,CAAC,KAAK,SAAS,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,MAAM,QAAQ,CACvG,CAAC;oBACJ,CAAC;gBACH,CAAC;qBAAM,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;oBACzB,IAAI,CAAC,IAAI,CACP,YAAY,YAAY,+CAA+C,CACxE,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,IAAI,CAAC,IAAI,CAAC,mCAAmC,YAAY,EAAE,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;QAED,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,IAAI,CAAC,4CAA4C,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QAED,8BAA8B;QAC9B,IAAI,UAAU,GAAG,mBAAmB,CAAC,YAAY,EAAE,YAAY,EAAE,KAAK,CAAC,CAAC;QAExE,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,GAAG,CACN,8BAA8B,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM,QAAQ,CAC3E,CAAC;QACJ,CAAC;QAED,gDAAgD;QAChD,IAAI,cAAoC,CAAC;QACzC,IAAI,KAAK,CAAC,eAAe,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,eAAe,CAAC,EAAE,OAAO,CAAC,CAAC;gBACjE,cAAc,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YACrE,CAAC;YAAC,MAAM,CAAC;gBACP,IAAI,CAAC,IAAI,CACP,yCAAyC,KAAK,CAAC,eAAe,CAAC,EAAE,CAClE,CAAC;YACJ,CAAC;QACH,CAAC;QAED,oBAAoB;QACpB,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC;QACzD,UAAU,GAAG,mBAAmB,CAC9B,UAAU,EACV,KAAK,CAAC,YAAY,CAAC,EACnB,cAAc,CACf,CAAC;QACF,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC;QAExD,IAAI,KAAK,CAAC,OAAO,IAAI,WAAW,KAAK,UAAU,EAAE,CAAC;YAChD,IAAI,CAAC,GAAG,CAAC,UAAU,WAAW,GAAG,UAAU,cAAc,CAAC,CAAC;QAC7D,CAAC;QAED,mBAAmB;QACnB,cAAc,CAAC,KAAK,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAEzC,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,GAAG,CAAC,+BAA+B,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QAC1D,CAAC;QAED,iBAAiB;QACjB,IAAI,CAAC,GAAG,CACN,IAAI,CAAC,SAAS,CAAC;YACb,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM;YAC3C,SAAS,EAAE,UAAU,CAAC,SAAS;YAC/B,OAAO,EAAE,UAAU,CAAC,OAAO;SAC5B,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,KAOC,EACD,KAAa;QAEb,4BAA4B;QAC5B,MAAM,YAAY,GAAG,KAAK,CAAC,QAAQ;YACjC,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,QAAQ,CAAC;YAChC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAU,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QAE5E,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,MAAM,SAAS,GACb,OAAO,IAAI,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YACvE,IAAI,CAAC,GAAG,CAAC,oCAAoC,SAAS,QAAQ,CAAC,CAAC;QAClE,CAAC;QAED,4BAA4B;QAC5B,MAAM,YAAY,GAA0B,EAAE,CAAC;QAE/C,KAAK,MAAM,YAAY,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;YACrC,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,OAAO,CAAC,CAAC;gBACrE,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAwB,CAAC;gBAE5D,6BAA6B;gBAC7B,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;oBACnB,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;oBAC5B,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;wBAClB,IAAI,CAAC,GAAG,CACN,qCAAqC,QAAQ,CAAC,KAAK,SAAS,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,MAAM,QAAQ,CACvG,CAAC;oBACJ,CAAC;gBACH,CAAC;qBAAM,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;oBACzB,IAAI,CAAC,IAAI,CACP,YAAY,YAAY,+CAA+C,CACxE,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,IAAI,CAAC,IAAI,CAAC,mCAAmC,YAAY,EAAE,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;QAED,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,IAAI,CAAC,4CAA4C,CAAC,CAAC;YACxD,OAAO;QACT,CAAC;QAED,8BAA8B;QAC9B,IAAI,UAAU,GAAG,eAAe,CAAC,YAAY,EAAE,YAAY,EAAE,KAAK,CAAC,CAAC;QAEpE,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,GAAG,CACN,8BAA8B,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM,QAAQ,CAC3E,CAAC;QACJ,CAAC;QAED,6CAA6C;QAC7C,IAAI,YAAkC,CAAC;QACvC,IAAI,KAAK,CAAC,eAAe,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,eAAe,CAAC,EAAE,OAAO,CAAC,CAAC;gBACjE,YAAY,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YACnE,CAAC;YAAC,MAAM,CAAC;gBACP,IAAI,CAAC,IAAI,CACP,sCAAsC,KAAK,CAAC,eAAe,CAAC,EAAE,CAC/D,CAAC;YACJ,CAAC;QACH,CAAC;QAED,oBAAoB;QACpB,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC;QACzD,MAAM,UAAU,GAAG,eAAe,CAChC,UAAU,EACV,KAAK,CAAC,YAAY,CAAC,EACnB,YAAY,CACb,CAAC;QACF,uEAAuE;QACvE,IAAI,OAAO,IAAI,UAAU,EAAE,CAAC;YAC1B,UAAU,GAAG,UAA+B,CAAC;QAC/C,CAAC;QACD,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC;QAExD,IAAI,KAAK,CAAC,OAAO,IAAI,WAAW,KAAK,UAAU,EAAE,CAAC;YAChD,IAAI,CAAC,GAAG,CAAC,UAAU,WAAW,GAAG,UAAU,cAAc,CAAC,CAAC;QAC7D,CAAC;QAED,mBAAmB;QACnB,cAAc,CAAC,KAAK,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAEzC,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,GAAG,CAAC,+BAA+B,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QAC1D,CAAC;QAED,iBAAiB;QACjB,IAAI,CAAC,GAAG,CACN,IAAI,CAAC,SAAS,CAAC;YACb,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM;YAC3C,SAAS,EAAE,UAAU,CAAC,SAAS;YAC/B,OAAO,EAAE,UAAU,CAAC,OAAO;SAC5B,CAAC,CACH,CAAC;IACJ,CAAC"}

package/dist/core/ckk-algorithm.d.ts

@@ -0,0 +1,38 @@
+import type { TestShardAssignment, TestWithDuration } from './types.js';
+/**
+ * Default timeout for CKK algorithm in milliseconds
+ */
+export declare const DEFAULT_CKK_TIMEOUT = 500;
+/**
+ * Result from the CKK algorithm
+ */
+export interface CKKResult {
+    /** Shard assignments */
+    assignments: TestShardAssignment[];
+    /** Maximum shard duration (makespan) */
+    makespan: number;
+    /** Whether the solution is provably optimal */
+    isOptimal: boolean;
+}
+/**
+ * Complete Karmarkar-Karp (CKK) inspired algorithm for multi-way number partitioning
+ *
+ * This algorithm finds the optimal distribution of tests across shards by:
+ * 1. Using branch and bound search with aggressive pruning
+ * 2. Starting with LPT solution as upper bound
+ * 3. Exploring assignment tree, pruning when partial makespan exceeds best
+ *
+ * Falls back to LPT if search exceeds timeout.
+ *
+ * @param tests - Tests with their durations
+ * @param numShards - Number of shards to distribute across
+ * @param timeoutMs - Maximum time to search for optimal solution
+ * @returns Optimal (or near-optimal) shard assignments
+ */
+export declare function assignWithCKK(tests: TestWithDuration[], numShards: number, timeoutMs?: number): CKKResult;
+/**
+ * Calculate theoretical lower bound for makespan
+ * This is the best possible makespan if we could partition perfectly
+ */
+export declare function calculateLowerBound(tests: TestWithDuration[], numShards: number): number;
+//# sourceMappingURL=ckk-algorithm.d.ts.map

package/dist/core/ckk-algorithm.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ckk-algorithm.d.ts","sourceRoot":"","sources":["../../src/core/ckk-algorithm.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAExE;;GAEG;AACH,eAAO,MAAM,mBAAmB,MAAM,CAAC;AAEvC;;GAEG;AACH,MAAM,WAAW,SAAS;IACxB,wBAAwB;IACxB,WAAW,EAAE,mBAAmB,EAAE,CAAC;IACnC,wCAAwC;IACxC,QAAQ,EAAE,MAAM,CAAC;IACjB,+CAA+C;IAC/C,SAAS,EAAE,OAAO,CAAC;CACpB;AAED;;;;;;;;;;;;;;GAcG;AACH,wBAAgB,aAAa,CAC3B,KAAK,EAAE,gBAAgB,EAAE,EACzB,SAAS,EAAE,MAAM,EACjB,SAAS,GAAE,MAA4B,GACtC,SAAS,CA8HX;AA4ED;;;GAGG;AACH,wBAAgB,mBAAmB,CACjC,KAAK,EAAE,gBAAgB,EAAE,EACzB,SAAS,EAAE,MAAM,GAChB,MAAM,CAQR"}