import time

from engine.events.agents import AgentReported
from features.parts import AgentContext, Handler
from features.status_bar.bar import EMPTY, bar, doing
from features.status_bar.usage import observe
from controllers.types import Agents



class WriteBar(Handler):
    def handle(self, context: AgentContext, event: AgentReported) -> None:
        newest = context.journal.get(Agents).primary_to_read()
        drawn = bar(newest, time.time()) if newest else EMPTY
        state = context.record.state("status_bar")
        if state.get("bar") != drawn:
            state.set("bar", drawn)


class RecordWhatItDoes(Handler):
    """The agent's own row keeps the bar's plain line for its latest activity, which the Home cells show in place of a tool name."""

    def handle(self, context: AgentContext, event: AgentReported) -> None:
        row = context.agent.row
        line = doing(row, time.time())
        if line != row.doing:
            context.journal.get(Agents).update(row.n, doing=line)


class RefreshUsage(Handler):
    behaviour = "usage"

    def handle(self, context: AgentContext, event: AgentReported) -> None:
        row = context.agent.row
        usage = observe(row.provider, row.transcript, row.usage)
        if usage is not None and usage != row.usage:
            context.journal.get(Agents).update(row.n, usage=usage)