@townco/ui 0.1.15 → 0.1.17

package/dist/core/hooks/index.d.ts

@@ -5,3 +5,4 @@ export * from "./use-chat-input.js";
 export * from "./use-chat-messages.js";
 export * from "./use-chat-session.js";
 export * from "./use-media-query.js";
+export * from "./use-tool-calls.js";

package/dist/core/hooks/index.js

@@ -5,3 +5,4 @@ export * from "./use-chat-input.js";
 export * from "./use-chat-messages.js";
 export * from "./use-chat-session.js";
 export * from "./use-media-query.js";
+export * from "./use-tool-calls.js";

package/dist/core/hooks/use-chat-messages.d.ts

@@ -3,15 +3,54 @@ import type { AcpClient } from "../../sdk/client/index.js";
  * Hook for managing chat messages
  */
 export declare function useChatMessages(client: AcpClient | null): {
-	messages: {
-		id: string;
-		role: "user" | "assistant" | "system";
-		content: string;
-		timestamp: string;
-		isStreaming: boolean;
-		streamingStartTime?: number | undefined;
-		metadata?: Record<string, unknown> | undefined;
-	}[];
-	isStreaming: boolean;
-	sendMessage: (content: string) => Promise<void>;
+    messages: {
+        id: string;
+        role: "user" | "assistant" | "system";
+        content: string;
+        timestamp: string;
+        isStreaming: boolean;
+        streamingStartTime?: number | undefined;
+        metadata?: Record<string, unknown> | undefined;
+        toolCalls?: {
+            id: string;
+            title: string;
+            kind: "read" | "edit" | "delete" | "move" | "search" | "execute" | "think" | "fetch" | "switch_mode" | "other";
+            status: "pending" | "in_progress" | "completed" | "failed";
+            locations?: {
+                path: string;
+                line?: number | null | undefined;
+            }[] | undefined;
+            rawInput?: Record<string, unknown> | undefined;
+            rawOutput?: Record<string, unknown> | undefined;
+            content?: ({
+                type: "content";
+                content: {
+                    type: "text";
+                    text: string;
+                };
+            } | {
+                type: "text";
+                text: string;
+            } | {
+                type: "diff";
+                path: string;
+                oldText: string;
+                newText: string;
+                line?: number | null | undefined;
+            } | {
+                type: "terminal";
+                terminalId: string;
+            })[] | undefined;
+            error?: string | undefined;
+            startedAt?: number | undefined;
+            completedAt?: number | undefined;
+            tokenUsage?: {
+                inputTokens?: number | undefined;
+                outputTokens?: number | undefined;
+                totalTokens?: number | undefined;
+            } | undefined;
+        }[] | undefined;
+    }[];
+    isStreaming: boolean;
+    sendMessage: (content: string) => Promise<void>;
 };

package/dist/core/hooks/use-chat-session.d.ts

@@ -3,9 +3,9 @@ import type { AcpClient } from "../../sdk/client/index.js";
  * Hook for managing chat session lifecycle
  */
 export declare function useChatSession(client: AcpClient | null): {
-	connectionStatus: "error" | "connecting" | "connected" | "disconnected";
-	sessionId: string | null;
-	connect: () => Promise<void>;
-	startSession: () => Promise<void>;
-	disconnect: () => Promise<void>;
+    connectionStatus: "connecting" | "connected" | "error" | "disconnected";
+    sessionId: string | null;
+    connect: () => Promise<void>;
+    startSession: () => Promise<void>;
+    disconnect: () => Promise<void>;
 };

package/dist/core/hooks/use-tool-calls.d.ts

