import { parseTimeScope } from '../temporal-recall-service/time-parser.mjs'; export const RAIN_DEFAULT_DAYS = 3; const TZ_OFFSET_MIN = Number(process.env.TEMPORAL_RECALL_TZ_OFFSET_MIN ?? 480); /** @param {Date} anchor @param {number} deltaDays */ function addDays(anchor, deltaDays) { const shifted = new Date(anchor.getTime() + TZ_OFFSET_MIN * 60_000); const ms = Date.UTC( shifted.getUTCFullYear(), shifted.getUTCMonth(), shifted.getUTCDate() + deltaDays, 0, 0, 0, 0, ) - TZ_OFFSET_MIN * 60_000; return new Date(ms); } /** @param {Date} anchor */ function startOfDay(anchor) { const p = new Date(anchor.getTime() + TZ_OFFSET_MIN * 60_000); const ms = Date.UTC(p.getUTCFullYear(), p.getUTCMonth(), p.getUTCDate(), 0, 0, 0, 0) - TZ_OFFSET_MIN * 60_000; return new Date(ms); } /** @param {Date} anchor */ function endOfDay(anchor) { const p = new Date(anchor.getTime() + TZ_OFFSET_MIN * 60_000); const ms = Date.UTC(p.getUTCFullYear(), p.getUTCMonth(), p.getUTCDate() + 1, 0, 0, 0, 0) - TZ_OFFSET_MIN * 60_000; return new Date(ms); } /** @param {Date} now */ function buildDefault3dRange(now) { return { start: addDays(now, -RAIN_DEFAULT_DAYS).toISOString(), end: endOfDay(now).toISOString(), }; } export const RAIN_TIME_PRESET_IDS = ['today', 'yesterday', '3d', '7d']; /** * @param {string} presetId * @param {Date} [now] */ export function resolveRainTimeRangeFromPreset(presetId, now = new Date()) { const id = String(presetId ?? '3d').trim(); if (id === 'today') { return { range: { start: startOfDay(now).toISOString(), end: endOfDay(now).toISOString() }, source: 'ui_preset', label: '今天', }; } if (id === 'yesterday') { const day = addDays(now, -1); return { range: { start: startOfDay(day).toISOString(), end: endOfDay(day).toISOString() }, source: 'ui_preset', label: '昨天', }; } if (id === '7d') { return { range: { start: addDays(now, -7).toISOString(), end: endOfDay(now).toISOString() }, source: 'ui_preset', label: '近7天', }; } return { range: buildDefault3dRange(now), source: 'ui_preset', label: `近${RAIN_DEFAULT_DAYS}天`, }; } /** Only truly unknowable time phrases trigger clarification; others fall back to default 3d. */ const STRICT_AMBIGUOUS_TIME_PATTERNS = [ /不太确定.{0,8}时间/u, /说不清.{0,6}(?:哪|什么).{0,4}(?:天|时间)/u, ]; const EXPLICIT_TIME_HINT = /昨天|昨日|今天|今日|明天|明日|前天|后天|上周|这周|本周|这个月|本月|上个月|\d{1,2}\s*月|\d{1,2}\s*[::]\d{2}|最近\s*\d+\s*天|最近一周|最近1周|\d{4}-\d{2}-\d{2}/u; /** * @param {string} query * @param {Date} [now] */ export function resolveRainTimeRange(query, now = new Date()) { const text = String(query ?? '').trim(); for (const pattern of STRICT_AMBIGUOUS_TIME_PATTERNS) { if (pattern.test(text)) { return { needsClarification: true, clarificationQuestion: '请说明要分析 MeInput 输入记录的起止时间(例如「9月2日 18:00 到 9月4日 10:00」,或「昨天全天」)。若你不补充,我将默认使用近 3 天。', range: null, source: 'ambiguous', label: null, }; } } if (!EXPLICIT_TIME_HINT.test(text)) { return { needsClarification: false, range: buildDefault3dRange(now), source: 'default_3d', label: `近${RAIN_DEFAULT_DAYS}天`, }; } const scope = parseTimeScope(text, now); const isDefaultFallback = scope.rule_hits?.includes('time:default_week') || scope.rule_hits?.includes('time:recent_fuzzy'); if (isDefaultFallback && /最近|近期|这几天|前几天|那段(?:时间|日子)/u.test(text) && !/最近\s*\d+\s*天/u.test(text)) { return { needsClarification: false, range: buildDefault3dRange(now), source: 'default_3d', label: `近${RAIN_DEFAULT_DAYS}天`, }; } return { needsClarification: false, range: scope.mention_range, source: 'user_explicit', label: scope.relative_label, }; } export function formatRainTimeRangeLabel(range, meta = {}) { if (!range?.start || !range?.end) return meta.label ?? '未指定'; const start = range.start.slice(0, 16).replace('T', ' '); const end = range.end.slice(0, 16).replace('T', ' '); const suffix = meta.source === 'default_3d' ? '(默认近3天)' : ''; return `${start} ~ ${end}${suffix}`; }