"""
Factory-boy factories for documents app models.
"""

from __future__ import annotations

import factory
from django.contrib.auth.models import User

from documents.models import Correspondent
from documents.models import Document
from documents.models import DocumentType
from documents.models import MatchingModel
from documents.models import PaperlessTask
from documents.models import StoragePath
from documents.models import Tag
from paperless_testing.typed_factory import TypedModelFactory


class CorrespondentFactory(TypedModelFactory[Correspondent]):
    class Meta:
        model = Correspondent

    name = factory.Sequence(lambda n: f"{factory.Faker('company')} {n}")
    match = ""
    matching_algorithm = MatchingModel.MATCH_NONE


class DocumentTypeFactory(TypedModelFactory[DocumentType]):
    class Meta:
        model = DocumentType

    name = factory.Sequence(lambda n: f"{factory.Faker('bs')} {n}")
    match = ""
    matching_algorithm = MatchingModel.MATCH_NONE


class TagFactory(TypedModelFactory[Tag]):
    class Meta:
        model = Tag

    name = factory.Sequence(lambda n: f"{factory.Faker('word')} {n}")
    match = ""
    matching_algorithm = MatchingModel.MATCH_NONE
    is_inbox_tag = False


class StoragePathFactory(TypedModelFactory[StoragePath]):
    class Meta:
        model = StoragePath

    name = factory.Sequence(
        lambda n: f"{factory.Faker('file_path', depth=2, extension='')} {n}",
    )
    path = factory.LazyAttribute(lambda o: f"{o.name}/{{title}}")
    match = ""
    matching_algorithm = MatchingModel.MATCH_NONE


class DocumentFactory(TypedModelFactory[Document]):
    class Meta:
        model = Document

    title = factory.Faker("sentence", nb_words=4)
    checksum = factory.Faker("sha256")
    content = factory.Faker("paragraph")
    correspondent = None
    document_type = None
    storage_path = None


class UserFactory(TypedModelFactory[User]):
    class Meta:
        model = User

    username = factory.Sequence(lambda n: f"user{n}")
    is_staff = False
    is_superuser = False
    # Hashing a real password costs about half a second per user, and no test
    # authenticates with its password. None gives the unusable password
    # create_user() gives.
    password = factory.django.Password(None)

    class Params:
        superuser = factory.Trait(is_staff=True, is_superuser=True)
        staff = factory.Trait(is_staff=True)


class PaperlessTaskFactory(TypedModelFactory[PaperlessTask]):
    class Meta:
        model = PaperlessTask

    task_id = factory.Faker("uuid4")
    task_type = PaperlessTask.TaskType.CONSUME_FILE
    trigger_source = PaperlessTask.TriggerSource.WEB_UI
    status = PaperlessTask.Status.PENDING
    input_data = factory.LazyFunction(dict)
    result_data = None
    acknowledged = False