DataTypes improved

lib/data-types.js

+var STRING = function(length, binary) {
+  if (this instanceof STRING) {
+    this._binary = !!binary;
+    if (typeof length === 'number') {
+      this._length = length;
+    } else {
+      this._length = 255;
+    }
+  } else {
+    return new STRING(length, binary);
+  }
+};
+
+STRING.prototype = {
+  get BINARY() {
+    this._binary = true;
+    return this;
+  },
+  get type() {
+    return this.toString();
+  },
+  toString: function() {
+    return 'VARCHAR(' + this._length + ')' + ((this._binary) ? ' BINARY' : '');
+  }
+};
+
+Object.defineProperty(STRING, 'BINARY', {
+  get: function() {
+    return new STRING(undefined, true);
+  }
+});
+
+var INTEGER = function() {
+  return INTEGER.prototype.construct.apply(this, [INTEGER].concat(Array.prototype.slice.apply(arguments)));
+};
+
+var BIGINT = function() {
+  return BIGINT.prototype.construct.apply(this, [BIGINT].concat(Array.prototype.slice.apply(arguments)));
+};
+
+var FLOAT = function() {
+  return FLOAT.prototype.construct.apply(this, [FLOAT].concat(Array.prototype.slice.apply(arguments)));
+};
+
+FLOAT._type = FLOAT;
+FLOAT._typeName = 'FLOAT';
+INTEGER._type = INTEGER;
+INTEGER._typeName = 'INTEGER';
+BIGINT._type = BIGINT;
+BIGINT._typeName = 'BIGINT';
+STRING._type = STRING;
+STRING._typeName = 'VARCHAR';
+
+STRING.toString = INTEGER.toString = FLOAT.toString = BIGINT.toString = function() {
+  return new this._type().toString();
+};
+
+FLOAT.prototype = BIGINT.prototype = INTEGER.prototype = {
+
+  construct: function(RealType, length, decimals, unsigned, zerofill) {
+    if (this instanceof RealType) {
+      this._typeName = RealType._typeName;
+      this._unsigned = !!unsigned;
+      this._zerofill = !!zerofill;
+      if (typeof length === 'number') {
+        this._length = length;
+      }
+      if (typeof decimals === 'number') {
+        this._decimals = decimals;
+      }
+    } else {
+      return new RealType(length, decimals, unsigned, zerofill);
+    }
+  },
+
+  get type() {
+    return this.toString();
+  },
+
+  get UNSIGNED() {
+    this._unsigned = true;
+    return this;
+  },
+
+  get ZEROFILL() {
+    this._zerofill = true;
+    return this;
+  },
+
+  toString: function() {
+    var result = this._typeName;
+    if (this._length) {
+      result += '(' + this._length;
+      if (typeof this._decimals === 'number') {
+        result += ',' + this._decimals;
+      }
+      result += ')';
+    }
+    if (this._unsigned) {
+      result += ' UNSIGNED';
+    }
+    if (this._zerofill) {
+      result += ' ZEROFILL';
+    }
+    return result;
+  }
+};
+
+var unsignedDesc = {
+  get: function() {
+    return new this._type(undefined, undefined, true);
+  }
+};
+
+var zerofillDesc = {
+  get: function() {
+    return new this._type(undefined, undefined, undefined, true);
+  }
+};
+
+var typeDesc = {
+  get: function() {
+    return new this._type().toString();
+  }
+};
+
+Object.defineProperty(STRING,  'type', typeDesc);
+Object.defineProperty(INTEGER, 'type', typeDesc);
+Object.defineProperty(BIGINT,  'type', typeDesc);
+Object.defineProperty(FLOAT,   'type', typeDesc);
+
+Object.defineProperty(INTEGER, 'UNSIGNED', unsignedDesc);
+Object.defineProperty(BIGINT,  'UNSIGNED', unsignedDesc);
+Object.defineProperty(FLOAT,   'UNSIGNED', unsignedDesc);
+
+Object.defineProperty(INTEGER, 'ZEROFILL', zerofillDesc);
+Object.defineProperty(BIGINT,  'ZEROFILL', zerofillDesc);
+Object.defineProperty(FLOAT,   'ZEROFILL', zerofillDesc);
+
 module.exports = {
 module.exports = {
-  STRING: 'VARCHAR(255)',
+  STRING: STRING,
+
   TEXT: 'TEXT',
   TEXT: 'TEXT',
-  INTEGER: 'INTEGER',
-  BIGINT:  'BIGINT',
+  INTEGER: INTEGER,
+  BIGINT:  BIGINT,
   DATE: 'DATETIME',
   DATE: 'DATETIME',
   BOOLEAN: 'TINYINT(1)',
   BOOLEAN: 'TINYINT(1)',
-  FLOAT: 'FLOAT',
+  FLOAT: FLOAT,
   NOW: 'NOW',
   NOW: 'NOW',
 
 
   get ENUM() {
   get ENUM() {