klaude-code 1.2.1__py3-none-any.whl → 1.2.3__py3-none-any.whl

klaude_code/session/templates/export_session.html

@@ -4,16 +4,29 @@
     <meta charset="UTF-8" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>Klaude Code - $first_user_message</title>
-    <link href="https://cdn.jsdelivr.net/npm/@fontsource/commit-mono/400.css" rel="stylesheet" />
-    <link href="https://cdn.jsdelivr.net/npm/@fontsource/commit-mono/500.css" rel="stylesheet" />
-    <link href="https://cdn.jsdelivr.net/npm/@fontsource/commit-mono/700.css" rel="stylesheet" />
+    <link
+      rel="icon"
+      href="data:image/svg+xml,<svg xmlns=%22http://www.w3.org/2000/svg%22 viewBox=%220 0 24 24%22 fill=%22none%22 stroke=%22%230851b2%22 stroke-width=%222%22 stroke-linecap=%22round%22 stroke-linejoin=%22round%22><polyline points=%2216 18 22 12 16 6%22></polyline><polyline points=%228 6 2 12 8 18%22></polyline></svg>"
+    />
+    <link
+      href="https://cdn.jsdelivr.net/npm/@fontsource/iosevka/400.css"
+      rel="stylesheet"
+    />
+    <link
+      href="https://cdn.jsdelivr.net/npm/@fontsource/iosevka/500.css"
+      rel="stylesheet"
+    />
+    <link
+      href="https://cdn.jsdelivr.net/npm/@fontsource/iosevka/800.css"
+      rel="stylesheet"
+    />
     <style>
       :root {
         --bg-body: #ededed;
         --bg-container: #f0f0f0;
         --bg-card: #f0f0f0;
         --border: #c8c8c8;
-        --text: #333333;
+        --text: #111111;
         --text-dim: #64748b;
         --accent: #0851b2;
         --accent-dim: rgba(8, 145, 178, 0.08);
@@ -21,12 +34,14 @@
         --error: #dc2626;
         --bg-error: #ffebee;
         --bg-code: #f3f3f3;
-        --font-sans: "Commit Mono", "SF Mono", Menlo, monospace;
-        --font-mono: "Commit Mono", "SF Mono", Menlo, monospace;
-        --font-size-xs: 11px;
-        --font-size-sm: 12px;
-        --font-size-base: 13px;
-        --font-size-lg: 16px;
+        --font-sans: "Iosevka", "SF Mono", Menlo, monospace;
+        --font-mono: "Iosevka", "SF Mono", Menlo, monospace;
+        --font-markdown: "Iosevka", "SF Mono", Menlo, monospace;
+        --font-weight-bold: 800;
+        --font-size-xs: 13px;
+        --font-size-sm: 14px;
+        --font-size-base: 16px;
+        --font-size-lg: 18px;
         --spacing-xs: 4px;
         --spacing-sm: 8px;
         --spacing-md: 12px;
@@ -66,7 +81,7 @@
 
       .header h1 {
         font-size: 20px;
-        font-weight: 700;
+        font-weight: var(--font-weight-bold);
         margin-bottom: 16px;
         color: var(--text);
         display: inline-block;
@@ -86,9 +101,8 @@
 
       .meta-label {
         font-size: var(--font-size-sm);
-        font-weight: 700;
+        font-weight: var(--font-weight-bold);
         text-transform: uppercase;
-        letter-spacing: 0.05em;
         color: var(--text-dim);
       }
 
@@ -116,8 +130,7 @@
         font-family: var(--font-mono);
         font-size: var(--font-size-sm);
         text-transform: uppercase;
-        letter-spacing: 0.05em;
-        font-weight: 700;
+        font-weight: var(--font-weight-bold);
         cursor: pointer;
         user-select: none;
         list-style: none;
@@ -184,14 +197,17 @@
 
       .role-label {
         font-size: var(--font-size-sm);
-        font-weight: 700;
+        font-weight: var(--font-weight-bold);
         text-transform: uppercase;
-        letter-spacing: 0.05em;
         margin-bottom: 4px;
         color: var(--text-dim);
         display: flex;
         align-items: center;
         gap: 8px;
+        width: 100%;
+      }
+      .message-header .role-label {
+        width: auto;
       }
       .role-label.user {
         color: var(--accent);
@@ -210,8 +226,7 @@
         font-family: var(--font-mono);
         font-size: var(--font-size-xs);
         text-transform: uppercase;
-        letter-spacing: 0.05em;
-        font-weight: 700;
+        font-weight: var(--font-weight-bold);
         cursor: pointer;
         user-select: none;
         list-style: none;
@@ -279,14 +294,14 @@
         border: 1px solid var(--border);
         background: transparent;
         color: var(--text-dim);
+        font-family: var(--font-mono);
         font-size: var(--font-size-xs);
         text-transform: uppercase;
-        letter-spacing: 0.08em;
         padding: 2px 10px;
         border-radius: 999px;
         cursor: pointer;
         transition: color 0.2s, border-color 0.2s, background 0.2s;
-        font-weight: 700;
+        font-weight: var(--font-weight-bold);
       }
       .raw-toggle:hover {
         color: var(--text);
@@ -303,14 +318,14 @@
         border: 1px solid var(--border);
         background: transparent;
         color: var(--text-dim);
+        font-family: var(--font-mono);
         font-size: var(--font-size-xs);
         text-transform: uppercase;
-        letter-spacing: 0.08em;
         padding: 2px 10px;
         border-radius: 999px;
         cursor: pointer;
         transition: color 0.2s, border-color 0.2s, background 0.2s;
-        font-weight: 700;
+        font-weight: var(--font-weight-bold);
       }
       .copy-raw-btn:hover {
         color: var(--text);
@@ -321,14 +336,14 @@
         border: 1px solid var(--border);
         background: transparent;
         color: var(--text-dim);
+        font-family: var(--font-mono);
         font-size: var(--font-size-xs);
         text-transform: uppercase;
-        letter-spacing: 0.08em;
         padding: 2px 10px;
         border-radius: 999px;
         cursor: pointer;
         transition: color 0.2s, border-color 0.2s, background 0.2s;
-        font-weight: 700;
+        font-weight: var(--font-weight-bold);
       }
       .copy-mermaid-btn:hover {
         color: var(--text);
@@ -361,7 +376,7 @@
         border-radius: 0;
         padding: 0;
         box-shadow: none;
-        font-weight: 700;
+        font-weight: var(--font-weight-bold);
       }
 
       .thinking-block {
@@ -371,9 +386,15 @@
         border-left: 2px solid var(--border);
         color: var(--text-dim);
         font-style: italic;
-        font-size: var(--font-size-base);
+        font-size: var(--font-size-sm);
         background: var(--bg-body);
       }
+      .thinking-block.markdown-body p {
+        margin-bottom: 8px;
+      }
+      .thinking-block.markdown-body p:last-child {
+        margin-bottom: 0;
+      }
 
       /* Response Metadata */
       .response-metadata {
@@ -392,7 +413,7 @@
         line-height: 1.4;
       }
       .metadata-model {
-        font-weight: 700;
+        font-weight: var(--font-weight-bold);
         color: var(--text);
       }
       .metadata-provider {
@@ -427,7 +448,7 @@
       }
 
       .tool-name {
-        font-weight: 700;
+        font-weight: var(--font-weight-bold);
         color: var(--accent);
       }
       .tool-id {
@@ -435,6 +456,24 @@
         font-size: var(--font-size-xs);
       }
 
+      .tool-header-right {
+        display: flex;
+        align-items: center;
+        gap: 8px;
+      }
+
+      .timestamp {
+        font-family: var(--font-mono);
+        font-size: var(--font-size-xs);
+        color: var(--text-dim);
+        margin-left: auto;
+        font-weight: normal;
+      }
+
+      .assistant-toolbar .timestamp {
+        margin-right: 8px;
+      }
+
       .tool-args {
         padding: 12px;
         background: var(--bg-body);
@@ -455,8 +494,7 @@
         font-family: var(--font-mono);
         font-size: var(--font-size-xs);
         text-transform: uppercase;
-        letter-spacing: 0.05em;
-        font-weight: 700;
+        font-weight: var(--font-weight-bold);
         cursor: pointer;
         user-select: none;
         list-style: none;
@@ -557,7 +595,7 @@
 
       /* Markdown Elements */
       .markdown-body {
-        font-family: var(--font-sans);
+        font-family: var(--font-markdown);
       }
       .markdown-body hr {
         height: 0;
@@ -574,7 +612,7 @@
         border: 1px solid var(--border);
       }
       .markdown-body code {
-        font-family: var(--font-mono);
+        font-family: var(--font-markdown);
         font-size: 0.9em;
         background: var(--bg-code);
         padding: 2px 4px;
@@ -648,8 +686,7 @@
         font-family: var(--font-mono);
         font-size: var(--font-size-xs);
         text-transform: uppercase;
-        letter-spacing: 0.05em;
-        font-weight: 700;
+        font-weight: var(--font-weight-bold);
         cursor: pointer;
         user-select: none;
         list-style: none;
@@ -689,6 +726,17 @@
         padding-top: 24px;
       }
 
+      .footer-link {
+        color: inherit;
+        text-decoration: none;
+        font-weight: 600;
+        transition: color 0.2s;
+      }
+
+      .footer-link:hover {
+        color: var(--accent);
+      }
+
       .expandable {
         cursor: pointer;
       }
@@ -757,6 +805,42 @@
         color: var(--text-dim);
       }
 
+      /* Scroll to Bottom Button */
+      .scroll-btn {
+        position: fixed;
+        bottom: 24px;
+        right: 24px;
+        width: 40px;
+        height: 40px;
+        border-radius: 50%;
+        background: var(--bg-card);
+        border: 1px solid var(--border);
+        box-shadow: 0 4px 12px rgba(0, 0, 0, 0.1);
+        color: var(--text-dim);
+        display: flex;
+        align-items: center;
+        justify-content: center;
+        cursor: pointer;
+        opacity: 0;
+        visibility: hidden;
+        transform: translateY(10px);
+        transition: all 0.3s cubic-bezier(0.4, 0, 0.2, 1);
+        z-index: 100;
+      }
+
+      .scroll-btn:hover {
+        color: var(--accent);
+        border-color: var(--accent);
+        transform: translateY(-2px);
+        box-shadow: 0 6px 16px rgba(0, 0, 0, 0.12);
+      }
+
+      .scroll-btn.visible {
+        opacity: 1;
+        visibility: visible;
+        transform: translateY(0);
+      }
+
       /* Tool details in Available Tools section */
       .tool-details {
         margin-bottom: 2px;
@@ -798,9 +882,8 @@
       }
       .tool-params-title {
         font-size: var(--font-size-xs);
-        font-weight: 700;
+        font-weight: var(--font-weight-bold);
         text-transform: uppercase;
-        letter-spacing: 0.05em;
         color: var(--text-dim);
         margin-bottom: 8px;
       }
@@ -858,9 +941,7 @@
         <div
           class="details-content system-prompt-content"
           style="font-family: var(--font-mono); white-space: pre-wrap"
-        >
-          $system_prompt
-        </div>
+        >$system_prompt</div>
       </details>
 
       <details class="collapsible-section">
@@ -871,17 +952,48 @@
       <div class="message-stream">$messages_html</div>
 
       <div class="footer">
-        Generated by klaude-code &bull; $footer_time &bull; $total_messages
-        messages
+        Generated by
+        <a
+          href="https://github.com/inspirepan/klaude-code"
+          class="footer-link"
+          target="_blank"
+          >klaude-code</a
+        >
+        &bull; $footer_time &bull; $total_messages messages
       </div>
     </div>
 
+    <div id="scroll-btn" class="scroll-btn" title="Scroll to bottom">
+      <svg
+        xmlns="http://www.w3.org/2000/svg"
+        width="20"
+        height="20"
+        viewBox="0 0 24 24"
+        fill="none"
+        stroke="currentColor"
+        stroke-width="2"
+        stroke-linecap="round"
+        stroke-linejoin="round"
+      >
+        <line x1="12" y1="5" x2="12" y2="19"></line>
+        <polyline points="19 12 12 19 5 12"></polyline>
+      </svg>
+    </div>
+
     <link
       rel="stylesheet"
       href="https://cdn.jsdelivr.net/gh/highlightjs/cdn-release@11.9.0/build/styles/github.min.css"
     />
     <script src="https://cdn.jsdelivr.net/gh/highlightjs/cdn-release@11.9.0/build/highlight.min.js"></script>
     <script src="https://cdn.jsdelivr.net/npm/marked/marked.min.js"></script>
+    <script type="module">
+      import mermaid from "https://cdn.jsdelivr.net/npm/mermaid@10/dist/mermaid.esm.min.mjs";
+      mermaid.initialize({
+        startOnLoad: true,
+        theme: "neutral",
+        fontFamily: '"Iosevka", "SF Mono", Menlo, monospace',
+      });
+    </script>
     <script>
       // Markdown rendering and Syntax Highlighting
       document.querySelectorAll(".markdown-content").forEach((el) => {
@@ -1077,6 +1189,32 @@
           }
         });
       });
