node-nlp
Version:
Library for NLU (Natural Language Understanding) done in Node.js
87 lines (74 loc) • 2.8 kB
JavaScript
/*
*/
var moment = require('moment');
var Parser = require('../parser').Parser;
var ParsedResult = require('../../result').ParsedResult;
var DAYS_OFFSET = {
'sonntag': 0, 'so': 0,
'montag': 1, 'mo': 1,
'dienstag': 2, 'di': 2,
'mittwoch': 3, 'mi': 3,
'donnerstag': 4, 'do': 4,
'freitag': 5, 'fr': 5,
'samstag': 6, 'sa': 6
};
var PATTERN = new RegExp('(\\W|^)' +
'(?:(?:\\,|\\(|\\()\\s*)?' +
'(?:a[mn]\\s*?)?' +
'(?:(diese[mn]|letzte[mn]|n(?:ä|ae)chste[mn])\\s*)?' +
'(' + Object.keys(DAYS_OFFSET).join('|') + ')' +
'(?:\\s*(?:\\,|\\)|\\)))?' +
'(?:\\s*(diese|letzte|n(?:ä|ae)chste)\\s*woche)?' +
'(?=\\W|$)', 'i');
var PREFIX_GROUP = 2;
var WEEKDAY_GROUP = 3;
var POSTFIX_GROUP = 4;
exports.Parser = function DEWeekdayParser() {
Parser.apply(this, arguments);
this.pattern = function() { return PATTERN; };
this.extract = function(text, ref, match, opt){
var index = match.index + match[1].length;
var text = match[0].substr(match[1].length, match[0].length - match[1].length);
var result = new ParsedResult({
index: index,
text: text,
ref: ref,
});
var dayOfWeek = match[WEEKDAY_GROUP].toLowerCase();
var offset = DAYS_OFFSET[dayOfWeek];
if(offset === undefined) return null;
var startMoment = moment(ref);
var prefix = match[PREFIX_GROUP];
var postfix = match[POSTFIX_GROUP];
var refOffset = startMoment.day();
var norm = prefix || postfix;
norm = norm || '';
norm = norm.toLowerCase();
if (/letzte/.test(norm)) {
startMoment.day(offset - 7);
} else if (/n(?:ä|ae)chste/.test(norm)) {
startMoment.day(offset + 7);
} else if (/diese/.test(norm)) {
if ( opt.forwardDatesOnly && refOffset > offset ) {
startMoment.day(offset + 7);
} else {
startMoment.day(offset);
}
} else {
if ( opt.forwardDatesOnly && refOffset > offset ) {
startMoment.day(offset + 7);
} else if (!opt.forwardDatesOnly && Math.abs(offset - 7 - refOffset) < Math.abs(offset - refOffset)) {
startMoment.day(offset - 7);
} else if (!opt.forwardDatesOnly && Math.abs(offset + 7 - refOffset) < Math.abs(offset - refOffset)) {
startMoment.day(offset + 7);
} else {
startMoment.day(offset);
}
}
result.start.assign('weekday', offset);
result.start.imply('day', startMoment.date());
result.start.imply('month', startMoment.month() + 1);
result.start.imply('year', startMoment.year());
return result;
}
};