chrono-node
Version:
A natural language date parser in Javascript
69 lines (57 loc) • 2.56 kB
text/typescript
import { TIME_UNIT_DICTIONARY } from "../constants";
import { ParsingContext } from "../../../chrono";
import { ParsingComponents } from "../../../results";
import { AbstractParserWithWordBoundaryChecking } from "../../../common/parsers/AbstractParserWithWordBoundary";
import { matchAnyPattern } from "../../../utils/pattern";
const PATTERN = new RegExp(
`(dit|deze|(?:aan)?komend|volgend|afgelopen|vorig)e?\\s*(${matchAnyPattern(TIME_UNIT_DICTIONARY)})(?=\\s*)` +
"(?=\\W|$)",
"i"
);
const MODIFIER_WORD_GROUP = 1;
const RELATIVE_WORD_GROUP = 2;
export default class NLRelativeDateFormatParser extends AbstractParserWithWordBoundaryChecking {
innerPattern(): RegExp {
return PATTERN;
}
innerExtract(context: ParsingContext, match: RegExpMatchArray): ParsingComponents {
const modifier = match[MODIFIER_WORD_GROUP].toLowerCase();
const unitWord = match[RELATIVE_WORD_GROUP].toLowerCase();
const timeunit = TIME_UNIT_DICTIONARY[unitWord];
if (modifier == "volgend" || modifier == "komend" || modifier == "aankomend") {
const timeUnits = {};
timeUnits[timeunit] = 1;
return ParsingComponents.createRelativeFromReference(context.reference, timeUnits);
}
if (modifier == "afgelopen" || modifier == "vorig") {
const timeUnits = {};
timeUnits[timeunit] = -1;
return ParsingComponents.createRelativeFromReference(context.reference, timeUnits);
}
const components = context.createParsingComponents();
let date = new Date(context.reference.instant.getTime());
// This week
if (unitWord.match(/week/i)) {
date.setDate(date.getDate() - date.getDay());
components.imply("day", date.getDate());
components.imply("month", date.getMonth() + 1);
components.imply("year", date.getFullYear());
}
// This month
else if (unitWord.match(/maand/i)) {
date.setDate(1);
components.imply("day", date.getDate());
components.assign("year", date.getFullYear());
components.assign("month", date.getMonth() + 1);
}
// This year
else if (unitWord.match(/jaar/i)) {
date.setDate(1);
date.setMonth(0);
components.imply("day", date.getDate());
components.imply("month", date.getMonth() + 1);
components.assign("year", date.getFullYear());
}
return components;
}
}