104 lines
3.1 KiB
Python
104 lines
3.1 KiB
Python
import os
|
|
from datetime import datetime
|
|
from datetime import timezone
|
|
from typing import Any
|
|
from typing import Dict
|
|
|
|
from little_boxes import activitypub as ap
|
|
from poussetaches import PousseTaches
|
|
|
|
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 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 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
|