+
+      // Scroll to bottom button
+      const scrollBtn = document.getElementById("scroll-btn");
+
+      function updateScrollBtn() {
+        // Show button if we are not at the bottom
+        const isAtBottom =
+          window.innerHeight + window.scrollY >=
+          document.body.offsetHeight - 100;
+        if (isAtBottom) {
+          scrollBtn.classList.remove("visible");
+        } else {
+          scrollBtn.classList.add("visible");
+        }
+      }
+
+      window.addEventListener("scroll", updateScrollBtn);
+      window.addEventListener("resize", updateScrollBtn);
+      updateScrollBtn(); // Initial check
+
+      scrollBtn.addEventListener("click", () => {
+        window.scrollTo({
+          top: document.body.scrollHeight,
+          behavior: "smooth",
+        });
+      });
     </script>
   </body>
 </html>

klaude_code/trace/__init__.py

@@ -1,3 +1,3 @@
-from .log import DebugType, log, log_debug, logger, set_debug_logging
+from .log import DebugType, is_debug_enabled, log, log_debug, logger, set_debug_logging
 
-__all__ = ["log", "log_debug", "logger", "set_debug_logging", "DebugType"]
+__all__ = ["log", "log_debug", "logger", "set_debug_logging", "DebugType", "is_debug_enabled"]

