ToolJet/plugins/packages/mariadb/lib/index.ts

import { QueryError, QueryResult, QueryService, ConnectionTestResult } from '@tooljet-plugins/common';
import { SourceOptions, QueryOptions } from './types';
const mariadb = require('mariadb');
export default class Mariadb implements QueryService {
  async run(sourceOptions: SourceOptions, queryOptions: QueryOptions): Promise<QueryResult> {
    let result = {};
    const mariadbClient = await this.getConnection(sourceOptions);
    try {
      const rows = await mariadbClient.query(queryOptions.query);
      result = this.toJson(rows);
    } catch (error) {
      throw new QueryError('Query could not be completed', error.message, {});
    }
    return {
      status: 'ok',
      data: result,
    };
  }

  async testConnection(sourceOptions: SourceOptions): Promise<ConnectionTestResult> {
    const conn = await this.getConnection(sourceOptions);
    const rows = await conn.query('SELECT 1 as val');

    if (!rows) {
      throw new Error('Error');
    }
    return {
      status: 'ok',
    };
  }

  async getConnection(sourceOptions: SourceOptions): Promise<any> {
    const poolConfig = {
      host: sourceOptions.host,
      user: sourceOptions.user,
      password: sourceOptions.password,
      connectionLimit: sourceOptions.connectionLimit,
      port: sourceOptions.port,
      database: sourceOptions.database,
    };

    const sslObject = { rejectUnauthorized: (sourceOptions.ssl_certificate ?? 'none') != 'none' };
    if (sourceOptions.ssl_certificate === 'ca_certificate') {
      sslObject['ca'] = sourceOptions.ca;
    }
    if (sourceOptions.ssl_certificate === 'self_signed') {
      sslObject['ca'] = sourceOptions.ca;
      sslObject['cert'] = sourceOptions.cert;
      sslObject['key'] = sourceOptions.key;
    }

    if (sourceOptions.ssl_enabled) poolConfig['ssl'] = sslObject;

    const mariadbPool = mariadb.createPool(poolConfig);

    return mariadbPool.getConnection();
  }

  private toJson(data) {
    return JSON.parse(
      JSON.stringify(data, (_, v) => (typeof v === 'bigint' ? `${v}n` : v)).replace(/"(-?\d+)n"/g, (_, a) => a)
    );
  }
}