@@ -0,0 +1,52 @@
+import type { AcpClient } from "../../sdk/client/acp-client.js";
+import type { ToolCall } from "../schemas/tool-call.js";
+/**
+ * Hook to track and manage tool calls from ACP sessions
+ *
+ * Subscribes to session updates and tracks tool call lifecycle:
+ * - Receives initial tool_call notifications
+ * - Merges incremental tool_call_update notifications
+ * - Associates tool calls with sessions
+ */
+export declare function useToolCalls(client: AcpClient | null): {
+    toolCalls: Record<string, {
+        id: string;
+        title: string;
+        kind: "read" | "edit" | "delete" | "move" | "search" | "execute" | "think" | "fetch" | "switch_mode" | "other";
+        status: "pending" | "in_progress" | "completed" | "failed";
+        locations?: {
+            path: string;
+            line?: number | null | undefined;
+        }[] | undefined;
+        rawInput?: Record<string, unknown> | undefined;
+        rawOutput?: Record<string, unknown> | undefined;
+        content?: ({
+            type: "content";
+            content: {
+                type: "text";
+                text: string;
+            };
+        } | {
+            type: "text";
+            text: string;
+        } | {
+            type: "diff";
+            path: string;
+            oldText: string;
+            newText: string;
+            line?: number | null | undefined;
+        } | {
+            type: "terminal";
+            terminalId: string;
+        })[] | undefined;
+        error?: string | undefined;
+        startedAt?: number | undefined;
+        completedAt?: number | undefined;
+        tokenUsage?: {
+            inputTokens?: number | undefined;
+            outputTokens?: number | undefined;
+            totalTokens?: number | undefined;
+        } | undefined;
+    }[]>;
+    getToolCallsForSession: (sessionId: string) => ToolCall[];
+};

package/dist/core/hooks/use-tool-calls.js

@@ -0,0 +1,61 @@
+import { useEffect } from "react";
+import { useChatStore } from "../store/chat-store.js";
+/**
+ * Hook to track and manage tool calls from ACP sessions
+ *
+ * Subscribes to session updates and tracks tool call lifecycle:
+ * - Receives initial tool_call notifications
+ * - Merges incremental tool_call_update notifications
+ * - Associates tool calls with sessions
+ */
+export function useToolCalls(client) {
+    const toolCalls = useChatStore((state) => state.toolCalls);
+    const addToolCall = useChatStore((state) => state.addToolCall);
+    const updateToolCall = useChatStore((state) => state.updateToolCall);
+    const addToolCallToCurrentMessage = useChatStore((state) => state.addToolCallToCurrentMessage);
+    const updateToolCallInCurrentMessage = useChatStore((state) => state.updateToolCallInCurrentMessage);
+    useEffect(() => {
+        if (!client)
+            return;
+        // Subscribe to session updates for tool calls
+        const unsubscribe = client.onSessionUpdate((update) => {
+            if (update.type === "tool_call") {
+                // Initial tool call notification
+                // Add to session-level tool calls (for sidebar)
+                addToolCall(update.sessionId, update.toolCall);
+                // Also add to current assistant message (for inline display)
+                addToolCallToCurrentMessage(update.toolCall);
+            }
+            else if (update.type === "tool_call_update") {
+                // Tool call update notification
+                // Update session-level tool calls (for sidebar)
+                updateToolCall(update.sessionId, update.toolCallUpdate);
+                // Also update in current assistant message (for inline display)
+                updateToolCallInCurrentMessage(update.toolCallUpdate);
+            }
+        });
+        return () => {
+            unsubscribe();
+        };
+    }, [
+        client,
+        addToolCall,
+        updateToolCall,
+        addToolCallToCurrentMessage,
+        updateToolCallInCurrentMessage,
+    ]);
+    return {
+        toolCalls,
+        getToolCallsForSession: (sessionId) => {
+            // First try the exact session ID
+            const exactMatch = toolCalls[sessionId];
+            if (exactMatch && exactMatch.length > 0) {
+                return exactMatch;
+            }
+            // If no exact match, return all tool calls from all sessions
+            // This handles the case where the UI session ID doesn't match the ACP session ID
+            const allToolCalls = Object.values(toolCalls).flat();
+            return allToolCalls;
+        },
+    };
+}

package/dist/core/schemas/chat.d.ts

@@ -5,101 +5,184 @@ import { z } from "zod";
 /**
  * Display message schema (UI representation of messages)
  */
