wrongmove/services/decision_service.py

"""Unified decision service - shared between CLI and HTTP API.

This module provides the core business logic for listing decision operations
(like/dislike). Both the CLI and HTTP API should use these functions.
"""
from models.decision import ListingDecision
from repositories.decision_repository import DecisionRepository

VALID_DECISIONS = {"liked", "disliked"}
VALID_LISTING_TYPES = {"RENT", "BUY"}


def set_decision(
    repository: DecisionRepository,
    user_id: int,
    listing_id: int,
    listing_type: str,
    decision: str,
) -> ListingDecision:
    """Set or update a like/dislike decision for a listing."""
    if decision not in VALID_DECISIONS:
        raise ValueError(
            f"Invalid decision: {decision}. Must be one of {VALID_DECISIONS}"
        )
    if listing_type not in VALID_LISTING_TYPES:
        raise ValueError(
            f"Invalid listing_type: {listing_type}. Must be one of {VALID_LISTING_TYPES}"
        )
    return repository.upsert_decision(user_id, listing_id, listing_type, decision)


def get_user_decisions(
    repository: DecisionRepository,
    user_id: int,
) -> list[ListingDecision]:
    """Get all decisions for a user."""
    return repository.get_decisions_for_user(user_id)


def remove_decision(
    repository: DecisionRepository,
    user_id: int,
    listing_id: int,
    listing_type: str,
) -> bool:
    """Remove a decision (un-like/un-dislike). Returns False if not found."""
    if listing_type not in VALID_LISTING_TYPES:
        raise ValueError(
            f"Invalid listing_type: {listing_type}. Must be one of {VALID_LISTING_TYPES}"
        )
    return repository.delete_decision(user_id, listing_id, listing_type)


def get_disliked_ids(
    repository: DecisionRepository,
    user_id: int,
    listing_type: str,
) -> set[int]:
    """Get all disliked listing IDs for a user and listing type."""
    return repository.get_disliked_listing_ids(user_id, listing_type)


def get_liked_ids(
    repository: DecisionRepository,
    user_id: int,
    listing_type: str,
) -> set[int]:
    """Get all liked listing IDs for a user and listing type."""
    return repository.get_liked_listing_ids(user_id, listing_type)