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

import type {
	ICredentialsDecrypted,
	ICredentialTestFunctions,
	IDataObject,
	INodeCredentialTestResult,
} from 'n8n-workflow';

import { Client } from 'ssh2';
import { configurePostgres } from '../transport';

import type { PgpClient } from '../helpers/interfaces';

export async function postgresConnectionTest(
	this: ICredentialTestFunctions,
	credential: ICredentialsDecrypted,
): Promise<INodeCredentialTestResult> {
	const credentials = credential.data as IDataObject;

	let sshClientCreated: Client | undefined = new Client();
	let pgpClientCreated: PgpClient | undefined;

	try {
		const { db, pgp, sshClient } = await configurePostgres(credentials, {}, sshClientCreated);

		sshClientCreated = sshClient;
		pgpClientCreated = pgp;

		await db.connect();
	} catch (error) {
		let message = error.message as string;

		if (error.message.includes('ECONNREFUSED')) {
			message = 'Connection refused';
		}

		if (error.message.includes('ENOTFOUND')) {
			message = 'Host not found, please check your host name';
		}

		if (error.message.includes('ETIMEDOUT')) {
			message = 'Connection timed out';
		}

		return {
			status: 'Error',
			message,
		};
	} finally {
		if (sshClientCreated) {
			sshClientCreated.end();
		}
		if (pgpClientCreated) {
			pgpClientCreated.end();
		}
	}
	return {
		status: 'OK',
		message: 'Connection successful!',
	};
}