Extract resolve_output, publish_gist, open_in_browser into commands/__init__.py

Eliminates identical output-resolution, gist-publishing, and browser-opening
blocks duplicated across local, file, and web commands.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: zackman0010

src/claude_code_transcripts/commands/__init__.py

@@ -1,15 +1,60 @@
 """Shared Click option decorators and utilities for commands."""
 
 import shutil
+import tempfile
+import webbrowser
 from pathlib import Path
 
 import click
 
-from claude_code_transcripts.html_generation import Project, Session
+from claude_code_transcripts.html_generation import (
+    Project,
+    Session,
+    create_gist,
+    inject_gist_preview_js,
+)
 from claude_code_transcripts.parser import parse_session_file
 from claude_code_transcripts.sessions import find_all_sessions, find_cowork_sessions
 
 
+def resolve_output(output, output_auto, gist, stem):
+    """Resolve the output directory and whether to auto-open the browser.
+
+    Args:
+        output: Explicit output path string, or None.
+        output_auto: Whether to auto-name the output dir using stem.
+        gist: Whether the gist flag is set (suppresses auto_open).
+        stem: Filename stem used for default naming.
+
+    Returns:
+        (output_path, auto_open) tuple.
+    """
+    auto_open = output is None and not gist and not output_auto
+    if output_auto:
+        parent_dir = Path(output) if output else Path(".")
+        output = parent_dir / stem
+    elif output is None:
+        output = Path(tempfile.gettempdir()) / f"claude-session-{stem}"
+    return Path(output), auto_open
+
+
+def publish_gist(output):
+    """Inject gist preview JS, upload to GitHub Gist, and print URLs."""
+    inject_gist_preview_js(output)
+    click.echo("Creating GitHub gist...")
+    gist_id, gist_url = create_gist(output)
+    preview_url = f"https://gisthost.github.io/?{gist_id}/index.html"
+    click.echo(f"Gist: {gist_url}")
+    click.echo(f"Preview: {preview_url}")
+
+
+def open_in_browser(output, open_browser, auto_open):
+    """Open output/index.html in the default browser if requested."""
+    if open_browser or auto_open:
+        index_url = (output / "index.html").resolve().as_uri()
+        webbrowser.open(index_url)
+
+
 def collect_raw_projects(source, include_agents=False):
     """Discover raw project dicts from the given source.
 

src/claude_code_transcripts/commands/file.py

@@ -1,18 +1,20 @@
 import shutil
 import tempfile
-import webbrowser
 from pathlib import Path
 
 import click
 import httpx
 
 from claude_code_transcripts.cli import cli
-from claude_code_transcripts.commands import interactive_options
+from claude_code_transcripts.commands import (
+    interactive_options,
+    open_in_browser,
+    publish_gist,
+    resolve_output,
+)
 
 from claude_code_transcripts.html_generation import (
-    create_gist,
     generate_html,
-    inject_gist_preview_js,
 )
 from claude_code_transcripts.parser import parse_session_file
 
@@ -70,17 +72,9 @@ def file_cmd(json_file, output, output_auto, repo, gist, include_json, open_brow
             raise click.ClickException(f"File not found: {json_file}")
         url_name = None
 
-    auto_open = output is None and not gist and not output_auto
-    if output_auto:
-        parent_dir = Path(output) if output else Path(".")
-        output = parent_dir / (url_name or json_file_path.stem)
-    elif output is None:
-        output = (
-            Path(tempfile.gettempdir())
-            / f"claude-session-{url_name or json_file_path.stem}"
-        )
+    stem = url_name or json_file_path.stem
+    output, auto_open = resolve_output(output, output_auto, gist, stem)
 
-    output = Path(output)
     generate_html(parse_session_file(json_file_path), output, github_repo=repo)
 
     click.echo(f"Output: {output.resolve()}")
@@ -93,13 +87,6 @@ def file_cmd(json_file, output, output_auto, repo, gist, include_json, open_brow
         click.echo(f"JSON: {json_dest} ({json_size_kb:.1f} KB)")
 
     if gist:
-        inject_gist_preview_js(output)
-        click.echo("Creating GitHub gist...")
-        gist_id, gist_url = create_gist(output)
-        preview_url = f"https://gisthost.github.io/?{gist_id}/index.html"
-        click.echo(f"Gist: {gist_url}")
-        click.echo(f"Preview: {preview_url}")
-
-    if open_browser or auto_open:
-        index_url = (output / "index.html").resolve().as_uri()
-        webbrowser.open(index_url)
+        publish_gist(output)
+
+    open_in_browser(output, open_browser, auto_open)

src/claude_code_transcripts/commands/local.py

@@ -1,19 +1,21 @@
 import shutil
-import tempfile
-import webbrowser
 from datetime import datetime
 from pathlib import Path
 
 import click
 import questionary
 
 from claude_code_transcripts.cli import cli
-from claude_code_transcripts.commands import local_options, source_option
+from claude_code_transcripts.commands import (
+    local_options,
+    open_in_browser,
+    publish_gist,
+    resolve_output,
+    source_option,
+)
 
 from claude_code_transcripts.html_generation import (
-    create_gist,
     generate_html,
-    inject_gist_preview_js,
 )
 from claude_code_transcripts.parser import parse_session_file
 from claude_code_transcripts.sessions import find_cowork_sessions, find_local_sessions
@@ -100,14 +102,8 @@ def local_cmd(
     session_file = selected["session_file"]
     transcript_label = selected["transcript_label"]
 
-    auto_open = output is None and not gist and not output_auto
-    if output_auto:
-        parent_dir = Path(output) if output else Path(".")
-        output = parent_dir / session_file.stem
-    elif output is None:
-        output = Path(tempfile.gettempdir()) / f"claude-session-{session_file.stem}"
+    output, auto_open = resolve_output(output, output_auto, gist, session_file.stem)
 
-    output = Path(output)
     generate_html(
         parse_session_file(session_file),
         output,
@@ -125,13 +121,6 @@ def local_cmd(
         click.echo(f"JSONL: {json_dest} ({json_size_kb:.1f} KB)")
 
     if gist:
-        inject_gist_preview_js(output)
-        click.echo("Creating GitHub gist...")
-        gist_id, gist_url = create_gist(output)
-        preview_url = f"https://gisthost.github.io/?{gist_id}/index.html"
-        click.echo(f"Gist: {gist_url}")
-        click.echo(f"Preview: {preview_url}")
-
-    if open_browser or auto_open:
-        index_url = (output / "index.html").resolve().as_uri()
-        webbrowser.open(index_url)
+        publish_gist(output)
+
+    open_in_browser(output, open_browser, auto_open)

src/claude_code_transcripts/commands/web.py

@@ -1,15 +1,17 @@
 import json
 import platform
-import tempfile
-import webbrowser
-from pathlib import Path
 
 import click
 import httpx
 import questionary
 
 from claude_code_transcripts.cli import cli
-from claude_code_transcripts.commands import interactive_options
+from claude_code_transcripts.commands import (
+    interactive_options,
+    open_in_browser,
+    publish_gist,
+    resolve_output,
+)
 
 from claude_code_transcripts.api import (
     enrich_sessions_with_repos,
@@ -20,9 +22,7 @@
     get_org_uuid_from_config,
 )
 from claude_code_transcripts.html_generation import (
-    create_gist,
     generate_html,
-    inject_gist_preview_js,
 )
 
 
@@ -146,14 +146,8 @@ def web_cmd(
     except httpx.RequestError as e:
         raise click.ClickException(f"Network error: {e}")
 
-    auto_open = output is None and not gist and not output_auto
-    if output_auto:
-        parent_dir = Path(output) if output else Path(".")
-        output = parent_dir / session_id
-    elif output is None:
-        output = Path(tempfile.gettempdir()) / f"claude-session-{session_id}"
+    output, auto_open = resolve_output(output, output_auto, gist, session_id)
 
-    output = Path(output)
     click.echo(f"Generating HTML in {output}/...")
     generate_html(session_data.get("loglines", []), output, github_repo=repo)
 
@@ -168,13 +162,6 @@ def web_cmd(
         click.echo(f"JSON: {json_dest} ({json_size_kb:.1f} KB)")
 
     if gist:
-        inject_gist_preview_js(output)
-        click.echo("Creating GitHub gist...")
-        gist_id, gist_url = create_gist(output)
-        preview_url = f"https://gisthost.github.io/?{gist_id}/index.html"
-        click.echo(f"Gist: {gist_url}")
-        click.echo(f"Preview: {preview_url}")
-
-    if open_browser or auto_open:
-        index_url = (output / "index.html").resolve().as_uri()
-        webbrowser.open(index_url)
+        publish_gist(output)
+
+    open_in_browser(output, open_browser, auto_open)

tests/test_commands_output.py

@@ -0,0 +1,100 @@
+"""Tests for shared output utilities in commands/__init__.py."""
+
+import tempfile
+import webbrowser
+from pathlib import Path
+from unittest.mock import patch, MagicMock
+
+import pytest
+
+from claude_code_transcripts.commands import (
+    open_in_browser,
+    publish_gist,
+    resolve_output,
+)
+
+
+class TestResolveOutput:
+    def test_explicit_output_returned_as_path(self):
+        output, auto_open = resolve_output("/tmp/mydir", False, False, "session123")
+        assert output == Path("/tmp/mydir")
+        assert auto_open is False
+
+    def test_output_auto_uses_stem_under_cwd(self):
+        output, auto_open = resolve_output(None, True, False, "session123")
+        assert output == Path(".") / "session123"
+        assert auto_open is False
+
+    def test_output_auto_with_explicit_parent(self):
+        output, auto_open = resolve_output("/tmp/parent", True, False, "session123")
+        assert output == Path("/tmp/parent") / "session123"
+        assert auto_open is False
+
+    def test_no_output_uses_tempdir(self):
+        output, auto_open = resolve_output(None, False, False, "session123")
+        assert output == Path(tempfile.gettempdir()) / "claude-session-session123"
+        assert auto_open is True
+
+    def test_no_output_with_gist_suppresses_auto_open(self):
+        output, auto_open = resolve_output(None, False, True, "session123")
+        assert auto_open is False
+
+    def test_no_output_with_output_auto_suppresses_auto_open(self):
+        output, auto_open = resolve_output(None, True, False, "session123")
+        assert auto_open is False
+
+
+class TestPublishGist:
+    def test_calls_inject_and_create_gist(self, tmp_path):
+        with (
+            patch(
+                "claude_code_transcripts.commands.inject_gist_preview_js"
+            ) as mock_inject,
+            patch("claude_code_transcripts.commands.create_gist") as mock_create,
+        ):
+            mock_create.return_value = ("abc123", "https://gist.github.com/abc123")
+            publish_gist(tmp_path)
+
+        mock_inject.assert_called_once_with(tmp_path)
+        mock_create.assert_called_once_with(tmp_path)
+
+    def test_prints_gist_and_preview_urls(self, tmp_path, capsys):
+        with (
+            patch("claude_code_transcripts.commands.inject_gist_preview_js"),
+            patch("claude_code_transcripts.commands.create_gist") as mock_create,
+        ):
+            mock_create.return_value = ("abc123", "https://gist.github.com/abc123")
+            publish_gist(tmp_path)
+
+        out = capsys.readouterr().out
+        assert "https://gist.github.com/abc123" in out
+        assert "gisthost.github.io" in out
+        assert "abc123" in out
+
+
+class TestOpenInBrowser:
+    def test_opens_when_open_browser_true(self, tmp_path):
+        (tmp_path / "index.html").write_text("<html/>")
+        with patch("webbrowser.open") as mock_open:
+            open_in_browser(tmp_path, open_browser=True, auto_open=False)
+        mock_open.assert_called_once()
+        assert "index.html" in mock_open.call_args[0][0]
+
+    def test_opens_when_auto_open_true(self, tmp_path):
+        (tmp_path / "index.html").write_text("<html/>")
+        with patch("webbrowser.open") as mock_open:
+            open_in_browser(tmp_path, open_browser=False, auto_open=True)
+        mock_open.assert_called_once()
+
+    def test_does_not_open_when_both_false(self, tmp_path):
+        with patch("webbrowser.open") as mock_open:
+            open_in_browser(tmp_path, open_browser=False, auto_open=False)
+        mock_open.assert_not_called()
+
+    def test_url_points_to_index_html(self, tmp_path):
+        (tmp_path / "index.html").write_text("<html/>")
+        with patch("webbrowser.open") as mock_open:
+            open_in_browser(tmp_path, open_browser=True, auto_open=False)
+        url = mock_open.call_args[0][0]
+        assert url.startswith("file://")
+        assert url.endswith("index.html")