from taleweave.models.base import Attributes, AttributeValue


def add_value(value: AttributeValue, offset: int | float) -> AttributeValue:
    """
    Add an offset to a value.
    """
    if isinstance(value, str):
        raise ValueError(f"Cannot add a number to a string attribute: {value}")

    return value + offset


def multiply_value(value: AttributeValue, factor: int | float) -> AttributeValue:
    """
    Multiply a value by a factor.
    """
    if isinstance(value, str):
        raise ValueError(f"Cannot multiply a string attribute: {value}")

    return value * factor


def append_value(value: AttributeValue, suffix: str) -> str:
    """
    Append a suffix to a string.
    """
    return str(value) + suffix


def prepend_value(value: AttributeValue, prefix: str) -> str:
    """
    Prepend a prefix to a string.
    """
    return prefix + str(value)


def value_contains(value: AttributeValue, substring: str) -> bool:
    """
    Check if a string contains a substring.
    """
    if not isinstance(value, str):
        raise ValueError(
            f"Cannot check for a substring in a non-string attribute: {value}"
        )

    return substring in value


def add_attribute(attributes: Attributes, name: str, value: int | float) -> Attributes:
    """
    Add an attribute to a set of attributes.
    """
    if name in attributes:
        previous_value = attributes[name]
        attributes[name] = add_value(previous_value, value)
    else:
        attributes[name] = value

    return attributes


def subtract_attribute(
    attributes: Attributes, name: str, value: int | float
) -> Attributes:
    """
    Subtract an attribute from a set of attributes.
    """
    if name in attributes:
        previous_value = attributes[name]
        if isinstance(previous_value, str):
            raise ValueError(
                f"Cannot subtract a number from a string attribute: {name}"
            )

        attributes[name] = value - previous_value
    else:
        attributes[name] = -value

    return attributes


def multiply_attribute(
    attributes: Attributes, name: str, value: int | float
) -> Attributes:
    """
    Multiply an attribute in a set of attributes.
    """
    if name in attributes:
        previous_value = attributes[name]
        if isinstance(previous_value, str):
            raise ValueError(f"Cannot multiply a string attribute: {name}")

        attributes[name] = previous_value * value
    else:
        attributes[name] = 0

    return attributes


def divide_attribute(
    attributes: Attributes, name: str, value: int | float
) -> Attributes:
    """
    Divide an attribute in a set of attributes.
    """
    if name in attributes:
        previous_value = attributes[name]
        if isinstance(previous_value, str):
            raise ValueError(f"Cannot divide a string attribute: {name}")

        attributes[name] = previous_value / value
    else:
        attributes[name] = 0

    return attributes


def set_attribute(
    attributes: Attributes, name: str, value: AttributeValue
) -> Attributes:
    """
    Set an attribute in a set of attributes.
    """
    attributes[name] = value

    return attributes


def append_attribute(attributes: Attributes, name: str, value: str) -> Attributes:
    """
    Append a string to an attribute in a set of attributes.
    """
    if name in attributes:
        previous_value = attributes[name]
        if isinstance(previous_value, str):
            attributes[name] = previous_value + value
        else:
            raise ValueError(
                f"Cannot append a string to a non-string attribute: {name}"
            )
    else:
        attributes[name] = value

    return attributes


def prepend_attribute(attributes: Attributes, name: str, value: str) -> Attributes:
    """
    Prepend a string to an attribute in a set of attributes.
    """
    if name in attributes:
        previous_value = attributes[name]
        if isinstance(previous_value, str):
            attributes[name] = value + previous_value
        else:
            raise ValueError(
                f"Cannot prepend a string to a non-string attribute: {name}"
            )
    else:
        attributes[name] = value

    return attributes