n8n
Version:
n8n Workflow Automation Tool
175 lines • 8.36 kB
JavaScript
;
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 __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
return c > 3 && r && Object.defineProperty(target, key, r), r;
};
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;
};
})();
var __metadata = (this && this.__metadata) || function (k, v) {
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.NodeCatalogService = void 0;
const backend_common_1 = require("@n8n/backend-common");
const di_1 = require("@n8n/di");
const fs = __importStar(require("fs/promises"));
const path = __importStar(require("path"));
const load_nodes_and_credentials_1 = require("../load-nodes-and-credentials");
const UNFILTERED = Symbol('unfiltered');
let NodeCatalogService = class NodeCatalogService {
constructor(loadNodesAndCredentials, logger) {
this.loadNodesAndCredentials = loadNodesAndCredentials;
this.logger = logger;
this.nodeDefinitionDirs = [];
this.searchStates = new Map();
this.getCache = new Map();
this.suggestCache = new Map();
this.loadNodesAndCredentials.addPostProcessor(async () => await this.refreshNodeTypes());
}
async initialize() {
this.initPromise ??= this.doInitialize();
await this.initPromise;
}
getNodeTypeParser() {
if (!this.nodeTypeParser) {
throw new Error('NodeCatalogService not initialized. Call initialize() first.');
}
return this.nodeTypeParser;
}
getNodeDefinitionDirs() {
return this.nodeDefinitionDirs;
}
async searchNodes(queries, options = {}) {
const { nodeFilter } = options;
const stateKey = nodeFilter ?? UNFILTERED;
let state = this.searchStates.get(stateKey);
if (!state) {
state = { cache: new Map() };
this.searchStates.set(stateKey, state);
}
const cacheKey = JSON.stringify([...queries].sort());
const cached = state.cache.get(cacheKey);
if (cached)
return cached;
if (!state.search) {
const { searchCodeBuilderNodes } = await Promise.resolve().then(() => __importStar(require('@n8n/ai-workflow-builder')));
const nodeTypeParser = this.getNodeTypeParser();
state.search = (searchQueries) => nodeFilter
? searchCodeBuilderNodes(nodeTypeParser, searchQueries, { nodeFilter })
: searchCodeBuilderNodes(nodeTypeParser, searchQueries);
}
const result = state.search(queries);
state.cache.set(cacheKey, result);
return result;
}
async getNodeTypes(nodeIds) {
const cacheKey = JSON.stringify(nodeIds.map((id) => (typeof id === 'string' ? id : JSON.stringify(id))).sort());
const cached = this.getCache.get(cacheKey);
if (cached)
return cached;
const { getNodeTypes } = await Promise.resolve().then(() => __importStar(require('@n8n/ai-workflow-builder')));
const result = getNodeTypes(nodeIds, { nodeDefinitionDirs: this.nodeDefinitionDirs });
this.getCache.set(cacheKey, result);
return result;
}
async getSuggestedNodes(categories) {
const cacheKey = JSON.stringify([...categories].sort());
const cached = this.suggestCache.get(cacheKey);
if (cached)
return cached;
if (!this.suggestTool) {
const { createGetSuggestedNodesTool } = await Promise.resolve().then(() => __importStar(require('@n8n/ai-workflow-builder')));
this.suggestTool = createGetSuggestedNodesTool(this.getNodeTypeParser());
}
const result = await this.suggestTool.invoke({ categories });
this.suggestCache.set(cacheKey, result);
return result;
}
async doInitialize() {
const { NodeTypeParser: NodeTypeParserClass } = await Promise.resolve().then(() => __importStar(require('@n8n/ai-workflow-builder')));
const { setSchemaBaseDirs } = await Promise.resolve().then(() => __importStar(require('@n8n/workflow-sdk')));
await this.loadNodesAndCredentials.postProcessLoaders();
const { nodes: nodeTypeDescriptions } = await this.loadNodesAndCredentials.collectTypes();
this.nodeTypeParser = new NodeTypeParserClass(nodeTypeDescriptions);
this.nodeDefinitionDirs = await this.resolveBuiltinNodeDefinitionDirs();
setSchemaBaseDirs(this.nodeDefinitionDirs);
this.logger.debug('NodeCatalogService initialized', {
nodeTypeCount: nodeTypeDescriptions.length,
nodeDefinitionDirs: this.nodeDefinitionDirs.length,
});
}
async refreshNodeTypes() {
if (!this.nodeTypeParser)
return;
const { NodeTypeParser: NodeTypeParserClass } = await Promise.resolve().then(() => __importStar(require('@n8n/ai-workflow-builder')));
const { nodes: nodeTypeDescriptions } = await this.loadNodesAndCredentials.collectTypes();
this.nodeTypeParser = new NodeTypeParserClass(nodeTypeDescriptions);
this.searchStates.clear();
this.suggestTool = undefined;
this.getCache.clear();
this.suggestCache.clear();
this.logger.debug('NodeCatalogService refreshed node types', {
nodeTypeCount: nodeTypeDescriptions.length,
});
}
async resolveBuiltinNodeDefinitionDirs() {
const dirs = [];
for (const packageId of ['n8n-nodes-base', '@n8n/n8n-nodes-langchain']) {
try {
const packageJsonPath = require.resolve(`${packageId}/package.json`);
const distDir = path.dirname(packageJsonPath);
let nodeDefsDir = path.join(distDir, 'dist', 'node-definitions');
const separator = process.platform === 'win32' ? '\\' : '/';
if (!nodeDefsDir.endsWith(separator)) {
nodeDefsDir += separator;
}
await fs.access(nodeDefsDir);
dirs.push(nodeDefsDir);
}
catch (error) {
this.logger.debug(`Could not resolve node definitions for ${packageId}`, { error });
}
}
return dirs;
}
};
exports.NodeCatalogService = NodeCatalogService;
exports.NodeCatalogService = NodeCatalogService = __decorate([
(0, di_1.Service)(),
__metadata("design:paramtypes", [load_nodes_and_credentials_1.LoadNodesAndCredentials,
backend_common_1.Logger])
], NodeCatalogService);
//# sourceMappingURL=node-catalog.service.js.map