pi-ynabro 2.0.0 → 2.1.0

package/README.md

@@ -28,4 +28,4 @@ pi install npm:pi-ynabro
 
 ## Requirements
 
-- `YNAB_TOKEN` environment variable must be set
+Run `ynabro_setup` once to complete interactive onboarding. The tool will prompt for a YNAB Personal Access Token (generate at https://app.ynab.com/settings/developer) and let you select a default plan. Both are stored securely in pi's `AuthStorage` (`~/.pi/agent/auth.json`).

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-ynabro",
-  "version": "2.0.0",
+  "version": "2.1.0",
   "description": "Pi extension that registers YNABro tools for YNAB integration",
   "type": "module",
   "main": "./src/index.ts",

package/skills/ynabro/prompts/ynabro.md

@@ -10,7 +10,7 @@ You are **YNABro**, a friendly and reliable YNAB assistant designed to help user
 
 Before performing any YNAB operations, check whether YNAB access has been set up:
 
-1. A YNAB Personal Access Token must be available (via `YNAB_TOKEN` environment variable or stored in `.ynabro/config.json`).
+1. A YNAB Personal Access Token must be available. In OpenClaw, the preferred storage location is `plugins.entries.openclaw-ynabro.config.token` in `openclaw.json` (surfaced as a sensitive field in the settings UI). The `YNAB_TOKEN` environment variable is also accepted as a fallback.
 2. A default plan must be selected and saved.
 
 If either is missing, call the `setupYnab()` tool first. This tool will guide the user through creating a token (if needed) and selecting a default plan. Do not attempt to read or modify transactions until setup is complete.

package/src/index.ts

@@ -1,5 +1,7 @@
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import { AuthStorage } from "@earendil-works/pi-coding-agent";
 import { Type } from "typebox";
+import type { YnabroConfigAdapter } from "ynabro";
 import {
   approveTransaction,
   getPendingTransactions,
@@ -11,18 +13,42 @@ import {
   YnabroClient,
 } from "ynabro";
 
-function getClient(): YnabroClient {
-  const token = process.env.YNAB_TOKEN;
-  if (!token) throw new Error("YNAB_TOKEN environment variable is not set");
+const authStorage = AuthStorage.create();
+
+const piConfigAdapter: YnabroConfigAdapter = {
+  async getDefaultPlanId(): Promise<string | undefined> {
+    const credential = authStorage.get("ynab-plan");
+    if (credential?.type === "api_key") {
+      return credential.key;
+    }
+    return undefined;
+  },
+  async setDefaultPlanId(planId: string): Promise<void> {
+    authStorage.set("ynab-plan", { type: "api_key", key: planId });
+  },
+};
+
+async function getClient(): Promise<YnabroClient> {
+  const token = await authStorage.getApiKey("ynab");
+  if (!token) {
+    throw new Error(
+      "YNAB token not configured. Run ynabro_setup to complete onboarding.",
+    );
+  }
   return new YnabroClient(token);
 }
 
-const planIdSchema = Type.Object({
-  planId: Type.String({ description: "The ID of the YNAB plan" }),
-});
+async function getDefaultPlanId(): Promise<string> {
+  const planId = await piConfigAdapter.getDefaultPlanId();
+  if (!planId) {
+    throw new Error(
+      "No default plan configured. Run ynabro_setup to complete onboarding.",
+    );
+  }
+  return planId;
+}
 
 const approveSchema = Type.Object({
-  planId: Type.String({ description: "The ID of the YNAB plan" }),
   transactionId: Type.String({
     description: "The ID of the transaction to approve",
   }),
@@ -52,10 +78,15 @@ export default function ynabroExtension(api: ExtensionAPI): void {
   api.registerTool({
     name: "ynabro_get_pending_transactions",
     label: "Get Pending Transactions",
-    description: "Get all pending (uncategorized) transactions for a plan",
-    parameters: planIdSchema,
-    async execute(_toolCallId, params) {
-      const result = await getPendingTransactions(getClient(), params.planId);
+    description:
+      "Get all pending (uncategorized) transactions for the default plan",
+    parameters: Type.Object({}),
+    async execute(_toolCallId, _params) {
+      const [client, planId] = await Promise.all([
+        getClient(),
+        getDefaultPlanId(),
+      ]);
+      const result = await getPendingTransactions(client, planId);
       return {
         content: [{ type: "text", text: JSON.stringify(result, null, 2) }],
         details: undefined,
@@ -66,10 +97,14 @@ export default function ynabroExtension(api: ExtensionAPI): void {
   api.registerTool({
     name: "ynabro_get_recent_transactions",
     label: "Get Recent Transactions",
-    description: "Get recent transactions for a plan",
-    parameters: planIdSchema,
-    async execute(_toolCallId, params) {
-      const result = await getRecentTransactions(getClient(), params.planId);
+    description: "Get recent transactions for the default plan",
+    parameters: Type.Object({}),
+    async execute(_toolCallId, _params) {
+      const [client, planId] = await Promise.all([
+        getClient(),
+        getDefaultPlanId(),
+      ]);
+      const result = await getRecentTransactions(client, planId);
       return {
         content: [{ type: "text", text: JSON.stringify(result, null, 2) }],
         details: undefined,
@@ -80,14 +115,14 @@ export default function ynabroExtension(api: ExtensionAPI): void {
   api.registerTool({
     name: "ynabro_approve_transaction",
     label: "Approve Transaction",
-    description: "Approve a specific transaction",
+    description: "Approve a specific transaction in the default plan",
     parameters: approveSchema,
     async execute(_toolCallId, params) {
-      await approveTransaction(
+      const [client, planId] = await Promise.all([
         getClient(),
-        params.planId,
-        params.transactionId,
-      );
+        getDefaultPlanId(),
+      ]);
+      await approveTransaction(client, planId, params.transactionId);
       return {
         content: [{ type: "text", text: JSON.stringify({ success: true }) }],
         details: undefined,
@@ -98,10 +133,14 @@ export default function ynabroExtension(api: ExtensionAPI): void {
   api.registerTool({
     name: "ynabro_get_plan_info",
     label: "Get Plan Info",
-    description: "Get basic information about a plan",
-    parameters: planIdSchema,
-    async execute(_toolCallId, params) {
-      const result = await getPlanInfo(getClient(), params.planId);
+    description: "Get basic information about the default plan",
+    parameters: Type.Object({}),
+    async execute(_toolCallId, _params) {
+      const [client, planId] = await Promise.all([
+        getClient(),
+        getDefaultPlanId(),
+      ]);
+      const result = await getPlanInfo(client, planId);
       return {
         content: [{ type: "text", text: JSON.stringify(result, null, 2) }],
         details: undefined,
@@ -113,12 +152,83 @@ export default function ynabroExtension(api: ExtensionAPI): void {
     name: "ynabro_setup",
     label: "Setup YNAB",
     description:
-      "Set up YNAB integration — checks for token and selects a default plan",
+      "Set up or reconfigure YNAB integration — stores API token and selects a default plan",
     parameters: Type.Object({}),
-    async execute() {
-      const result = await setupYnab();
+    async execute(_toolCallId, _params, _signal, _onUpdate, ctx) {
+      // Step 1: Token check
+      let token = await authStorage.getApiKey("ynab");
+      if (!token) {
+        const input = await ctx.ui.input(
+          "YNAB Personal Access Token",
+          "Paste your token from https://app.ynab.com/settings/developer",
+        );
+        if (!input) {
+          return {
+            content: [
+              {
+                type: "text",
+                text: JSON.stringify({
+                  error: "Setup cancelled — no token provided.",
+                }),
+              },
+            ],
+            details: undefined,
+          };
+        }
+        authStorage.set("ynab", { type: "api_key", key: input });
+        token = input;
+      }
+
+      // Step 2: Fetch plans
+      const client = new YnabroClient(token);
+      const plans = await client.getPlans();
+      if (plans.length === 0) {
+        return {
+          content: [
+            {
+              type: "text",
+              text: JSON.stringify({
+                error: "No plans found in your YNAB account.",
+              }),
+            },
+          ],
+          details: undefined,
+        };
+      }
+
+      // Step 3: Plan selector
+      const planOptions = plans.map((p) => `${p.name} (${p.id})`);
+      const selected = await ctx.ui.select(
+        "Select your default YNAB plan",
+        planOptions,
+      );
+      if (!selected) {
+        return {
+          content: [
+            {
+              type: "text",
+              text: JSON.stringify({
+                error: "Setup cancelled — no plan selected.",
+              }),
+            },
+          ],
+          details: undefined,
+        };
+      }
+
+      // Step 4: Store and return
+      const selectedPlan = plans[planOptions.indexOf(selected)];
+      await setupYnab(client, plans, selectedPlan.id, piConfigAdapter);
       return {
-        content: [{ type: "text", text: JSON.stringify(result, null, 2) }],
+        content: [
+          {
+            type: "text",
+            text: JSON.stringify({
+              message: `Setup complete! Default plan set to: ${selectedPlan.name}`,
+              defaultPlanId: selectedPlan.id,
+            }),
+          },
+        ],
         details: undefined,
       };
     },