不要怂,就是干,撸起袖子干!

range.js 1.91 KB
'use strict';

var Utils = require('../../utils'),
  moment = require('moment');

module.exports = {
  stringify: function (data) {
    if (data === null) return null;

    if (!Utils._.isArray(data) || data.length !== 2) return '';

    if (Utils._.any(data, Utils._.isNull)) return '';

    if (data.hasOwnProperty('inclusive')) {
      if (!data.inclusive) data.inclusive = [false, false];
      else if (data.inclusive === true) data.inclusive = [true, true];
    } else {
      data.inclusive = [false, false];
    }

    Utils._.each(data, function (value, index) {
      if (Utils._.isObject(value)) {
        if (value.hasOwnProperty('inclusive')) data.inclusive[index] = !!value.inclusive;
        if (value.hasOwnProperty('value')) data[index] = value.value;
      }
    });

    return (data.inclusive[0] ? '[' : '(') + JSON.stringify(data[0]) + ',' + JSON.stringify(data[1]) + (data.inclusive[1] ? ']' : ')');
  },
  parse: function (value, AttributeType) {
    if (value === null) return null;

    if(typeof AttributeType === 'function') AttributeType = new AttributeType();
    AttributeType = AttributeType || ''; // if attribute is not defined, assign empty string in order to prevent
                                         // AttributeType.toString() to fail with uncaught exception later in the code
    var result = value
      .substring(1, value.length - 1)
      .split(',', 2);

    if (result.length !== 2) return value;

    result = result
      .map(function (value) {
        switch (AttributeType.toString()) {
          case 'int4range':
            return parseInt(value, 10);
          case 'numrange':
            return parseFloat(value);
          case 'daterange':
          case 'tsrange':
          case 'tstzrange':
            return moment(value).toDate();
        }

        return value;
      });

    result.inclusive = [(value[0] === '['), (value[value.length - 1] === ']')];

    return result;
  }
};