from dataclasses import dataclass
from enum import Enum, StrEnum
from typing import TypedDict


class Tab(StrEnum):
    AGENT = "agent"
    FEATURES = "features"
    SYSTEM = "system"
    SHARING = "sharing"
    DEVELOPER = "developer"


class Section(StrEnum):
    AGENT = "Agent"
    WORK = "Work"
    MEMORY = "Memory"
    PROJECT = "Project"
    CHAT = "Chat"
    SYSTEM = "System"
    SHARING = "Sharing"
    DEVELOPER = "Developer"


class Described(TypedDict):
    key: str
    title: str
    line: str
    section: str
    tab: str
    lead: str


@dataclass(frozen=True)
class Grouping:
    title: str
    line: str
    section: Section
    tab: Tab = Tab.FEATURES
    lead: str = ""


class Group(Enum):
    AGENT = Grouping("Agent", "How the agent works and talks to you", Section.AGENT, Tab.AGENT)
    SESSIONS = Grouping("Sessions", "Agent sessions, subagents and helpers", Section.AGENT, Tab.AGENT)
    LONG_COMMANDS = Grouping(
        "Long commands", "Moves a command that blocks the agent's terminal to the background", Section.AGENT, Tab.AGENT, lead="long_commands"
    )
    LAWS = Grouping(
        "Rules", "Rules the journal gives every agent: how it starts subagents and how much it reads", Section.AGENT, Tab.AGENT, lead="journal_laws"
    )
    MODELS = Grouping("Models", "The model each provider starts agents on", Section.AGENT, Tab.AGENT)
    SKILLS = Grouping("Skills", "When the agent loads skills", Section.AGENT, Tab.AGENT, lead="skill_loading")
    WORK_TRACKING = Grouping(
        "Work tracking", "The agent opens work before it changes files, and logs it on its to-do", Section.WORK, Tab.AGENT, lead="work_tracking"
    )
    PLANS = Grouping("Plans", "How the agent moves through a plan", Section.WORK, Tab.AGENT, lead="plans")
    QUESTIONS = Grouping("Questions", "Choices only you can make are asked as questions", Section.WORK, Tab.AGENT, lead="ask_questions")
    MESSAGES = Grouping("Messages", "How the agent handles the messages you send", Section.WORK, Tab.AGENT, lead="messages")
    VOICE = Grouping("Voice", "How the agent talks to you and what it calls you, in every environment of this project", Section.AGENT, Tab.AGENT)
    SEQUENCES = Grouping("Sequences", "Steps the agent follows in order, in every environment of this project", Section.PROJECT, lead="sequences")
    BOARDS = Grouping("Boards", "Ticket boards and the agents that fill them, in every environment of this project", Section.PROJECT, lead="boards")
    TICKETS = Grouping("Tickets", "Each ticket runs in an environment of its own", Section.PROJECT, lead="tickets")
    INTEGRATIONS = Grouping("Integrations", "Outside services the journal reads from, each off until you switch it on and pick its key", Section.PROJECT)
    SHARED_RECORDS = Grouping("Checks, triggers and templates", "Kept once for the project and used by every environment", Section.PROJECT)
    MEMORY = Grouping("Memory", "Facts, rules and reminders repeated to the agent", Section.MEMORY)
    RECORDS = Grouping("Documents", "Documents, collections, checks and other things the agent files", Section.MEMORY)
    ARCHIVE = Grouping("Archive", "How long closed items and runtime files are kept", Section.MEMORY)
    CHAT = Grouping("Chat", "What the chat shows", Section.CHAT)
    VIEWER = Grouping("Viewer", "What the viewer shows, and when it opens", Section.CHAT)
    UPDATES = Grouping("Updates", "How new versions of the journal are installed in this project", Section.PROJECT, Tab.SYSTEM)
    PROJECT = Grouping("Project", "Settings for the whole project, in every environment", Section.PROJECT, Tab.SYSTEM)
    STOP = Grouping("Journal", "Stop the journal: closes the viewer and every plugin. Nothing is deleted.", Section.SYSTEM, Tab.SYSTEM)
    SHARING = Grouping(
        "Share links",
        "Share one document, report, collection or plan with someone outside the journal. The link opens that item and nothing else.",
        Section.SHARING,
        Tab.SHARING,
        "sharing",
    )
    DEVELOPER = Grouping("Developer", "For working on the journal itself", Section.DEVELOPER, Tab.DEVELOPER)

    @property
    def key(self) -> str:
        return self.name.lower()

    def describe(self) -> Described:
        g = self.value
        return Described(key=self.key, title=g.title, line=g.line, section=g.section.value, tab=g.tab.value, lead=g.lead)


def describe() -> list[Described]:
    return [group.describe() for group in Group]