import io
import json
import sys
import tempfile
import unittest
from pathlib import Path
from unittest.mock import patch

ROOT = Path(__file__).resolve().parents[1]
SRC = ROOT / "src"
if str(SRC) not in sys.path:
    sys.path.insert(0, str(SRC))

import aman
from diagnostics import DiagnosticCheck, DiagnosticReport


class AmanCliTests(unittest.TestCase):
    def test_parse_cli_args_defaults_to_run_command(self):
        args = aman._parse_cli_args(["--dry-run"])

        self.assertEqual(args.command, "run")
        self.assertTrue(args.dry_run)

    def test_parse_cli_args_doctor_command(self):
        args = aman._parse_cli_args(["doctor", "--json"])

        self.assertEqual(args.command, "doctor")
        self.assertTrue(args.json)

    def test_doctor_command_json_output_and_exit_code(self):
        report = DiagnosticReport(
            checks=[DiagnosticCheck(id="config.load", ok=True, message="ok", hint="")]
        )
        args = aman._parse_cli_args(["doctor", "--json"])
        out = io.StringIO()
        with patch("aman.run_diagnostics", return_value=report), patch("sys.stdout", out):
            exit_code = aman._doctor_command(args)

        self.assertEqual(exit_code, 0)
        payload = json.loads(out.getvalue())
        self.assertTrue(payload["ok"])
        self.assertEqual(payload["checks"][0]["id"], "config.load")

    def test_doctor_command_failed_report_returns_exit_code_2(self):
        report = DiagnosticReport(
            checks=[DiagnosticCheck(id="config.load", ok=False, message="broken", hint="fix")]
        )
        args = aman._parse_cli_args(["doctor"])
        out = io.StringIO()
        with patch("aman.run_diagnostics", return_value=report), patch("sys.stdout", out):
            exit_code = aman._doctor_command(args)

        self.assertEqual(exit_code, 2)
        self.assertIn("[FAIL] config.load", out.getvalue())

    def test_init_command_creates_default_config(self):
        with tempfile.TemporaryDirectory() as td:
            path = Path(td) / "config.json"
            args = aman._parse_cli_args(["init", "--config", str(path)])

            exit_code = aman._init_command(args)
            self.assertEqual(exit_code, 0)
            self.assertTrue(path.exists())
            payload = json.loads(path.read_text(encoding="utf-8"))
            self.assertIn("daemon", payload)

    def test_init_command_refuses_overwrite_without_force(self):
        with tempfile.TemporaryDirectory() as td:
            path = Path(td) / "config.json"
            path.write_text('{"daemon":{"hotkey":"Super+m"}}\n', encoding="utf-8")
            args = aman._parse_cli_args(["init", "--config", str(path)])

            exit_code = aman._init_command(args)
            self.assertEqual(exit_code, 1)
            self.assertIn("Super+m", path.read_text(encoding="utf-8"))

    def test_init_command_force_overwrites_existing_config(self):
        with tempfile.TemporaryDirectory() as td:
            path = Path(td) / "config.json"
            path.write_text('{"daemon":{"hotkey":"Super+m"}}\n', encoding="utf-8")
            args = aman._parse_cli_args(["init", "--config", str(path), "--force"])

            exit_code = aman._init_command(args)
            self.assertEqual(exit_code, 0)
            payload = json.loads(path.read_text(encoding="utf-8"))
            self.assertEqual(payload["daemon"]["hotkey"], "Cmd+m")



if __name__ == "__main__":
    unittest.main()