agent-dump 0.1.0__py3-none-any.whl

agent_dump/__init__.py

@@ -0,0 +1,9 @@
+"""
+Agent Dump - AI Coding Assistant Session Export Tool
+"""
+
+__version__ = "0.1.0"
+__all__ = ["find_db_path", "get_recent_sessions", "export_session", "export_sessions"]
+
+from agent_dump.db import find_db_path, get_recent_sessions
+from agent_dump.exporter import export_session, export_sessions

agent_dump/__main__.py

@@ -0,0 +1,8 @@
+"""
+Entry point for python -m agent_dump
+"""
+
+from agent_dump.cli import main
+
+if __name__ == "__main__":
+    main()

agent_dump/cli.py

@@ -0,0 +1,91 @@
+"""
+Command-line interface for agent-dump
+"""
+
+import argparse
+from pathlib import Path
+
+from agent_dump.db import find_db_path, get_recent_sessions
+from agent_dump.exporter import export_sessions
+from agent_dump.selector import select_sessions_interactive
+
+
+def main():
+    """Main entry point"""
+    parser = argparse.ArgumentParser(description="Export agent sessions to JSON")
+    parser.add_argument("--days", type=int, default=7, help="Number of days to look back (default: 7)")
+    parser.add_argument(
+        "--agent",
+        type=str,
+        default="opencode",
+        help="Agent tool name (default: opencode)",
+    )
+    parser.add_argument(
+        "--output",
+        type=str,
+        default="./sessions",
+        help="Output base directory (default: ./sessions)",
+    )
+    parser.add_argument(
+        "--export",
+        type=str,
+        metavar="IDS",
+        help="Export specific session IDs (comma-separated)",
+    )
+    parser.add_argument("--list", action="store_true", help="List sessions without exporting")
+    args = parser.parse_args()
+
+    print(f"🔍 {args.agent.title()} Session Exporter\n")
+
+    # Find database
+    try:
+        db_path = find_db_path()
+        print(f"📁 Database: {db_path}\n")
+    except FileNotFoundError as e:
+        print(f"❌ Error: {e}")
+        return
+
+    # Get recent sessions
+    print(f"📊 Loading sessions from the last {args.days} days...")
+    sessions = get_recent_sessions(db_path, days=args.days)
+    print(f"✓ Found {len(sessions)} sessions\n")
+
+    if not sessions:
+        print("No sessions found.")
+        return
+
+    # List mode
+    if args.list:
+        print("Available sessions:")
+        print("-" * 80)
+        for i, session in enumerate(sessions, 1):
+            print(f"{i}. {session['title']}")
+            print(f"   Time: {session['created_formatted']}")
+            print(f"   ID: {session['id']}")
+            print()
+        return
+
+    # Export specific IDs
+    if args.export:
+        target_ids = [sid.strip() for sid in args.export.split(",")]
+        selected = [s for s in sessions if s["id"] in target_ids]
+        if not selected:
+            print(f"❌ No sessions found with IDs: {args.export}")
+            return
+        print(f"✓ Selected {len(selected)} session(s) by ID\n")
+    else:
+        # Interactive selection
+        selected = select_sessions_interactive(sessions)
+        if not selected:
+            print("\n⚠️  No sessions selected. Exiting.")
+            return
+        print(f"\n✓ Selected {len(selected)} session(s)\n")
+
+    # Export
+    output_dir = Path(args.output) / args.agent
+    exported = export_sessions(db_path, selected, output_dir)
+    print(f"\n✅ Successfully exported {len(exported)} session(s) to {output_dir}/")
+
+
+if __name__ == "__main__":
+    main()

agent_dump/db.py

@@ -0,0 +1,74 @@
+"""
+Database operations for agent session export
+"""
+
+from datetime import datetime, timedelta
+import os
+from pathlib import Path
+import sqlite3
+from typing import Any
+
+
+def find_db_path() -> Path:
+    """Find the OpenCode database path"""
+    paths = [
+        os.path.expanduser("data/opencode/opencode.db"),
+        os.path.expanduser("~/.local/share/opencode/opencode.db"),
+    ]
+
+    for path in paths:
+        if os.path.exists(path):
+            return Path(path)
+
+    raise FileNotFoundError("Could not find opencode.db database")
+
+
+def get_recent_sessions(db_path: Path, days: int = 7) -> list[dict[str, Any]]:
+    """Get sessions from the last N days"""
+    conn = sqlite3.connect(db_path)
+    conn.row_factory = sqlite3.Row
+    cursor = conn.cursor()
+
+    # Calculate timestamp for N days ago (milliseconds)
+    cutoff_time = int((datetime.now() - timedelta(days=days)).timestamp() * 1000)
+
+    # Query sessions with basic info
+    cursor.execute(
+        """
+        SELECT 
+            s.id,
+            s.title,
+            s.time_created,
+            s.time_updated,
+            s.slug,
+            s.directory,
+            s.version,
+            s.summary_files
+        FROM session s
+        WHERE s.time_created >= ?
+        ORDER BY s.time_created DESC
+        """,
+        (cutoff_time,),
+    )
+
+    sessions = []
+    for row in cursor.fetchall():
+        # Convert timestamp to readable format
+        created_dt = datetime.fromtimestamp(row["time_created"] / 1000)
+
+        sessions.append(
+            {
+                "id": row["id"],
+                "title": row["title"],
+                "time_created": row["time_created"],
+                "time_updated": row["time_updated"],
+                "created_formatted": created_dt.strftime("%Y-%m-%d %H:%M:%S"),
+                "slug": row["slug"],
+                "directory": row["directory"],
+                "version": row["version"],
+                "summary_files": row["summary_files"],
+            }
+        )
+
+    conn.close()
+    return sessions

