from utils.policy.models import CaptureToggles, ChatKind

DEFAULTS: dict[ChatKind, CaptureToggles] = {
    ChatKind.CHANNEL: CaptureToggles(),
    ChatKind.GROUP: CaptureToggles(messages=True),
    ChatKind.DM: CaptureToggles(
        messages=True,
        media=True,
        self_destruct_media=True,
        stt=True,
        reactions=True,
        track_edits_deletes=True,
        profile_history=True,
        stories=True,
        presence=True,
        backfill=True,
    ),
}

TRACKING: dict[ChatKind, CaptureToggles] = {
    ChatKind.CHANNEL: CaptureToggles(
        messages=True,
        media=True,
        reactions=True,
        track_edits_deletes=True,
        backfill=True,
    ),
    ChatKind.GROUP: CaptureToggles(
        messages=True,
        media=True,
        reactions=True,
        track_edits_deletes=True,
        profile_history=True,
        backfill=True,
    ),
    ChatKind.DM: DEFAULTS[ChatKind.DM],
}