-export declare const DisplayMessage: z.ZodObject<
-	{
-		id: z.ZodString;
-		role: z.ZodEnum<{
-			user: "user";
-			assistant: "assistant";
-			system: "system";
-		}>;
-		content: z.ZodString;
-		timestamp: z.ZodISODateTime;
-		isStreaming: z.ZodDefault<z.ZodBoolean>;
-		streamingStartTime: z.ZodOptional<z.ZodNumber>;
-		metadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
-	},
-	z.core.$strip
->;
+export declare const DisplayMessage: z.ZodObject<{
+    id: z.ZodString;
+    role: z.ZodEnum<{
+        user: "user";
+        assistant: "assistant";
+        system: "system";
+    }>;
+    content: z.ZodString;
+    timestamp: z.ZodISODateTime;
+    isStreaming: z.ZodDefault<z.ZodBoolean>;
+    streamingStartTime: z.ZodOptional<z.ZodNumber>;
+    metadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+    toolCalls: z.ZodOptional<z.ZodArray<z.ZodObject<{
+        id: z.ZodString;
+        title: z.ZodString;
+        kind: z.ZodEnum<{
+            read: "read";
+            edit: "edit";
+            delete: "delete";
+            move: "move";
+            search: "search";
+            execute: "execute";
+            think: "think";
+            fetch: "fetch";
+            switch_mode: "switch_mode";
+            other: "other";
+        }>;
+        status: z.ZodEnum<{
+            pending: "pending";
+            in_progress: "in_progress";
+            completed: "completed";
+            failed: "failed";
+        }>;
+        locations: z.ZodOptional<z.ZodArray<z.ZodObject<{
+            path: z.ZodString;
+            line: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
+        }, z.core.$strip>>>;
+        rawInput: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+        rawOutput: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+        content: z.ZodOptional<z.ZodArray<z.ZodDiscriminatedUnion<[z.ZodObject<{
+            type: z.ZodLiteral<"content">;
+            content: z.ZodObject<{
+                type: z.ZodLiteral<"text">;
+                text: z.ZodString;
+            }, z.core.$strip>;
+        }, z.core.$strip>, z.ZodObject<{
+            type: z.ZodLiteral<"text">;
+            text: z.ZodString;
+        }, z.core.$strip>, z.ZodObject<{
+            type: z.ZodLiteral<"diff">;
+            path: z.ZodString;
+            oldText: z.ZodString;
+            newText: z.ZodString;
+            line: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
+        }, z.core.$strip>, z.ZodObject<{
+            type: z.ZodLiteral<"terminal">;
+            terminalId: z.ZodString;
+        }, z.core.$strip>], "type">>>;
+        error: z.ZodOptional<z.ZodString>;
+        startedAt: z.ZodOptional<z.ZodNumber>;
+        completedAt: z.ZodOptional<z.ZodNumber>;
+        tokenUsage: z.ZodOptional<z.ZodObject<{
+            inputTokens: z.ZodOptional<z.ZodNumber>;
+            outputTokens: z.ZodOptional<z.ZodNumber>;
+            totalTokens: z.ZodOptional<z.ZodNumber>;
+        }, z.core.$strip>>;
+    }, z.core.$strip>>>;
+}, z.core.$strip>;
 export type DisplayMessage = z.infer<typeof DisplayMessage>;
 /**
  * Input state schema
  */
-export declare const InputState: z.ZodObject<
-	{
-		value: z.ZodString;
-		isSubmitting: z.ZodBoolean;
-		attachedFiles: z.ZodArray<
-			z.ZodObject<
-				{
-					name: z.ZodString;
-					path: z.ZodString;
-					size: z.ZodNumber;
-					mimeType: z.ZodString;
-				},
-				z.core.$strip
-			>
-		>;
-	},
-	z.core.$strip
->;
+export declare const InputState: z.ZodObject<{
+    value: z.ZodString;
+    isSubmitting: z.ZodBoolean;
+    attachedFiles: z.ZodArray<z.ZodObject<{
+        name: z.ZodString;
+        path: z.ZodString;
+        size: z.ZodNumber;
+        mimeType: z.ZodString;
+    }, z.core.$strip>>;
+}, z.core.$strip>;
 export type InputState = z.infer<typeof InputState>;
 /**
  * Chat session UI state
  */
