openclaw-server 0.1.0

package/src/tasks/time-parser.ts

@@ -0,0 +1,610 @@
+import {
+  addDaysInTimeZone,
+  getTimeZoneDateParts,
+  startOfDayInTimeZone,
+  zonedDateTimeToDate,
+} from "./timezone.js";
+import type { ParsedTaskAction, ParsedTaskDraft, ReminderPreference, TaskScope } from "./types.js";
+
+const weekdayMap: Record<string, number> = {
+  一: 1,
+  二: 2,
+  三: 3,
+  四: 4,
+  五: 5,
+  六: 6,
+  日: 0,
+  天: 0,
+};
+
+const chineseDigitMap: Record<string, number> = {
+  零: 0,
+  〇: 0,
+  一: 1,
+  二: 2,
+  两: 2,
+  三: 3,
+  四: 4,
+  五: 5,
+  六: 6,
+  七: 7,
+  八: 8,
+  九: 9,
+};
+
+const chineseUnitMap: Record<string, number> = {
+  十: 10,
+  百: 100,
+  千: 1000,
+};
+
+const numberTokenSource = String.raw`(?:\d{1,4}|[零〇一二两三四五六七八九十百千]+)`;
+const clockPattern = new RegExp(
+  String.raw`(?:(凌晨|早上|早晨|上午|中午|下午|晚上|今晚|傍晚)\s*)?(${numberTokenSource})(?:[:：](${numberTokenSource})|点(半|一刻|三刻|(?:${numberTokenSource})分?)?)`,
+  "u",
+);
+const monthDayPattern = new RegExp(String.raw`(${numberTokenSource})月(${numberTokenSource})(?:日|号)`, "u");
+const fullDatePattern = new RegExp(
+  String.raw`(\d{4})[-/年](${numberTokenSource})[-/月](${numberTokenSource})(?:日|号)?`,
+  "u",
+);
+const minuteAfterPattern = new RegExp(String.raw`(${numberTokenSource})\s*(分钟|分)后`, "u");
+const hourAfterPattern = new RegExp(String.raw`(${numberTokenSource})\s*(小时|时)后`, "u");
+const hourAndHalfAfterPattern = new RegExp(
+  String.raw`(${numberTokenSource})\s*(?:(?:个)?半小时|(?:个)?(?:小时|时)半)后`,
+  "u",
+);
+const reminderMinutePattern = new RegExp(
+  String.raw`(提前\s*(${numberTokenSource})\s*(分钟|分)提醒)`,
+  "u",
+);
+const reminderHourPattern = new RegExp(String.raw`(提前\s*(${numberTokenSource})\s*(小时|时)提醒)`, "u");
+const reminderHourAndHalfPattern = new RegExp(
+  String.raw`(提前\s*(${numberTokenSource})\s*(?:(?:个)?半小时|(?:个)?(?:小时|时)半)提醒)`,
+  "u",
+);
+const snoozeMinutePattern = new RegExp(String.raw`^延后\s*(${numberTokenSource})\s*(分钟|分)$`, "u");
+const snoozeHourPattern = new RegExp(String.raw`^延后\s*(${numberTokenSource})\s*(小时|时)$`, "u");
+const snoozeHourAndHalfPattern = new RegExp(
+  String.raw`^延后\s*(${numberTokenSource})\s*(?:(?:个)?半小时|(?:个)?(?:小时|时)半)$`,
+  "u",
+);
+
+function addMinutes(date: Date, minutes: number): Date {
+  return new Date(date.getTime() + minutes * 60_000);
+}
+
+function stripText(text: string): string {
+  return text.replace(/\s+/g, " ").trim();
+}
+
+function parseNumberToken(token: string): number | undefined {
+  const normalized = token.normalize("NFKC").trim();
+  if (!normalized) {
+    return undefined;
+  }
+  if (/^\d+$/u.test(normalized)) {
+    return Number(normalized);
+  }
+  if (!/^[零〇一二两三四五六七八九十百千]+$/u.test(normalized)) {
+    return undefined;
+  }
+
+  if (!/[十百千]/u.test(normalized)) {
+    const digits = [...normalized].map((char) => chineseDigitMap[char]);
+    if (digits.some((value) => value === undefined)) {
+      return undefined;
+    }
+    return Number(digits.join(""));
+  }
+
+  let total = 0;
+  let current = 0;
+  for (const char of normalized) {
+    const digit = chineseDigitMap[char];
+    if (digit !== undefined) {
+      current = digit;
+      continue;
+    }
+
+    const unit = chineseUnitMap[char];
+    if (!unit) {
+      return undefined;
+    }
+    total += (current || 1) * unit;
+    current = 0;
+  }
+
+  return total + current;
+}
+
+function parseClockMinuteToken(token?: string): number | undefined {
+  if (!token) {
+    return 0;
+  }
+  if (token === "半") {
+    return 30;
+  }
+  if (token === "一刻") {
+    return 15;
+  }
+  if (token === "三刻") {
+    return 45;
+  }
+  return parseNumberToken(token.replace(/分$/u, ""));
+}
+
+function removeFragments(text: string, fragments: string[]): string {
+  let output = text;
+  for (const fragment of fragments) {
+    if (!fragment) {
+      continue;
+    }
+    output = output.replace(fragment, " ");
+  }
+  return stripText(output);
+}
+
+function cleanupTitle(text: string): string {
+  return stripText(
+    text
+      .replace(/提醒我一下|提醒我|提醒一下|提醒/g, " ")
+      .replace(/需要提醒吗|准时提醒|到点提醒(?:就行|即可|就好|就可以)?|不用提醒(?:了)?|不需要提醒|别提醒(?:了)?/g, " ")
+      .replace(/[，。,！!？?]/g, " "),
+  );
+}
+
+function parseClock(text: string): { hour: number; minute: number; matchedText: string } | undefined {
+  const match = clockPattern.exec(text);
+  if (!match) {
+    return undefined;
+  }
+
+  const rawPeriod = match[1] ?? "";
+  const period = rawPeriod === "今晚" || rawPeriod === "傍晚" ? "晚上" : rawPeriod === "早晨" ? "早上" : rawPeriod;
+  let hour = parseNumberToken(match[2]);
+  const minute = match[3] ? parseNumberToken(match[3]) : parseClockMinuteToken(match[4]);
+
+  if (hour === undefined || minute === undefined) {
+    return undefined;
+  }
+
+  if (["下午", "晚上"].includes(period) && hour < 12) {
+    hour += 12;
+  }
+  if (period === "中午" && hour < 11) {
+    hour += 12;
+  }
+  if ((period === "凌晨" || period === "晚上") && hour === 12) {
+    hour = 0;
+  }
+
+  if (hour > 23 || minute > 59) {
+    return undefined;
+  }
+
+  return {
+    hour,
+    minute,
+    matchedText: match[0],
+  };
+}
+
+function shiftCalendarDate(
+  parts: Pick<ReturnType<typeof getTimeZoneDateParts>, "year" | "month" | "day">,
+  days: number,
+) {
+  const shifted = new Date(Date.UTC(parts.year, parts.month - 1, parts.day + days));
+  return {
+    year: shifted.getUTCFullYear(),
+    month: shifted.getUTCMonth() + 1,
+    day: shifted.getUTCDate(),
+  };
+}
+
+function weekdayFromDateParts(parts: Pick<ReturnType<typeof getTimeZoneDateParts>, "year" | "month" | "day">): number {
+  return new Date(Date.UTC(parts.year, parts.month - 1, parts.day)).getUTCDay();
+}
+
+function parseDay(
+  text: string,
+  now: Date,
+  timeZone: string,
+): { baseDate: Date; matchedText: string; rollForwardIfPast?: boolean } | undefined {
+  const todayStart = startOfDayInTimeZone(now, timeZone);
+  const todayParts = getTimeZoneDateParts(todayStart, timeZone);
+  const relativeMap: Array<{ pattern: RegExp; days: number }> = [
+    { pattern: /大后天/u, days: 3 },
+    { pattern: /后天/u, days: 2 },
+    { pattern: /明天/u, days: 1 },
+    { pattern: /今天/u, days: 0 },
+  ];
+  for (const item of relativeMap) {
+    const match = item.pattern.exec(text);
+    if (match) {
+      return {
+        baseDate: addDaysInTimeZone(todayStart, item.days, timeZone),
+        matchedText: match[0],
+      };
+    }
+  }
+
+  const nextWeekMatch = /下(?:周|星期|礼拜)([一二三四五六日天])/u.exec(text);
+  if (nextWeekMatch) {
+    const target = weekdayMap[nextWeekMatch[1]];
+    const base = shiftCalendarDate(todayParts, 7);
+    const delta = (target - weekdayFromDateParts(base) + 7) % 7;
+    const candidate = shiftCalendarDate(base, delta);
+    return {
+      baseDate: zonedDateTimeToDate(candidate, timeZone),
+      matchedText: nextWeekMatch[0],
+    };
+  }
+
+  const weekdayMatch = /(?:本周|这周|本星期|这星期|本礼拜|这礼拜|周|星期|礼拜)([一二三四五六日天])/u.exec(text);
+  if (weekdayMatch) {
+    const target = weekdayMap[weekdayMatch[1]];
+    const delta = (target - weekdayFromDateParts(todayParts) + 7) % 7;
+    const candidate = shiftCalendarDate(todayParts, delta);
+    return {
+      baseDate: zonedDateTimeToDate(candidate, timeZone),
+      matchedText: weekdayMatch[0],
+      rollForwardIfPast: true,
+    };
+  }
+
+  const monthDayMatch = monthDayPattern.exec(text);
+  if (monthDayMatch) {
+    const month = parseNumberToken(monthDayMatch[1]);
+    const day = parseNumberToken(monthDayMatch[2]);
+    if (month === undefined || day === undefined) {
+      return undefined;
+    }
+    let year = todayParts.year;
+    if (month < todayParts.month || (month === todayParts.month && day < todayParts.day)) {
+      year += 1;
+    }
+    const candidate = zonedDateTimeToDate({ year, month, day }, timeZone);
+    const candidateParts = getTimeZoneDateParts(candidate, timeZone);
+    if (candidateParts.month !== month || candidateParts.day !== day) {
+      return undefined;
+    }
+    return {
+      baseDate: candidate,
+      matchedText: monthDayMatch[0],
+    };
+  }
+
+  const fullDateMatch = fullDatePattern.exec(text);
+  if (fullDateMatch) {
+    const year = Number(fullDateMatch[1]);
+    const month = parseNumberToken(fullDateMatch[2]);
+    const day = parseNumberToken(fullDateMatch[3]);
+    if (month === undefined || day === undefined) {
+      return undefined;
+    }
+    const candidate = zonedDateTimeToDate({ year, month, day }, timeZone);
+    const candidateParts = getTimeZoneDateParts(candidate, timeZone);
+    if (candidateParts.year !== year || candidateParts.month !== month || candidateParts.day !== day) {
+      return undefined;
+    }
+    return {
+      baseDate: candidate,
+      matchedText: fullDateMatch[0],
+    };
+  }
+
+  return undefined;
+}
+
+function hasDateHint(text: string): boolean {
+  return (
+    /(今天|明天|后天|大后天|下(?:周|星期|礼拜)[一二三四五六日天]|(?:本周|这周|本星期|这星期|本礼拜|这礼拜|周|星期|礼拜)[一二三四五六日天])/u.test(
+      text,
+    ) ||
+    monthDayPattern.test(text) ||
+    fullDatePattern.test(text) ||
+    clockPattern.test(text) ||
+    /半(?:个)?小时后|一刻钟后|三刻钟后/u.test(text) ||
+    hourAndHalfAfterPattern.test(text) ||
+    minuteAfterPattern.test(text) ||
+    hourAfterPattern.test(text)
+  );
+}
+
+function parseAbsoluteDueAt(
+  text: string,
+  now: Date,
+  timeZone: string,
+): { dueAt: Date; fragments: string[] } | undefined {
+  const clock = parseClock(text);
+  const day = parseDay(text, now, timeZone);
+  if (!clock && !day) {
+    return undefined;
+  }
+  if (!clock) {
+    return undefined;
+  }
+
+  const baseDate = day?.baseDate ?? startOfDayInTimeZone(now, timeZone);
+  const baseParts = getTimeZoneDateParts(baseDate, timeZone);
+  let dueAt = zonedDateTimeToDate(
+    {
+      year: baseParts.year,
+      month: baseParts.month,
+      day: baseParts.day,
+      hour: clock.hour,
+      minute: clock.minute,
+    },
+    timeZone,
+  );
+
+  if (!day && dueAt.getTime() <= now.getTime()) {
+    dueAt = addDaysInTimeZone(dueAt, 1, timeZone);
+  }
+  if (day?.rollForwardIfPast && dueAt.getTime() <= now.getTime()) {
+    dueAt = addDaysInTimeZone(dueAt, 7, timeZone);
+  }
+
+  return {
+    dueAt,
+    fragments: [day?.matchedText ?? "", clock.matchedText].filter(Boolean),
+  };
+}
+
+function parseRelativeDueAt(text: string, now: Date): { dueAt: Date; fragments: string[] } | undefined {
+  const halfHourMatch = /半(?:个)?小时后/u.exec(text);
+  if (halfHourMatch) {
+    return {
+      dueAt: addMinutes(now, 30),
+      fragments: [halfHourMatch[0]],
+    };
+  }
+
+  const quarterMatch = /(一刻钟|三刻钟)后/u.exec(text);
+  if (quarterMatch) {
+    return {
+      dueAt: addMinutes(now, quarterMatch[1] === "一刻钟" ? 15 : 45),
+      fragments: [quarterMatch[0]],
+    };
+  }
+
+  const hourAndHalfMatch = hourAndHalfAfterPattern.exec(text);
+  if (hourAndHalfMatch) {
+    const hours = parseNumberToken(hourAndHalfMatch[1]);
+    if (hours === undefined) {
+      return undefined;
+    }
+    return {
+      dueAt: addMinutes(now, hours * 60 + 30),
+      fragments: [hourAndHalfMatch[0]],
+    };
+  }
+
+  const minuteMatch = minuteAfterPattern.exec(text);
+  if (minuteMatch) {
+    const minutes = parseNumberToken(minuteMatch[1]);
+    if (minutes === undefined) {
+      return undefined;
+    }
+    return {
+      dueAt: addMinutes(now, minutes),
+      fragments: [minuteMatch[0]],
+    };
+  }
+
+  const hourMatch = hourAfterPattern.exec(text);
+  if (hourMatch) {
+    const hours = parseNumberToken(hourMatch[1]);
+    if (hours === undefined) {
+      return undefined;
+    }
+    return {
+      dueAt: addMinutes(now, hours * 60),
+      fragments: [hourMatch[0]],
+    };
+  }
+
+  return undefined;
+}
+
+export function parseReminderPreference(text: string): ReminderPreference | undefined {
+  const halfHourMatch = /(提前\s*半(?:个)?小时提醒)/u.exec(text);
+  if (halfHourMatch) {
+    return {
+      offsetMinutes: 30,
+      matchedText: halfHourMatch[1],
+    };
+  }
+
+  const quarterMatch = /(提前\s*(一刻钟|三刻钟)提醒)/u.exec(text);
+  if (quarterMatch) {
+    return {
+      offsetMinutes: quarterMatch[2] === "一刻钟" ? 15 : 45,
+      matchedText: quarterMatch[1],
+    };
+  }
+
+  const hourAndHalfMatch = reminderHourAndHalfPattern.exec(text);
+  if (hourAndHalfMatch) {
+    const hours = parseNumberToken(hourAndHalfMatch[2]);
+    if (hours === undefined) {
+      return undefined;
+    }
+    return {
+      offsetMinutes: hours * 60 + 30,
+      matchedText: hourAndHalfMatch[1],
+    };
+  }
+
+  const minuteMatch = reminderMinutePattern.exec(text);
+  if (minuteMatch) {
+    const minutes = parseNumberToken(minuteMatch[2]);
+    if (minutes === undefined) {
+      return undefined;
+    }
+    return {
+      offsetMinutes: minutes,
+      matchedText: minuteMatch[1],
+    };
+  }
+
+  const hourMatch = reminderHourPattern.exec(text);
+  if (hourMatch) {
+    const hours = parseNumberToken(hourMatch[2]);
+    if (hours === undefined) {
+      return undefined;
+    }
+    return {
+      offsetMinutes: hours * 60,
+      matchedText: hourMatch[1],
+    };
+  }
+
+  const immediateMatch = /(到点提醒(?:就行|即可|就好|就可以)?|准时提醒)/u.exec(text);
+  if (immediateMatch) {
+    return {
+      offsetMinutes: 0,
+      matchedText: immediateMatch[1],
+    };
+  }
+
+  const noneMatch = /(不用提醒(?:了)?|不需要提醒|别提醒(?:了)?)/u.exec(text);
+  if (noneMatch) {
+    return {
+      offsetMinutes: null,
+      matchedText: noneMatch[1],
+    };
+  }
+
+  return undefined;
+}
+
+export function parseTaskAction(text: string): ParsedTaskAction | undefined {
+  const normalized = stripText(text);
+  if (/^(完成|已完成)$/u.test(normalized)) {
+    return { kind: "done" };
+  }
+  if (/^(取消|取消任务)$/u.test(normalized)) {
+    return { kind: "cancel" };
+  }
+
+  const halfHourMatch = /^延后\s*半(?:个)?小时$/u.exec(normalized);
+  if (halfHourMatch) {
+    return { kind: "snooze", minutes: 30 };
+  }
+
+  const quarterMatch = /^延后\s*(一刻钟|三刻钟)$/u.exec(normalized);
+  if (quarterMatch) {
+    return { kind: "snooze", minutes: quarterMatch[1] === "一刻钟" ? 15 : 45 };
+  }
+
+  const hourAndHalfMatch = snoozeHourAndHalfPattern.exec(normalized);
+  if (hourAndHalfMatch) {
+    const hours = parseNumberToken(hourAndHalfMatch[1]);
+    if (hours === undefined) {
+      return undefined;
+    }
+    return { kind: "snooze", minutes: hours * 60 + 30 };
+  }
+
+  const minuteMatch = snoozeMinutePattern.exec(normalized);
+  if (minuteMatch) {
+    const minutes = parseNumberToken(minuteMatch[1]);
+    if (minutes === undefined) {
+      return undefined;
+    }
+    return { kind: "snooze", minutes };
+  }
+
+  const hourMatch = snoozeHourPattern.exec(normalized);
+  if (hourMatch) {
+    const hours = parseNumberToken(hourMatch[1]);
+    if (hours === undefined) {
+      return undefined;
+    }
+    return { kind: "snooze", minutes: hours * 60 };
+  }
+
+  return undefined;
+}
+
+export function parseTaskQuery(text: string): TaskScope | undefined {
+  const normalized = stripText(text);
+  const compact = normalized.replace(/[，。！？?,.!]/gu, "");
+  if (/(今天.*(任务|待办|安排|有什么)|今日.*(任务|待办|安排))/u.test(normalized)) {
+    return "today";
+  }
+  if (/(明天.*(任务|待办|安排|有什么))/u.test(normalized)) {
+    return "tomorrow";
+  }
+  if (/(本周.*(任务|待办|安排|计划)|这周.*(任务|待办|安排))/u.test(normalized)) {
+    return "week";
+  }
+  if (/(逾期.*(任务|待办)|过期.*(任务|待办))/u.test(normalized)) {
+    return "overdue";
+  }
+  if (/(全部.*任务|所有.*任务)/u.test(normalized)) {
+    return "all";
+  }
+  if (
+    /^(?:我(?:的)?(?:任务|待办|安排)|我(?:现在)?有哪(?:些)?(?:任务|待办|安排)|我(?:现在)?有什么(?:任务|待办|安排)|我(?:的)?(?:任务|待办|安排)有哪(?:些)?|(?:查看|看看|列出|显示)(?:一下)?我(?:的)?(?:任务|待办|安排)|(?:任务|待办|安排)列表|有哪些(?:任务|待办|安排)|有什么(?:任务|待办|安排))$/u.test(
+      compact,
+    )
+  ) {
+    return "all";
+  }
+  return undefined;
+}
+
+export function isConfirmation(text: string): boolean {
+  return /^(确认|是|好的|好|可以|创建吧)$/u.test(stripText(text));
+}
+
+export { formatLocalDateTime } from "./timezone.js";
+
+export function parseTaskDraft(text: string, now: Date, timeZone: string): ParsedTaskDraft | undefined {
+  const reminder = parseReminderPreference(text);
+  const withoutReminder = reminder ? text.replace(reminder.matchedText, " ") : text;
+  const relative = parseRelativeDueAt(withoutReminder, now);
+  const absolute = relative ? undefined : parseAbsoluteDueAt(withoutReminder, now, timeZone);
+  const parsed = relative ?? absolute;
+
+  if (!parsed) {
+    if (hasDateHint(withoutReminder)) {
+      const title = cleanupTitle(withoutReminder);
+      if (!title) {
+        return { kind: "missing_title" };
+      }
+      return { kind: "missing_time", title };
+    }
+    return undefined;
+  }
+
+  const title = cleanupTitle(removeFragments(withoutReminder, parsed.fragments));
+  if (!title) {
+    return { kind: "missing_title" };
+  }
+
+  return {
+    kind: "ready",
+    title,
+    dueAt: parsed.dueAt,
+    reminderOffsetMinutes: reminder?.offsetMinutes,
+  };
+}
+
+export function summarizeScope(scope: TaskScope): string {
+  switch (scope) {
+    case "today":
+      return "今天";
+    case "tomorrow":
+      return "明天";
+    case "week":
+      return "本周";
+    case "overdue":
+      return "逾期";
+    case "all":
+      return "全部";
+  }
+}