velocious 1.0.47 → 1.0.49

package/src/database/table-data/table-column.js

@@ -0,0 +1,145 @@
+import * as inflection from "inflection"
+import restArgsError from "../../utils/rest-args-error.js"
+import TableForeignKey from "./table-foreign-key.js"
+
+export default class TableColumn {
+  constructor(name, args) {
+    if (args) {
+      const {autoIncrement, default: columnDefault, foreignKey, index, isNewColumn, maxLength, name, null: argsNull, primaryKey, type, ...restArgs} = args
+
+      if (Object.keys(args).length == 0) {
+        throw new Error("Empty args given")
+      }
+
+      restArgsError(restArgs)
+    }
+
+    this.args = args || {}
+    this.name = name
+  }
+
+  getName() { return this.name }
+
+  getNewName() { return this._newName }
+  setNewName(newName) { this._newName = newName }
+
+  getActualName() { return this.getNewName() || this.getName() }
+
+  getAutoIncrement() { return this.args?.autoIncrement }
+  setAutoIncrement(newAutoIncrement) { this.args.autoIncrement = newAutoIncrement }
+
+  getDefault() { return this.args?.default }
+  setDefault(newDefault) { this.args.default = newDefault }
+
+  getForeignKey() { return this.args?.foreignKey }
+  setForeignKey(newForeignKey) { this.args.foreignKey = newForeignKey }
+
+  getIndex() { return this.args?.index }
+  setIndex(newIndex) { this.args.index = newIndex }
+
+  getMaxLength() { return this.args?.maxLength }
+  setMaxLength(newMaxLength) { this.args.maxLength = newMaxLength }
+
+  getNull() { return this.args?.null }
+  setNull(nullable) { this.args.null = nullable }
+
+  getPrimaryKey() { return this.args?.primaryKey }
+  setPrimaryKey(newPrimaryKey) { this.args.primaryKey = newPrimaryKey }
+
+  getType() { return this.args?.type }
+  setType(newType) { this.args.type = newType }
+
+  isNewColumn() { return this.args?.isNewColumn }
+
+  getSQL({forAlterTable, driver}) {
+    const databaseType = driver.getType()
+    const options = driver.options()
+    let maxlength = this.getMaxLength()
+    let type = this.getType()?.toUpperCase()
+
+    if (type == "DATETIME" && databaseType == "pgsql") {
+      type = "TIMESTAMP"
+    }
+
+    if (type == "STRING") {
+      type = "VARCHAR"
+      maxlength ||= 255
+    }
+
+    if (databaseType == "mssql" && type == "BOOLEAN") {
+      type = "BIT"
+    } else if (databaseType == "mssql" && type == "UUID") {
+      type = "VARCHAR"
+      maxlength ||= 36
+    }
+
+    if (databaseType == "sqlite" && this.getAutoIncrement() && this.getPrimaryKey()) {
+      type = "INTEGER"
+    }
+
+    if (databaseType == "pgsql" && this.getAutoIncrement() && this.getPrimaryKey()) {
+      type = "SERIAL"
+    }
+
+    let sql = `${options.quoteColumnName(this.getActualName())} `
+
+    if (databaseType == "pgsql" && forAlterTable) sql += "TYPE "
+    if (type) sql += type
+    if (type && maxlength !== undefined && maxlength !== null) sql += `(${maxlength})`
+
+    if (this.getAutoIncrement() && driver.shouldSetAutoIncrementWhenPrimaryKey()) {
+      if (databaseType == "mssql") {
+        sql += " IDENTITY"
+      } else if (databaseType == "pgsql") {
+        if (this.getAutoIncrement() && this.getPrimaryKey()) {
+          // Do nothing
+        } else {
+          throw new Error("pgsql auto increment must be primary key")
+        }
+      } else {
+        sql += " AUTO_INCREMENT"
+      }
+    }
+
+    if (typeof this.getDefault() == "function") {
+      const defaultValue = this.getDefault()()
+
+      sql += ` DEFAULT (`
+
+      if (databaseType == "pgsql" && defaultValue == "UUID()") {
+        sql += "gen_random_uuid()"
+      } else if (databaseType == "mssql" && defaultValue == "UUID()") {
+        sql += "NEWID()"
+      } else {
+        sql += defaultValue
+      }
+
+      sql += ")"
+    } else if (this.getDefault()) {
+      sql += ` DEFAULT ${options.quote(this.getDefault())}`
+    }
+
+    if (this.getPrimaryKey()) sql += " PRIMARY KEY"
+    if (this.getNull() === false) sql += " NOT NULL"
+
+    const foreignKey = this.getForeignKey()
+
+    if (foreignKey) {
+      let foreignKeyTable, foreignKeyColumn
+
+      if (foreignKey === true) {
+        foreignKeyColumn = "id"
+        foreignKeyTable = inflection.pluralize(this.getActualName().replace(/_id$/, ""))
+      } else if (foreignKey instanceof TableForeignKey) {
+        foreignKeyColumn = foreignKey.getReferencedColumnName()
+        foreignKeyTable = foreignKey.getReferencedTableName()
+      } else {
+        throw new Error(`Unknown foreign key type given: ${foreignKey} (${typeof foreignKey})`)
+      }
+
+      sql += ` REFERENCES ${options.quoteTableName(foreignKeyTable)}(${options.quoteColumnName(foreignKeyColumn)})`
+    }
+
+    return sql
+  }
+}

package/src/database/table-data/table-foreign-key.js

@@ -0,0 +1,23 @@
+import restArgsError from "../../utils/rest-args-error.js"
+
+export default class TableForeignKey {
+  constructor({columnName, isNewForeignKey, name, tableName, referencedColumnName, referencedTableName, ...restArgs}) {
+    restArgsError(restArgs)
+
+    this._columnName = columnName
+    this._isNewForeignKey = isNewForeignKey
+    this._name = name
+    this._tableName = tableName
+    this._referencedColumnName = referencedColumnName
+    this._referencedTableName = referencedTableName
+  }
+
+  getColumnName() { return this._columnName }
+  getIsNewForeignKey() { return this._isNewForeignKey }
+  getTableName() { return this._tableName }
+  getReferencedColumnName() { return this._referencedColumnName }
+  getReferencedTableName() { return this._referencedTableName }
+
+  getName() { return this._name }
+  setName(newName) { this._name = newName }
+}

package/src/database/table-data/table-index.js

@@ -0,0 +1,18 @@
+import restArgsError from "../../utils/rest-args-error.js"
+
+export default class TableIndex {
+  constructor(columns, args) {
+    if (args) {
+      const {name, unique, ...restArgs} = args
+
+      restArgsError(restArgs)
+    }
+
+    this.args = args
+    this.columns = columns
+  }
+
+  getColumns() { return this.columns }
+  getName() { return this.args?.name }
+  getUnique() { return Boolean(this.args?.unique) }
+}

package/src/database/table-data/table-reference.js

@@ -0,0 +1,6 @@
+export default class TableReference {
+  constructor(name, args) {
+    this.args = args
+    this.name = name
+  }
+}