chrono-node
Version:
A natural language date parser in Javascript
70 lines • 3.51 kB
JavaScript
import dayjs from "dayjs";
import { implySimilarDate } from "../../utils/dayjs.js";
export default class ForwardDateRefiner {
refine(context, results) {
if (!context.option.forwardDate) {
return results;
}
results.forEach((result) => {
let refMoment = dayjs(context.refDate);
if (result.start.isOnlyTime() && refMoment.isAfter(result.start.dayjs())) {
refMoment = refMoment.add(1, "day");
implySimilarDate(result.start, refMoment);
if (result.end && result.end.isOnlyTime()) {
implySimilarDate(result.end, refMoment);
if (result.start.dayjs().isAfter(result.end.dayjs())) {
refMoment = refMoment.add(1, "day");
implySimilarDate(result.end, refMoment);
}
}
context.debug(() => {
console.log(`${this.constructor.name} adjusted ${result} time result (${result.start})`);
});
}
if (result.start.isOnlyWeekdayComponent() && refMoment.isAfter(result.start.dayjs())) {
if (refMoment.day() >= result.start.get("weekday")) {
refMoment = refMoment.day(result.start.get("weekday") + 7);
}
else {
refMoment = refMoment.day(result.start.get("weekday"));
}
result.start.imply("day", refMoment.date());
result.start.imply("month", refMoment.month() + 1);
result.start.imply("year", refMoment.year());
context.debug(() => {
console.log(`${this.constructor.name} adjusted ${result} weekday (${result.start})`);
});
if (result.end && result.end.isOnlyWeekdayComponent()) {
if (refMoment.day() > result.end.get("weekday")) {
refMoment = refMoment.day(result.end.get("weekday") + 7);
}
else {
refMoment = refMoment.day(result.end.get("weekday"));
}
result.end.imply("day", refMoment.date());
result.end.imply("month", refMoment.month() + 1);
result.end.imply("year", refMoment.year());
context.debug(() => {
console.log(`${this.constructor.name} adjusted ${result} weekday (${result.end})`);
});
}
}
if (result.start.isDateWithUnknownYear() && refMoment.isAfter(result.start.dayjs())) {
for (let i = 0; i < 3 && refMoment.isAfter(result.start.dayjs()); i++) {
result.start.imply("year", result.start.get("year") + 1);
context.debug(() => {
console.log(`${this.constructor.name} adjusted ${result} year (${result.start})`);
});
if (result.end && !result.end.isCertain("year")) {
result.end.imply("year", result.end.get("year") + 1);
context.debug(() => {
console.log(`${this.constructor.name} adjusted ${result} month (${result.start})`);
});
}
}
}
});
return results;
}
}
//# sourceMappingURL=ForwardDateRefiner.js.map