initial commit

src/utils/logged.ts

@@ -0,0 +1,32 @@
+import { randomUUID } from "node:crypto";
+import { TextResult } from "../types";
+
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+export function logged<A extends any[]>(
+    name: string,
+    handler: (...args: A) => Promise<TextResult>,
+): (...args: A) => Promise<TextResult> {
+    return async (...args: A) => {
+        const id = randomUUID().slice(0, 8);
+        const t0 = process.hrtime.bigint();
+        console.log(`[tool>] ${name} #${id} args=${JSON.stringify(args)}`);
+        try {
+            const result = await handler(...args);
+            const ms = Number(process.hrtime.bigint() - t0) / 1e6;
+            console.log(
+                `[tool<] ${name} #${id} ${ms.toFixed(1)}ms` +
+                `${result.isError ? " ERROR" : ""} result=${JSON.stringify(result).slice(0, 2000)}`,
+            );
+            return result;
+        } catch (err) {
+            const ms = Number(process.hrtime.bigint() - t0) / 1e6;
+            console.error(`[tool!] ${name} #${id} ${ms.toFixed(1)}ms threw:`, err);
+            throw err; // let the SDK turn it into a JSON-RPC error
+        }
+    };
+}
+
+export function isValidHost(host: string): boolean {
+    // allow IPv4, IPv6, and hostnames; reject anything with shell metacharacters
+    return /^[a-zA-Z0-9.:_-]+$/.test(host) && host.length <= 255;
+}