63 lines
1.2 KiB
Python
63 lines
1.2 KiB
Python
from enum import Enum
|
|
from enum import unique
|
|
from typing import Any
|
|
from typing import Dict
|
|
|
|
from little_boxes import activitypub as ap
|
|
|
|
_SubQuery = Dict[str, Any]
|
|
|
|
|
|
@unique
|
|
class Box(Enum):
|
|
INBOX = "inbox"
|
|
OUTBOX = "outbox"
|
|
REPLIES = "replies"
|
|
|
|
|
|
@unique
|
|
class MetaKey(Enum):
|
|
NOTIFICATION = "notification"
|
|
NOTIFICATION_UNREAD = "notification_unread"
|
|
NOTIFICATION_FOLLOWS_BACK = "notification_follows_back"
|
|
POLL_ANSWER = "poll_answer"
|
|
STREAM = "stream"
|
|
ACTOR_ID = "actor_id"
|
|
UNDO = "undo"
|
|
PUBLISHED = "published"
|
|
GC_KEEP = "gc_keep"
|
|
OBJECT = "object"
|
|
OBJECT_ACTOR = "object_actor"
|
|
PUBLIC = "public"
|
|
|
|
|
|
def _meta(mk: MetaKey) -> str:
|
|
return f"meta.{mk.value}"
|
|
|
|
|
|
def by_remote_id(remote_id: str) -> _SubQuery:
|
|
return {"remote_id": remote_id}
|
|
|
|
|
|
def in_inbox() -> _SubQuery:
|
|
return {"box": Box.INBOX.value}
|
|
|
|
|
|
def in_outbox() -> _SubQuery:
|
|
return {"box": Box.OUTBOX.value}
|
|
|
|
|
|
def by_type(type_: ap.ActivityType) -> _SubQuery:
|
|
return {"type": type_.value}
|
|
|
|
|
|
def not_undo() -> _SubQuery:
|
|
return {_meta(MetaKey.UNDO): False}
|
|
|
|
|
|
def by_actor(actor: ap.BaseActivity) -> _SubQuery:
|
|
return {_meta(MetaKey.ACTOR_ID): actor.id}
|
|
|
|
|
|
def is_public() -> _SubQuery:
|
|
return {_meta(MetaKey.PUBLIC): True}
|