comictagger/comicapi/merge.py

from __future__ import annotations

import dataclasses
from collections.abc import Collection
from enum import auto
from typing import Any

from comicapi.utils import DefaultDict, StrEnum, norm_fold


@dataclasses.dataclass
class Credit:
    person: str = ""
    role: str = ""
    primary: bool = False
    language: str = ""  # Should be ISO 639 language code

    def __str__(self) -> str:
        lang = ""
        if self.language:
            lang = f" [{self.language}]"
        return f"{self.role}: {self.person}{lang}"


class Mode(StrEnum):
    OVERLAY = auto()
    ADD_MISSING = auto()


def merge_lists(old: Collection[Any], new: Collection[Any]) -> list[Any] | set[Any]:
    """Dedupes normalised (NFKD), casefolded values using 'new' values on collisions"""
    if len(new) == 0:
        return old if isinstance(old, set) else list(old)
    if len(old) == 0:
        return new if isinstance(new, set) else list(new)

    # Create dict to preserve case
    new_dict = {norm_fold(str(n)): n for n in new}
    old_dict = {norm_fold(str(c)): c for c in old}

    old_dict.update(new_dict)

    if isinstance(old, set):
        return set(old_dict.values())

    return list(old_dict.values())


def overlay(old: Any, new: Any) -> Any:
    """overlay - When the `new` object is not empty, replace `old` with `new`."""
    if new is None or (isinstance(new, Collection) and len(new) == 0):
        return old

    return new


attribute = DefaultDict(
    {
        Mode.OVERLAY: overlay,
        Mode.ADD_MISSING: lambda old, new: overlay(new, old),
    },
    default=lambda x: overlay,
)


lists = DefaultDict(
    {
        Mode.OVERLAY: merge_lists,
        Mode.ADD_MISSING: lambda old, new: merge_lists(new, old),
    },
    default=lambda x: overlay,
)