mingo/dist/esm/operators/expression/date/dateAdd.js

MongoDB query language for in-memory objects

import { computeValue } from "../../../core";
import {
  adjustDate,
  computeDate,
  DURATION_IN_MILLIS,
  parseTimezone
} from "./_internal";
const $dateAdd = (obj, expr, options) => {
  const args = computeValue(obj, expr, null, options);
  const d = computeDate(obj, expr.startDate, options);
  switch (args.unit) {
    case "year":
      d.setUTCFullYear(d.getUTCFullYear() + args.amount);
      break;
    case "quarter":
      addMonth(d, 3 * args.amount);
      break;
    case "month":
      addMonth(d, args.amount);
      break;
    default:
      d.setTime(d.getTime() + DURATION_IN_MILLIS[args.unit] * args.amount);
  }
  if (args.timezone) {
    const tz = parseTimezone(args.timezone);
    adjustDate(d, tz);
  }
  return d;
};
function addMonth(d, amount) {
  const m = d.getUTCMonth() + amount;
  const yearOffset = Math.floor(m / 12);
  if (m < 0) {
    const month = m % 12 + 12;
    d.setUTCFullYear(d.getUTCFullYear() + yearOffset, month, d.getUTCDate());
  } else {
    d.setUTCFullYear(d.getUTCFullYear() + yearOffset, m % 12, d.getUTCDate());
  }
}
export {
  $dateAdd
};