import path from 'node:path';
import type { RepomixConfigMerged } from '../../../config/configSchema.js';
import { type PackResult, pack } from '../../../core/packager.js';
import { logger, setLogLevelByWorkerData } from '../../../shared/logger.js';
import { Spinner } from '../../cliSpinner.js';
import type { CliOptions } from '../../types.js';
// Initialize logger configuration from workerData at module load time
// This must be called before any logging operations in the worker
setLogLevelByWorkerData();
export interface DefaultActionTask {
directories: string[];
cwd: string;
config: RepomixConfigMerged;
cliOptions: CliOptions;
stdinFilePaths?: string[];
}
export interface PingTask {
ping: true;
}
export interface DefaultActionWorkerResult {
packResult: PackResult;
config: RepomixConfigMerged;
}
export interface PingResult {
ping: true;
}
// Function overloads for better type inference
function defaultActionWorker(task: DefaultActionTask): Promise<DefaultActionWorkerResult>;
function defaultActionWorker(task: PingTask): Promise<PingResult>;
async function defaultActionWorker(
task: DefaultActionTask | PingTask,
): Promise<DefaultActionWorkerResult | PingResult> {
// Handle ping requests for Bun compatibility check
if ('ping' in task) {
return {
ping: true,
};
}
// Validate task structure
if (!task || typeof task !== 'object') {
throw new Error(`Invalid task: expected object, got ${typeof task}`);
}
// At this point, task is guaranteed to be DefaultActionTask
const { directories, cwd, config, cliOptions, stdinFilePaths } = task;
if (!directories || !Array.isArray(directories)) {
throw new Error('Invalid task: directories must be an array');
}
// Provide defaults for bundled environments where cliOptions might be undefined
const safeCliOptions: CliOptions = cliOptions ?? {};
logger.trace('Worker: Using pre-loaded config:', config);
const spinner = new Spinner('Initializing...', safeCliOptions);
spinner.start();
let packResult: PackResult;
try {
const { skillName, skillDir, skillProjectName, skillSourceUrl } = safeCliOptions;
const packOptions = { skillName, skillDir, skillProjectName, skillSourceUrl };
if (stdinFilePaths) {
// Handle stdin processing with file paths from main process
// File paths were already read from stdin in the main process
logger.trace(`Worker: Processing ${stdinFilePaths.length} files from stdin`);
// Use pack with predefined files from stdin
packResult = await pack(
[cwd],
config,
(message) => {
spinner.update(message);
},
{},
stdinFilePaths,
packOptions,
);
} else {
// Handle directory processing
const targetPaths = directories.map((directory) => path.resolve(cwd, directory));
packResult = await pack(
targetPaths,
config,
(message) => {
spinner.update(message);
},
{},
undefined,
packOptions,
);
}
spinner.succeed('Packing completed successfully!');
return {
packResult,
config,
};
} catch (error) {
spinner.fail('Error during packing');
throw error;
}
}
export default defaultActionWorker;
// Export cleanup function for Tinypool teardown
export const onWorkerTermination = async (): Promise<void> => {
// Any cleanup needed when worker terminates
// Currently no specific cleanup required for defaultAction worker
};
