h-orm/lib/model/index.js

H-orm is a mysql orm for nodejs

const mysql = require('../mysql')
const { toArray, handleParam } = require('../util')

/**
 * @param {*} tableName - tablename
 * @param {*} schema - schema
 * @param {*} exInfo - ex info
 */
function Model (tableName, schema = {}, exInfo = {}) {
  this.table = tableName
  this.schema = schema
  this.schemaKey = Object.keys(schema).join(',')
  this.conditions = {}
  this.type = 'select'
  this.pri = exInfo.pri || 'id'
  this.status = exInfo.status || 'status'
  const self = new Promise((resolve, reject) => {
    setTimeout(() => {
      let SQL = ''
      switch (this.type) {
        case 'insert':
          const insertKey = Object.keys(this.conditions['insert'][0]).join(',')
          const insertValue = []
          this.conditions['insert'].forEach(data => {
            insertValue.push(`(${Object.values(data).map( d => mysql.escape(d)).join(',')})`)
          })
          SQL = `insert into ${this.table} (${insertKey}) VALUES ${insertValue.join(',')}`
          break
        case 'select':
          SQL = `select ${this.schemaKey} from ${this.table} `
          // where select params must be a object
          if (this.conditions['select'] && Object.keys(this.conditions['select']).length) {
            let linkMode = 'and'
            const conditions = []
            for (let key in this.conditions['select']) {
              if(Array.isArray(this.conditions['select'][key])) {
                conditions.push(`${key} in ( ${this.conditions['select'][key].map(d => mysql.escape(d)).join(',') } )`)
              } else {
                if (/_not_like$/.test(key)) {
                  conditions.push(`${key.split('_not_like')[0]} not like ${mysql.escape(this.conditions['select'][key])}`)
                } else if (/_like$/.test(key)) {
                  conditions.push(`${key.split('_like')[0]} like ${mysql.escape(this.conditions['select'][key])}`)
                } else {
                  conditions.push(`${key} in ( ${mysql.escape(this.conditions['select'][key]) } )`)
                }
              }
            }
            if (this.conditions['selectMode'].toLocaleUpperCase() !== 'AND') {
              linkMode = 'or'
            }
            SQL = SQL + 'where ' + conditions.join(` ${linkMode} `)
          }
          break
        case 'like':
          SQL = `select ${this.schemaKey} from ${this.table} `
          if (this.conditions['like']['key']) {
            let likeMode = 'like'
            let linkMode = 'and'
            const conditions = []
            for (let key in this.conditions['like']['key']) {
              if (this.conditions['like']['likeMode'].toLocaleUpperCase() !== 'LIKE') {
                likeMode = 'not like'
              }
              conditions.push(`${key} ${likeMode} ${mysql.escape(this.conditions['like']['key'][key]) }`)
            }
            if (this.conditions['like']['linkMode'].toLocaleUpperCase() !== 'AND') {
              linkMode = 'or'
            }
            SQL = SQL + 'where ' + conditions.join(` ${linkMode} `)
          }
          break
        case 'update':
          const updateconditions = []
          const whereconditions = []
          for (let key in this.conditions['update']) {
            const updateValue = this.conditions['update'][key]
            updateconditions.push(`${key} = ${mysql.escape(updateValue)} `)
          }
          for (let key in this.conditions['updateWhere']) {
            const whereValue = this.conditions['updateWhere'][key]
            if(Array.isArray(whereValue)) {
              whereconditions.push(`${key} in ( ${whereValue.map(d => mysql.escape(d)).join(',') } )`)
            } else {
              whereconditions.push(`${key} in ( ${mysql.escape(whereValue)} )`)
            }
          }
          SQL = `update ${this.table} set ${updateconditions.join(', ')} where ${whereconditions.join(' and ')}`
          break
        default:
          break
      }
      if ((this.type === 'select' || this.type === 'like') && this.conditions['sort']) {
        SQL = SQL + ' order by ' + this.conditions['sort'].map(d => `${d.key} ${d.rule}`).join(',')
      }
      mysql.query(SQL)
        .then(res => {
          let result = JSON.parse(JSON.stringify(res.results))
          // limit
          if (this.type === 'select' && this.conditions['limit']) {
            if (!this.conditions['limit'].count) {
              result = result.slice(0, Number(this.conditions['limit'].num))
            } else {
              result = result.slice(Number(this.conditions['limit'].num), this.conditions['limit'].count + Number(this.conditions['limit'].num))
            }
          }
          // reset
          this.conditions = {}
          this.type = 'select'
          resolve(result)
        })
        .catch(err => reject(err))
    }, 0);
  })
  // create
  self.create = obj => {
    if (handleParam(obj)) {
      this.type = 'insert'
      this.conditions['insert'] = toArray(obj)
    } else {
      throw Error(`create function mast have params`)
    }
    return self
  }
  // delete
  self.delete = (obj = {}) => {
    if (handleParam(obj)) {
      if(typeof obj !== 'object') {
        let pri_val = obj
        obj = {}
        obj[this.pri] =pri_val
      }
      let updateVal = {}
      updateVal[this.status] = 0
      self.update(updateVal, obj)
    } else {
      throw Error(`delete function mast have parmas`)
    }
    return self
  }
  // update
  self.update = (obj = {}, whereObj = {}) => {
    if (handleParam(obj)) {
      this.type = 'update'
      this.conditions['update'] = Object.assign({}, this.conditions['update'], obj)
      this.conditions['updateWhere'] = Object.assign({}, this.conditions['updateWhere'], whereObj)
    } else {
      throw Error(`update function mast have parmas`)
    }
    return self
  }
  // select
  self.where = (obj, mode = 'AND') => {
    this.type = 'select'
    this.conditions['select'] = Object.assign({}, this.conditions['select'], obj)
    this.conditions['selectMode'] = mode
    return self
  }
  // select
  self.find = pris => {
    this.type = 'select'
    const query = {}
    query[this.pri] = toArray(pris)
    self.where(query)
    return self
  }
  // select
  self.like = (obj, likeMode = 'LIKE', linkMode = 'AND') => {
    if (handleParam(obj)) {
      this.type = 'like'
      this.conditions['like'] = {
        key: obj,
        likeMode,
        linkMode,
      }
    } else {
      throw Error(`like function mast have parmas`)
    }
    return self
  }
  // limit
  self.limit = (num, count) => {
    this.conditions['limit'] = {
      num,
      count
    }
    return self
  }
  // sort
  self.sort = data => {
    this.conditions['sort'] = toArray(data)
    return self
  }
  return self
}

module.exports = Model