@case-framework/survey-assistant 0.1.0 → 0.2.0

package/dist/ui.mjs

@@ -1,6 +1,7 @@
 "use client";
-import { Dt as parseSurveyAssistantError } from "./protocol-DV0ka9WT.mjs";
-import { f as unwrapSurveyAssistantToolOutput, i as createSurveyAssistantThreadClient, l as parseSurveyChangeSet, p as isSurveyAssistantTurnActive, r as useSurveyAssistantModelCatalog, t as useSurveyAssistantSession } from "./react-VuYe-cPh.mjs";
+import { kt as parseSurveyAssistantError } from "./protocol-7EEgcYja.mjs";
+import { f as unwrapSurveyAssistantToolOutput, i as createSurveyAssistantThreadClient, l as parseSurveyChangeSet, p as isSurveyAssistantTurnActive, r as useSurveyAssistantModelCatalog, t as useSurveyAssistantSession } from "./react-B_5smfiK.mjs";
+import { o as connectSurveyAssistantThreadLauncher } from "./controller-proxy-ClsY4lFI.mjs";
 import { getToolName, isFileUIPart, isReasoningUIPart, isToolUIPart } from "ai";
 import { createContext, createElement, memo, useCallback, useContext, useEffect, useMemo, useRef, useState } from "react";
 import { Collapsible, Select, Slot } from "radix-ui";
@@ -14,7 +15,6 @@ import { createLocalizationTranslator, mergeLocalizationCatalogs } from "@case-f
 import { cva } from "class-variance-authority";
 import { StickToBottom, useStickToBottomContext } from "use-stick-to-bottom";
 import { Streamdown } from "streamdown";
-import { createHighlighter } from "shiki";
 //#region src/ui/internal/collapsible.tsx
 function Collapsible$1({ ...props }) {
 	return /* @__PURE__ */ jsx(Collapsible.Root, {
@@ -1041,10 +1041,10 @@ const getTokensCacheKey = (code, language) => {
 const getHighlighter = (language) => {
 	const cached = highlighterCache.get(language);
 	if (cached) return cached;
-	const highlighterPromise = createHighlighter({
+	const highlighterPromise = import("shiki").then(({ createHighlighter }) => createHighlighter({
 		langs: [language],
 		themes: ["github-light", "github-dark"]
-	});
+	}));
 	highlighterCache.set(language, highlighterPromise);
 	return highlighterPromise;
 };
@@ -1227,8 +1227,13 @@ const getAssistantQuickPrompts = (t) => [
 		prompt: t("quickActions.refine.prompt")
 	}
 ];
+const isSurveyAssistantAttachmentReferencePart = (part) => {
+	if (part.type !== "data-survey-assistant-attachment" || typeof part.data !== "object" || part.data === null) return false;
+	const data = part.data;
+	return typeof data.filename === "string" && typeof data.mediaType === "string" && typeof data.bytes === "number";
+};
 const hasRenderableParts = (message) => {
-	return message.parts.some((part) => part.type === "text" && part.text.length > 0 || isFileUIPart(part) || isReasoningUIPart(part) && part.text.trim().length > 0 || isToolUIPart(part));
+	return message.parts.some((part) => part.type === "text" && part.text.length > 0 || isFileUIPart(part) || isSurveyAssistantAttachmentReferencePart(part) || isReasoningUIPart(part) && part.text.trim().length > 0 || isToolUIPart(part));
 };
 const getFilePartLabel = (part) => part.filename ?? part.mediaType;
 const getToolOutput = (part) => part.state === "output-available" ? part.output : void 0;
@@ -1489,6 +1494,15 @@ const buildMessageBlocks = (parts) => {
 				key: `file-${index}`,
 				part
 			});
+			return;
+		}
+		if (isSurveyAssistantAttachmentReferencePart(part)) {
+			flushChain();
+			blocks.push({
+				type: "attachment-reference",
+				key: `attachment-reference-${index}`,
+				attachment: part.data
+			});
 		}
 	});
 	flushChain();
@@ -1525,6 +1539,21 @@ const SurveyAssistantMessageParts = ({ applyResultsByChangeId, isAborted, isLast
 				})
 			]
 		}, block.key);