agent_dump/exporter.py

@@ -0,0 +1,126 @@
+"""
+Session export functionality
+"""
+
+import json
+from pathlib import Path
+import sqlite3
+from typing import Any
+
+
+def export_session(db_path: Path, session: dict[str, Any], output_dir: Path) -> Path:
+    """Export a single session to JSON"""
+    conn = sqlite3.connect(db_path)
+    conn.row_factory = sqlite3.Row
+    cursor = conn.cursor()
+
+    # Build session data
+    session_data = {
+        "id": session["id"],
+        "title": session["title"],
+        "slug": session["slug"],
+        "directory": session["directory"],
+        "version": session["version"],
+        "time_created": session["time_created"],
+        "time_updated": session["time_updated"],
+        "summary_files": session["summary_files"],
+        "stats": {
+            "total_cost": 0,
+            "total_input_tokens": 0,
+            "total_output_tokens": 0,
+            "message_count": 0,
+        },
+        "messages": [],
+    }
+
+    # Get messages for this session
+    cursor.execute(
+        """
+        SELECT * FROM message
+        WHERE session_id = ?
+        ORDER BY time_created ASC
+        """,
+        (session["id"],),
+    )
+
+    for msg_row in cursor.fetchall():
+        msg_data = json.loads(msg_row["data"])
+
+        message = {
+            "id": msg_row["id"],
+            "role": msg_data.get("role", "unknown"),
+            "agent": msg_data.get("agent"),
+            "mode": msg_data.get("mode"),
+            "model": msg_data.get("modelID"),
+            "provider": msg_data.get("providerID"),
+            "time_created": msg_row["time_created"],
+            "time_completed": msg_data.get("time", {}).get("completed"),
+            "tokens": msg_data.get("tokens", {}),
+            "cost": msg_data.get("cost", 0),
+            "parts": [],
+        }
+
+        # Update session stats
+        session_data["stats"]["message_count"] += 1
+        if message["cost"]:
+            session_data["stats"]["total_cost"] += message["cost"]
+        tokens = message["tokens"] or {}
+        session_data["stats"]["total_input_tokens"] += tokens.get("input", 0)
+        session_data["stats"]["total_output_tokens"] += tokens.get("output", 0)
+
+        # Get parts for this message
+        cursor.execute(
+            """
+            SELECT * FROM part
+            WHERE message_id = ?
+            ORDER BY time_created ASC
+            """,
+            (msg_row["id"],),
+        )
+
+        for part_row in cursor.fetchall():
+            part_data = json.loads(part_row["data"])
+            part = {
+                "type": part_data.get("type"),
+                "time_created": part_row["time_created"],
+            }
+
+            if part["type"] == "text" or part["type"] == "reasoning":
+                part["text"] = part_data.get("text", "")
+            elif part["type"] == "tool":
+                part["tool"] = part_data.get("tool")
+                part["callID"] = part_data.get("callID")
+                part["title"] = part_data.get("title", "")
+                part["state"] = part_data.get("state", {})
+            elif part["type"] in ["step-start", "step-finish"]:
+                part["reason"] = part_data.get("reason")
+                part["tokens"] = part_data.get("tokens")
+                part["cost"] = part_data.get("cost")
+
+            message["parts"].append(part)
+
+        session_data["messages"].append(message)
+
+    conn.close()
+
+    # Save to file
+    output_path = output_dir / f"{session['id']}.json"
+    with open(output_path, "w", encoding="utf-8") as f:
+        json.dump(session_data, f, ensure_ascii=False, indent=2)
+
+    return output_path
+
+
+def export_sessions(db_path: Path, sessions: list[dict[str, Any]], output_dir: Path) -> list[Path]:
+    """Export multiple sessions"""
+    output_dir = Path(output_dir)
+    output_dir.mkdir(parents=True, exist_ok=True)
+
+    print("📤 Exporting sessions...")
+    exported = []
+    for session in sessions:
+        output_path = export_session(db_path, session, output_dir)
+        exported.append(output_path)
+        print(f"  ✓ {session['title'][:50]}... → {output_path.name}")
+
+    return exported

