iris-chatbot 0.2.4

package/template/src/lib/tooling/tools/notes_plus.ts

@@ -0,0 +1,294 @@
+import { runCommandSafe } from "../runtime";
+import { ensureMacOS } from "../safety";
+import type { ToolDefinition, ToolExecutionContext } from "../types";
+import { normalizeNotesBody } from "./notes";
+
+type NotesGetByTitleInput = {
+  title?: string;
+  folder?: string;
+};
+
+type NotesReplaceSectionInput = {
+  title?: string;
+  heading?: string;
+  content?: string;
+  folder?: string;
+  createIfMissing?: boolean;
+};
+
+type NotesStructuredSection = {
+  heading: string;
+  bullets?: string[];
+  paragraphs?: string[];
+};
+
+type NotesCreateStructuredInput = {
+  title?: string;
+  folder?: string;
+  sections?: NotesStructuredSection[];
+};
+
+const APPLESCRIPT_TIMEOUT_MS = 30_000;
+
+function asObject(input: unknown): Record<string, unknown> {
+  if (!input || typeof input !== "object") {
+    throw new Error("Tool input must be an object.");
+  }
+  return input as Record<string, unknown>;
+}
+
+function asString(input: unknown, field: string): string {
+  if (typeof input !== "string" || !input.trim()) {
+    throw new Error(`Missing required string field: ${field}`);
+  }
+  return input.trim();
+}
+
+async function runAppleScript(script: string, args: string[] = [], signal?: AbortSignal) {
+  const { stdout } = await runCommandSafe({
+    command: "osascript",
+    args: ["-e", script, ...args],
+    signal,
+    timeoutMs: APPLESCRIPT_TIMEOUT_MS,
+  });
+  return stdout;
+}
+
+function buildStructuredBody(sections: NotesStructuredSection[]): string {
+  return sections
+    .map((section) => {
+      const lines = [`## ${section.heading}`];
+      if (Array.isArray(section.paragraphs)) {
+        lines.push(...section.paragraphs.filter(Boolean));
+      }
+      if (Array.isArray(section.bullets)) {
+        lines.push(...section.bullets.filter(Boolean).map((bullet) => `- ${bullet}`));
+      }
+      return lines.join("\n");
+    })
+    .join("\n\n");
+}
+
+async function runNotesGetByTitle(input: unknown) {
+  ensureMacOS("Apple Notes automation");
+  const payload = asObject(input) as NotesGetByTitleInput;
+  const title = asString(payload.title, "title");
+  const folder = typeof payload.folder === "string" ? payload.folder.trim() : "";
+
+  const script =
+    'on run argv\n' +
+    'set targetTitle to item 1 of argv\n' +
+    'set targetFolderName to item 2 of argv\n' +
+    'tell application "Notes"\n' +
+    'repeat with acc in accounts\n' +
+    'repeat with f in folders of acc\n' +
+    'if targetFolderName is "" or name of f is targetFolderName then\n' +
+    'repeat with n in notes of f\n' +
+    'if name of n is targetTitle then\n' +
+    'set noteBody to body of n\n' +
+    'return (name of n as text) & tab & (name of f as text) & tab & noteBody\n' +
+    'end if\n' +
+    'end repeat\n' +
+    'end if\n' +
+    'end repeat\n' +
+    'end repeat\n' +
+    'end tell\n' +
+    'return ""\n' +
+    'end run';
+  const output = await runAppleScript(script, [title, folder]);
+  if (!output) {
+    return { found: false, title, folder: folder || null };
+  }
+
+  const [foundTitle, foundFolder, body] = output.split("\t");
+  return {
+    found: true,
+    title: foundTitle,
+    folder: foundFolder,
+    bodyPreview: (body ?? "").replace(/\s+/g, " ").slice(0, 600),
+  };
+}
+
+async function runNotesReplaceSection(input: unknown, context: ToolExecutionContext) {
+  ensureMacOS("Apple Notes automation");
+  const payload = asObject(input) as NotesReplaceSectionInput;
+  const title = asString(payload.title, "title");
+  const heading = asString(payload.heading, "heading");
+  const content = asString(payload.content, "content");
+  const folder = typeof payload.folder === "string" ? payload.folder.trim() : "";
+  const createIfMissing = payload.createIfMissing !== false;
+  const sectionHtml = normalizeNotesBody(`## ${heading}\n${content}`);
+
+  if (context.localTools.dryRun) {
+    return {
+      dryRun: true,
+      action: "notes_replace_section",
+      title,
+      heading,
+      folder,
+      createIfMissing,
+      contentPreview: sectionHtml.slice(0, 200),
+    };
+  }
+
+  const script =
+    'on run argv\n' +
+    'set targetTitle to item 1 of argv\n' +
+    'set folderName to item 2 of argv\n' +
+    'set headingText to item 3 of argv\n' +
+    'set newSection to item 4 of argv\n' +
+    'set createMissing to (item 5 of argv) is "true"\n' +
+    'tell application "Notes"\n' +
+    'repeat with acc in accounts\n' +
+    'repeat with f in folders of acc\n' +
+    'if folderName is "" or name of f is folderName then\n' +
+    'repeat with n in notes of f\n' +
+    'if name of n is targetTitle then\n' +
+    'set noteBody to body of n as text\n' +
+    'if noteBody contains headingText then\n' +
+    'set body of n to noteBody & "<br/>" & newSection\n' +
+    'return "updated"\n' +
+    'else if createMissing then\n' +
+    'set body of n to noteBody & "<br/>" & newSection\n' +
+    'return "appended"\n' +
+    'else\n' +
+    'return "heading_missing"\n' +
+    'end if\n' +
+    'end if\n' +
+    'end repeat\n' +
+    'end if\n' +
+    'end repeat\n' +
+    'end repeat\n' +
+    'end tell\n' +
+    'return "not_found"\n' +
+    'end run';
+  const status = await runAppleScript(
+    script,
+    [title, folder, heading, sectionHtml, createIfMissing ? "true" : "false"],
+    context.signal,
+  );
+  return { status, title, heading, folder: folder || null };
+}
+
+async function runNotesCreateStructured(input: unknown, context: ToolExecutionContext) {
+  const payload = asObject(input) as NotesCreateStructuredInput;
+  const title = asString(payload.title, "title");
+  const folder = typeof payload.folder === "string" ? payload.folder.trim() : "";
+  if (!Array.isArray(payload.sections) || payload.sections.length === 0) {
+    throw new Error("Missing required array field: sections");
+  }
+
+  const sections = payload.sections
+    .filter((section): section is NotesStructuredSection => {
+      return Boolean(section && typeof section.heading === "string" && section.heading.trim());
+    })
+    .map((section) => ({
+      heading: section.heading.trim(),
+      bullets: Array.isArray(section.bullets) ? section.bullets.filter(Boolean) : [],
+      paragraphs: Array.isArray(section.paragraphs) ? section.paragraphs.filter(Boolean) : [],
+    }));
+
+  const body = buildStructuredBody(sections);
+  const formattedBody = normalizeNotesBody(body);
+  if (context.localTools.dryRun) {
+    return {
+      dryRun: true,
+      action: "notes_create_structured",
+      title,
+      folder,
+      sectionCount: sections.length,
+      bodyPreview: formattedBody.slice(0, 240),
+    };
+  }
+
+  // Reuse base notes tool behavior through AppleScript create mode.
+  const script =
+    'on run argv\n' +
+    'set noteTitle to item 1 of argv\n' +
+    'set noteBody to item 2 of argv\n' +
+    'set folderName to item 3 of argv\n' +
+    'tell application "Notes"\n' +
+    'if not (exists account 1) then error "No Notes account is available."\n' +
+    'set targetAccount to account 1\n' +
+    'set targetFolder to missing value\n' +
+    'if folderName is not "" then\n' +
+    'repeat with f in folders of targetAccount\n' +
+    'if name of f is folderName then set targetFolder to f\n' +
+    'end repeat\n' +
+    'if targetFolder is missing value then\n' +
+    'set targetFolder to make new folder at targetAccount with properties {name:folderName}\n' +
+    'end if\n' +
+    'else\n' +
+    'set targetFolder to folder "Notes" of targetAccount\n' +
+    'end if\n' +
+    'set newNote to make new note at targetFolder with properties {name:noteTitle, body:noteBody}\n' +
+    'return "created"\n' +
+    'end tell\n' +
+    'end run';
+  const status = await runAppleScript(script, [title, formattedBody, folder], context.signal);
+  return { status, title, folder: folder || null, sectionCount: sections.length };
+}
+
+export const notesPlusTools: ToolDefinition[] = [
+  {
+    name: "notes_get_by_title",
+    description: "Find a note by exact title and optional folder.",
+    inputSchema: {
+      type: "object",
+      required: ["title"],
+      properties: {
+        title: { type: "string" },
+        folder: { type: "string" },
+      },
+      additionalProperties: false,
+    },
+    risk: "read",
+    execute: runNotesGetByTitle,
+  },
+  {
+    name: "notes_replace_section",
+    description: "Replace or append a heading-based section in a note.",
+    inputSchema: {
+      type: "object",
+      required: ["title", "heading", "content"],
+      properties: {
+        title: { type: "string" },
+        heading: { type: "string" },
+        content: { type: "string" },
+        folder: { type: "string" },
+        createIfMissing: { type: "boolean" },
+      },
+      additionalProperties: false,
+    },
+    risk: "write",
+    execute: runNotesReplaceSection,
+  },
+  {
+    name: "notes_create_structured",
+    description: "Create a structured note from headings, bullets, and paragraphs.",
+    inputSchema: {
+      type: "object",
+      required: ["title", "sections"],
+      properties: {
+        title: { type: "string" },
+        folder: { type: "string" },
+        sections: {
+          type: "array",
+          items: {
+            type: "object",
+            required: ["heading"],
+            properties: {
+              heading: { type: "string" },
+              bullets: { type: "array", items: { type: "string" } },
+              paragraphs: { type: "array", items: { type: "string" } },
+            },
+            additionalProperties: false,
+          },
+        },
+      },
+      additionalProperties: false,
+    },
+    risk: "write",
+    execute: runNotesCreateStructured,
+  },
+];

