kyd-shared-badge 0.3.7 → 0.3.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kyd-shared-badge",
-  "version": "0.3.7",
+  "version": "0.3.8",
   "private": false,
   "main": "./src/index.ts",
   "module": "./src/index.ts",

package/src/chat/ChatWidget.tsx

@@ -53,6 +53,19 @@ export default function ChatWidget({ api = '/api/chat', title = 'KYD Bot', hintT
     if (listRef.current) listRef.current.scrollTop = listRef.current.scrollHeight;
   }, [messages, open]);
 
+  // Prefill input from URL chatPrompt and open the sidebar when present
+  useEffect(() => {
+    try {
+      if (typeof window === 'undefined') return;
+      const url = new URL(window.location.href);
+      const prompt = url.searchParams.get('chatPrompt');
+      if (prompt && prompt.trim()) {
+        setInput(prompt);
+        setOpen(true);
+      }
+    } catch {}
+  }, [setInput]);
+
   // Optional hint (only shows if user hasn't dismissed before and when collapsed)
   useEffect(() => {
     try {

package/src/chat/ChatWindowStreaming.tsx

@@ -13,6 +13,18 @@ export default function ChatWindowStreaming({ api = '/api/chat', badgeId }: { ap
     if (listRef.current) listRef.current.scrollTop = listRef.current.scrollHeight;
   }, [messages]);
 
+  // Prefill input from URL chatPrompt if present
+  useEffect(() => {
+    try {
+      if (typeof window === 'undefined') return;
+      const url = new URL(window.location.href);
+      const prompt = url.searchParams.get('chatPrompt');
+      if (prompt && prompt.trim()) {
+        setInput(prompt);
+      }
+    } catch {}
+  }, [setInput]);
+
   return (
     <div className="flex flex-col border rounded-lg overflow-hidden" style={{ background: 'var(--content-card-background)', borderColor: 'var(--icon-button-secondary)'}}>
       <div ref={listRef} className="p-4 space-y-3 h-72 overflow-auto">

package/src/chat/useChatStreaming.ts

@@ -28,6 +28,16 @@ export function useChatStreaming(cfg: UseChatStreamingConfig) {
     const assistantId = crypto.randomUUID();
     setMessages(m => [...m, userMsg, { id: assistantId, role: 'assistant', content: '' }]);
 
+    const redirectToLogin = (promptText: string) => {
+      // Inform the user then redirect to login carrying callbackUrl and prompt
+      setMessages(m => m.map(msg => msg.id === assistantId ? { ...msg, content: 'You need to log in to use chat. Redirecting to login…' } : msg));
+      const currentPath = typeof window !== 'undefined' ? (window.location.pathname + window.location.search + window.location.hash) : '/';
+      const loginUrl = `/auth/login?callbackUrl=${encodeURIComponent(currentPath)}&chatPrompt=${encodeURIComponent(promptText)}`;
+      setTimeout(() => {
+        if (typeof window !== 'undefined') window.location.href = loginUrl;
+      }, 700);
+    };
+
     try {
       abortRef.current?.abort();
       abortRef.current = new AbortController();
@@ -35,6 +45,7 @@ export function useChatStreaming(cfg: UseChatStreamingConfig) {
       let sid = sessionId;
       if (!sid) {
         const sres = await fetch(`${api}/session`, { method: 'POST' });
+        if (sres.status === 401) { redirectToLogin(content); return; }
         if (!sres.ok) throw new Error('session');
         const sj = await sres.json();
         sid = sj.sessionId as string;
@@ -46,6 +57,7 @@ export function useChatStreaming(cfg: UseChatStreamingConfig) {
         body: JSON.stringify({ content, sessionId: sid, badgeId }),
         signal: abortRef.current.signal,
       });
+      if (res.status === 401) { redirectToLogin(content); return; }
       if (!res.ok || !res.body) throw new Error(`send: ${res.status}`);
 
       const reader = res.body.getReader();

package/src/lib/routes.ts

@@ -5,7 +5,6 @@ import { NextRequest } from 'next/server';
 import { streamText } from 'ai';
 import { bedrock } from '@ai-sdk/amazon-bedrock';
 
-// import { verifyCognito } from './auth-verify';
 import { getHistory, putMessage } from './chat-store';
 import { aggregateUserData, cleanDeveloperProfile, buildAllContextPrompt, getReportGraphData } from './context';
 import { checkAndConsumeToken } from './rate-limit';
@@ -17,7 +16,6 @@ export const runtime = 'nodejs';
 
 export async function chatStreamRoute(req: NextRequest, userId: string, companyId?: string) {
   try {
-    // const user = await verifyCognito(req);
 
     const { sessionId, content, badgeId } = await req.json();
     if (!content || !sessionId) {
@@ -70,7 +68,6 @@ export async function chatStreamRoute(req: NextRequest, userId: string, companyI
 
 export async function createSessionRoute(req: NextRequest, userId: string, companyId?: string) {
   try {
-    // const user = await verifyCognito(req);
     const body = await req.json().catch(() => ({}));
     const sessionId = await createSession({
       userId: userId,