-export declare const ChatSessionState: z.ZodObject<
-	{
-		sessionId: z.ZodNullable<z.ZodString>;
-		isConnected: z.ZodBoolean;
-		isStreaming: z.ZodBoolean;
-		messages: z.ZodArray<
-			z.ZodObject<
-				{
-					id: z.ZodString;
-					role: z.ZodEnum<{
-						user: "user";
-						assistant: "assistant";
-						system: "system";
-					}>;
-					content: z.ZodString;
-					timestamp: z.ZodISODateTime;
-					isStreaming: z.ZodDefault<z.ZodBoolean>;
-					streamingStartTime: z.ZodOptional<z.ZodNumber>;
-					metadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
-				},
-				z.core.$strip
-			>
-		>;
-		input: z.ZodObject<
-			{
-				value: z.ZodString;
-				isSubmitting: z.ZodBoolean;
-				attachedFiles: z.ZodArray<
-					z.ZodObject<
-						{
-							name: z.ZodString;
-							path: z.ZodString;
-							size: z.ZodNumber;
-							mimeType: z.ZodString;
-						},
-						z.core.$strip
-					>
-				>;
-			},
-			z.core.$strip
-		>;
-		error: z.ZodNullable<z.ZodString>;
-	},
-	z.core.$strip
->;
+export declare const ChatSessionState: z.ZodObject<{
+    sessionId: z.ZodNullable<z.ZodString>;
+    isConnected: z.ZodBoolean;
+    isStreaming: z.ZodBoolean;
+    messages: z.ZodArray<z.ZodObject<{
+        id: z.ZodString;
+        role: z.ZodEnum<{
+            user: "user";
+            assistant: "assistant";
+            system: "system";
+        }>;
+        content: z.ZodString;
+        timestamp: z.ZodISODateTime;
+        isStreaming: z.ZodDefault<z.ZodBoolean>;
+        streamingStartTime: z.ZodOptional<z.ZodNumber>;
+        metadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+        toolCalls: z.ZodOptional<z.ZodArray<z.ZodObject<{
+            id: z.ZodString;
+            title: z.ZodString;
+            kind: z.ZodEnum<{
+                read: "read";
+                edit: "edit";
+                delete: "delete";
+                move: "move";
+                search: "search";
+                execute: "execute";
+                think: "think";
+                fetch: "fetch";
+                switch_mode: "switch_mode";
+                other: "other";
+            }>;
+            status: z.ZodEnum<{
+                pending: "pending";
+                in_progress: "in_progress";
+                completed: "completed";
+                failed: "failed";
+            }>;
+            locations: z.ZodOptional<z.ZodArray<z.ZodObject<{
+                path: z.ZodString;
+                line: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
+            }, z.core.$strip>>>;
+            rawInput: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+            rawOutput: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+            content: z.ZodOptional<z.ZodArray<z.ZodDiscriminatedUnion<[z.ZodObject<{
+                type: z.ZodLiteral<"content">;
+                content: z.ZodObject<{
+                    type: z.ZodLiteral<"text">;
+                    text: z.ZodString;
+                }, z.core.$strip>;
+            }, z.core.$strip>, z.ZodObject<{
+                type: z.ZodLiteral<"text">;
+                text: z.ZodString;
+            }, z.core.$strip>, z.ZodObject<{
+                type: z.ZodLiteral<"diff">;
+                path: z.ZodString;
+                oldText: z.ZodString;
+                newText: z.ZodString;
+                line: z.ZodOptional<z.ZodNullable<z.ZodNumber>>;
+            }, z.core.$strip>, z.ZodObject<{
+                type: z.ZodLiteral<"terminal">;
+                terminalId: z.ZodString;
+            }, z.core.$strip>], "type">>>;
+            error: z.ZodOptional<z.ZodString>;
+            startedAt: z.ZodOptional<z.ZodNumber>;
+            completedAt: z.ZodOptional<z.ZodNumber>;
+            tokenUsage: z.ZodOptional<z.ZodObject<{
+                inputTokens: z.ZodOptional<z.ZodNumber>;
+                outputTokens: z.ZodOptional<z.ZodNumber>;
+                totalTokens: z.ZodOptional<z.ZodNumber>;
+            }, z.core.$strip>>;
+        }, z.core.$strip>>>;
+    }, z.core.$strip>>;
+    input: z.ZodObject<{
+        value: z.ZodString;
+        isSubmitting: z.ZodBoolean;
+        attachedFiles: z.ZodArray<z.ZodObject<{
+            name: z.ZodString;
+            path: z.ZodString;
+            size: z.ZodNumber;
+            mimeType: z.ZodString;
+        }, z.core.$strip>>;
+    }, z.core.$strip>;
+    error: z.ZodNullable<z.ZodString>;
+}, z.core.$strip>;
 export type ChatSessionState = z.infer<typeof ChatSessionState>;
 /**
  * Connection status
  */
 export declare const ConnectionStatus: z.ZodEnum<{
-	error: "error";
-	disconnected: "disconnected";
-	connecting: "connecting";
-	connected: "connected";
+    error: "error";
+    connecting: "connecting";
+    connected: "connected";
+    disconnected: "disconnected";
 }>;
 export type ConnectionStatus = z.infer<typeof ConnectionStatus>;

