import path from 'path';
import picocolors from 'picocolors';

function traceSync<T>(prefix: string, fn: () => T): T {
  const start = Bun.nanoseconds();
  const result = fn();
  const end = Bun.nanoseconds();
  console.log(`${picocolors.gray(`[${((end - start) / 1e6).toFixed(3)}ms]`)} ${prefix}`);
  return result;
}
traceSync.skip = <T>(prefix: string, fn: () => T): T => fn();
export { traceSync };

const traceAsync = async <T>(prefix: string, fn: () => Promise<T>): Promise<T> => {
  const start = Bun.nanoseconds();
  const result = await fn();
  const end = Bun.nanoseconds();
  console.log(`${picocolors.gray(`[${((end - start) / 1e6).toFixed(3)}ms]`)} ${prefix}`);
  return result;
};
export { traceAsync };

export interface TaskResult {
  readonly start: number,
  readonly end: number,
  readonly taskName: string
}

const task = <T>(importMetaPath: string, fn: () => Promise<T>, customname: string | null = null) => {
  const taskName = customname ?? path.basename(importMetaPath, path.extname(importMetaPath));
  return async () => {
    console.log(`🏃 [${taskName}] Start executing`);
    const start = Bun.nanoseconds();
    await fn();
    const end = Bun.nanoseconds();
    console.log(`✅ [${taskName}] [${((end - start) / 1e6).toFixed(3)}ms] Executed successfully`);

    return { start, end, taskName } as TaskResult;
  };
};
export { task };