WIP Knex dialect for Snowflake in

knex-snowflake-dialect.ts

import Knex, { MySqlConnectionConfig } from "knex";
import * as Snowflake from "snowflake-sdk";
import { promisify } from "util";

const Dialect = require("knex/lib/dialects/mysql/index.js");

Dialect.prototype._driver = () => Snowflake;

Dialect.prototype.wrapIdentifierImpl = (value: string) => value;
Dialect.prototype.validateConnection = () => true;

Dialect.prototype.acquireRawConnection = async function () {
  const connection = Snowflake.createConnection({
    account: this.connectionSettings.account,
    username: this.connectionSettings.username,
    password: this.connectionSettings.password,
    warehouse: this.connectionSettings.warehouse,
    role: this.connectionSettings.role,
    database: this.connectionSettings.database
  });

  await promisify(connection.connect)();

  return connection;
};

Dialect.prototype.destroyRawConnection = async function (connection: any) {
  return promisify(connection.destroy)();
};

Dialect.prototype._query = async function (
  connection: any,
  obj: any
) {
  const queryConfig = {
    sqlText: obj.sql,
    binds: obj.bindings || []
  };

  if (obj.options) {
    Object.assign(queryConfig, obj.options);
  }

  const rows = await new Promise((resolve, reject) => {
    connection.execute({
      ...queryConfig,
      complete: (err, _, rows) => {
        if (err) reject(err);
        else resolve(rows);
      }
    });
  });

  obj.response = [rows, []];

  return obj;
};

export default Knex({
  dialect: Dialect,
  client: "mysql",
  connection: {
    account: "",
    username: "",
    password: "",
    warehouse: "",
    role: "",
    database: ""
  } as MySqlConnectionConfig
});