114 lines
7.9 KiB
JavaScript
114 lines
7.9 KiB
JavaScript
"use strict";
|
|
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
if (k2 === undefined) k2 = k;
|
|
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
}
|
|
Object.defineProperty(o, k2, desc);
|
|
}) : (function(o, m, k, k2) {
|
|
if (k2 === undefined) k2 = k;
|
|
o[k2] = m[k];
|
|
}));
|
|
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
}) : function(o, v) {
|
|
o["default"] = v;
|
|
});
|
|
var __importStar = (this && this.__importStar) || (function () {
|
|
var ownKeys = function(o) {
|
|
ownKeys = Object.getOwnPropertyNames || function (o) {
|
|
var ar = [];
|
|
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
|
|
return ar;
|
|
};
|
|
return ownKeys(o);
|
|
};
|
|
return function (mod) {
|
|
if (mod && mod.__esModule) return mod;
|
|
var result = {};
|
|
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
|
|
__setModuleDefault(result, mod);
|
|
return result;
|
|
};
|
|
})();
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.load = exports.resolve_package_path = exports.defaultTemplate = void 0;
|
|
const osr_commons_1 = require("@plastichub/osr-commons");
|
|
const exists_1 = require("@plastichub/fs/exists");
|
|
const path = __importStar(require("node:path"));
|
|
const tools_1 = require("@plastichub/osr-ai-tools/lib/tools/tools");
|
|
const index_1 = require("./index");
|
|
const aspects_simple_1 = require("./utils/aspects_simple");
|
|
exports.defaultTemplate = {
|
|
include: [
|
|
"!node_modules/**/*",
|
|
"!dist/**/*",
|
|
"!build/**/*",
|
|
"!coverage/**/*",
|
|
"!*.log",
|
|
"!.kbot",
|
|
"!.git",
|
|
"*"
|
|
],
|
|
messages: [],
|
|
tools: ['fs', 'git', 'interact', 'terminal', 'search', 'web', 'email', 'user']
|
|
};
|
|
const resolve_package_path = (dir) => (0, osr_commons_1.resolve)(path.join(__dirname, dir || ''));
|
|
exports.resolve_package_path = resolve_package_path;
|
|
const loadToolFromPath = async (toolPath, target, options) => {
|
|
try {
|
|
const absolutePath = path.resolve(toolPath);
|
|
if (!(0, exists_1.sync)(absolutePath)) {
|
|
index_1.logger.warn(`Tool path not found: ${absolutePath}`);
|
|
return null;
|
|
}
|
|
const toolModule = await Promise.resolve(`${absolutePath}`).then(s => __importStar(require(s)));
|
|
if (typeof toolModule.default === 'function') {
|
|
return toolModule.default(target, options);
|
|
}
|
|
else {
|
|
index_1.logger.warn(`Invalid tool module at ${toolPath}: no default export function`);
|
|
return null;
|
|
}
|
|
}
|
|
catch (e) {
|
|
index_1.logger.error(`Error loading tool from path ${toolPath}:`, e.message);
|
|
return null;
|
|
}
|
|
};
|
|
const load = async (options) => {
|
|
const target = path.resolve(options.output || options.path);
|
|
const toolsList = options.tools || exports.defaultTemplate.tools;
|
|
const enabledTools = toolsList.filter(tool => !options.disable.includes(tool));
|
|
const loadedTools = await Promise.all(enabledTools.map(async (tool) => {
|
|
if (tools_1.tools[tool]) {
|
|
try {
|
|
return tools_1.tools[tool](target, options);
|
|
}
|
|
catch (e) {
|
|
index_1.logger.error(`Error loading built-in tool ${tool}`, e.message);
|
|
return null;
|
|
}
|
|
}
|
|
return loadToolFromPath(tool, target, options);
|
|
}));
|
|
const tools = loadedTools
|
|
.flat()
|
|
.filter((tool) => {
|
|
if (!tool)
|
|
return false;
|
|
if (options.disableTools.includes(tool.function.name))
|
|
return false;
|
|
if (tool.function.name === 'completion')
|
|
return false;
|
|
return true;
|
|
});
|
|
tools.forEach(tool => {
|
|
tool.function.function = (0, aspects_simple_1.before)(tool.function.function, options.collector.onToolBefore);
|
|
tool.function.function = (0, aspects_simple_1.after)(tool.function.function, options.collector.onToolAfter);
|
|
});
|
|
return tools;
|
|
};
|
|
exports.load = load;
|
|
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidG9vbHMuanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyJzcmMvdG9vbHMudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6Ijs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7O0FBQ0EseURBQWlEO0FBRWpELGtEQUFzRDtBQUN0RCxnREFBaUM7QUFFakMsb0VBQWdGO0FBQ2hGLG1DQUFpQztBQUlqQywyREFBc0Q7QUFLekMsUUFBQSxlQUFlLEdBQUc7SUFDM0IsT0FBTyxFQUFFO1FBQ0wsb0JBQW9CO1FBQ3BCLFlBQVk7UUFDWixhQUFhO1FBQ2IsZ0JBQWdCO1FBQ2hCLFFBQVE7UUFDUixRQUFRO1FBQ1IsT0FBTztRQUNQLEdBQUc7S0FDTjtJQUNELFFBQVEsRUFBRSxFQUFFO0lBQ1osS0FBSyxFQUFFLENBQUMsSUFBSSxFQUFFLEtBQUssRUFBRSxVQUFVLEVBQUUsVUFBVSxFQUFFLFFBQVEsRUFBRSxLQUFLLEVBQUUsT0FBTyxFQUFFLE1BQU0sQ0FBQztDQUNqRixDQUFBO0FBQ00sTUFBTSxvQkFBb0IsR0FBRyxDQUFDLEdBQVcsRUFBVSxFQUFFLENBQUMsSUFBQSxxQkFBTyxFQUFDLElBQUksQ0FBQyxJQUFJLENBQUMsU0FBUyxFQUFFLEdBQUcsSUFBSSxFQUFFLENBQUMsQ0FBQyxDQUFBO0FBQXhGLFFBQUEsb0JBQW9CLHdCQUFvRTtBQUNyRyxNQUFNLGdCQUFnQixHQUFHLEtBQUssRUFBRSxRQUFnQixFQUFFLE1BQWMsRUFBRSxPQUFxQixFQUFFLEVBQUU7SUFDdkYsSUFBSSxDQUFDO1FBQ0QsTUFBTSxZQUFZLEdBQUcsSUFBSSxDQUFDLE9BQU8sQ0FBQyxRQUFRLENBQUMsQ0FBQTtRQUMzQyxJQUFJLENBQUMsSUFBQSxhQUFNLEVBQUMsWUFBWSxDQUFDLEVBQUUsQ0FBQztZQUN4QixjQUFNLENBQUMsSUFBSSxDQUFDLHdCQUF3QixZQUFZLEVBQUUsQ0FBQyxDQUFBO1lBQ25ELE9BQU8sSUFBSSxDQUFBO1FBQ2YsQ0FBQztRQUNELE1BQU0sVUFBVSxHQUFHLHlCQUFhLFlBQVksdUNBQUMsQ0FBQTtRQUM3QyxJQUFJLE9BQU8sVUFBVSxDQUFDLE9BQU8sS0FBSyxVQUFVLEVBQUUsQ0FBQztZQUMzQyxPQUFPLFVBQVUsQ0FBQyxPQUFPLENBQUMsTUFBTSxFQUFFLE9BQU8sQ0FBQyxDQUFBO1FBQzlDLENBQUM7YUFBTSxDQUFDO1lBQ0osY0FBTSxDQUFDLElBQUksQ0FBQywwQkFBMEIsUUFBUSw4QkFBOEIsQ0FBQyxDQUFBO1lBQzdFLE9BQU8sSUFBSSxDQUFBO1FBQ2YsQ0FBQztJQUNMLENBQUM7SUFBQyxPQUFPLENBQUMsRUFBRSxDQUFDO1FBQ1QsY0FBTSxDQUFDLEtBQUssQ0FBQyxnQ0FBZ0MsUUFBUSxHQUFHLEVBQUUsQ0FBQyxDQUFDLE9BQU8sQ0FBQyxDQUFBO1FBQ3BFLE9BQU8sSUFBSSxDQUFBO0lBQ2YsQ0FBQztBQUNMLENBQUMsQ0FBQTtBQUNNLE1BQU0sSUFBSSxHQUFHLEtBQUssRUFBRSxPQUFrQixFQUFpRCxFQUFFO0lBQzVGLE1BQU0sTUFBTSxHQUFHLElBQUksQ0FBQyxPQUFPLENBQUMsT0FBTyxDQUFDLE1BQU0sSUFBSSxPQUFPLENBQUMsSUFBSSxDQUFDLENBQUE7SUFDM0QsTUFBTSxTQUFTLEdBQUcsT0FBTyxDQUFDLEtBQUssSUFBSSx1QkFBZSxDQUFDLEtBQUssQ0FBQTtJQUN4RCxNQUFNLFlBQVksR0FBSSxTQUFnQixDQUFDLE1BQU0sQ0FBQyxJQUFJLENBQUMsRUFBRSxDQUFDLENBQUMsT0FBTyxDQUFDLE9BQU8sQ0FBQyxRQUFRLENBQUMsSUFBSSxDQUFDLENBQUMsQ0FBQTtJQUN0RixNQUFNLFdBQVcsR0FBRyxNQUFNLE9BQU8sQ0FBQyxHQUFHLENBQUMsWUFBWSxDQUFDLEdBQUcsQ0FBQyxLQUFLLEVBQUUsSUFBWSxFQUFFLEVBQUU7UUFDMUUsSUFBSSxhQUFZLENBQUMsSUFBSSxDQUFDLEVBQUUsQ0FBQztZQUNyQixJQUFJLENBQUM7Z0JBQ0QsT0FBTyxhQUFZLENBQUMsSUFBSSxDQUFDLENBQUMsTUFBTSxFQUFFLE9BQU8sQ0FBQyxDQUFBO1lBQzlDLENBQUM7WUFBQyxPQUFPLENBQUMsRUFBRSxDQUFDO2dCQUNULGNBQU0sQ0FBQyxLQUFLLENBQUMsK0JBQStCLElBQUksRUFBRSxFQUFFLENBQUMsQ0FBQyxPQUFPLENBQUMsQ0FBQTtnQkFDOUQsT0FBTyxJQUFJLENBQUE7WUFDZixDQUFDO1FBQ0wsQ0FBQztRQUNELE9BQU8sZ0JBQWdCLENBQUMsSUFBSSxFQUFFLE1BQU0sRUFBRSxPQUFPLENBQUMsQ0FBQTtJQUNsRCxDQUFDLENBQUMsQ0FBQyxDQUFBO0lBRUgsTUFBTSxLQUFLLEdBQUcsV0FBVztTQUNwQixJQUFJLEVBQUU7U0FDTixNQUFNLENBQUMsQ0FBQyxJQUFJLEVBQThDLEVBQUU7UUFDekQsSUFBSSxDQUFDLElBQUk7WUFBRSxPQUFPLEtBQUssQ0FBQTtRQUN2QixJQUFJLE9BQU8sQ0FBQyxZQUFZLENBQUMsUUFBUSxDQUFDLElBQUksQ0FBQyxRQUFRLENBQUMsSUFBSSxDQUFDO1lBQUUsT0FBTyxLQUFLLENBQUE7UUFDbkUsSUFBSSxJQUFJLENBQUMsUUFBUSxDQUFDLElBQUksS0FBSyxZQUFZO1lBQUUsT0FBTyxLQUFLLENBQUE7UUFDckQsT0FBTyxJQUFJLENBQUE7SUFDZixDQUFDLENBQUMsQ0FBQTtJQUVOLEtBQUssQ0FBQyxPQUFPLENBQUMsSUFBSSxDQUFDLEVBQUU7UUFDakIsSUFBSSxDQUFDLFFBQVEsQ0FBQyxRQUFRLEdBQUcsSUFBQSx1QkFBTSxFQUFDLElBQUksQ0FBQyxRQUFRLENBQUMsUUFBUSxFQUFDLE9BQU8sQ0FBQyxTQUFTLENBQUMsWUFBWSxDQUFDLENBQUE7UUFDdEYsSUFBSSxDQUFDLFFBQVEsQ0FBQyxRQUFRLEdBQUcsSUFBQSxzQkFBSyxFQUFDLElBQUksQ0FBQyxRQUFRLENBQUMsUUFBUSxFQUFDLE9BQU8sQ0FBQyxTQUFTLENBQUMsV0FBVyxDQUFDLENBQUE7SUFDeEYsQ0FBQyxDQUFDLENBQUE7SUFDRixPQUFPLEtBQUssQ0FBQTtBQUNoQixDQUFDLENBQUE7QUE5QlksUUFBQSxJQUFJLFFBOEJoQiJ9
|