import json
import tempfile
import unittest
from pathlib import Path

import aite
import loop
import seiri
import yarukoto


class HumanFeaturesTest(unittest.TestCase):
    def test_loop_accepts_nfkc_and_japanese_units(self):
        self.assertEqual(loop.parse("/loop　３０　新しい技術を作る"),
                         {"topic": "新しい技術を作る", "interval": 1800})
        self.assertEqual(loop.parse("/loop 30分 お題")["interval"], 1800)
        self.assertEqual(loop.parse("/loop 1時間 お題")["interval"], 3600)
        self.assertEqual(loop.parse("/loop 90秒 お題")["interval"], 90)
        self.assertEqual(loop.parse("/loop 30m お題")["interval"], 1800)

    def test_loop_start_message_explains_schedule(self):
        self.assertEqual(loop.start_message(1800, "新しい技術", hour=19),
                         "わかりました。30分ごとに『新しい技術』を最大20周やります。1周目は今から。")
        self.assertIn("7時から", loop.start_message(3600, "題", hour=2))

    def test_aite_is_private_bounded_and_deletable(self):
        with tempfile.TemporaryDirectory() as temp:
            self.assertTrue(aite.capture("返事は短くお願いします", temp))
            self.assertEqual(len(aite.list_items(temp)), 1)
            self.assertFalse(aite.add("APIキーは秘密です", folder=temp))
            removed = aite.delete(0, temp)
            self.assertIn("返事は短く", removed["文"])
            self.assertEqual(aite.list_items(temp), [])

    def test_seiri_reviews_cards_and_writes_diary_without_llm(self):
        with tempfile.TemporaryDirectory() as temp:
            base = Path(temp)
            (base / "kyoukun.json").write_text(json.dumps([
                {"知らせ": "同じ教訓", "強さ": 4, "次に見直す日": "2020-01-01"},
                {"知らせ": "同じ教訓", "強さ": 4, "次に見直す日": "2020-01-01"}]), encoding="utf-8")
            result = seiri.run(folder=base, now=1791158400)
            cards = json.loads((base / "kyoukun.json").read_text(encoding="utf-8"))
            self.assertEqual(len(cards), 1)
            self.assertEqual(result["思い出した数"], 2)
            self.assertTrue((base / "nikki.jsonl").exists())
            self.assertTrue((base / "seichou.json").exists())

    def test_seiri_accepts_a_replaceable_bounded_summarizer(self):
        with tempfile.TemporaryDirectory() as temp:
            calls = []
            def ask(prompt):
                calls.append(prompt)
                return {"相手": [{"文": "返事は短く"}]}
            result = seiri.run(folder=Path(temp), recent=[{"役": "user", "文": "これから整理"}], ask=ask)
            self.assertEqual(len(calls), 1)
            self.assertEqual(result["相手の数"], 1)
            self.assertIn("返事は短く", aite.list_items(temp)[0]["文"])

    def test_yarukoto_respects_hourly_limit(self):
        with tempfile.TemporaryDirectory() as temp:
            base = Path(temp)
            (base / "state.json").write_text(json.dumps({"興味": [{"テーマ": "計算", "題": "計算", "問い": "なぜ?"}]}), encoding="utf-8")
            first = yarukoto.run_once(folder=base, now=100000)
            self.assertEqual(first["種類"], "問いに答える")
            self.assertIsNone(yarukoto.run_once(folder=base, now=100001))
            self.assertEqual(len((base / "yarukoto.jsonl").read_text(encoding="utf-8").splitlines()), 1)


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