UNPKG

node-nlp

Version:

Library for NLU (Natural Language Understanding) done in Node.js

120 lines (93 loc) 3.38 kB
/* Allows future dates according to 'days from now' eg: 90 days from now 90 days forward 90 days out */ var moment = require('moment'); var Parser = require('../parser').Parser; var ParsedResult = require('../../result').ParsedResult; var util = require('../../utils/EN'); var PATTERN = new RegExp('' + '(\\W|^)' + '(?:within\\s*)?' + '(' + util.INTEGER_WORDS_PATTERN + '|[0-9]+|an?(?:\\s*few)?|half(?:\\s*an?)?)\\s*' + '(seconds?|min(?:ute)?s?|hours?|weeks?|days?|months?|years?)\\s*' + '(?:from now|henceforth|forward|out)(?=(?:\\W|$))', 'i'); var STRICT_PATTERN = new RegExp('' + '(\\W|^)' + '(?:within\\s*)?' + '([0-9]+|an?)\\s*' + '(seconds?|minutes?|hours?|days?)\\s*' + 'from now(?=(?:\\W|$))', 'i'); exports.Parser = function ENTimeFromNowFormatParser() { Parser.apply(this, arguments); this.pattern = function() { return this.isStrictMode() ? STRICT_PATTERN : PATTERN; }; this.extract = function(text, ref, match, opt) { if (match.index > 0 && text[match.index-1].match(/\w/)) { return null; } var text = match[0]; text = match[0].substr(match[1].length, match[0].length - match[1].length); index = match.index + match[1].length; var result = new ParsedResult({ index: index, text: text, ref: ref }); var num = match[2].toLowerCase() ; if (util.INTEGER_WORDS[num] !== undefined) { num = util.INTEGER_WORDS[num]; } else if(num === 'a' || num === 'an'){ num = 1; } else if (num.match(/few/)) { num = 3; } else if (num.match(/half/)) { num = 0.5; } else { num = parseInt(num); } var date = moment(ref); if (match[3].match(/hour|min|second/i)) { if (match[3].match(/hour/i)) { date.add(+num, 'hour'); } else if (match[3].match(/min/i)) { date.add(+num, 'minute'); } else if (match[3].match(/second/i)) { date.add(+num, 'second'); } result.start.imply('day', date.date()); result.start.imply('month', date.month() + 1); result.start.imply('year', date.year()); result.start.assign('hour', date.hour()); result.start.assign('minute', date.minute()); result.start.assign('second', date.second()); result.tags['ENTimeFromNowFormatParser'] = true; return result; } if (match[3].match(/week/i)) { date.add(+num, 'week'); result.start.imply('day', date.date()); result.start.imply('month', date.month() + 1); result.start.imply('year', date.year()); result.start.imply('weekday', date.day()); return result; } if (match[3].match(/day/i)) { date.add(+num, 'd'); } if (match[3].match(/month/i)) { date.add(+num, 'month'); } if (match[3].match(/year/i)) { date.add(+num, 'year'); } result.start.assign('day', date.date()); result.start.assign('month', date.month() + 1); result.start.assign('year', date.year()); return result; }; }