opencode-chat-exporter 1.0.0__py3-none-any.whl

chat_exporter/render.py

@@ -0,0 +1,272 @@
+import base64
+import json
+import re
+from pathlib import Path
+from typing import Any
+
+from .db import OpenCodeDB
+
+ROLE_LABELS = {
+    "user": "USER",
+    "assistant": "ASSISTANT",
+    "system": "SYSTEM",
+    "tool": "TOOL",
+}
+
+
+def _ts(db: OpenCodeDB, ms: int | None) -> str:
+    t = db.ts(ms)
+    return t.strftime("%Y-%m-%d %H:%M:%S") if t else ""
+
+
+def _fence(text: str) -> str:
+    for f in ("```", "~~~"):
+        if f not in text:
+            return f
+    return "````"
+
+
+def _is_base64url(value: Any) -> bool:
+    return isinstance(value, str) and value.startswith("data:")
+
+
+def _pretty_json(obj: Any) -> str:
+    try:
+        return json.dumps(obj, ensure_ascii=False, indent=2, default=str)
+    except Exception:
+        return str(obj)
+
+
+def _safe_filename(name: str) -> str:
+    return re.sub(r'[\\/:*?"<>|]', "_", name)
+
+
+def _tool_title(state: dict) -> str:
+    """One-line human summary of a tool call from its input."""
+    inp = state.get("input") or {}
+    for key in ("command", "pattern", "filePath", "path", "url", "query"):
+        if inp.get(key) is not None:
+            return str(inp[key])
+    return ""
+
+
+class Exporter:
+    def __init__(
+        self,
+        db: OpenCodeDB,
+        include_reasoning: bool = False,
+        include_metadata: bool = False,
+        output_dir: Path | None = None,
+        max_output: int | None = None,
+    ) -> None:
+        self.db = db
+        self.include_reasoning = include_reasoning
+        self.include_metadata = include_metadata
+        self.output_dir = output_dir
+        self.max_output = max_output
+        self.attachments_dir = (output_dir / "attachments") if output_dir else None
+
+    # ---------------------------------------------------------------- sessions
+    def render_session(self, session_id: str) -> str:
+        session = self.db.get_session(session_id)
+        if session is None:
+            raise KeyError(f"session not found: {session_id}")
+        messages = self.db.get_session_messages(session_id)
+
+        out: list[str] = []
+        self._header(out, session)
+        out.append("")
+        out.append("---")
+        out.append("")
+
+        n_user = n_asst = 0
+        for msg in messages:
+            role = msg["data"].get("role")
+            if role == "user":
+                n_user += 1
+            elif role == "assistant":
+                n_asst += 1
+            self._message(out, msg)
+
+        if n_user == 0 and n_asst == 0:
+            out.append("_(no messages in this session)_")
+
+        return "\n".join(out).rstrip() + "\n"
+
+    def render_all(self, include_archived: bool = False) -> str:
+        chunks = []
+        for s in self.db.list_sessions(include_archived=include_archived):
+            chunks.append(self.render_session(s["id"]))
+            chunks.append("\n\n---\n\n# ┄┄┄ NEXT SESSION ┄┄┄\n\n")
+        return "\n".join(chunks).rstrip() + "\n"
+
+    # ------------------------------------------------------------------ header
+    def _header(self, out: list[str], s: dict) -> None:
+        title = s.get("title") or "Untitled session"
+        out.append(f"# {title}")
+        out.append("")
+        out.append(f"> Session `{s.get('id')}`")
+        agent = s.get("agent")
+        model = s.get("model")
+        if isinstance(model, dict):
+            bits = [x for x in (model.get("providerID"), model.get("modelID")) if x]
+            model = " · ".join(bits)
+        if agent:
+            out.append(f"**Agent:** `{agent}`")
+        if model:
+            out.append(f"**Model:** `{model}`")
+        directory = s.get("directory")
+        if directory:
+            out.append(f"**Directory:** `{directory}`")
+        created = _ts(self.db, s.get("time_created"))
+        updated = _ts(self.db, s.get("time_updated"))
+        if created:
+            out.append(f"**Created:** {created}")
+        if updated:
+            out.append(f"**Updated:** {updated}")
+        share_url = s.get("share_url")
+        if share_url:
+            out.append(f"**Share:** {share_url}")
+        total = (int(s.get("tokens_input") or 0) + int(s.get("tokens_output") or 0)
+                 + int(s.get("tokens_reasoning") or 0))
+        if total:
+            out.append(f"**Tokens:** {total:,}")
+        cost = s.get("cost")
+        if isinstance(cost, (int, float)) and cost:
+            out.append(f"**Cost:** ${float(cost):.4f}")
+        out.append("")
+
+    # ---------------------------------------------------------------- messages
+    def _message(self, out: list[str], msg: dict) -> None:
+        role = msg["data"].get("role", "unknown")
+        label = ROLE_LABELS.get(role, role.upper())
+        header = f"### {label}"
+        t = _ts(self.db, msg["time_created"])
+        if t:
+            header += f"  `{t}`"
+        model = msg["data"].get("model")
+        if isinstance(model, dict) and model.get("modelID"):
+            header += f"  · {model['modelID']}"
+        out.append(header)
+        out.append("")
+
+        for part in msg["parts"]:
+            data = part["data"]
+            ptype = data.get("type", "")
+            if ptype == "text":
+                text = str(data.get("text", "")).strip()
+                if text:
+                    out.append(text)
+                    out.append("")
+            elif ptype == "reasoning":
+                if self.include_reasoning:
+                    rtext = str(data.get("text", "")).strip()
+                    if rtext:
+                        out.append("<details><summary>🤔 reasoning</summary>")
+                        out.append("")
+                        out.append(rtext)
+                        out.append("")
+                        out.append("</details>")
+                        out.append("")
+            elif ptype == "tool":
+                self._tool(out, data)
+            elif ptype == "file":
+                self._file(out, data)
+            elif ptype == "compaction":
+                out.append("> 🧹 *context compacted*")
+                out.append("")
+            elif ptype in ("step-start", "step-finish"):
+                pass
+            else:
+                if self.include_metadata:
+                    out.append(f"<!-- part type: {ptype} -->")
+                    out.append(_pretty_json(data))
+                    out.append("")
+
+        out.append("")
+        out.append("---")
+        out.append("")
+
+    # ------------------------------------------------------------------- tools
+    def _tool(self, out: list[str], data: dict) -> None:
+        tool = data.get("tool") or data.get("name") or "tool"
+        state = data.get("state") or {}
+        status = state.get("status", "")
+        inp = state.get("input") or {}
+        out_value = state.get("output")
+        metadata = state.get("metadata") if isinstance(state.get("metadata"), dict) else {}
+
+        icon = {"completed": "✅", "error": "⛔", "cancelled": "🚫"}.get(status, "⏳")
+        out.append(f"#### {icon} Tool: `{tool}` · `{status}`")
+        out.append("")
+
+        title = _tool_title(state)
+        if title:
+            out.append(f"**Input:**")
+            out.append("")
+            out.append(self._fence_block(title))
+            out.append("")
+        elif inp:
+            out.append(f"**Input:**")
+            out.append("")
+            out.append(self._fence_block(_pretty_json(inp)))
+            out.append("")
+
+        text = self._format_output(out_value, tool)
+        if text:
+            out.append("**Output:**")
+            out.append("")
+            out.append(self._fence_block(text))
+            out.append("")
+
+        if metadata.get("exit") is not None:
+            out.append(f"*exit code: {metadata['exit']}*")
+            out.append("")
+
+    def _format_output(self, out_value: Any, tool: str) -> str | None:
+        if out_value is None or out_value == "":
+            return None
+        if _is_base64url(out_value):
+            try:
+                b64 = out_value.split(",", 1)[1]
+                return base64.b64decode(b64).decode("utf-8", errors="replace")
+            except Exception:
+                return None
+        if isinstance(out_value, (dict, list)):
+            text = _pretty_json(out_value)
+        else:
+            text = str(out_value)
+        if self.max_output and len(text) > self.max_output:
+            text = text[: self.max_output] + "\n…[output truncated]"
+        return text
+
+    def _fence_block(self, text: str) -> str:
+        f = _fence(text)
+        return f"{f}\n{text}\n{f}"
+
+    # ------------------------------------------------------------------- files
+    def _file(self, out: list[str], data: dict) -> None:
+        filename = data.get("filename") or "(attachment)"
+        mime = data.get("mime") or ""
+        url = data.get("url") or ""
+        line = f"#### 📎 {filename}"
+        if mime:
+            line += f"  `{mime}`"
+        out.append(line)
+        out.append("")
+
+        if self.attachments_dir and _is_base64url(url):
+            try:
+                b64 = url.split(",", 1)[1]
+                raw = base64.b64decode(b64)
+                dest = self.attachments_dir / _safe_filename(filename)
+                dest.parent.mkdir(parents=True, exist_ok=True)
+                dest.write_bytes(raw)
+                out.append(f"- saved to `{dest}`")
+            except Exception as e:
+                out.append(f"- *could not save attachment: {e}*")
+        elif url.startswith("http"):
+            out.append(f"- `{url}`")
+        elif url:
+            out.append(f"- `{url[:200]}`")
+        out.append("")

