mingo
Version:
MongoDB query language for in-memory objects
82 lines (81 loc) • 3 kB
JavaScript
import { computeValue } from "../../../core/_internal";
import { assert, isNil, isObject } from "../../../util";
import {
adjustDate,
DATE_FORMAT,
DATE_FORMAT_SEP_RE,
DATE_FORMAT_SYM_RE,
DATE_SYM_TABLE,
MINUTES_PER_HOUR,
MONTHS,
parseTimezone
} from "./_internal";
function tzLetterOffset(c) {
if (c === "Z") return 0;
if (c >= "A" && c < "N") return c.charCodeAt(0) - 64;
return 77 - c.charCodeAt(0);
}
const regexStrip = (s) => s.replace(/^\//, "").replace(/\/$/, "").replace(/\/i/, "");
const REGEX_SPECIAL_CHARS = ["^", ".", "-", "*", "?", "$"];
function regexQuote(s) {
REGEX_SPECIAL_CHARS.forEach((c) => {
s = s.replace(c, `\\${c}`);
});
return s;
}
const $dateFromString = (obj, expr, options) => {
const args = computeValue(obj, expr, null, options);
args.format = args.format || DATE_FORMAT;
args.onNull = args.onNull || null;
let dateString = args.dateString;
if (isNil(dateString)) return args.onNull;
const separators = args.format.split(DATE_FORMAT_SEP_RE);
separators.reverse();
const matches = args.format.match(DATE_FORMAT_SYM_RE);
const dateParts = {};
let expectedPattern = "";
for (let i = 0, len = matches.length; i < len; i++) {
const formatSpecifier = matches[i];
const props = DATE_SYM_TABLE[formatSpecifier];
if (isObject(props)) {
const m2 = props.re.exec(dateString);
const delimiter = separators.pop() || "";
if (m2 !== null) {
dateParts[props.name] = /^\d+$/.exec(m2[0]) ? parseInt(m2[0]) : m2[0];
dateString = dateString.substring(m2.index + m2[0].length + 1);
expectedPattern += regexQuote(delimiter) + regexStrip(props.re.toString());
} else {
dateParts[props.name] = null;
}
}
}
if (isNil(dateParts.month)) {
const abbrMonth = (dateParts.full_month?.slice(0, 3) ?? dateParts.abbr_month ?? "").toLowerCase();
if (MONTHS[abbrMonth]) {
dateParts.month = MONTHS[abbrMonth];
}
}
if (isNil(dateParts.year) || isNil(dateParts.month) || isNil(dateParts.day) || !new RegExp("^" + expectedPattern + "[A-Z]?$").test(args.dateString)) {
return args.onError;
}
const m = args.dateString.match(/([A-Z])$/);
assert(
// only one of in-date timeone or timezone argument but not both.
!(m && args.timezone),
`$dateFromString: you cannot pass in a date/time string with time zone information ('${m && m[0]}') together with a timezone argument`
);
const minuteOffset = m ? tzLetterOffset(m[0]) * MINUTES_PER_HOUR : parseTimezone(args.timezone);
const d = new Date(
Date.UTC(dateParts.year, dateParts.month - 1, dateParts.day, 0, 0, 0)
);
if (!isNil(dateParts.hour)) d.setUTCHours(dateParts.hour);
if (!isNil(dateParts.minute)) d.setUTCMinutes(dateParts.minute);
if (!isNil(dateParts.second)) d.setUTCSeconds(dateParts.second);
if (!isNil(dateParts.millisecond))
d.setUTCMilliseconds(dateParts.millisecond);
adjustDate(d, -minuteOffset);
return d;
};
export {
$dateFromString
};