@aliou/pi-linkup 0.2.0 → 0.3.0

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @aliou/pi-linkup
 
+## 0.3.0
+
+### Minor Changes
+
+- e57f6fe: Enhanced balance command to show remaining requests per operation type (standard search, deep search, fetch with/without JS) using a themed custom message renderer instead of a transient notification.
+
 ## 0.2.0
 
 ### Minor Changes

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aliou/pi-linkup",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "repository": {
     "type": "git",
     "url": "https://github.com/aliou/pi-linkup"

package/src/commands/balance.ts

@@ -9,10 +9,12 @@ export function registerBalanceCommand(pi: ExtensionAPI) {
 
       try {
         const response = await client.getBalance();
-        ctx.ui.notify(
-          `Linkup Balance: ${response.balance.toFixed(2)} credits`,
-          "info",
-        );
+        pi.sendMessage({
+          customType: "linkup-balance",
+          content: `Linkup Balance: ${response.balance.toFixed(2)} credits`,
+          display: true,
+          details: { balance: response.balance },
+        });
       } catch (error) {
         const message =
           error instanceof Error ? error.message : "Unknown error";

package/src/components/balance-renderer.ts

@@ -0,0 +1,45 @@
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { Box, Text } from "@mariozechner/pi-tui";
+import { LINKUP_PRICING } from "../types";
+
+export function registerBalanceRenderer(pi: ExtensionAPI) {
+  pi.registerMessageRenderer("linkup-balance", (message, _options, theme) => {
+    const details = message.details as { balance: number };
+    const balance = details.balance;
+
+    // Build header line
+    const header = [
+      theme.fg("accent", "Linkup"),
+      theme.fg("muted", " · "),
+      theme.fg("muted", `${balance} credits`),
+    ].join("");
+
+    // Calculate remaining requests for each operation type
+    const remaining = {
+      standardSearch: Math.floor(balance / LINKUP_PRICING.standardSearch),
+      deepSearch: Math.floor(balance / LINKUP_PRICING.deepSearch),
+      fetchNoJs: Math.floor(balance / LINKUP_PRICING.fetchNoJs),
+      fetchWithJs: Math.floor(balance / LINKUP_PRICING.fetchWithJs),
+    };
+
+    // Format number with locale string and ~ prefix
+    const fmt = (num: number): string => `~${num.toLocaleString()}`;
+
+    // Table rows with right-aligned numbers (padded to 7 chars)
+    const row = (num: number, label: string) =>
+      `${theme.fg("accent", fmt(num).padStart(7))}  ${theme.fg("dim", label)}`;
+
+    const lines = [
+      header,
+      "",
+      row(remaining.standardSearch, "standard searches"),
+      row(remaining.deepSearch, "deep searches"),
+      row(remaining.fetchNoJs, "fetches (no JS)"),
+      row(remaining.fetchWithJs, "fetches (with JS)"),
+    ].join("\n");
+
+    const box = new Box(1, 1, (t) => theme.bg("customMessageBg", t));
+    box.addChild(new Text(lines, 0, 0));
+    return box;
+  });
+}

package/src/components/index.ts

@@ -0,0 +1,6 @@
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { registerBalanceRenderer } from "./balance-renderer";
+
+export function registerRenderers(pi: ExtensionAPI) {
+  registerBalanceRenderer(pi);
+}

package/src/index.ts

@@ -1,5 +1,6 @@
 import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 import { registerBalanceCommand } from "./commands/balance";
+import { registerRenderers } from "./components";
 import { registerWebAnswerTool } from "./tools/web-answer";
 import { registerWebFetchTool } from "./tools/web-fetch";
 import { registerWebSearchTool } from "./tools/web-search";
@@ -30,4 +31,7 @@ export default function (pi: ExtensionAPI) {
 
   // Register commands
   registerBalanceCommand(pi);
+
+  // Register renderers
+  registerRenderers(pi);
 }

package/src/types.ts

@@ -32,3 +32,14 @@ export interface LinkupErrorResponse {
     message?: string;
   };
 }
+
+/**
+ * Credit cost per request by operation type.
+ * Source: https://docs.linkup.so/pages/documentation/development/pricing
+ */
+export const LINKUP_PRICING = {
+  standardSearch: 0.005,
+  deepSearch: 0.05,
+  fetchNoJs: 0.001,
+  fetchWithJs: 0.005,
+} as const;