microblog.pub/core/tasks.py

import os
from datetime import datetime
from datetime import timezone
from typing import Any
from typing import Dict
from typing import Set

from little_boxes import activitypub as ap
from poussetaches import PousseTaches

from config import DISABLE_WEBMENTIONS
from config import MEDIA_CACHE
from utils import parse_datetime

p = PousseTaches(
    os.getenv("MICROBLOGPUB_POUSSETACHES_HOST", "http://localhost:7991"),
    os.getenv("MICROBLOGPUB_INTERNAL_HOST", "http://localhost:5000"),
)


class Tasks:
    @staticmethod
    def cache_object(iri: str) -> None:
        p.push(iri, "/task/cache_object")

    @staticmethod
    def cache_actor(iri: str, also_cache_attachments: bool = True) -> None:
        p.push(
            {"iri": iri, "also_cache_attachments": also_cache_attachments},
            "/task/cache_actor",
        )

    @staticmethod
    def cache_actor_icon(icon_url: str, actor_iri: str) -> None:
        if MEDIA_CACHE.is_actor_icon_cached(icon_url):
            return None

    @staticmethod
    def cache_emoji(url: str, iri: str) -> None:
        if MEDIA_CACHE.is_emoji_cached(iri):
            return None

        p.push({"url": url, "iri": iri}, "/task/cache_emoji")

    @staticmethod
    def send_webmentions(activity: ap.Create, links: Set[str]) -> None:
        if DISABLE_WEBMENTIONS:
            return None

        for link in links:
            p.push(
                {
                    "link": link,
                    "note_url": activity.get_object().get_url(),
                    "remote_id": activity.id,
                },
                "/task/send_webmention",
            )

    @staticmethod
    def cache_emojis(activity: ap.BaseActivity) -> None:
        for emoji in activity.get_emojis():
            try:
                Tasks.cache_emoji(emoji.get_icon_url(), emoji.id)
            except KeyError:
                # TODO(tsileo): log invalid emoji
                pass

    @staticmethod
    def post_to_remote_inbox(payload: str, recp: str) -> None:
        p.push({"payload": payload, "to": recp}, "/task/post_to_remote_inbox")

    @staticmethod
    def forward_activity(iri: str) -> None:
        p.push(iri, "/task/forward_activity")

    @staticmethod
    def fetch_og_meta(iri: str) -> None:
        p.push(iri, "/task/fetch_og_meta")

    @staticmethod
    def process_reply(iri: str) -> None:
        p.push(iri, "/task/process_reply")

    @staticmethod
    def process_new_activity(iri: str) -> None:
        p.push(iri, "/task/process_new_activity")

    @staticmethod
    def cache_attachments(iri: str) -> None:
        p.push(iri, "/task/cache_attachments")

    @staticmethod
    def cache_attachment(attachment: Dict[str, Any], iri: str) -> None:
        p.push({"iri": iri, "attachment": attachment}, "/task/cache_attachment")

    @staticmethod
    def finish_post_to_inbox(iri: str) -> None:
        p.push(iri, "/task/finish_post_to_inbox")

    @staticmethod
    def finish_post_to_outbox(iri: str) -> None:
        p.push(iri, "/task/finish_post_to_outbox")

    @staticmethod
    def send_actor_update() -> None:
        p.push({}, "/task/send_actor_update", delay=2)

    @staticmethod
    def update_question_outbox(iri: str, open_for: int) -> None:
        p.push(
            iri, "/task/update_question", delay=open_for
        )  # XXX: delay expects minutes

    @staticmethod
    def fetch_remote_question(question) -> None:
        now = datetime.now(timezone.utc)
        dt = parse_datetime(question.closed or question.endTime)
        minutes = int((dt - now).total_seconds() / 60)

        if minutes > 0:
            # Only push the task if the poll is not ended yet
            p.push(
                question.id, "/task/fetch_remote_question", delay=minutes
            )  # XXX: delay expects minutes