@radnine/storybook-addon-claude 0.2.2 → 0.2.4

package/dist/useClaudeSession.js

@@ -7,11 +7,32 @@ exports.useClaudeSession = useClaudeSession;
 var _react = require("react");
 var _WebSocketClient = require("./WebSocketClient");
 var _constants = require("./constants");
+const SESSION_STORAGE_KEY = `${_constants.ADDON_ID}:sessionId`;
+function loadPersistedSessionId() {
+  try {
+    return localStorage.getItem(SESSION_STORAGE_KEY) || null;
+  } catch {
+    return null;
+  }
+}
+function persistSessionId(id) {
+  try {
+    if (id) {
+      localStorage.setItem(SESSION_STORAGE_KEY, id);
+    } else {
+      localStorage.removeItem(SESSION_STORAGE_KEY);
+    }
+  } catch {
+    // localStorage may be unavailable
+  }
+}
+
 /**
  * React hook that manages a Claude session via the standalone daemon.
  *
  * Handles WebSocket connection lifecycle, message accumulation,
- * session activation, and command sending.
+ * session activation, and command sending. Persists the session ID
+ * in localStorage so chat survives page refreshes.
  */
 function useClaudeSession(options = {}) {
   const {
@@ -22,7 +43,7 @@ function useClaudeSession(options = {}) {
   } = options;
   const [connectionState, setConnectionState] = (0, _react.useState)(_constants.CONNECTION_STATES.DISCONNECTED);
   const [messages, setMessages] = (0, _react.useState)([]);
-  const [sessionId, setSessionId] = (0, _react.useState)(initialSessionId || null);
+  const [sessionId, setSessionId] = (0, _react.useState)(initialSessionId || loadPersistedSessionId());
   const [isProcessing, setIsProcessing] = (0, _react.useState)(false);
   const [authFailed, setAuthFailed] = (0, _react.useState)(false);
   const clientRef = (0, _react.useRef)(null);
@@ -43,6 +64,7 @@ function useClaudeSession(options = {}) {
       setConnectionState(newState);
     }), client.on('session_activated', msg => {
       setSessionId(msg.sessionId);
+      persistSessionId(msg.sessionId);
     }), client.on('output', msg => {
       if (msg.sessionId !== sessionIdRef.current) return;
       setMessages(prev => [...prev, {
@@ -116,6 +138,7 @@ function useClaudeSession(options = {}) {
     // Synchronously update the ref so sendMessage sees it immediately
     sessionIdRef.current = sid;
     setSessionId(sid);
+    persistSessionId(sid);
     setMessages([]);
     if (client && client.isConnected) {
       client.activate(sid, workingDir);
@@ -132,6 +155,7 @@ function useClaudeSession(options = {}) {
     }
     sessionIdRef.current = null;
     setSessionId(null);
+    persistSessionId(null);
     setMessages([]);
     setIsProcessing(false);
   }, []);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@radnine/storybook-addon-claude",
-  "version": "0.2.2",
+  "version": "0.2.4",
   "description": "Storybook addon panel for chatting with Claude via the standalone daemon",
   "license": "SEE LICENSE IN LICENSE",
   "repository": {