agent_dump/selector.py

@@ -0,0 +1,75 @@
+"""
+Session selection utilities
+"""
+
+import sys
+from typing import Any
+
+import questionary
+
+
+def is_terminal() -> bool:
+    """Check if running in a terminal"""
+    return sys.stdin.isatty() and sys.stdout.isatty()
+
+
+def select_sessions_interactive(sessions: list[dict[str, Any]]) -> list[dict[str, Any]]:
+    """Let user select sessions interactively"""
+    if not sessions:
+        print("No sessions found in the specified time range.")
+        return []
+
+    # If not in terminal, use simple selection
+    if not is_terminal():
+        return select_sessions_simple(sessions)
+
+    # Prepare choices for questionary
+    choices = []
+    for session in sessions:
+        # Format: Title (Date) - ID
+        label = f"{session['title'][:60]}{'...' if len(session['title']) > 60 else ''}"
+        description = f"{session['created_formatted']} | {session['id']}"
+
+        choices.append(questionary.Choice(title=label, value=session, description=description))
+
+    # Show interactive checkbox
+    selected = questionary.checkbox(
+        "选择要导出的会话 (空格选择/取消, 回车确认):",
+        choices=choices,
+        instruction="\n使用 ↑↓ 移动, 空格 选择/取消, 回车 确认导出",
+    ).ask()
+
+    return selected or []
+
+
+def select_sessions_simple(sessions: list[dict[str, Any]]) -> list[dict[str, Any]]:
+    """Simple selection for non-terminal environments"""
+    print("Available sessions:")
+    print("-" * 80)
+    for i, session in enumerate(sessions, 1):
+        print(f"{i}. {session['title'][:60]}")
+        print(f"   {session['created_formatted']} | {session['id']}")
+        print()
+
+    print("Enter session numbers to export (comma-separated, e.g., '1,3,5' or 'all'):")
+    try:
+        selection = input("> ").strip()
+    except EOFError:
+        print("\n⚠️  No input provided. Exiting.")
+        return []
+
+    if selection.lower() == "all":
+        return sessions
+
+    try:
+        indices = [int(x.strip()) - 1 for x in selection.split(",")]
+        selected = []
+        for idx in indices:
+            if 0 <= idx < len(sessions):
+                selected.append(sessions[idx])
+            else:
+                print(f"⚠️  Invalid selection: {idx + 1}")
+        return selected
+    except ValueError:
+        print("⚠️  Invalid input. Please enter numbers separated by commas.")
+        return []

agent_dump-0.1.0.dist-info/METADATA