opencode_chat_exporter-1.0.0.dist-info/METADATA

@@ -0,0 +1,109 @@
+Metadata-Version: 2.4
+Name: opencode-chat-exporter
+Version: 1.0.0
+Summary: Export OpenCode / AntiGravity local chats to a single connected Markdown file (GUI + CLI)
+Author: nasro2123
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/nasro2123/chat-exporter
+Project-URL: Repository, https://github.com/nasro2123/chat-exporter
+Project-URL: Issues, https://github.com/nasro2123/chat-exporter/issues
+Keywords: opencode,antigravity,chat,export,markdown,cli
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Topic :: Utilities
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: customtkinter==5.2.2
+Dynamic: license-file
+
+# Chat Exporter
+
+A Python tool that exports your local chats — from **OpenCode (Desktop)** or **AntiGravity (Google)** — into one **single connected Markdown file**, preserving every detail verbatim: user and assistant messages, CMD / PowerShell commands, tool outputs, file & search operations, and attachments.
+
+## Features
+
+- **Two chat sources**: OpenCode or AntiGravity (switch from the GUI or CLI)
+- **Simple, clean GUI** (CustomTkinter) to browse and preview conversations
+- **Live preview** with color-coded messages (you / assistant / tools)
+- **Full export** of each conversation into one connected Markdown file — not separate sections
+- Export all conversations to a folder, or everything into a single file
+- **Live watch mode**: automatically re-export when the source updates
+- Fully equivalent command-line interface
+
+## Requirements
+
+- Python 3.10+
+
+## Building the EXE
+
+**Easiest way:** double-click `build.bat`
+
+Or from PowerShell:
+
+```powershell
+.\build.ps1            # onedir build (faster startup)
+.\build.ps1 -OneFile   # single-file exe (easier to share)
+```
+
+The script handles everything automatically: creating a `.venv`, installing dependencies, then building with PyInstaller. Output goes to `dist\`.
+
+## Usage
+
+```
+pip install -r requirements.txt
+
+# GUI
+python run.py
+
+# CLI — OpenCode (default)
+python -m chat_exporter --cli list
+python -m chat_exporter --cli export --latest
+
+# CLI — AntiGravity
+python -m chat_exporter --cli --source antigravity list
+python -m chat_exporter --cli --source antigravity export --latest
+```
+
+### Default locations
+
+| Source | Path |
+|---|---|
+| OpenCode | `%USERPROFILE%\.local\share\opencode\opencode.db` |
+| AntiGravity | `%USERPROFILE%\.gemini\antigravity` |
+
+The path can be changed from the "…" button in the GUI or via `--db` (for AntiGravity, pass the `antigravity` folder containing `conversations\` and `brain\`).
+
+### How does AntiGravity reading work?
+
+Conversations are stored in a SQLite database with a `step_payload` column encoded as **protobuf**. The reader first tries the plain-text `transcript.jsonl` generated by the app under `brain\<uuid>\.system_generated\logs\` (easy and readable); if not found, it decodes the protobuf directly from the database. Both paths produce identical output.
+
+## Project structure
+
+```
+Chat Exporter/
+├── chat_exporter/
+│   ├── db.py          # OpenCode database reader (sessions / messages / parts) + factory
+│   ├── antigravity.py # AntiGravity reader (transcript.jsonl + protobuf fallback)
+│   ├── render.py      # Conversation -> full Markdown renderer
+│   ├── gui.py         # Graphical interface
+│   ├── cli.py         # Command-line interface
+│   └── __main__.py
+├── run.py             # Launch the GUI
+├── run_gui.py
+├── build.bat          # One-click EXE build
+├── build.ps1          # Build script (auto venv + PyInstaller)
+└── requirements.txt
+```
+
+## Privacy
+
+Everything runs locally — the tool reads data on your machine and writes Markdown files only. Nothing is sent to the internet. Review exported files before sharing them; they may contain sensitive paths and data from your sessions.

opencode_chat_exporter-1.0.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+chat_exporter/__init__.py,sha256=ZE9dmbrNoJ7tBAOISIu5mK8xZEOzBgVYYxOAKtRphkg,109
+chat_exporter/__main__.py,sha256=s--TO_TfzC1TZRCuZtu4HQBObfAnv_hIVhWAgHmHpLU,447
+chat_exporter/antigravity.py,sha256=ozmz2HImVRCRBS1wr50geNFohNEUWRR-cWSgidM9rMw,21876
+chat_exporter/cli.py,sha256=E1aJan1WUnQ2tmmxxWuN2pKdpXIv4SD7NYGzUa25nZ0,7341
+chat_exporter/db.py,sha256=1SEW9dY4K9uTGZpAYEH6J9dvbaGXHuAIJkZyODVZQxo,5756
+chat_exporter/gui.py,sha256=jMgcvVnuMihFYVKqhOxFR7qxdAjZ2JcZaxX1Y1ZxFmI,22034
+chat_exporter/render.py,sha256=PDXoglDjMj8Qquv60ufWJcDfz_MpJkI7a0KaZ0ImwIM,9502
+opencode_chat_exporter-1.0.0.dist-info/licenses/LICENSE,sha256=WKBSpMZp6NRMM0kGUKXFH2OpDKtSQJtnaOvte0bETOo,1066
+opencode_chat_exporter-1.0.0.dist-info/METADATA,sha256=eYINvNg2MQa747qJvJJ9IvHvOgQ4ai6sOik9snMbhSs,4450
+opencode_chat_exporter-1.0.0.dist-info/WHEEL,sha256=YVMoNqKzERt-wjUZwJ33xBGAwnFl-4cqbYkTtWa4itE,91
+opencode_chat_exporter-1.0.0.dist-info/entry_points.txt,sha256=T_EglGYkzIcXoeQqbELA6pi5O6Vk90gGNG8Dn_bGRVg,66
+opencode_chat_exporter-1.0.0.dist-info/top_level.txt,sha256=X2tacK0ysdURlsoPuW4TzGvcDt2cXhbHPlto5Qc5WQE,14
+opencode_chat_exporter-1.0.0.dist-info/RECORD,,

opencode_chat_exporter-1.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (84.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

opencode_chat_exporter-1.0.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+opencode-chat-exporter = chat_exporter.cli:main

opencode_chat_exporter-1.0.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 nasro2123
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

opencode_chat_exporter-1.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+chat_exporter