MysqlClient.ts

import type { Connection as MySqlConnection } from 'mysql2'
import { Connection } from '../SettingStore'
import AbstractClient, { RawField } from './AbstractClient'
// eslint-disable-next-line @typescript-eslint/no-var-requires
const mysql = require('mysql2')

export default class MysqlClient extends AbstractClient {
  connection: MySqlConnection | null = null

  constructor(settings: Connection) {
    super(settings)
  }

  get DefaultPort() {
    return 3306
  }
  get DefaultHost() {
    return '127.0.0.1'
  }
  get DefaultUser() {
    return 'root'
  }

  async connect() {
    this.connection = mysql.createConnection({
      host: this.settings.host || this.DefaultHost,
      password: this.settings.password || '',
      user: this.settings.user || this.DefaultUser,
      port: this.settings.port || this.DefaultPort,
      database: this.settings.database || '',
    })
    return true
  }

  disconnect() {
    if (this.connection) {
      this.connection.end()
    }
    this.connection = null
  }

  getTables(): Promise<string[]> {
    const sql = `
      SELECT table_name 
      FROM information_schema.tables
      WHERE table_schema = '${this.settings.database}'
    `
    return new Promise((resolve, reject) => {
      if (!this.connection) {
        reject(new Error("Don't have database connection."))
        return
      }
      this.connection.query(
        sql,
        (err, results: { [key: string]: string }[]) => {
          if (err) {
            reject(new Error(err.message))
            return
          }
          const tables = results.map((v) => v['table_name'] || v['TABLE_NAME'])
          resolve(tables)
        }
      )
    })
  }

  getColumns(tableName: string): Promise<RawField[]> {
    const sql = `SHOW FULL FIELDS FROM ${tableName}`
    return new Promise((resolve, reject) => {
      if (!this.connection) {
        reject(new Error("Don't have database connection."))
        return
      }
      this.connection.query(sql, (err, results) => {
        if (err) {
          reject(new Error(err.message))
          return
        }
        // eslint-disable-next-line @typescript-eslint/no-explicit-any
        const columns: RawField[] = (results as any).map((v: any) => ({
          field: v.Field,
          type: v.Type,
          null: v.Null,
          default: v.Default,
          comment: v.Comment,
        }))
        resolve(columns)
      })
    })
  }
}