klaude_code/trace/log.py

@@ -7,7 +7,7 @@ from rich.console import Console
 from rich.logging import RichHandler
 from rich.text import Text
 
-from klaude_code.const import DEFAULT_DEBUG_LOG_FILE, LOG_BACKUP_COUNT, LOG_MAX_BYTES
+from klaude_code import const
 
 # Module-level logger
 logger = logging.getLogger("klaude_code")
@@ -87,13 +87,13 @@ def set_debug_logging(
 
     # Determine output mode
     use_file = write_to_file if write_to_file is not None else True
-    file_path = log_file if log_file is not None else DEFAULT_DEBUG_LOG_FILE
+    file_path = log_file if log_file is not None else const.DEFAULT_DEBUG_LOG_FILE
 
     if use_file:
         _file_handler = RotatingFileHandler(
             file_path,
-            maxBytes=LOG_MAX_BYTES,
-            backupCount=LOG_BACKUP_COUNT,
+            maxBytes=const.LOG_MAX_BYTES,
+            backupCount=const.LOG_BACKUP_COUNT,
             encoding="utf-8",
         )
         _file_handler.setLevel(logging.DEBUG)
@@ -121,7 +121,8 @@ def log(*objects: str | tuple[str, str], style: str = "") -> None:
         style: Default style for all objects
     """
     log_console.print(
-        *((Text(obj[0], style=obj[1]) if isinstance(obj, tuple) else Text(obj)) for obj in objects), style=style
+        *((Text(obj[0], style=obj[1]) if isinstance(obj, tuple) else Text(obj)) for obj in objects),
+        style=style,
     )
 
 
@@ -160,3 +161,8 @@ def _build_message(objects: Iterable[str | tuple[str, str]]) -> str:
         else:
             parts.append(obj)
     return " ".join(parts)
+
+
+def is_debug_enabled() -> bool:
+    """Check if debug logging is currently enabled."""
+    return _debug_enabled

klaude_code/ui/__init__.py

@@ -1,8 +1,91 @@
-from .base.debug_event_display import DebugEventDisplay
-from .base.display_abc import DisplayABC
-from .base.exec_display import ExecDisplay
-from .base.input_abc import InputProviderABC
-from .repl.display import REPLDisplay
-from .repl.input import PromptToolkitInput
-
-__all__ = ["DisplayABC", "InputProviderABC", "REPLDisplay", "PromptToolkitInput", "DebugEventDisplay", "ExecDisplay"]
+"""
+UI Module - Display and Input Abstractions for klaude-code
+
+This module provides the UI layer for klaude-code, including display modes
+and input providers. The UI is designed around three main concepts:
+
+Display Modes:
+- REPLDisplay: Interactive terminal mode with Rich rendering, spinners, and live updates
+- ExecDisplay: Non-interactive exec mode that only outputs task results
+- DebugEventDisplay: Decorator that logs all events for debugging purposes
+
+Input Providers:
+- PromptToolkitInput: Interactive input with prompt-toolkit (completions, keybindings)
+
+Factory Functions:
+- create_default_display(): Creates the appropriate display for interactive mode
+- create_exec_display(): Creates the appropriate display for exec mode
+"""
+
+# --- Abstract Interfaces ---
+from .core.display import DisplayABC
+from .core.input import InputProviderABC
+from .modes.debug.display import DebugEventDisplay
+from .modes.exec.display import ExecDisplay, StreamJsonDisplay
+
+# --- Display Mode Implementations ---
+from .modes.repl.display import REPLDisplay
+
+# --- Input Implementations ---
+from .modes.repl.input_prompt_toolkit import PromptToolkitInput
+from .terminal.notifier import TerminalNotifier
+
+
+def create_default_display(
+    debug: bool = False,
+    theme: str | None = None,
+    notifier: TerminalNotifier | None = None,
+) -> DisplayABC:
+    """
+    Create the default display for interactive REPL mode.
+
+    Args:
+        debug: If True, wrap the display with DebugEventDisplay to log all events.
+        theme: Optional theme name ("light" or "dark") for syntax highlighting.
+        notifier: Optional terminal notifier for desktop notifications.
+
+    Returns:
+        A DisplayABC implementation suitable for interactive use.
+    """
+    repl_display = REPLDisplay(theme=theme, notifier=notifier)
+    if debug:
+        return DebugEventDisplay(repl_display)
+    return repl_display
+
+
+def create_exec_display(debug: bool = False, stream_json: bool = False) -> DisplayABC:
+    """
+    Create a display for exec (non-interactive) mode.
+
+    Args:
+        debug: If True, wrap the display with DebugEventDisplay to log all events.
+        stream_json: If True, stream all events as JSON lines instead of normal output.
+
+    Returns:
+        A DisplayABC implementation that only outputs task results.
+    """
+    if stream_json:
+        return StreamJsonDisplay()
+    exec_display = ExecDisplay()
+    if debug:
+        return DebugEventDisplay(exec_display)
+    return exec_display
+
+
+__all__ = [
+    # Abstract interfaces
+    "DisplayABC",
+    "InputProviderABC",
+    # Display mode implementations
+    "REPLDisplay",
+    "ExecDisplay",
+    "StreamJsonDisplay",
+    "DebugEventDisplay",
+    # Input implementations
+    "PromptToolkitInput",
+    # Factory functions
+    "create_default_display",
+    "create_exec_display",
+    # Supporting types
+    "TerminalNotifier",
+]

klaude_code/ui/core/__init__.py

@@ -0,0 +1 @@
+# Core UI abstractions

klaude_code/ui/core/display.py

@@ -0,0 +1,103 @@
+from abc import ABC, abstractmethod
+from asyncio import Queue
+
+from klaude_code.protocol import events
+from klaude_code.trace import log
+
+
+class DisplayABC(ABC):
+    """
+    Abstract base class for UI display implementations.
+
+    A Display is responsible for rendering events from the executor to the user.
+    Implementations can range from simple text output (ExecDisplay) to rich
+    interactive terminals (REPLDisplay) or debugging wrappers (DebugEventDisplay).
+
+    Lifecycle:
+        1. start() is called once before any events are consumed.
+        2. consume_event() is called for each event from the executor.
+        3. stop() is called once when the display is shutting down (after EndEvent).
+
+    Typical Usage:
+        display = create_default_display()
+        await display.consume_event_loop(event_queue)
+
+        # Or manually:
+        await display.start()
+        try:
+            async for event in events:
+                if isinstance(event, EndEvent):
+                    break
+                await display.consume_event(event)
+        finally:
+            await display.stop()
+
+    Thread Safety:
+        Display implementations should be used from a single async task.
+        The consume_event_loop method handles the standard event loop pattern.
+    """
+
+    @abstractmethod
+    async def consume_event(self, event: events.Event) -> None:
+        """
+        Process a single event from the executor.
+
+        This method is called for each event except EndEvent, which triggers stop().
+        Implementations should handle all relevant event types and render them
+        appropriately for the user.
+
+        Args:
+            event: The event to process. See klaude_code.protocol.events for types.
+        """
+
+    @abstractmethod
+    async def start(self) -> None:
+        """
+        Initialize the display before processing events.
+
+        Called once before any consume_event calls. Use this for any setup
+        that needs to happen before rendering begins (e.g., initializing
+        terminal state, starting background tasks).
+        """
+
+    @abstractmethod
+    async def stop(self) -> None:
+        """
+        Clean up the display after all events have been processed.
+
+        Called once after EndEvent is received. Use this for cleanup such as
+        stopping spinners, restoring terminal state, or flushing output buffers.
+        """
+
+    async def consume_event_loop(self, q: Queue[events.Event]) -> None:
+        """
+        Main event loop that processes events from a queue.
+
+        This is the standard entry point for running a display. It handles:
+        - Calling start() before processing
+        - Consuming events until EndEvent is received
+        - Calling stop() after EndEvent
+        - Error handling and logging for individual events
+
+        Args:
+            q: An asyncio Queue of events to process. The loop exits when
+               an EndEvent is received.
+        """
+        await self.start()
+        while True:
+            event = await q.get()
+            try:
+                if isinstance(event, events.EndEvent):
+                    await self.stop()
+                    break
+                await self.consume_event(event)
+            except Exception as e:
+                import traceback
+
+                log(
+                    f"Error in consume_event_loop, {e.__class__.__name__}, {e}",
+                    style="red",
+                )
+                log(traceback.format_exc(), style="red")
+            finally:
+                q.task_done()