@jovercao/mssql/lib/cli.js

Microsoft SQL Server client for Node.js.

'use strict'

const fs = require('fs')
const path = require('path')
const sql = require('./tedious')

const exit = process.exit
const write = text => process.stdout.write(text)
Buffer.prototype.toJSON = () => {
  return `0x${this.toString('hex')}`
}

// Resolve config path

let cfgPath = path.resolve(process.argv[2] || process.cwd())
if (fs.lstatSync(cfgPath).isDirectory()) {
  cfgPath = path.resolve(cfgPath, './.mssql.json')
}
if (!fs.existsSync(cfgPath)) {
  console.error('Config file not found.')
  exit(1)
}

// Config checks & parse

let config
try {
  config = fs.readFileSync(cfgPath)
} catch (err) {
  console.error(`Failed to load config file. ${err.message}`)
  exit(1)
}

try {
  config = JSON.parse(config)
} catch (err) {
  console.error(`Failed to parse config file. ${err.message}`)
  exit(1)
}

// Read stdin

const buffer = []

process.stdin.setEncoding('utf8')
process.stdin.on('readable', () => buffer.push(process.stdin.read()))

process.stdin.on('end', () => {
  const statement = buffer.join('')
  let rst = 0
  let index = 0

  if (!statement.length) {
    console.error('Statement is empty.')
    exit(1)
  }

  sql.connect(config, err => {
    if (err) {
      console.error(err.message)
      exit(1)
    }

    write('[')

    const request = new sql.Request()
    request.stream = true
    request.on('recordset', () => {
      index = 0
      if (rst++ > 0) {
        write('],')
      }

      return write('[')
    })

    request.on('error', err => {
      console.error(err.message)
      sql.close()
      return exit(1)
    })

    request.on('row', row => {
      if (index++ > 0) {
        write(',')
      }

      return write(JSON.stringify(row))
    })

    request.on('done', () => {
      if (rst > 0) {
        write(']')
      }

      write(']\n')
      sql.close()
      return exit(0)
    })

    request.query(statement)
  })
})

process.on('uncaughtException', err => {
  if (err.code === 'EPIPE') {
    console.error('Failed to pipe output stream.')
  } else {
    console.error(err.message)
  }

  exit(1)
})