query.js

const Parameters = require('./parameters')
const Extra = require('./extra')

class Query {
  constructor (table, data, where) {
    this.table = table
    this.db = table.db
    this.name = this.table.name
    this.data = new Parameters(data)
    this.columnString = this.data.getColumnString()
    this.assignString = this.data.getAssignString()
    this.where = new Parameters(where)

    this.whereString = this.where.getAssignString(' AND ')
    this.questionMarkString = this.data.getQuestionMarkString()

    this.extra = new Extra(where)
    this.extraString = this.extra.string

    this.values = this.data.values

    if (this.where.isArray) {
      this.where.keys.forEach(key => {
        this.values.push(this.data.obj[key.name])
      })
    } else {
      this.values = this.values.concat(this.where.getValues())
    }
  }

  async execute () {
    return this.table.ensure(this.data.keyNames).then(() => {
      return this.table.ensure(this.where.keyNames)
    }).then(() => {
      return this.db.run(this.getQueryString(), this.values)
    }).catch(error => {
      console.error(error)
      console.error(this.getQueryString())
      console.error(this.values)
      return error
    })
  }

  all () {
    return this.table.ensure(this.data.keyNames).then(() => {
      return this.table.ensure(this.where.keyNames)
    }).then(() => {
      return this.table.ensureIndex(this.where.getIndexableColumnNames())
    }).then(() => {
      return this.db.all(this.getQueryString(), this.values)
    }).catch(error => {
      console.error(error)
      console.error(this.getQueryString())
      console.error(this.values)
      return error
    })
  }
}

class SelectQuery extends Query {
  getQueryString () {
    return [
      'SELECT',
      this.columnString,
      'FROM',
            `"${this.name}"`,
            this.whereString ? 'WHERE' : '',
            this.whereString,
            this.extraString
    ].join(' ')
  }

  async execute () {
    return await this.all()
  }
}

class UpdateQuery extends Query {
  getQueryString () {
    return [
      'UPDATE',
            `"${this.name}"`,
            'SET',
            `${this.assignString}`,
            this.where.keys.length !== 0 ? 'WHERE' : '',
            this.where.keys.length !== 0 ? this.whereString : ''
    ].join(' ')
  }

  async execute () {
    const result = await super.execute()
    return result.changes
  }
}

class DeleteQuery extends Query {
  getQueryString () {
    return [
      'DELETE',
      'FROM',
            `"${this.name}"`,
            this.where.keys.length !== 0 ? 'WHERE' : '',
            this.where.keys.length !== 0 ? this.whereString : ''
    ].join(' ')
  }

  async execute () {
    const result = await super.execute()
    return result.changes
  }
}

class CountQuery extends Query {
  getQueryString () {
    return [
      'SELECT',
      'count(0) as "count"',
      'FROM',
            `"${this.name}"`,
            this.where.keys.length !== 0 ? 'WHERE' : '',
            this.where.keys.length !== 0 ? this.whereString : ''
    ].join(' ')
  }

  async execute () {
    const result = await super.all()
    return result[0].count
  }
}

class InsertQuery extends Query {
  getQueryString () {
    return [
      'INSERT INTO',
            `"${this.name}"`,
            `(${this.columnString})`,
            'VALUES',
            `(${this.questionMarkString})`
    ].join(' ')
  }

  async execute () {
    const result = await super.execute()
    return result.lastID
  }
}

module.exports = {
  Query,
  InsertQuery,
  SelectQuery,
  CountQuery,
  DeleteQuery,
  UpdateQuery
}