--json option for saving session JSON

For every command that supports --gist add a --json option, if you pass --json
then the .json version of the session that was retrieved from the API is
included in the output directory (or temporary output directory) and teh path
to that JSON file is shown in the output, along with its size in KB

https://gistpreview.github.io/?5e11fdc8622af967450f790f052042c1

Author: simonw

README.md

@@ -40,6 +40,7 @@ This will generate:
 - `-o, --output DIRECTORY` - output directory (default: current directory)
 - `--repo OWNER/NAME` - GitHub repo for commit links (auto-detected from git push output if not specified)
 - `--gist` - upload the generated HTML files to a GitHub Gist and output a preview URL
+- `--json` - include the original JSON session file in the output directory
 
 ### Publishing to GitHub Gist
 
@@ -74,6 +75,21 @@ claude-code-publish session.json -o ./my-transcript --gist
 
 **Requirements:** The `--gist` option requires the [GitHub CLI](https://cli.github.com/) (`gh`) to be installed and authenticated (`gh auth login`).
 
+### Including the JSON source
+
+Use the `--json` option to include the original session JSON file in the output directory:
+
+```bash
+claude-code-publish session.json -o ./my-transcript --json
+```
+
+This will output:
+```
+JSON: ./my-transcript/session_ABC.json (245.3 KB)
+```
+
+The JSON file preserves its original filename. This is useful for archiving the source data alongside the HTML output.
+
 ## Importing from Claude API
 
 You can import sessions directly from the Claude API without needing to export a `session.json` file:
@@ -90,10 +106,15 @@ claude-code-publish import
 
 # Import and publish to gist
 claude-code-publish import SESSION_ID --gist
+
+# Import and save the JSON session data
+claude-code-publish import SESSION_ID --json
 ```
 
 On macOS, the API credentials are automatically retrieved from your keychain (requires being logged into Claude Code). On other platforms, provide `--token` and `--org-uuid` manually.
 
+The `--json` option for the import command saves the session data fetched from the API as `{session_id}.json` in the output directory.
+
 ## Development
 
 To contribute to this tool, first checkout the code. You can run the tests using `uv run`:

src/claude_code_publish/__init__.py

@@ -5,6 +5,7 @@
 import os
 import platform
 import re
+import shutil
 import subprocess
 import tempfile
 from pathlib import Path
@@ -640,8 +641,7 @@ def inject_gist_preview_js(output_dir):
         # Insert the gist preview JS before the closing </body> tag
         if "</body>" in content:
             content = content.replace(
-                "</body>",
-                f"<script>{GIST_PREVIEW_JS}</script>\n</body>"
+                "</body>", f"<script>{GIST_PREVIEW_JS}</script>\n</body>"
             )
             html_file.write_text(content)
 
@@ -931,7 +931,13 @@ def cli():
     is_flag=True,
     help="Upload to GitHub Gist and output a gistpreview.github.io URL.",
 )
-def session(json_file, output, repo, gist):
+@click.option(
+    "--json",
+    "include_json",
+    is_flag=True,
+    help="Include the original JSON session file in the output directory.",
+)
+def session(json_file, output, repo, gist, include_json):
     """Convert a Claude Code session JSON file to HTML."""
     # Determine output directory
     if gist and output is None:
@@ -943,8 +949,18 @@ def session(json_file, output, repo, gist):
     elif output is None:
         output = "."
 
+    output = Path(output)
     generate_html(json_file, output, github_repo=repo)
 
+    # Copy JSON file to output directory if requested
+    if include_json:
+        output.mkdir(exist_ok=True)
+        json_source = Path(json_file)
+        json_dest = output / json_source.name
+        shutil.copy(json_file, json_dest)
+        json_size_kb = json_dest.stat().st_size / 1024
+        click.echo(f"JSON: {json_dest} ({json_size_kb:.1f} KB)")
+
     if gist:
         # Inject gist preview JS and create gist
         inject_gist_preview_js(output)
@@ -1233,7 +1249,13 @@ def generate_html_from_session_data(session_data, output_dir, github_repo=None):
     is_flag=True,
     help="Upload to GitHub Gist and output a gistpreview.github.io URL.",
 )
-def import_session(session_id, output, token, org_uuid, repo, gist):
+@click.option(
+    "--json",
+    "include_json",
+    is_flag=True,
+    help="Include the JSON session data in the output directory.",
+)
+def import_session(session_id, output, token, org_uuid, repo, gist, include_json):
     """Import a session from the Claude API and convert to HTML.
 
     If SESSION_ID is not provided, displays an interactive picker to select a session.
@@ -1298,9 +1320,19 @@ def import_session(session_id, output, token, org_uuid, repo, gist):
     elif output is None:
         output = session_id
 
+    output = Path(output)
     click.echo(f"Generating HTML in {output}/...")
     generate_html_from_session_data(session_data, output, github_repo=repo)
 
+    # Save JSON session data if requested
+    if include_json:
+        output.mkdir(exist_ok=True)
+        json_dest = output / f"{session_id}.json"
+        with open(json_dest, "w") as f:
+            json.dump(session_data, f, indent=2)
+        json_size_kb = json_dest.stat().st_size / 1024
+        click.echo(f"JSON: {json_dest} ({json_size_kb:.1f} KB)")
+
     if gist:
         # Inject gist preview JS and create gist
         inject_gist_preview_js(output)

tests/test_generate_html.py

@@ -577,7 +577,9 @@ def mock_run(*args, **kwargs):
         monkeypatch.setattr(subprocess, "run", mock_run)
 
         # Mock tempfile.gettempdir to use our tmp_path
-        monkeypatch.setattr("claude_code_publish.tempfile.gettempdir", lambda: str(tmp_path))
+        monkeypatch.setattr(
+            "claude_code_publish.tempfile.gettempdir", lambda: str(tmp_path)
+        )
 
         runner = CliRunner()
         result = runner.invoke(
@@ -624,6 +626,93 @@ def mock_run(*args, **kwargs):
         assert "gistpreview.github.io" in index_content
 
 
+class TestSessionJsonOption:
+    """Tests for the session command --json option."""
+
+    def test_session_json_copies_file(self, output_dir):
+        """Test that session --json copies the JSON file to output."""
+        from click.testing import CliRunner
+        from claude_code_publish import cli
+
+        fixture_path = Path(__file__).parent / "sample_session.json"
+
+        runner = CliRunner()
+        result = runner.invoke(
+            cli,
+            ["session", str(fixture_path), "-o", str(output_dir), "--json"],
+        )
+
+        assert result.exit_code == 0
+        json_file = output_dir / "sample_session.json"
+        assert json_file.exists()
+        assert "JSON:" in result.output
+        assert "KB" in result.output
+
+    def test_session_json_preserves_original_name(self, output_dir):
+        """Test that --json preserves the original filename."""
+        from click.testing import CliRunner
+        from claude_code_publish import cli
+
+        fixture_path = Path(__file__).parent / "sample_session.json"
+
+        runner = CliRunner()
+        result = runner.invoke(
+            cli,
+            ["session", str(fixture_path), "-o", str(output_dir), "--json"],
+        )
+
+        assert result.exit_code == 0
+        # Should use original filename, not "session.json"
+        assert (output_dir / "sample_session.json").exists()
+        assert not (output_dir / "session.json").exists()
+
+
+class TestImportJsonOption:
+    """Tests for the import command --json option."""
+
+    def test_import_json_saves_session_data(self, httpx_mock, output_dir):
+        """Test that import --json saves the session JSON."""
+        from click.testing import CliRunner
+        from claude_code_publish import cli
+
+        # Load sample session to mock API response
+        fixture_path = Path(__file__).parent / "sample_session.json"
+        with open(fixture_path) as f:
+            session_data = json.load(f)
+
+        httpx_mock.add_response(
+            url="https://api.anthropic.com/v1/session_ingress/session/test-session-id",
+            json=session_data,
+        )
+
+        runner = CliRunner()
+        result = runner.invoke(
+            cli,
+            [
+                "import",
+                "test-session-id",
+                "--token",
+                "test-token",
+                "--org-uuid",
+                "test-org",
+                "-o",
+                str(output_dir),
+                "--json",
+            ],
+        )
+
+        assert result.exit_code == 0
+        json_file = output_dir / "test-session-id.json"
+        assert json_file.exists()
+        assert "JSON:" in result.output
+        assert "KB" in result.output
+
+        # Verify JSON content is valid
+        with open(json_file) as f:
+            saved_data = json.load(f)
+        assert saved_data == session_data
+
+
 class TestImportGistOption:
     """Tests for the import command --gist option."""
 
@@ -657,7 +746,9 @@ def mock_run(*args, **kwargs):
         monkeypatch.setattr(subprocess, "run", mock_run)
 
         # Mock tempfile.gettempdir
-        monkeypatch.setattr("claude_code_publish.tempfile.gettempdir", lambda: str(tmp_path))
+        monkeypatch.setattr(
+            "claude_code_publish.tempfile.gettempdir", lambda: str(tmp_path)
+        )
 
         runner = CliRunner()
         result = runner.invoke(