smartclick-pgsqlorm/pqorm.js

orm for postgresql

'use strict';

const process = require('node:process');
const mo = require('./model.js');
const TableTrigger = require('./tableTrigger.js');
const types = require('./dataTypes.js');
const makeId = require('./makeId.js');

function PsqlORM(db) {
  if (!(this instanceof PsqlORM)) {
    return new PsqlORM(db);
  }

  this.db = db;

  this.schema = 'public';

  this.field = '*'

  this.setdb = (db) => {
    this.db = db;
  };

  Object.defineProperty(this, '__max__', {
    configurable: false,
    enumerable: false,
    writable: true,
    value: 2560
  });

  Object.defineProperty(this, 'max', {
    get: () => {
      return this.__max__;
    },

    set: (v) => {
      if (typeof v === 'number' && v > 0) {
        this.__max__ = v;
      }
    }
  });

  Object.defineProperty(this, 'pool', {
    enumerable: true,
    configurable: false,
    writable: false,
    value: [],
  });

  Object.defineProperty(this, '__register__', {
    enumerable: false,
    configurable: false,
    writable: false,
    value: {},
  });

  this.tableTrigger = new TableTrigger();
}

PsqlORM.prototype.free = function (mdb) {
  if (this.pool.length < this.__max__) {
    if (mdb.__state__ === mdb.state.USING) {
      mdb.init();
      mdb.resetIdInfo();
      mdb.commitTriggers = [];
      mdb.__state__ = mdb.state.FREE;
      mdb.__transaction__ = false;
      mdb.tableName = '';
      this.pool.push(mdb);
    }
  }
}

PsqlORM.prototype.getm = function (tablename, schema) {
  if (this.pool.length > 0) {
    let t = this.pool.pop();
    t.odb = t.db = this.db;
    t.tableName = tablename;
    t.__schema__ = t._schema = schema;
    t.__fetch_sql__ = false;
    t.__free_lock__ = false;
    t.__state__ = t.state.USING;
    t.__trigger_commit__ = false;
    return t;
  }

  return null;
}

PsqlORM.prototype.ignoreCopyWarning = function (igr = true) {
  process.env.PSQLORM_IGNORE_COPY_WARNING = igr
}

PsqlORM.prototype.setSchema = function (name, runsql = false) {
  this.schema = name

  if (runsql) {
    this.db.query(`set search_path to ${name}`).catch(err => { })
  }
}

PsqlORM.prototype.model = function (tablename, schema = '') {
  let mdb = this.getm(tablename, schema || this.schema)

  if (mdb) return mdb

  return new mo(this.db, tablename, schema || this.schema, this, this.tableTrigger)
}
PsqlORM.prototype.select = function (field = "*") {

  this.field = field
  return this
}
PsqlORM.prototype.from = function (tablename, schema = '') {
  let mdb = this.getm(tablename, schema || this.schema)
  if (mdb) return mdb

  return new mo(this.db, tablename, schema || this.schema, this, this.tableTrigger, this.field)
}


PsqlORM.prototype.connect = function (tablename = '', schema = '') {
  return this.model(tablename, schema).connect();
}

PsqlORM.prototype.transaction = async function (callback, schema = '') {
  let m = this.getm('', schema || this.schema)

  if (m) {
    return m.transaction(callback)
  }

  m = new mo(this.db, '', schema || this.schema, this, this.tableTrigger)

  return m.transaction(callback)
}

PsqlORM.prototype.end = function () {
  this.db.end()
}

PsqlORM.prototype.query = function (sql, args) {
  return this.db.query(sql, args || [])
}

PsqlORM.prototype.insert = function (tableName, data) {
  let mdb = this.getm(tableName, this.schema)

  if (!mdb) mdb = new mo(this.db, tableName, this.schema, this, this.tableTrigger)

  return mdb.insert(data);

}

PsqlORM.prototype.update = function (tableName, data) {
  let mdb = this.getm(tableName, this.schema)

  if (!mdb) mdb = new mo(this.db, tableName, this.schema, this, this.tableTrigger)

  return mdb.update(data);

}

PsqlORM.prototype.save = function (tableName, data) {
  let mdb = this.getm(tableName, this.schema)

  if (!mdb) mdb = new mo(this.db, tableName, this.schema, this, this.tableTrigger)
  const id = data.id;
  if (!id) return mdb.insert(data);
  delete data.id;
  return mdb.save(id, data);

}

PsqlORM.prototype.sql = async function (sql, args) {
  const res = await this.db.query(sql, args || []);
  return res.rows
}

PsqlORM.initORM = (config, schema = null) => {
  let pg = require('pg')
  let orm = new PsqlORM(new pg.Pool(config))
  if (schema) orm.schema = schema;
  return orm
}

PsqlORM.dataTypes = types
PsqlORM.makeId = makeId

module.exports = PsqlORM