home-automation-backend/src/cloud_util/homeassistant.py

from __future__ import annotations

import ast
from datetime import datetime, timedelta, timezone

from pydantic import BaseModel

from src.cloud_util.ticktick import TickTick
from src.config import Config


class HomeAssistant:
    class PublishMessage(BaseModel):
        target: str
        action: str
        content: str

    def __init__(self, ticktick: TickTick) -> None:
        self._ticktick = ticktick

    def process_publish_message(self, message: PublishMessage) -> dict[str, str]:
        if message.target == "ticktick":
            return self._process_ticktick_message(message=message)

        return {"Status": "Unknown target"}

    def _process_ticktick_message(self, message: PublishMessage) -> dict[str, str]:
        if message.action == "create_shopping_list":
            return self._create_shopping_list(content=message.content)
        if message.action == "create_action_task":
            return self._create_action_task(content=message.content)

        return {"Status": "Unknown action"}

    def _create_shopping_list(self, content: str) -> dict[str, str]:
        project_id = Config.get_env("TICKTICK_SHOPPING_LIST")
        item: dict[str, str] = ast.literal_eval(content)
        task = TickTick.Task(projectId=project_id, title=item["item"])
        return self._ticktick.create_task(task=task)

    def _create_action_task(self, content: str) -> dict[str, str]:
        detail: dict[str, str] = ast.literal_eval(content)
        project_id = Config.get_env("TICKTICK_HOME_TASK_LIST")
        due_hour = detail["due_hour"]
        due = datetime.now(tz=datetime.now().astimezone().tzinfo) + timedelta(hours=due_hour)
        due = (due + timedelta(days=1)).replace(hour=0, minute=0, second=0, microsecond=0)
        due = due.astimezone(timezone.utc)
        task = TickTick.Task(projectId=project_id, title=detail["action"], dueDate=TickTick.datetime_to_ticktick_format(due))
        return self._ticktick.create_task(task=task)