package/dist/core/schemas/chat.js

@@ -1,4 +1,5 @@
 import { z } from "zod";
+import { ToolCallSchema } from "./tool-call.js";
 /**
  * Chat UI state schemas
  */
@@ -6,46 +7,45 @@ import { z } from "zod";
  * Display message schema (UI representation of messages)
  */
 export const DisplayMessage = z.object({
-	id: z.string(),
-	role: z.enum(["user", "assistant", "system"]),
-	content: z.string(),
-	timestamp: z.iso.datetime(),
-	isStreaming: z.boolean().default(false),
-	streamingStartTime: z.number().optional(), // Unix timestamp when streaming started
-	metadata: z.record(z.string(), z.unknown()).optional(),
+    id: z.string(),
+    role: z.enum(["user", "assistant", "system"]),
+    content: z.string(),
+    timestamp: z.iso.datetime(),
+    isStreaming: z.boolean().default(false),
+    streamingStartTime: z.number().optional(), // Unix timestamp when streaming started
+    metadata: z.record(z.string(), z.unknown()).optional(),
+    toolCalls: z.array(ToolCallSchema).optional(),
 });
 /**
  * Input state schema
  */
 export const InputState = z.object({
-	value: z.string(),
-	isSubmitting: z.boolean(),
-	attachedFiles: z.array(
-		z.object({
-			name: z.string(),
-			path: z.string(),
-			size: z.number(),
-			mimeType: z.string(),
-		}),
-	),
+    value: z.string(),
+    isSubmitting: z.boolean(),
+    attachedFiles: z.array(z.object({
+        name: z.string(),
+        path: z.string(),
+        size: z.number(),
+        mimeType: z.string(),
+    })),
 });
 /**
  * Chat session UI state
  */
 export const ChatSessionState = z.object({
-	sessionId: z.string().nullable(),
-	isConnected: z.boolean(),
-	isStreaming: z.boolean(),
-	messages: z.array(DisplayMessage),
-	input: InputState,
-	error: z.string().nullable(),
+    sessionId: z.string().nullable(),
+    isConnected: z.boolean(),
+    isStreaming: z.boolean(),
+    messages: z.array(DisplayMessage),
+    input: InputState,
+    error: z.string().nullable(),
 });
 /**
  * Connection status
  */
 export const ConnectionStatus = z.enum([
-	"disconnected",
-	"connecting",
-	"connected",
-	"error",
+    "disconnected",
+    "connecting",
+    "connected",
+    "error",
 ]);

package/dist/core/schemas/index.d.ts

@@ -2,3 +2,4 @@
  * Export all core schemas
  */
 export * from "./chat.js";
+export * from "./tool-call.js";

package/dist/core/schemas/index.js

@@ -2,3 +2,4 @@
  * Export all core schemas
  */
 export * from "./chat.js";
+export * from "./tool-call.js";