@n1k1t/unit-generator/lib/src/commands/generate.js

Coverage based unit tests AI generator

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const utils_1 = require("./utils");
const models_1 = require("../models");
const env_1 = __importDefault(require("../env"));
exports.default = async (parameters = {}) => {
    const intervals = [];
    const render = (0, utils_1.buildRenderer)();
    const target = parameters.target ?? Number(env_1.default.target);
    const cwd = process.cwd();
    const ignore = await (0, utils_1.extractIgnoredPaths)(cwd);
    const extracted = await (0, utils_1.extractCoberturaItems)(env_1.default.cobertura, {
        ignore,
        target,
        cwd,
        paths: parameters.paths,
        limit: parameters.limit,
    });
    const assistants = await Promise.all(extracted.map((item) => models_1.Assistant.build(item.path, {
        cwd,
        target,
        model: parameters.model,
        iterations: parameters.iterations,
        rate: item.rate,
    })));
    intervals.push(setInterval(() => render(assistants), 100));
    if (env_1.default.parallel) {
        await Promise.all(assistants.map((assistant) => assistant.run()));
    }
    else {
        for (const assistant of assistants) {
            await assistant.run();
        }
    }
    intervals.forEach((interval) => interval.unref());
    render(assistants);
};
//# sourceMappingURL=generate.js.map