+		if (block.type === "attachment-reference") return /* @__PURE__ */ jsxs("span", {
+			className: "inline-flex max-w-full items-center gap-2 rounded-md border border-border bg-muted px-2 py-1 text-xs",
+			title: block.attachment.filename,
+			children: [
+				/* @__PURE__ */ jsx(FileIcon, { className: "size-3.5 shrink-0 text-muted-foreground" }),
+				/* @__PURE__ */ jsx("span", {
+					className: "min-w-0 truncate",
+					children: block.attachment.filename
+				}),
+				/* @__PURE__ */ jsx("span", {
+					className: "shrink-0 text-muted-foreground",
+					children: block.attachment.mediaType
+				})
+			]
+		}, block.key);
 		const isChainActive = isLastMessage && isStreaming && isLastBlock;
 		return /* @__PURE__ */ jsxs(ChainOfThought, {
 			autoClose: true,
@@ -1596,7 +1625,7 @@ const SurveyAssistantHeaderIdentity = ({ indicatorClassName, status, title }) =>
 		})]
 	})]
 });
-const SurveyAssistantChatPanel = ({ api, applyPolicy = "auto", controller, embedded = false, initialMessages, modelId, models, onModelChange, onFirstSubmit, onTurnSettled, onApplyReceipts, onReviewChangeSets, onTurnActiveChange, resourceId, threadControls, threadId, threadTitle }) => {
+const SurveyAssistantChatPanel = ({ api, applyPolicy, controller, embedded = false, initialLaunch, initialMessages, modelId, models, onInitialLaunchHandled, onModelChange, onFirstSubmit, onTurnSettled, onApplyReceipts, onSettleApply, onReviewChangeSets, onTurnActiveChange, resourceId, surveyId, threadControls, threadId, threadTitle }) => {
 	const t = useSurveyAssistantTranslation();
 	const assistantQuickPrompts = getAssistantQuickPrompts(t);
 	const { applyResultsByChangeId, authorizationError, dismissError, error, isStreaming, latestTurnOutcome, messages, state, status, stop, submit, retry } = useSurveyAssistantSession({
@@ -1606,9 +1635,11 @@ const SurveyAssistantChatPanel = ({ api, applyPolicy = "auto", controller, embed
 		initialMessages,
 		modelId,
 		onApplyReceipts,
+		onSettleApply,
 		onReviewChangeSets,
 		onTurnActiveChange,
 		resourceId,
+		surveyId,
 		threadId
 	});
 	const showAssistantError = Boolean(authorizationError) || Boolean(error);
@@ -1653,6 +1684,22 @@ const SurveyAssistantChatPanel = ({ api, applyPolicy = "auto", controller, embed
 		files: [],
 		text: prompt
 	}), [handleSubmit]);
+	useEffect(() => {
+		if (!initialLaunch) return;
+		if (submit({
+			files: initialLaunch.files ?? [],
+			text: initialLaunch.prompt,
+			uiContext: initialLaunch.uiContext
+		})) {
+			onFirstSubmit(initialLaunch.prompt);
+			onInitialLaunchHandled();
+		}
+	}, [
+		initialLaunch,
+		onFirstSubmit,
+		onInitialLaunchHandled,
+		submit
+	]);
 	return /* @__PURE__ */ jsxs("aside", {
 		"aria-label": t("panel.ariaLabel"),
 		className: cn("flex h-full min-h-0 flex-col overflow-visible bg-background", !embedded && "rounded-xl border border-border shadow-sm"),
@@ -1847,10 +1894,10 @@ const createThreadTitle = (text, fallback) => {
 	const words = normalized.split(" ").slice(0, 8).join(" ");
 	return words.length > 64 ? `${words.slice(0, 61).trimEnd()}...` : words;
 };
-const SurveyAssistantPanelContent = ({ embedded = false, modelId, models = [], onConfirmDeleteThread, onModelChange = () => {}, resourceId, selectedThreadStorageKey, threadApi, threadClient: providedThreadClient, ...sessionProps }) => {
+const SurveyAssistantPanelContent = ({ embedded = false, modelId, models = [], onConfirmDeleteThread, onModelChange = () => {}, resourceId, selectedThreadStorageKey, surveyId, threadApi, threadClient: providedThreadClient, threadLauncher, ...sessionProps }) => {
 	const t = useSurveyAssistantTranslation();
 	const threadClient = useMemo(() => providedThreadClient ?? createSurveyAssistantThreadClient(threadApi), [providedThreadClient, threadApi]);
-	const storageKey = selectedThreadStorageKey === false ? null : selectedThreadStorageKey ?? `${SELECTED_THREAD_STORAGE_KEY}:${encodeURIComponent(resourceId)}`;
+	const storageKey = selectedThreadStorageKey === false ? null : selectedThreadStorageKey ?? `${SELECTED_THREAD_STORAGE_KEY}:${encodeURIComponent(resourceId)}${surveyId ? `:${encodeURIComponent(surveyId)}` : ""}`;
 	const [threads, setThreads] = useState([]);
 	const [selectedThread, setSelectedThread] = useState(null);
 	const [initialMessages, setInitialMessages] = useState([]);
@@ -1859,6 +1906,7 @@ const SurveyAssistantPanelContent = ({ embedded = false, modelId, models = [], o
 	const [editingTitle, setEditingTitle] = useState("");
 	const [threadError, setThreadError] = useState(null);
 	const [loadingThread, setLoadingThread] = useState(true);
+	const [initialLaunch, setInitialLaunch] = useState();
 	const historyControlsRef = useRef(null);
 	const selectModel = useCallback((modelId) => {
 		if (!models.some((model) => model.id === modelId)) return;
@@ -1883,18 +1931,21 @@ const SurveyAssistantPanelContent = ({ embedded = false, modelId, models = [], o
 		t,
 		threadClient
 	]);
-	const createThread = useCallback(async () => {
+	const createThread = useCallback(async (launch) => {
 		setLoadingThread(true);
 		setThreadError(null);
 		try {
-			const thread = await threadClient.create();
+			const thread = await threadClient.create(launch?.title);
 			setThreads((current) => [thread, ...current]);
 			setSelectedThread(thread);
 			setInitialMessages([]);
 			setHistoryOpen(false);
 			if (storageKey) window.localStorage.setItem(storageKey, thread.id);
+			return thread;
 		} catch (error) {
 			setThreadError(parseSurveyAssistantError(error).detail ?? t("errors.fallbackDetail"));
+			if (launch) throw error;
+			return null;
 		} finally {
 			setLoadingThread(false);
 		}
@@ -1928,6 +1979,23 @@ const SurveyAssistantPanelContent = ({ embedded = false, modelId, models = [], o
 		t,
 		threadClient
 	]);
+	useEffect(() => {
+		if (!threadLauncher || loadingThread) return;
+		return connectSurveyAssistantThreadLauncher(threadLauncher, async (launch) => {
+			const thread = await createThread(launch);
+			if (!thread) throw new Error("The survey assistant thread could not be created.");
+			return new Promise((resolve) => {
+				setInitialLaunch({
+					input: launch,
+					onSubmitted: () => resolve(thread)
+				});
+			});
+		});
+	}, [
+		createThread,
+		loadingThread,
+		threadLauncher
+	]);
 	const renameThread = useCallback(async (thread, title) => {
 		const normalized = title.trim();
 		if (!normalized || normalized === thread.title) {
@@ -2004,6 +2072,11 @@ const SurveyAssistantPanelContent = ({ embedded = false, modelId, models = [], o
 		setAutomaticTitle,
 		t
 	]);
+	const handleInitialLaunchHandled = useCallback(() => {
+		if (!initialLaunch) return;
+		setInitialLaunch(void 0);
+		initialLaunch.onSubmitted();
+	}, [initialLaunch]);
 	const refreshThreadList = useCallback(async (reportError) => {
 		try {
 			const availableThreads = await threadClient.list();
@@ -2174,14 +2247,17 @@ const SurveyAssistantPanelContent = ({ embedded = false, modelId, models = [], o
 		children: [/* @__PURE__ */ createElement(SurveyAssistantChatPanel, {
 			...sessionProps,
 			embedded,
+			initialLaunch: initialLaunch?.input,
 			initialMessages,
 			key: selectedThread.id,
 			modelId,
 			models,
+			onInitialLaunchHandled: handleInitialLaunchHandled,
 			onModelChange: selectModel,
 			onFirstSubmit: handleFirstSubmit,
 			onTurnSettled: handleTurnSettled,
 			resourceId,
+			surveyId,
 			threadControls,
 			threadId: selectedThread.id,
 			threadTitle: selectedThread.title
@@ -2211,7 +2287,10 @@ const SurveyAssistantPanel = ({ language, localizations, ...props }) => /* @__PU
 	children: /* @__PURE__ */ jsx(SurveyAssistantLocalizationProvider, {
 		language,
 		localizations,
-		children: /* @__PURE__ */ jsx(SurveyAssistantPanelContent, { ...props })
+		children: /* @__PURE__ */ createElement(SurveyAssistantPanelContent, {
+			...props,
+			key: JSON.stringify([props.resourceId, props.surveyId])
+		})
 	})
 });
 //#endregion