package/template/src/lib/tooling/tools/numbers.ts

@@ -0,0 +1,175 @@
+import { ensureMacOS } from "../safety";
+import { runCommandSafe } from "../runtime";
+import type { ToolDefinition, ToolExecutionContext } from "../types";
+
+type NumbersSetInput = {
+  document?: string;
+  sheet?: string;
+  range?: string;
+  values?: string | number | boolean;
+};
+
+type NumbersReadInput = {
+  document?: string;
+  sheet?: string;
+  range?: string;
+};
+
+const APPLESCRIPT_TIMEOUT_MS = 30_000;
+
+function asObject(input: unknown): Record<string, unknown> {
+  if (!input || typeof input !== "object") {
+    throw new Error("Tool input must be an object.");
+  }
+  return input as Record<string, unknown>;
+}
+
+async function runCommand(command: string, args: string[], signal?: AbortSignal): Promise<string> {
+  const { stdout } = await runCommandSafe({
+    command,
+    args,
+    signal,
+    timeoutMs: APPLESCRIPT_TIMEOUT_MS,
+  });
+  return stdout;
+}
+
+async function runNumbersSetCells(input: unknown, context: ToolExecutionContext) {
+  const payload = asObject(input) as NumbersSetInput;
+
+  ensureMacOS("Numbers automation");
+
+  const document = typeof payload.document === "string" ? payload.document.trim() : "";
+  const sheet = typeof payload.sheet === "string" && payload.sheet.trim() ? payload.sheet.trim() : "Sheet 1";
+  const range = typeof payload.range === "string" && payload.range.trim() ? payload.range.trim() : "A1";
+  const value = payload.values ?? "";
+
+  if (context.localTools.dryRun) {
+    return {
+      dryRun: true,
+      action: "numbers_set_cells",
+      document,
+      sheet,
+      range,
+      value,
+    };
+  }
+
+  const script =
+    'on run argv\n' +
+    'set docPath to item 1 of argv\n' +
+    'set sheetName to item 2 of argv\n' +
+    'set targetRange to item 3 of argv\n' +
+    'set targetValue to item 4 of argv\n' +
+    'tell application "Numbers"\n' +
+    'activate\n' +
+    'if docPath is "" then\n' +
+    'if (count of documents) = 0 then\n' +
+    'set targetDoc to make new document\n' +
+    'else\n' +
+    'set targetDoc to front document\n' +
+    'end if\n' +
+    'else\n' +
+    'set targetDoc to open POSIX file docPath\n' +
+    'end if\n' +
+    'tell targetDoc\n' +
+    'tell sheet sheetName\n' +
+    'tell table 1\n' +
+    'set value of range targetRange to targetValue\n' +
+    'end tell\n' +
+    'end tell\n' +
+    'save\n' +
+    'end tell\n' +
+    'end tell\n' +
+    'return "ok"\n' +
+    'end run';
+
+  await runCommand("osascript", ["-e", script, document, sheet, range, String(value)], context.signal);
+
+  return {
+    updated: true,
+    document,
+    sheet,
+    range,
+    value,
+  };
+}
+
+async function runNumbersReadCells(input: unknown, context: ToolExecutionContext) {
+  const payload = asObject(input) as NumbersReadInput;
+
+  ensureMacOS("Numbers automation");
+
+  const document = typeof payload.document === "string" ? payload.document.trim() : "";
+  const sheet = typeof payload.sheet === "string" && payload.sheet.trim() ? payload.sheet.trim() : "Sheet 1";
+  const range = typeof payload.range === "string" && payload.range.trim() ? payload.range.trim() : "A1";
+
+  const script =
+    'on run argv\n' +
+    'set docPath to item 1 of argv\n' +
+    'set sheetName to item 2 of argv\n' +
+    'set targetRange to item 3 of argv\n' +
+    'tell application "Numbers"\n' +
+    'activate\n' +
+    'if docPath is "" then\n' +
+    'if (count of documents) = 0 then error "No Numbers documents are open."\n' +
+    'set targetDoc to front document\n' +
+    'else\n' +
+    'set targetDoc to open POSIX file docPath\n' +
+    'end if\n' +
+    'tell targetDoc\n' +
+    'tell sheet sheetName\n' +
+    'tell table 1\n' +
+    'set readValue to value of range targetRange\n' +
+    'end tell\n' +
+    'end tell\n' +
+    'end tell\n' +
+    'end tell\n' +
+    'return readValue as text\n' +
+    'end run';
+
+  const value = await runCommand("osascript", ["-e", script, document, sheet, range], context.signal);
+
+  return {
+    document,
+    sheet,
+    range,
+    value,
+  };
+}
+
+export const numbersTools: ToolDefinition[] = [
+  {
+    name: "numbers_set_cells",
+    description: "Set a value in Numbers sheet range.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        document: { type: "string" },
+        sheet: { type: "string" },
+        range: { type: "string" },
+        values: {
+          oneOf: [{ type: "string" }, { type: "number" }, { type: "boolean" }],
+        },
+      },
+      additionalProperties: false,
+    },
+    risk: "write",
+    execute: runNumbersSetCells,
+  },
+  {
+    name: "numbers_read_cells",
+    description: "Read a value from Numbers sheet range.",
+    inputSchema: {
+      type: "object",
+      properties: {
+        document: { type: "string" },
+        sheet: { type: "string" },
+        range: { type: "string" },
+      },
+      additionalProperties: false,
+    },
+    risk: "read",
+    execute: runNumbersReadCells,
+  },
+];