#!/usr/bin/env python3
"""Select a full SDK Session's permission preset and continue without resetting it."""

from __future__ import annotations

import argparse
import os
from pathlib import Path

from deepseek_harness import DeepSeekHarness, RunResult


def display(result: RunResult) -> None:
    """Print the input's result and stop on a reported failure."""
    if result.finish_reason != "completed":
        raise RuntimeError(f"Input ended with {result.finish_reason}: {result.final_response}")
    print(result.final_response)


def main() -> None:
    """Send an explicit permission command, then run another prompt in the same Session."""
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("--dsh-home", required=True, type=Path)
    parser.add_argument("--workspace", type=Path, default=Path.cwd())
    parser.add_argument("--preset", required=True)
    parser.add_argument("--provider", default="deepseek-official")
    parser.add_argument("--model", default=os.environ.get("DSH_MODEL", "deepseek-v4-flash"))
    args = parser.parse_args()
    if not args.preset.strip():
        parser.error("--preset must be non-blank")

    with DeepSeekHarness(
        dsh_home=str(args.dsh_home.resolve()), cwd=str(args.workspace.resolve()),
        profile="sdk", provider=args.provider, model=args.model,
        env={"DSH_PERMISSION_MODE": "workspace-write"},
    ) as harness:
        session = harness.start_session()
        display(session.run("Summarize the current file and approval policy in one sentence. Do not use tools."))
        display(session.run(f"/permission {args.preset}"))
        print(f"Launch setting: {harness.config.env['DSH_PERMISSION_MODE']}")
        display(session.run("Summarize the current file and approval policy again. Do not use tools."))


if __name__ == "__main__":
    main()