ytdl-sub/src/ytdl_sub/script/functions/boolean_functions.py

from ytdl_sub.script.types.array import Array
from ytdl_sub.script.types.map import Map
from ytdl_sub.script.types.resolvable import AnyArgument, Boolean, Float, Integer, String

# pylint: disable=invalid-name


class BooleanFunctions:
    """
    Comparison functions that output Booleans.
    """

    @staticmethod
    def bool(value: AnyArgument) -> Boolean:
        """
        :description:
          Cast any type to a Boolean.
        """
        return Boolean(bool(value.value))

    @staticmethod
    def eq(left: AnyArgument, right: AnyArgument) -> Boolean:
        """
        :description:
          ``==`` operator. Returns True if left == right. False otherwise.
        """
        return Boolean(left.value == right.value)

    @staticmethod
    def ne(left: AnyArgument, right: AnyArgument) -> Boolean:
        """
        :description:
          ``!=`` operator. Returns True if left != right. False otherwise.
        """
        return Boolean(left.value != right.value)

    @staticmethod
    def lt(left: AnyArgument, right: AnyArgument) -> Boolean:
        """
        :description:
          ``<`` operator. Returns True if left < right. False otherwise.
        """
        return Boolean(left.value < right.value)

    @staticmethod
    def lte(left: AnyArgument, right: AnyArgument) -> Boolean:
        """
        :description:
          ``<=`` operator. Returns True if left <= right. False otherwise.
        """
        return Boolean(left.value <= right.value)

    @staticmethod
    def gt(left: AnyArgument, right: AnyArgument) -> Boolean:
        """
        :description:
          ``>`` operator. Returns True if left > right. False otherwise.
        """
        return Boolean(left.value > right.value)

    @staticmethod
    def gte(left: AnyArgument, right: AnyArgument) -> Boolean:
        """
        :description:
          ``>=`` operator. Returns True if left >= right. False otherwise.
        """
        return Boolean(left.value >= right.value)

    @staticmethod
    def and_(*values: AnyArgument) -> Boolean:
        """
        :description:
          ``and`` operator. Returns True if all values evaluate to True. False otherwise.
        """
        return Boolean(all(bool(val.value) for val in values))

    @staticmethod
    def or_(*values: AnyArgument) -> Boolean:
        """
        :description:
          ``or`` operator. Returns True if any value evaluates to True. False otherwise.
        """
        return Boolean(any(bool(val.value) for val in values))

    @staticmethod
    def xor(*values: AnyArgument) -> Boolean:
        """
        :description:
          ``^`` operator. Returns True if exactly one value is set to True. False otherwise.
        """
        bit_array = [bool(val.value) for val in values]

        return Boolean(sum(bit_array) == 1)

    @staticmethod
    def not_(value: Boolean) -> Boolean:
        """
        :description:
          ``not`` operator. Returns the opposite of value.
        """
        return Boolean(not value.value)

    @staticmethod
    def is_null(value: AnyArgument) -> Boolean:
        """
        :description:
          Returns True if a value is null (i.e. an empty string). False otherwise.
        """
        return Boolean(isinstance(value, String) and value.value == "")

    @staticmethod
    def is_map(value: AnyArgument) -> Boolean:
        """
        :description:
          Returns True if a value is a Map. False otherwise.
        """
        return Boolean(isinstance(value, Map))

    @staticmethod
    def is_array(value: AnyArgument) -> Boolean:
        """
        :description:
          Returns True if a value is a Map. False otherwise.
        """
        return Boolean(isinstance(value, Array))

    @staticmethod
    def is_string(value: AnyArgument) -> Boolean:
        """
        :description:
          Returns True if a value is a String. False otherwise.
        """
        return Boolean(isinstance(value, String))

    @staticmethod
    def is_numeric(value: AnyArgument) -> Boolean:
        """
        :description:
          Returns True if a value is either an Integer or Float. False otherwise.
        """
        return Boolean(isinstance(value, (Integer, Float)))

    @staticmethod
    def is_int(value: AnyArgument) -> Boolean:
        """
        :description:
          Returns True if a value is an Integer. False otherwise.
        """
        return Boolean(isinstance(value, Integer))

    @staticmethod
    def is_float(value: AnyArgument) -> Boolean:
        """
        :description:
          Returns True if a value is a Float. False otherwise.
        """
        return Boolean(isinstance(value, Float))

    @staticmethod
    def is_bool(value: AnyArgument) -> Boolean:
        """
        :description:
          Returns True if a value is a Float. False otherwise.
        """
        return Boolean(isinstance(value, Boolean))