from __future__ import annotations

from pathlib import Path
import sys
from typing import List, Optional

# Support both package imports and direct script execution.
if __package__ in (None, ""):
    sys.path.insert(0, str(Path(__file__).resolve().parent.parent))

from core_agent.compression import RollingContextCompressor
from core_agent.memory import SimpleMemoryStore
from core_agent.providers.base import AgentProvider
from core_agent.session import AgentRunResult, ConversationSession


class CoreAgent:
    """Factory-style facade around a reusable multi-turn conversation session."""

    def __init__(
        self,
        *,
        provider: AgentProvider,
        workspace: str | Path,
        skill_dirs: Optional[List[str | Path]] = None,
        tool_registry=None,
        dispatcher=None,
        system_prompt: Optional[str] = None,
        max_iterations: int = 12,
        max_history_turns: int = 5,
        memory_store: Optional[SimpleMemoryStore] = None,
        context_compressor: Optional[RollingContextCompressor] = None,
    ) -> None:
        self.provider = provider
        self.workspace = Path(workspace).resolve()
        self.skill_dirs = list(skill_dirs or [self.workspace / "skills"])
        self.tool_registry = tool_registry
        self.dispatcher = dispatcher
        self.system_prompt = system_prompt
        self.max_iterations = max_iterations
        self.max_history_turns = max_history_turns
        self.memory_store = memory_store or SimpleMemoryStore(self.workspace / ".core_agent" / "memory.json")
        self.context_compressor = context_compressor or RollingContextCompressor()

    def run(self, user_message: str, *, active_skills: Optional[List[str]] = None) -> AgentRunResult:
        session = self.new_session(active_skills=active_skills)
        return session.ask(user_message)

    def new_session(self, *, active_skills: Optional[List[str]] = None) -> ConversationSession:
        return ConversationSession(
            provider=self.provider,
            workspace=self.workspace,
            skill_dirs=self.skill_dirs,
            tool_registry=self.tool_registry,
            dispatcher=self.dispatcher,
            system_prompt=self.system_prompt,
            max_iterations=self.max_iterations,
            max_history_turns=self.max_history_turns,
            memory_store=self.memory_store,
            context_compressor=self.context_compressor,
            active_skills=active_skills,
        )


if __name__ == "__main__":
    from core_agent.chat_cli import main

    main()