@@ -0,0 +1,165 @@
+Metadata-Version: 2.4
+Name: agent-dump
+Version: 0.1.0
+Summary: AI Coding Assistant Session Export Tool
+Project-URL: Homepage, https://github.com/xingkaixin/agent-dump
+Project-URL: Repository, https://github.com/xingkaixin/agent-dump
+Project-URL: Issues, https://github.com/xingkaixin/agent-dump/issues
+Author-email: XingKaiXin <xingkaixin@gmail.com>
+License-Expression: MIT
+License-File: LICENSE
+Keywords: ai,chat,cli,export,opencode
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Topic :: Utilities
+Requires-Python: >=3.14
+Requires-Dist: prompt-toolkit>=3.0.0
+Requires-Dist: questionary>=2.1.1
+Description-Content-Type: text/markdown
+
+# Agent Dump
+
+AI 编码助手会话导出工具 - 支持从多种 AI 编码工具的会话数据导出会话为 JSON 格式。
+
+## 支持的 AI 工具
+
+- **OpenCode** - 开源 AI 编程助手
+- **Claude Code** - Anthropic 的 AI 编码工具 *(计划中)*
+- **Code X** - GitHub Copilot Chat *(计划中)*
+- **更多工具** - 欢迎提交 PR 支持其他 AI 编码工具
+
+## 功能特性
+
+- **交互式选择**: 使用 questionary 提供友好的命令行交互界面
+- **批量导出**: 支持导出最近 N 天的所有会话
+- **指定导出**: 通过会话 ID 导出特定会话
+- **会话列表**: 仅列出会话而不导出
+- **统计数据**: 导出包含 tokens 使用量、成本等统计信息
+- **消息详情**: 完整保留会话消息、工具调用等详细信息
+
+## 安装
+
+### 方式一：使用 uv tool 安装（推荐）
+
+```bash
+# 从 PyPI 安装（发布后可使用）
+uv tool install agent-dump
+
+# 从 GitHub 直接安装
+uv tool install git+https://github.com/xingkaixin/agent-dump
+```
+
+### 方式二：使用 uvx 直接运行（无需安装）
+
+```bash
+# 从 PyPI 运行（发布后可使用）
+uvx agent-dump --help
+
+# 从 GitHub 直接运行
+uvx --from git+https://github.com/xingkaixin/agent-dump agent-dump --help
+```
+
+### 方式三：本地开发
+
+```bash
+# 克隆仓库
+git clone https://github.com/xingkaixin/agent-dump.git
+cd agent-dump
+
+# 使用 uv 安装依赖
+uv sync
+
+# 本地安装测试
+uv tool install . --force
+```
+
+## 使用方法
+
+### 交互式导出（默认）
+
+```bash
+# 方式一：使用命令行入口
+uv run agent-dump
+
+# 方式二：使用模块运行
+uv run python -m agent_dump
+```
+
+运行后会显示最近 7 天的会话列表，使用空格选择/取消，回车确认导出。
+
+### 命令行参数
+
+```bash
+uv run agent-dump --days 3                    # 导出最近 3 天的会话
+uv run agent-dump --agent claude              # 指定 Agent 工具名称
+uv run agent-dump --output ./my-sessions      # 指定输出目录
+uv run agent-dump --list                      # 仅列出会话
+uv run agent-dump --export ses_abc,ses_xyz    # 导出指定 ID 的会话
+```
+
+### 完整参数说明
+
+| 参数 | 说明 | 默认值 |
+|------|------|--------|
+| `--days` | 查询最近 N 天的会话 | 7 |
+| `--agent` | Agent 工具名称 | opencode |
+| `--output` | 输出目录 | ./sessions |
+| `--export` | 导出指定会话 ID（逗号分隔） | - |
+| `--list` | 仅列出会话，不导出 | - |
+
+## 项目结构
+
+```
+.
+├── src/
+│   └── agent_dump/      # 主包目录
+│       ├── __init__.py  # 包初始化
+│       ├── __main__.py  # python -m agent_dump 入口
+│       ├── cli.py       # 命令行接口
+│       ├── db.py        # 数据库操作
+│       ├── exporter.py  # 导出逻辑
+│       └── selector.py  # 交互式选择
+├── tests/               # 测试目录
+├── pyproject.toml       # 项目配置
+├── Makefile            # 自动化命令
+├── ruff.toml           # 代码风格配置
+├── data/               # 数据库目录
+│   └── opencode/
+│       └── opencode.db
+└── sessions/           # 导出目录
+    └── {agent-name}/   # 按工具分类的导出文件
+        └── ses_xxx.json
+```
+
+## 开发
+
+```bash
+# 代码检查
+make lint
+
+# 自动修复
+make lint.fix
+
+# 代码格式化
+make lint.fmt
+
+# 类型检查
+make check
+```
+
+## 依赖
+
+- Python >= 3.14
+- prompt-toolkit >= 3.0.0
+- questionary >= 2.1.1
+- ruff >= 0.15.2 (开发)
+- ty >= 0.0.18 (开发)
+
+## 许可证
+
+MIT

agent_dump-0.1.0.dist-info/RECORD

@@ -0,0 +1,11 @@
+agent_dump/__init__.py,sha256=CLmupjtbZNOU8E8zFMlm1qO7rffivPhP2AEH7_aDtN0,296
+agent_dump/__main__.py,sha256=ytB3s7vhkD3unQ__hRErh1gPHHSGFvElzphEguPyFSg,117
+agent_dump/cli.py,sha256=VHOuCfmXMwu41J74C3arN0Gdjgux7OXJA70lOqC96I4,2814
+agent_dump/db.py,sha256=Cz8C2TYlUbtoBgQWE-RgXQNIrflm2UySZGZ8v83Nfwg,2028
+agent_dump/exporter.py,sha256=Ks_myYHe79OHkkPe-byssoSgsepLd7NZFwGRTjtmDdU,4071
+agent_dump/selector.py,sha256=E93js3DDuCqsTQaYK5dz_BE8HBrMgH8eHADdYGVB7aQ,2368
+agent_dump-0.1.0.dist-info/METADATA,sha256=LxPPr9WRtOEl5nG2rgzqZ3O2-MW5NuBe_vSC4fV2iXs,4496
+agent_dump-0.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+agent_dump-0.1.0.dist-info/entry_points.txt,sha256=uiaMVYf0AejYHNGi_nJDsveFMk5rwWd--3ugoKq8f4o,51
+agent_dump-0.1.0.dist-info/licenses/LICENSE,sha256=0sQoYHhVUn3z4TLBQaZI1eBnejacFvxuAadX3YvnoEw,1081
+agent_dump-0.1.0.dist-info/RECORD,,

agent_dump-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

agent_dump-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+agent-dump = agent_dump.cli:main

agent_dump-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 XingKaiXin contributors
+
+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, standard 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.