n8n/packages/nodes-base/nodes/Postgres/v2/methods/loadOptions.ts

53 lines
1.6 KiB
TypeScript
Raw Normal View History

2023-04-03 08:18:01 -07:00
import type { ILoadOptionsFunctions, INodePropertyOptions } from 'n8n-workflow';
2023-04-03 08:18:01 -07:00
import { getTableSchema } from '../helpers/utils';
import { configurePostgres } from '../transport';
2023-04-03 08:18:01 -07:00
export async function getColumns(this: ILoadOptionsFunctions): Promise<INodePropertyOptions[]> {
const credentials = await this.getCredentials('postgres');
const options = { nodeVersion: this.getNode().typeVersion };
2023-04-03 08:18:01 -07:00
const { db, pgp, sshClient } = await configurePostgres(credentials, options);
2023-04-03 08:18:01 -07:00
const schema = this.getNodeParameter('schema', 0, {
extractValue: true,
}) as string;
const table = this.getNodeParameter('table', 0, {
extractValue: true,
}) as string;
try {
const columns = await getTableSchema(db, schema, table);
return columns.map((column) => ({
name: column.column_name,
value: column.column_name,
description: `Type: ${column.data_type.toUpperCase()}, Nullable: ${column.is_nullable}`,
}));
} catch (error) {
throw error;
} finally {
if (sshClient) {
sshClient.end();
}
pgp.end();
2023-04-03 08:18:01 -07:00
}
}
export async function getColumnsMultiOptions(
this: ILoadOptionsFunctions,
): Promise<INodePropertyOptions[]> {
const returnData = await getColumns.call(this);
const returnAll = { name: '*', value: '*', description: 'All columns' };
return [returnAll, ...returnData];
}
export async function getColumnsWithoutColumnToMatchOn(
this: ILoadOptionsFunctions,
): Promise<INodePropertyOptions[]> {
const columnToMatchOn = this.getNodeParameter('columnToMatchOn') as string;
const returnData = await getColumns.call(this);
return returnData.filter((column) => column.value !== columnToMatchOn);
}