prompt_objects 0.1.0

data/lib/prompt_objects/server/public/assets/index-DXU5uRXQ.css

@@ -0,0 +1 @@
+*,:before,:after{--tw-border-spacing-x: 0;--tw-border-spacing-y: 0;--tw-translate-x: 0;--tw-translate-y: 0;--tw-rotate: 0;--tw-skew-x: 0;--tw-skew-y: 0;--tw-scale-x: 1;--tw-scale-y: 1;--tw-pan-x: ;--tw-pan-y: ;--tw-pinch-zoom: ;--tw-scroll-snap-strictness: proximity;--tw-gradient-from-position: ;--tw-gradient-via-position: ;--tw-gradient-to-position: ;--tw-ordinal: ;--tw-slashed-zero: ;--tw-numeric-figure: ;--tw-numeric-spacing: ;--tw-numeric-fraction: ;--tw-ring-inset: ;--tw-ring-offset-width: 0px;--tw-ring-offset-color: #fff;--tw-ring-color: rgb(59 130 246 / .5);--tw-ring-offset-shadow: 0 0 #0000;--tw-ring-shadow: 0 0 #0000;--tw-shadow: 0 0 #0000;--tw-shadow-colored: 0 0 #0000;--tw-blur: ;--tw-brightness: ;--tw-contrast: ;--tw-grayscale: ;--tw-hue-rotate: ;--tw-invert: ;--tw-saturate: ;--tw-sepia: ;--tw-drop-shadow: ;--tw-backdrop-blur: ;--tw-backdrop-brightness: ;--tw-backdrop-contrast: ;--tw-backdrop-grayscale: ;--tw-backdrop-hue-rotate: ;--tw-backdrop-invert: ;--tw-backdrop-opacity: ;--tw-backdrop-saturate: ;--tw-backdrop-sepia: ;--tw-contain-size: ;--tw-contain-layout: ;--tw-contain-paint: ;--tw-contain-style: }::backdrop{--tw-border-spacing-x: 0;--tw-border-spacing-y: 0;--tw-translate-x: 0;--tw-translate-y: 0;--tw-rotate: 0;--tw-skew-x: 0;--tw-skew-y: 0;--tw-scale-x: 1;--tw-scale-y: 1;--tw-pan-x: ;--tw-pan-y: ;--tw-pinch-zoom: ;--tw-scroll-snap-strictness: proximity;--tw-gradient-from-position: ;--tw-gradient-via-position: ;--tw-gradient-to-position: ;--tw-ordinal: ;--tw-slashed-zero: ;--tw-numeric-figure: ;--tw-numeric-spacing: ;--tw-numeric-fraction: ;--tw-ring-inset: ;--tw-ring-offset-width: 0px;--tw-ring-offset-color: #fff;--tw-ring-color: rgb(59 130 246 / .5);--tw-ring-offset-shadow: 0 0 #0000;--tw-ring-shadow: 0 0 #0000;--tw-shadow: 0 0 #0000;--tw-shadow-colored: 0 0 #0000;--tw-blur: ;--tw-brightness: ;--tw-contrast: ;--tw-grayscale: ;--tw-hue-rotate: ;--tw-invert: ;--tw-saturate: ;--tw-sepia: ;--tw-drop-shadow: ;--tw-backdrop-blur: ;--tw-backdrop-brightness: ;--tw-backdrop-contrast: ;--tw-backdrop-grayscale: ;--tw-backdrop-hue-rotate: ;--tw-backdrop-invert: ;--tw-backdrop-opacity: ;--tw-backdrop-saturate: ;--tw-backdrop-sepia: ;--tw-contain-size: ;--tw-contain-layout: ;--tw-contain-paint: ;--tw-contain-style: }*,:before,:after{box-sizing:border-box;border-width:0;border-style:solid;border-color:#e5e7eb}:before,:after{--tw-content: ""}html,:host{line-height:1.5;-webkit-text-size-adjust:100%;-moz-tab-size:4;-o-tab-size:4;tab-size:4;font-family:ui-sans-serif,system-ui,sans-serif,"Apple Color Emoji","Segoe UI Emoji",Segoe UI Symbol,"Noto Color Emoji";font-feature-settings:normal;font-variation-settings:normal;-webkit-tap-highlight-color:transparent}body{margin:0;line-height:inherit}hr{height:0;color:inherit;border-top-width:1px}abbr:where([title]){-webkit-text-decoration:underline dotted;text-decoration:underline dotted}h1,h2,h3,h4,h5,h6{font-size:inherit;font-weight:inherit}a{color:inherit;text-decoration:inherit}b,strong{font-weight:bolder}code,kbd,samp,pre{font-family:ui-monospace,SFMono-Regular,Menlo,Monaco,Consolas,Liberation Mono,Courier New,monospace;font-feature-settings:normal;font-variation-settings:normal;font-size:1em}small{font-size:80%}sub,sup{font-size:75%;line-height:0;position:relative;vertical-align:baseline}sub{bottom:-.25em}sup{top:-.5em}table{text-indent:0;border-color:inherit;border-collapse:collapse}button,input,optgroup,select,textarea{font-family:inherit;font-feature-settings:inherit;font-variation-settings:inherit;font-size:100%;font-weight:inherit;line-height:inherit;letter-spacing:inherit;color:inherit;margin:0;padding:0}button,select{text-transform:none}button,input:where([type=button]),input:where([type=reset]),input:where([type=submit]){-webkit-appearance:button;background-color:transparent;background-image:none}:-moz-focusring{outline:auto}:-moz-ui-invalid{box-shadow:none}progress{vertical-align:baseline}::-webkit-inner-spin-button,::-webkit-outer-spin-button{height:auto}[type=search]{-webkit-appearance:textfield;outline-offset:-2px}::-webkit-search-decoration{-webkit-appearance:none}::-webkit-file-upload-button{-webkit-appearance:button;font:inherit}summary{display:list-item}blockquote,dl,dd,h1,h2,h3,h4,h5,h6,hr,figure,p,pre{margin:0}fieldset{margin:0;padding:0}legend{padding:0}ol,ul,menu{list-style:none;margin:0;padding:0}dialog{padding:0}textarea{resize:vertical}input::-moz-placeholder,textarea::-moz-placeholder{opacity:1;color:#9ca3af}input::placeholder,textarea::placeholder{opacity:1;color:#9ca3af}button,[role=button]{cursor:pointer}:disabled{cursor:default}img,svg,video,canvas,audio,iframe,embed,object{display:block;vertical-align:middle}img,video{max-width:100%;height:auto}[hidden]:where(:not([hidden=until-found])){display:none}.fixed{position:fixed}.absolute{position:absolute}.relative{position:relative}.sticky{position:sticky}.-right-1{right:-.25rem}.-top-1{top:-.25rem}.bottom-4{bottom:1rem}.left-2{left:.5rem}.right-0{right:0}.right-4{right:1rem}.top-0{top:0}.top-16{top:4rem}.top-full{top:100%}.z-10{z-index:10}.z-50{z-index:50}.my-3{margin-top:.75rem;margin-bottom:.75rem}.my-4{margin-top:1rem;margin-bottom:1rem}.-mb-px{margin-bottom:-1px}.mb-1{margin-bottom:.25rem}.mb-2{margin-bottom:.5rem}.mb-3{margin-bottom:.75rem}.mb-4{margin-bottom:1rem}.mb-6{margin-bottom:1.5rem}.ml-1{margin-left:.25rem}.ml-2{margin-left:.5rem}.ml-3{margin-left:.75rem}.ml-auto{margin-left:auto}.mt-0\.5{margin-top:.125rem}.mt-1{margin-top:.25rem}.mt-2{margin-top:.5rem}.mt-3{margin-top:.75rem}.mt-4{margin-top:1rem}.mt-6{margin-top:1.5rem}.line-clamp-2{overflow:hidden;display:-webkit-box;-webkit-box-orient:vertical;-webkit-line-clamp:2}.inline-block{display:inline-block}.flex{display:flex}.table{display:table}.grid{display:grid}.hidden{display:none}.h-1\.5{height:.375rem}.h-14{height:3.5rem}.h-2{height:.5rem}.h-4{height:1rem}.h-5{height:1.25rem}.h-8{height:2rem}.h-full{height:100%}.h-screen{height:100vh}.max-h-64{max-height:16rem}.max-h-\[60vh\]{max-height:60vh}.w-1\.5{width:.375rem}.w-2{width:.5rem}.w-4{width:1rem}.w-5{width:1.25rem}.w-56{width:14rem}.w-64{width:16rem}.w-8{width:2rem}.w-80{width:20rem}.w-96{width:24rem}.w-full{width:100%}.min-w-full{min-width:100%}.max-w-\[80\%\]{max-width:80%}.max-w-none{max-width:none}.flex-1{flex:1 1 0%}.flex-shrink-0{flex-shrink:0}.rotate-180{--tw-rotate: 180deg;transform:translate(var(--tw-translate-x),var(--tw-translate-y)) rotate(var(--tw-rotate)) skew(var(--tw-skew-x)) skewY(var(--tw-skew-y)) scaleX(var(--tw-scale-x)) scaleY(var(--tw-scale-y))}@keyframes bounce{0%,to{transform:translateY(-25%);animation-timing-function:cubic-bezier(.8,0,1,1)}50%{transform:none;animation-timing-function:cubic-bezier(0,0,.2,1)}}.animate-bounce{animation:bounce 1s infinite}@keyframes pulse{50%{opacity:.5}}.animate-pulse{animation:pulse 2s cubic-bezier(.4,0,.6,1) infinite}.cursor-not-allowed{cursor:not-allowed}.cursor-pointer{cursor:pointer}.list-inside{list-style-position:inside}.list-decimal{list-style-type:decimal}.list-disc{list-style-type:disc}.grid-cols-1{grid-template-columns:repeat(1,minmax(0,1fr))}.flex-row-reverse{flex-direction:row-reverse}.flex-col{flex-direction:column}.flex-wrap{flex-wrap:wrap}.items-start{align-items:flex-start}.items-center{align-items:center}.justify-center{justify-content:center}.justify-between{justify-content:space-between}.gap-1{gap:.25rem}.gap-2{gap:.5rem}.gap-3{gap:.75rem}.gap-4{gap:1rem}.space-y-1>:not([hidden])~:not([hidden]){--tw-space-y-reverse: 0;margin-top:calc(.25rem * calc(1 - var(--tw-space-y-reverse)));margin-bottom:calc(.25rem * var(--tw-space-y-reverse))}.space-y-2>:not([hidden])~:not([hidden]){--tw-space-y-reverse: 0;margin-top:calc(.5rem * calc(1 - var(--tw-space-y-reverse)));margin-bottom:calc(.5rem * var(--tw-space-y-reverse))}.space-y-4>:not([hidden])~:not([hidden]){--tw-space-y-reverse: 0;margin-top:calc(1rem * calc(1 - var(--tw-space-y-reverse)));margin-bottom:calc(1rem * var(--tw-space-y-reverse))}.overflow-auto{overflow:auto}.overflow-hidden{overflow:hidden}.overflow-x-auto{overflow-x:auto}.truncate{overflow:hidden;text-overflow:ellipsis;white-space:nowrap}.whitespace-pre-wrap{white-space:pre-wrap}.break-words{overflow-wrap:break-word}.break-all{word-break:break-all}.rounded{border-radius:.25rem}.rounded-full{border-radius:9999px}.rounded-lg{border-radius:.5rem}.rounded-t{border-top-left-radius:.25rem;border-top-right-radius:.25rem}.border{border-width:1px}.border-b{border-bottom-width:1px}.border-b-0{border-bottom-width:0px}.border-b-2{border-bottom-width:2px}.border-l{border-left-width:1px}.border-l-4{border-left-width:4px}.border-r{border-right-width:1px}.border-t{border-top-width:1px}.border-po-accent{--tw-border-opacity: 1;border-color:rgb(124 58 237 / var(--tw-border-opacity, 1))}.border-po-border{--tw-border-opacity: 1;border-color:rgb(45 45 68 / var(--tw-border-opacity, 1))}.border-po-border\/50{border-color:#2d2d4480}.border-po-warning\/30{border-color:#f59e0b4d}.border-transparent{border-color:transparent}.bg-blue-600\/30{background-color:#2563eb4d}.bg-gray-500{--tw-bg-opacity: 1;background-color:rgb(107 114 128 / var(--tw-bg-opacity, 1))}.bg-gray-600\/30{background-color:#4b55634d}.bg-green-500{--tw-bg-opacity: 1;background-color:rgb(34 197 94 / var(--tw-bg-opacity, 1))}.bg-green-600{--tw-bg-opacity: 1;background-color:rgb(22 163 74 / var(--tw-bg-opacity, 1))}.bg-po-accent{--tw-bg-opacity: 1;background-color:rgb(124 58 237 / var(--tw-bg-opacity, 1))}.bg-po-accent\/20{background-color:#7c3aed33}.bg-po-bg{--tw-bg-opacity: 1;background-color:rgb(15 15 26 / var(--tw-bg-opacity, 1))}.bg-po-bg\/50{background-color:#0f0f1a80}.bg-po-bg\/80{background-color:#0f0f1acc}.bg-po-border{--tw-bg-opacity: 1;background-color:rgb(45 45 68 / var(--tw-bg-opacity, 1))}.bg-po-surface{--tw-bg-opacity: 1;background-color:rgb(26 26 46 / var(--tw-bg-opacity, 1))}.bg-po-warning{--tw-bg-opacity: 1;background-color:rgb(245 158 11 / var(--tw-bg-opacity, 1))}.bg-po-warning\/10{background-color:#f59e0b1a}.bg-purple-600\/30{background-color:#9333ea4d}.bg-red-500{--tw-bg-opacity: 1;background-color:rgb(239 68 68 / var(--tw-bg-opacity, 1))}.p-2{padding:.5rem}.p-3{padding:.75rem}.p-4{padding:1rem}.p-6{padding:1.5rem}.px-1\.5{padding-left:.375rem;padding-right:.375rem}.px-2{padding-left:.5rem;padding-right:.5rem}.px-3{padding-left:.75rem;padding-right:.75rem}.px-4{padding-left:1rem;padding-right:1rem}.py-0\.5{padding-top:.125rem;padding-bottom:.125rem}.py-1{padding-top:.25rem;padding-bottom:.25rem}.py-1\.5{padding-top:.375rem;padding-bottom:.375rem}.py-2{padding-top:.5rem;padding-bottom:.5rem}.py-3{padding-top:.75rem;padding-bottom:.75rem}.py-4{padding-top:1rem;padding-bottom:1rem}.py-8{padding-top:2rem;padding-bottom:2rem}.pb-2{padding-bottom:.5rem}.pl-4{padding-left:1rem}.text-left{text-align:left}.text-center{text-align:center}.font-mono{font-family:ui-monospace,SFMono-Regular,Menlo,Monaco,Consolas,Liberation Mono,Courier New,monospace}.text-2xl{font-size:1.5rem;line-height:2rem}.text-4xl{font-size:2.25rem;line-height:2.5rem}.text-\[10px\]{font-size:10px}.text-base{font-size:1rem;line-height:1.5rem}.text-lg{font-size:1.125rem;line-height:1.75rem}.text-sm{font-size:.875rem;line-height:1.25rem}.text-xl{font-size:1.25rem;line-height:1.75rem}.text-xs{font-size:.75rem;line-height:1rem}.font-bold{font-weight:700}.font-medium{font-weight:500}.font-semibold{font-weight:600}.uppercase{text-transform:uppercase}.normal-case{text-transform:none}.italic{font-style:italic}.leading-relaxed{line-height:1.625}.tracking-wide{letter-spacing:.025em}.text-black{--tw-text-opacity: 1;color:rgb(0 0 0 / var(--tw-text-opacity, 1))}.text-blue-300{--tw-text-opacity: 1;color:rgb(147 197 253 / var(--tw-text-opacity, 1))}.text-blue-400{--tw-text-opacity: 1;color:rgb(96 165 250 / var(--tw-text-opacity, 1))}.text-gray-100{--tw-text-opacity: 1;color:rgb(243 244 246 / var(--tw-text-opacity, 1))}.text-gray-200{--tw-text-opacity: 1;color:rgb(229 231 235 / var(--tw-text-opacity, 1))}.text-gray-300{--tw-text-opacity: 1;color:rgb(209 213 219 / var(--tw-text-opacity, 1))}.text-gray-400{--tw-text-opacity: 1;color:rgb(156 163 175 / var(--tw-text-opacity, 1))}.text-gray-500{--tw-text-opacity: 1;color:rgb(107 114 128 / var(--tw-text-opacity, 1))}.text-gray-600{--tw-text-opacity: 1;color:rgb(75 85 99 / var(--tw-text-opacity, 1))}.text-po-accent{--tw-text-opacity: 1;color:rgb(124 58 237 / var(--tw-text-opacity, 1))}.text-po-warning{--tw-text-opacity: 1;color:rgb(245 158 11 / var(--tw-text-opacity, 1))}.text-purple-300{--tw-text-opacity: 1;color:rgb(216 180 254 / var(--tw-text-opacity, 1))}.text-purple-400{--tw-text-opacity: 1;color:rgb(192 132 252 / var(--tw-text-opacity, 1))}.text-red-400{--tw-text-opacity: 1;color:rgb(248 113 113 / var(--tw-text-opacity, 1))}.text-white{--tw-text-opacity: 1;color:rgb(255 255 255 / var(--tw-text-opacity, 1))}.placeholder-gray-500::-moz-placeholder{--tw-placeholder-opacity: 1;color:rgb(107 114 128 / var(--tw-placeholder-opacity, 1))}.placeholder-gray-500::placeholder{--tw-placeholder-opacity: 1;color:rgb(107 114 128 / var(--tw-placeholder-opacity, 1))}.shadow-xl{--tw-shadow: 0 20px 25px -5px rgb(0 0 0 / .1), 0 8px 10px -6px rgb(0 0 0 / .1);--tw-shadow-colored: 0 20px 25px -5px var(--tw-shadow-color), 0 8px 10px -6px var(--tw-shadow-color);box-shadow:var(--tw-ring-offset-shadow, 0 0 #0000),var(--tw-ring-shadow, 0 0 #0000),var(--tw-shadow)}.filter{filter:var(--tw-blur) var(--tw-brightness) var(--tw-contrast) var(--tw-grayscale) var(--tw-hue-rotate) var(--tw-invert) var(--tw-saturate) var(--tw-sepia) var(--tw-drop-shadow)}.transition-colors{transition-property:color,background-color,border-color,text-decoration-color,fill,stroke;transition-timing-function:cubic-bezier(.4,0,.2,1);transition-duration:.15s}.transition-transform{transition-property:transform;transition-timing-function:cubic-bezier(.4,0,.2,1);transition-duration:.15s}html{color-scheme:dark}body{-webkit-font-smoothing:antialiased;-moz-osx-font-smoothing:grayscale}::-webkit-scrollbar{width:8px;height:8px}::-webkit-scrollbar-track{--tw-bg-opacity: 1;background-color:rgb(15 15 26 / var(--tw-bg-opacity, 1))}::-webkit-scrollbar-thumb{border-radius:9999px;--tw-bg-opacity: 1;background-color:rgb(45 45 68 / var(--tw-bg-opacity, 1))}::-webkit-scrollbar-thumb:hover{--tw-bg-opacity: 1;background-color:rgb(124 58 237 / var(--tw-bg-opacity, 1))}.first\:mt-0:first-child{margin-top:0}.last\:mb-0:last-child{margin-bottom:0}.last\:border-0:last-child{border-width:0px}.hover\:border-po-accent:hover{--tw-border-opacity: 1;border-color:rgb(124 58 237 / var(--tw-border-opacity, 1))}.hover\:border-po-accent\/50:hover{border-color:#7c3aed80}.hover\:bg-po-accent\/50:hover{background-color:#7c3aed80}.hover\:bg-po-accent\/80:hover{background-color:#7c3aedcc}.hover\:bg-po-bg:hover{--tw-bg-opacity: 1;background-color:rgb(15 15 26 / var(--tw-bg-opacity, 1))}.hover\:bg-po-bg\/70:hover{background-color:#0f0f1ab3}.hover\:bg-po-border:hover{--tw-bg-opacity: 1;background-color:rgb(45 45 68 / var(--tw-bg-opacity, 1))}.hover\:bg-po-surface:hover{--tw-bg-opacity: 1;background-color:rgb(26 26 46 / var(--tw-bg-opacity, 1))}.hover\:bg-po-warning\/20:hover{background-color:#f59e0b33}.hover\:text-po-accent:hover{--tw-text-opacity: 1;color:rgb(124 58 237 / var(--tw-text-opacity, 1))}.hover\:text-white:hover{--tw-text-opacity: 1;color:rgb(255 255 255 / var(--tw-text-opacity, 1))}.hover\:underline:hover{text-decoration-line:underline}.focus\:border-po-accent:focus{--tw-border-opacity: 1;border-color:rgb(124 58 237 / var(--tw-border-opacity, 1))}.focus\:outline-none:focus{outline:2px solid transparent;outline-offset:2px}.disabled\:cursor-not-allowed:disabled{cursor:not-allowed}.disabled\:opacity-50:disabled{opacity:.5}.group:hover .group-hover\:text-po-accent{--tw-text-opacity: 1;color:rgb(124 58 237 / var(--tw-text-opacity, 1))}@media(min-width:768px){.md\:grid-cols-2{grid-template-columns:repeat(2,minmax(0,1fr))}}@media(min-width:1024px){.lg\:grid-cols-3{grid-template-columns:repeat(3,minmax(0,1fr))}}@media(min-width:1280px){.xl\:grid-cols-4{grid-template-columns:repeat(4,minmax(0,1fr))}}

data/lib/prompt_objects/server/public/index.html

@@ -0,0 +1,14 @@
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <link rel="icon" type="image/svg+xml" href="/vite.svg" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>PromptObjects</title>
+    <script type="module" crossorigin src="/assets/index-2acS2FYZ.js"></script>
+    <link rel="stylesheet" crossorigin href="/assets/index-DXU5uRXQ.css">
+  </head>
+  <body class="bg-po-bg text-gray-100">
+    <div id="root"></div>
+  </body>
+</html>

data/lib/prompt_objects/server/websocket_handler.rb

@@ -0,0 +1,619 @@
+# frozen_string_literal: true
+
+require "json"
+require "async"
+
+module PromptObjects
+  module Server
+    # Handles WebSocket connections for real-time communication with the frontend.
+    # Subscribes to MessageBus for state updates and handles client messages.
+    #
+    # == Real-time UI Feedback Pattern ==
+    #
+    # IMPORTANT: All user actions must provide immediate visual feedback BEFORE
+    # async operations complete. This makes the UI feel "alive" and responsive.
+    #
+    # When implementing new features, follow this pattern:
+    #
+    # 1. SWITCH CONTEXT FIRST
+    #    When creating/switching threads, sessions, etc., send the context switch
+    #    message FIRST so the frontend displays the new context immediately.
+    #    Example: thread_created → frontend switches to new thread
+    #
+    # 2. UPDATE NAVIGATION
+    #    Send updated lists (sessions, threads) immediately so sidebars reflect
+    #    the change without waiting for async work to complete.
+    #    Example: po_state with sessions list → ThreadsSidebar shows new thread
+    #
+    # 3. SHOW USER INPUT
+    #    Send the user's input back immediately so they see it in the UI.
+    #    Don't wait for the AI to respond before showing what the user typed.
+    #    Example: session_updated with user message → chat shows "You: ..."
+    #
+    # 4. SHOW PROGRESS
+    #    Update status indicators during work (thinking, calling_tool, etc.).
+    #    Use streaming for incremental content when supported.
+    #    Example: po_state status "thinking" → spinner/animation shown
+    #
+    # 5. CONFIRM COMPLETION
+    #    Send final authoritative state after async work completes.
+    #    Example: session_updated with full messages → final chat state
+    #
+    # Message flow for "send message with new thread":
+    #   thread_created       → switch to new thread
+    #   po_state (sessions)  → update sidebar
+    #   session_updated      → show user message
+    #   po_state (thinking)  → show progress indicator
+    #   [stream chunks]      → incremental AI response
+    #   po_response          → complete AI response
+    #   session_updated      → final messages
+    #   po_state (idle)      → clear progress indicator
+    class WebSocketHandler
+      def initialize(runtime:, connection:, app: nil)
+        @runtime = runtime
+        @connection = connection
+        @app = app
+        @subscribed = false
+        @bus_subscription = nil
+      end
+
+      def run
+        subscribe_to_bus
+        send_initial_state
+        read_loop
+      ensure
+        unsubscribe_from_bus
+      end
+
+      # Send a message to this client (public for broadcasting).
+      def send_message(data)
+        json = JSON.generate(data)
+        @connection.write(json)
+        @connection.flush
+      rescue => e
+        puts "WebSocket write error: #{e.message}" if ENV["DEBUG"]
+      end
+
+      private
+
+      # === MessageBus Integration ===
+
+      def subscribe_to_bus
+        @bus_subscription = ->(entry) { on_bus_message(entry) }
+        @runtime.bus.subscribe(&@bus_subscription)
+
+        # Subscribe to HumanQueue for notification events
+        @human_queue_subscription = ->(event, request) { on_human_queue_event(event, request) }
+        @runtime.human_queue.subscribe(&@human_queue_subscription)
+
+        @subscribed = true
+      end
+
+      def unsubscribe_from_bus
+        return unless @subscribed
+
+        @runtime.bus.unsubscribe(@bus_subscription) if @bus_subscription
+        @runtime.human_queue.unsubscribe(@human_queue_subscription) if @human_queue_subscription
+
+        @subscribed = false
+      end
+
+      def on_human_queue_event(event, request)
+        case event
+        when :added
+          send_message(
+            type: "notification",
+            payload: request_to_hash(request)
+          )
+        when :resolved
+          send_message(
+            type: "notification_resolved",
+            payload: { id: request.id }
+          )
+        end
+      rescue => e
+        puts "WebSocket notification error: #{e.message}" if ENV["DEBUG"]
+      end
+
+      def on_bus_message(entry)
+        send_message(
+          type: "bus_message",
+          payload: {
+            from: entry[:from],
+            to: entry[:to],
+            content: entry[:message],
+            timestamp: entry[:timestamp].iso8601
+          }
+        )
+      rescue => e
+        # Connection may be closed, ignore errors
+        puts "WebSocket send error: #{e.message}" if ENV["DEBUG"]
+      end
+
+      # === Initial State ===
+
+      def send_initial_state
+        # Send environment info
+        send_message(
+          type: "environment",
+          payload: {
+            name: @runtime.name,
+            path: @runtime.env_path,
+            po_count: @runtime.registry.prompt_objects.size,
+            primitive_count: @runtime.registry.primitives.size
+          }
+        )
+
+        # Send LLM config
+        handle_get_llm_config
+
+        # Send state of all POs
+        # Always send "idle" status for initial state - from this connection's
+        # perspective, no work is pending (even if another connection has a request)
+        @runtime.registry.prompt_objects.each do |po|
+          send_message(
+            type: "po_state",
+            payload: {
+              name: po.name,
+              state: po_state_hash(po).merge(status: "idle")
+            }
+          )
+        end
+
+        # Send pending human requests (notifications)
+        @runtime.human_queue.all_pending.each do |request|
+          send_message(
+            type: "notification",
+            payload: request_to_hash(request)
+          )
+        end
+
+        # Send recent bus messages (last 50)
+        @runtime.bus.log.last(50).each do |entry|
+          send_message(
+            type: "bus_message",
+            payload: {
+              from: entry[:from],
+              to: entry[:to],
+              content: entry[:message],
+              timestamp: entry[:timestamp].iso8601
+            }
+          )
+        end
+      end
+
+      # === Client Message Handling ===
+
+      def read_loop
+        while (message = @connection.read)
+          handle_client_message(message)
+        end
+      rescue Async::Stop, EOFError, IOError
+        # Connection closed, exit gracefully
+      rescue => e
+        puts "WebSocket error: #{e.message}"
+        puts e.backtrace.first(5).join("\n") if ENV["DEBUG"]
+      end
+
+      def handle_client_message(raw_message)
+        # raw_message is a Protocol::WebSocket::TextMessage
+        data = raw_message.respond_to?(:buffer) ? raw_message.buffer : raw_message.to_s
+        message = JSON.parse(data)
+
+        case message["type"]
+        when "send_message"
+          handle_send_message(message["payload"])
+        when "respond_to_notification"
+          handle_notification_response(message["payload"])
+        when "update_po"
+          handle_update_po(message["payload"])
+        when "create_session"
+          handle_create_session(message["payload"])
+        when "switch_session"
+          handle_switch_session(message["payload"])
+        when "create_thread"
+          handle_create_thread(message["payload"])
+        when "get_thread_tree"
+          handle_get_thread_tree(message["payload"])
+        when "get_llm_config"
+          handle_get_llm_config
+        when "switch_llm"
+          handle_switch_llm(message["payload"])
+        when "ping"
+          send_message(type: "pong", payload: {})
+        else
+          send_error("Unknown message type: #{message['type']}")
+        end
+      rescue JSON::ParserError => e
+        send_error("Invalid JSON: #{e.message}")
+      rescue => e
+        send_error("Error: #{e.message}")
+        puts "Handler error: #{e.message}"
+        puts e.backtrace.first(5).join("\n") if ENV["DEBUG"]
+      end
+
+      def handle_send_message(payload)
+        po_name = payload["target"]
+        content = payload["content"]
+        new_thread = payload["new_thread"] # If true, create a new thread first
+
+        po = @runtime.registry.get(po_name)
+
+        unless po.is_a?(PromptObject)
+          send_error("Unknown prompt object: #{po_name}")
+          return
+        end
+
+        # If new_thread requested, create one first
+        if new_thread
+          thread_id = po.new_thread
+          # Notify client of new thread immediately so they see it
+          send_message(
+            type: "thread_created",
+            payload: {
+              target: po_name,
+              thread_id: thread_id,
+              thread_type: "root"
+            }
+          )
+          # Also send updated sessions list so ThreadsSidebar shows it immediately
+          send_message(
+            type: "po_state",
+            payload: {
+              name: po_name,
+              state: { sessions: po.list_sessions.map { |s| session_summary(s) } }
+            }
+          )
+        end
+
+        # Capture the session_id at request start so response goes to correct session
+        request_session_id = po.session_id
+
+        # Update PO state to working AND send the user message for immediate UI feedback
+        send_message(
+          type: "po_state",
+          payload: { name: po_name, state: { status: "thinking" } }
+        )
+
+        # Send immediate session update showing the user's message
+        # This gives instant feedback before the AI responds
+        # Include existing messages so we don't clear the chat when continuing a thread
+        existing_messages = session_messages(po, request_session_id)
+        new_user_message = { role: "user", content: content, from: "human" }
+        send_message(
+          type: "session_updated",
+          payload: {
+            target: po_name,
+            session_id: request_session_id,
+            messages: existing_messages + [new_user_message]
+          }
+        )
+
+        # Run in async context
+        Async do
+          # Use tui_mode: true so ask_human uses HumanQueue (blocking) instead of REPL mode
+          context = @runtime.context(tui_mode: true)
+          context.current_capability = "human"
+
+          begin
+            # TODO: Add streaming support to receive()
+            # For now, just get the full response
+            response = po.receive(content, context: context)
+
+            # Auto-name the thread if it doesn't have a name yet
+            auto_name_thread_if_needed(po, request_session_id, content)
+
+            # Send the complete response with session_id for correct routing
+            send_message(
+              type: "po_response",
+              payload: {
+                target: po_name,
+                session_id: request_session_id,
+                content: response
+              }
+            )
+
+            # Send session update for the session where messages were added
+            # This ensures the correct session is updated even if user switched
+            if request_session_id
+              send_message(
+                type: "session_updated",
+                payload: {
+                  target: po_name,
+                  session_id: request_session_id,
+                  messages: session_messages(po, request_session_id)
+                }
+              )
+            end
+
+            # Also send updated sessions list (for thread panel)
+            send_message(
+              type: "po_state",
+              payload: {
+                name: po_name,
+                state: { sessions: po.list_sessions.map { |s| session_summary(s) } }
+              }
+            )
+          rescue => e
+            send_error("Error from #{po_name}: #{e.message}")
+          ensure
+            # Update PO state back to idle (status only, not session data)
+            send_message(
+              type: "po_state",
+              payload: { name: po_name, state: { status: "idle" } }
+            )
+          end
+        end
+      end
+
+      # Auto-name a thread based on the user's first message if it doesn't have a name
+      def auto_name_thread_if_needed(po, session_id, user_message)
+        return unless @runtime.session_store && session_id
+
+        session = @runtime.session_store.get_session(session_id)
+        return if session.nil? || session[:name] # Already has a name
+
+        @runtime.session_store.auto_name_thread(session_id, user_message)
+      end
+
+      def handle_notification_response(payload)
+        request_id = payload["id"]
+        response = payload["response"]
+
+        @runtime.human_queue.respond(request_id, response)
+
+        send_message(
+          type: "notification_resolved",
+          payload: { id: request_id }
+        )
+      end
+
+      def handle_update_po(payload)
+        po_name = payload["name"]
+        updates = payload["updates"]
+
+        po = @runtime.registry.get(po_name)
+        return send_error("Unknown prompt object: #{po_name}") unless po.is_a?(PromptObject)
+
+        # TODO: Implement PO updates (capabilities, etc.)
+        send_message(
+          type: "po_updated",
+          payload: { name: po_name }
+        )
+      end
+
+      def handle_create_session(payload)
+        po_name = payload["target"]
+        session_name = payload["name"]
+
+        po = @runtime.registry.get(po_name)
+        return send_error("Unknown prompt object: #{po_name}") unless po.is_a?(PromptObject)
+
+        session_id = po.new_session(name: session_name)
+
+        send_message(
+          type: "session_created",
+          payload: {
+            target: po_name,
+            session_id: session_id,
+            name: session_name
+          }
+        )
+
+        # Also send updated PO state
+        send_message(
+          type: "po_state",
+          payload: { name: po_name, state: po_state_hash(po) }
+        )
+      end
+
+      def handle_switch_session(payload)
+        po_name = payload["target"]
+        session_id = payload["session_id"]
+
+        po = @runtime.registry.get(po_name)
+        return send_error("Unknown prompt object: #{po_name}") unless po.is_a?(PromptObject)
+
+        if po.switch_session(session_id)
+          send_message(
+            type: "session_switched",
+            payload: { target: po_name, session_id: session_id }
+          )
+
+          # Send updated PO state with new session's messages
+          send_message(
+            type: "po_state",
+            payload: { name: po_name, state: po_state_hash(po) }
+          )
+        else
+          send_error("Could not switch to session: #{session_id}")
+        end
+      end
+
+      def handle_create_thread(payload)
+        po_name = payload["target"]
+        thread_name = payload["name"]
+        thread_type = payload["thread_type"] || "root"
+
+        po = @runtime.registry.get(po_name)
+        return send_error("Unknown prompt object: #{po_name}") unless po.is_a?(PromptObject)
+
+        thread_id = po.new_thread(name: thread_name)
+
+        send_message(
+          type: "thread_created",
+          payload: {
+            target: po_name,
+            thread_id: thread_id,
+            name: thread_name,
+            thread_type: thread_type
+          }
+        )
+
+        # Also send updated PO state
+        send_message(
+          type: "po_state",
+          payload: { name: po_name, state: po_state_hash(po) }
+        )
+      end
+
+      def handle_get_thread_tree(payload)
+        session_id = payload["session_id"]
+        return send_error("Session ID required") unless session_id
+        return send_error("No session store available") unless @runtime.session_store
+
+        tree = @runtime.session_store.get_thread_tree(session_id)
+        return send_error("Session not found: #{session_id}") unless tree
+
+        send_message(
+          type: "thread_tree",
+          payload: { tree: serialize_thread_tree(tree) }
+        )
+      end
+
+      def handle_get_llm_config
+        config = @runtime.llm_config
+
+        # Get models for each provider
+        providers_info = LLM::Factory.providers.map do |provider|
+          info = LLM::Factory.provider_info(provider)
+          {
+            name: provider,
+            models: info[:models],
+            default_model: info[:default_model],
+            available: LLM::Factory.available_providers[provider]
+          }
+        end
+
+        send_message(
+          type: "llm_config",
+          payload: {
+            current_provider: config[:provider],
+            current_model: config[:model],
+            providers: providers_info
+          }
+        )
+      end
+
+      def handle_switch_llm(payload)
+        provider = payload["provider"]
+        model = payload["model"]
+
+        begin
+          result = @runtime.switch_llm(provider: provider, model: model)
+
+          send_message(
+            type: "llm_switched",
+            payload: {
+              provider: result[:provider],
+              model: result[:model]
+            }
+          )
+
+          # Broadcast to all connected clients via app
+          @app&.broadcast(
+            type: "llm_switched",
+            payload: {
+              provider: result[:provider],
+              model: result[:model]
+            }
+          )
+        rescue PromptObjects::Error => e
+          send_error("Failed to switch LLM: #{e.message}")
+        end
+      end
+
+      # === Helpers ===
+
+      def send_error(message)
+        send_message(type: "error", payload: { message: message })
+      end
+
+      def po_state_hash(po)
+        {
+          status: po.instance_variable_get(:@state) || "idle",
+          description: po.description,
+          capabilities: po.config["capabilities"] || [],
+          current_session: current_session_hash(po),
+          sessions: po.list_sessions.map { |s| session_summary(s) },
+          # Include full prompt for inspection
+          prompt: po.body,
+          config: po.config
+        }
+      end
+
+      def current_session_hash(po)
+        return nil unless po.session_id
+
+        {
+          id: po.session_id,
+          messages: po.history.map { |m| message_to_hash(m) }
+        }
+      end
+
+      # Get messages for a specific session (may not be current session)
+      def session_messages(po, session_id)
+        return [] unless @runtime.session_store
+
+        messages = @runtime.session_store.get_messages(session_id)
+        messages.map { |m| message_to_hash(m) }
+      end
+
+      def session_summary(session)
+        {
+          id: session[:id],
+          name: session[:name],
+          message_count: session[:message_count] || 0,
+          updated_at: session[:updated_at]&.iso8601,
+          # Thread fields
+          parent_session_id: session[:parent_session_id],
+          parent_po: session[:parent_po],
+          thread_type: session[:thread_type] || "root"
+        }
+      end
+
+      # Recursively serialize a thread tree for JSON
+      def serialize_thread_tree(tree)
+        return nil unless tree
+
+        {
+          session: session_summary(tree[:session]),
+          children: (tree[:children] || []).map { |child| serialize_thread_tree(child) }
+        }
+      end
+
+      def message_to_hash(msg)
+        case msg[:role]
+        when :user
+          { role: "user", content: msg[:content], from: msg[:from] }
+        when :assistant
+          hash = { role: "assistant", content: msg[:content] }
+          if msg[:tool_calls]
+            hash[:tool_calls] = msg[:tool_calls].map do |tc|
+              # Handle both ToolCall objects and Hashes
+              tc_id = tc.respond_to?(:id) ? tc.id : (tc[:id] || tc["id"])
+              tc_name = tc.respond_to?(:name) ? tc.name : (tc[:name] || tc["name"])
+              tc_args = tc.respond_to?(:arguments) ? tc.arguments : (tc[:arguments] || tc["arguments"] || {})
+              { id: tc_id, name: tc_name, arguments: tc_args }
+            end
+          end
+          hash
+        when :tool
+          { role: "tool", results: msg[:results] }
+        else
+          { role: msg[:role].to_s, content: msg[:content] }
+        end
+      end
+
+      def request_to_hash(request)
+        {
+          id: request.id,
+          po_name: request.capability,  # capability is the PO name
+          type: "ask_human",            # hardcode type since it's always ask_human
+          message: request.question,    # question is the message
+          options: request.options || []
+        }
+      end
+    end
+  end
+end