n8n-nodes-base
Version:
Base nodes of n8n
366 lines • 19.5 kB
JavaScript
"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.Elasticsearch = void 0;
const omit_1 = __importDefault(require("lodash/omit"));
const n8n_workflow_1 = require("n8n-workflow");
const descriptions_1 = require("./descriptions");
const GenericFunctions_1 = require("./GenericFunctions");
class Elasticsearch {
description = {
displayName: 'Elasticsearch',
name: 'elasticsearch',
icon: 'file:elasticsearch.svg',
group: ['transform'],
version: 1,
subtitle: '={{$parameter["operation"] + ": " + $parameter["resource"]}}',
description: 'Consume the Elasticsearch API',
schemaPath: 'Elastic/Elasticsearch',
defaults: {
name: 'Elasticsearch',
},
usableAsTool: true,
inputs: [n8n_workflow_1.NodeConnectionTypes.Main],
outputs: [n8n_workflow_1.NodeConnectionTypes.Main],
credentials: [
{
name: 'elasticsearchApi',
required: true,
},
],
properties: [
{
displayName: 'Resource',
name: 'resource',
type: 'options',
noDataExpression: true,
options: [
{
name: 'Document',
value: 'document',
},
{
name: 'Index',
value: 'index',
},
],
default: 'document',
},
...descriptions_1.documentOperations,
...descriptions_1.documentFields,
...descriptions_1.indexOperations,
...descriptions_1.indexFields,
],
};
async execute() {
const items = this.getInputData();
const returnData = [];
const resource = this.getNodeParameter('resource', 0);
const operation = this.getNodeParameter('operation', 0);
let responseData;
let bulkBody = {};
for (let i = 0; i < items.length; i++) {
const bulkOperation = this.getNodeParameter('options.bulkOperation', i, false);
if (resource === 'document') {
// **********************************************************************
// document
// **********************************************************************
if (operation === 'delete') {
// ----------------------------------------
// document: delete
// ----------------------------------------
// https://www.elastic.co/guide/en/elasticsearch/reference/current/docs-delete.html
const indexId = this.getNodeParameter('indexId', i);
const documentId = this.getNodeParameter('documentId', i);
if (bulkOperation) {
bulkBody[i] = JSON.stringify({
delete: {
_index: indexId,
_id: documentId,
},
});
}
else {
const endpoint = `/${indexId}/_doc/${documentId}`;
responseData = await GenericFunctions_1.elasticsearchApiRequest.call(this, 'DELETE', endpoint);
}
}
else if (operation === 'get') {
// ----------------------------------------
// document: get
// ----------------------------------------
// https://www.elastic.co/guide/en/elasticsearch/reference/current/docs-get.html
const indexId = this.getNodeParameter('indexId', i);
const documentId = this.getNodeParameter('documentId', i);
const qs = {};
const options = this.getNodeParameter('options', i);
if (Object.keys(options).length) {
Object.assign(qs, options);
qs._source = true;
}
const endpoint = `/${indexId}/_doc/${documentId}`;
responseData = await GenericFunctions_1.elasticsearchApiRequest.call(this, 'GET', endpoint, {}, qs);
const simple = this.getNodeParameter('simple', i);
if (simple) {
responseData = {
_id: responseData._id,
...responseData._source,
};
}
}
else if (operation === 'getAll') {
// ----------------------------------------
// document: getAll
// ----------------------------------------
// https://www.elastic.co/guide/en/elasticsearch/reference/current/search-search.html
const indexId = this.getNodeParameter('indexId', i);
const body = {};
const qs = {};
const options = this.getNodeParameter('options', i);
// const paginate = this.getNodeParameter('paginate', i) as boolean;
if (Object.keys(options).length) {
const { query, ...rest } = options;
if (query) {
Object.assign(body, (0, n8n_workflow_1.jsonParse)(query, { errorMessage: "Invalid JSON in 'Query' option" }));
}
Object.assign(qs, rest);
qs._source = true;
}
const returnAll = this.getNodeParameter('returnAll', 0);
if (returnAll) {
//Defines the number of hits to return. Defaults to 10. By default, you cannot page through more than 10,000 hits
qs.size = 10000;
if (qs.sort) {
responseData = await GenericFunctions_1.elasticsearchApiRequestAllItems.call(this, indexId, body, qs);
}
else {
responseData = await GenericFunctions_1.elasticsearchApiRequest.call(this, 'POST', `/${indexId}/_search`, body, qs);
responseData = responseData.hits.hits;
}
}
else {
qs.size = this.getNodeParameter('limit', 0);
responseData = await GenericFunctions_1.elasticsearchApiRequest.call(this, 'POST', `/${indexId}/_search`, body, qs);
responseData = responseData.hits.hits;
}
const simple = this.getNodeParameter('simple', 0);
if (simple) {
responseData = responseData.map((item) => {
return {
_id: item._id,
...item._source,
};
});
}
}
else if (operation === 'create') {
// ----------------------------------------
// document: create
// ----------------------------------------
// https://www.elastic.co/guide/en/elasticsearch/reference/current/docs-index_.html
const body = {};
const dataToSend = this.getNodeParameter('dataToSend', 0);
if (dataToSend === 'defineBelow') {
const fields = this.getNodeParameter('fieldsUi.fieldValues', i, []);
fields.forEach(({ fieldId, fieldValue }) => (body[fieldId] = fieldValue));
}
else {
const inputData = items[i].json;
const rawInputsToIgnore = this.getNodeParameter('inputsToIgnore', i);
const inputsToIgnore = rawInputsToIgnore.split(',').map((c) => c.trim());
for (const key of Object.keys(inputData)) {
if (inputsToIgnore.includes(key))
continue;
body[key] = inputData[key];
}
}
const qs = {};
const additionalFields = this.getNodeParameter('additionalFields', i);
if (Object.keys(additionalFields).length) {
Object.assign(qs, (0, omit_1.default)(additionalFields, ['documentId']));
}
const indexId = this.getNodeParameter('indexId', i);
const { documentId } = additionalFields;
if (bulkOperation) {
bulkBody[i] = JSON.stringify({
index: {
_index: indexId,
_id: documentId,
},
});
bulkBody[i] += `\n${JSON.stringify(body)}`;
}
else {
if (documentId) {
const endpoint = `/${indexId}/_doc/${documentId}`;
responseData = await GenericFunctions_1.elasticsearchApiRequest.call(this, 'PUT', endpoint, body);
}
else {
const endpoint = `/${indexId}/_doc`;
responseData = await GenericFunctions_1.elasticsearchApiRequest.call(this, 'POST', endpoint, body);
}
}
}
else if (operation === 'update') {
// ----------------------------------------
// document: update
// ----------------------------------------
// https://www.elastic.co/guide/en/elasticsearch/reference/current/docs-update.html
const body = { doc: {} };
const dataToSend = this.getNodeParameter('dataToSend', 0);
if (dataToSend === 'defineBelow') {
const fields = this.getNodeParameter('fieldsUi.fieldValues', i, []);
fields.forEach(({ fieldId, fieldValue }) => (body.doc[fieldId] = fieldValue));
}
else {
const inputData = items[i].json;
const rawInputsToIgnore = this.getNodeParameter('inputsToIgnore', i);
const inputsToIgnore = rawInputsToIgnore.split(',').map((c) => c.trim());
for (const key of Object.keys(inputData)) {
if (inputsToIgnore.includes(key))
continue;
body.doc[key] = inputData[key];
}
}
const indexId = this.getNodeParameter('indexId', i);
const documentId = this.getNodeParameter('documentId', i);
const endpoint = `/${indexId}/_update/${documentId}`;
if (bulkOperation) {
bulkBody[i] = JSON.stringify({
update: {
_index: indexId,
_id: documentId,
},
});
bulkBody[i] += `\n${JSON.stringify(body)}`;
}
else {
responseData = await GenericFunctions_1.elasticsearchApiRequest.call(this, 'POST', endpoint, body);
}
}
}
else if (resource === 'index') {
// **********************************************************************
// index
// **********************************************************************
// https://www.elastic.co/guide/en/elasticsearch/reference/current/indices.html
if (operation === 'create') {
// ----------------------------------------
// index: create
// ----------------------------------------
// https://www.elastic.co/guide/en/elasticsearch/reference/current/indices-create-index.html
const indexId = this.getNodeParameter('indexId', i);
const body = {};
const qs = {};
const additionalFields = this.getNodeParameter('additionalFields', i);
if (Object.keys(additionalFields).length) {
const { aliases, mappings, settings, ...rest } = additionalFields;
Object.assign(body, aliases, mappings, settings);
Object.assign(qs, rest);
}
responseData = await GenericFunctions_1.elasticsearchApiRequest.call(this, 'PUT', `/${indexId}`);
responseData = { id: indexId, ...responseData };
delete responseData.index;
}
else if (operation === 'delete') {
// ----------------------------------------
// index: delete
// ----------------------------------------
// https://www.elastic.co/guide/en/elasticsearch/reference/current/indices-delete-index.html
const indexId = this.getNodeParameter('indexId', i);
responseData = await GenericFunctions_1.elasticsearchApiRequest.call(this, 'DELETE', `/${indexId}`);
responseData = { success: true };
}
else if (operation === 'get') {
// ----------------------------------------
// index: get
// ----------------------------------------
// https://www.elastic.co/guide/en/elasticsearch/reference/current/indices-get-index.html
const indexId = this.getNodeParameter('indexId', i);
const qs = {};
const additionalFields = this.getNodeParameter('additionalFields', i);
if (Object.keys(additionalFields).length) {
Object.assign(qs, additionalFields);
}
responseData = await GenericFunctions_1.elasticsearchApiRequest.call(this, 'GET', `/${indexId}`, {}, qs);
responseData = { id: indexId, ...responseData[indexId] };
}
else if (operation === 'getAll') {
// ----------------------------------------
// index: getAll
// ----------------------------------------
// https://www.elastic.co/guide/en/elasticsearch/reference/current/indices-aliases.html
responseData = await GenericFunctions_1.elasticsearchApiRequest.call(this, 'GET', '/_aliases');
responseData = Object.keys(responseData).map((index) => ({
indexId: index,
}));
const returnAll = this.getNodeParameter('returnAll', i);
if (!returnAll) {
const limit = this.getNodeParameter('limit', i);
responseData = responseData.slice(0, limit);
}
}
}
if (!bulkOperation) {
const executionData = this.helpers.constructExecutionMetaData(this.helpers.returnJsonArray(responseData), { itemData: { item: i } });
returnData.push(...executionData);
}
if (Object.keys(bulkBody).length >= 50) {
responseData = (await GenericFunctions_1.elasticsearchBulkApiRequest.call(this, bulkBody));
for (let j = 0; j < responseData.length; j++) {
const itemData = responseData[j];
if (itemData.error) {
const errorData = itemData.error;
const message = errorData.type;
const description = errorData.reason;
const itemIndex = parseInt(Object.keys(bulkBody)[j]);
if (this.continueOnFail()) {
returnData.push(...this.helpers.constructExecutionMetaData(this.helpers.returnJsonArray({ error: message, message: itemData.error }), { itemData: { item: itemIndex } }));
continue;
}
else {
throw new n8n_workflow_1.NodeApiError(this.getNode(), {
message,
description,
itemIndex,
});
}
}
const executionData = this.helpers.constructExecutionMetaData(this.helpers.returnJsonArray(itemData), { itemData: { item: parseInt(Object.keys(bulkBody)[j]) } });
returnData.push(...executionData);
}
bulkBody = {};
}
}
if (Object.keys(bulkBody).length) {
responseData = (await GenericFunctions_1.elasticsearchBulkApiRequest.call(this, bulkBody));
for (let j = 0; j < responseData.length; j++) {
const itemData = responseData[j];
if (itemData.error) {
const errorData = itemData.error;
const message = errorData.type;
const description = errorData.reason;
const itemIndex = parseInt(Object.keys(bulkBody)[j]);
if (this.continueOnFail()) {
returnData.push(...this.helpers.constructExecutionMetaData(this.helpers.returnJsonArray({ error: message, message: itemData.error }), { itemData: { item: itemIndex } }));
continue;
}
else {
throw new n8n_workflow_1.NodeApiError(this.getNode(), {
message,
description,
itemIndex,
});
}
}
const executionData = this.helpers.constructExecutionMetaData(this.helpers.returnJsonArray(itemData), { itemData: { item: parseInt(Object.keys(bulkBody)[j]) } });
returnData.push(...executionData);
}
}
return [returnData];
}
}
exports.Elasticsearch = Elasticsearch;
//# sourceMappingURL=Elasticsearch.node.js.map