import logging
from typing import Any

logger: logging.Logger = logging.getLogger("notification-tools")


class NotificationTools:
    """Tools exposed to the Notification Reader Agent for routing decisions."""

    def __init__(self) -> None:
        self.tool_called: bool = False
        self.action: str = "NO_ACTION"
        self.arguments: dict[str, Any] = {}

    def process_issue(self, owner: str, repo: str, issue_number: int, reason: str) -> str:
        """Process the notification as an issue.
        Use this when a notification refers to an issue that requires agent action.

        Args:
            owner: The repository owner (organization).
            repo: The repository name.
            issue_number: The Gitea issue number.
            reason: Why this notification should be processed.
        """
        logger.info(f"Notification tool 'process_issue' called for {owner}/{repo}#{issue_number}: {reason}")
        self.tool_called = True
        self.action = "PROCESS_ISSUE"
        self.arguments = {
            "owner": owner,
            "repo": repo,
            "issue_number": issue_number,
            "reason": reason,
        }
        return "Issue notification marked for processing."

    def process_pr(self, owner: str, repo: str, pr_number: int, reason: str) -> str:
        """Process the notification as a pull request.
        Use this when a notification refers to a PR that requires agent action.

        Args:
            owner: The repository owner (organization).
            repo: The repository name.
            pr_number: The Gitea PR number.
            reason: Why this notification should be processed.
        """
        logger.info(f"Notification tool 'process_pr' called for {owner}/{repo}#{pr_number}: {reason}")
        self.tool_called = True
        self.action = "PROCESS_PR"
        self.arguments = {
            "owner": owner,
            "repo": repo,
            "pr_number": pr_number,
            "reason": reason,
        }
        return "PR notification marked for processing."

    def skip_notification(self, reason: str) -> str:
        """Skip the notification and take no action.
        Use this if the notification is irrelevant, already handled, or does not require agent intervention.

        Args:
            reason: The reason for skipping this notification.
        """
        logger.info(f"Notification tool 'skip_notification' called: {reason}")
        self.tool_called = True
        self.action = "SKIP"
        self.arguments = {"reason": reason}
        return "Notification marked to be skipped."