"""DMS AI — IAM permission definitions."""

from __future__ import annotations

from simorgh.apps.iam.registry import register_permission

# OCR permissions
PERM_OCR_VIEW = "dms.ocr.view"
PERM_OCR_SUBMIT = "dms.ocr.submit"

# AI classification permissions
PERM_AI_VIEW = "dms.ai.view"
PERM_AI_SUBMIT = "dms.ai.submit"

# Extracted entity permissions
PERM_ENTITIES_VIEW = "dms.entities.view"


def register_all() -> None:
    register_permission(
        codename=PERM_OCR_VIEW,
        label="DMS: View OCR results",
        description="Allows viewing OCR processing results for document versions.",
    )
    register_permission(
        codename=PERM_OCR_SUBMIT,
        label="DMS: Submit OCR job",
        description="Allows submitting an OCR processing job for a document version.",
    )
    register_permission(
        codename=PERM_AI_VIEW,
        label="DMS: View AI classifications",
        description="Allows viewing AI classification results for document versions.",
    )
    register_permission(
        codename=PERM_AI_SUBMIT,
        label="DMS: Submit AI classification",
        description="Allows submitting an AI classification job for a document version.",
    )
    register_permission(
        codename=PERM_ENTITIES_VIEW,
        label="DMS: View extracted entities",
        description="Allows viewing named entities extracted from document versions.",
    )