# pylint: disable=too-few-public-methods, missing-module-docstring, broad-exception-caught,invalid-name

from enum import Enum
from typing import Optional, Dict, Any, Union, Literal, List
from dataclasses import dataclass, field
from .version import VERSION


# pylint: disable=too-many-instance-attributes

RenderType = Union[Literal["video-or-audio"], Literal["still"]]


class ValidStillImageFormats(str, Enum):
    """
    Enumeration of valid image formats for still images.

    Attributes:
        PNG: Represents the PNG image format.
        JPEG: Represents the JPEG image format.
        PDF: Represents the PDF format for images.
        WEBP: Represents the WEBP image format.
    """

    PNG = 'png'
    JPEG = 'jpeg'
    PDF = 'pdf'
    WEBP = 'webp'


class Privacy(str, Enum):
    """
    Enumeration of privacy settings.

    Attributes:
        PUBLIC: Indicates a public setting.
        PRIVATE: Indicates a private setting.
    """

    PUBLIC = 'public'
    PRIVATE = 'private'
    NO_ACL = 'no-acl'


class LogLevel(str, Enum):
    """
    Enumeration for different levels of logging.

    Attributes:
        VERBOSE: Verbose logging level.
        INFO: Informational logging level.
        WARN: Warning logging level.
        ERROR: Error logging level.
    """

    VERBOSE = 'verbose'
    INFO = 'info'
    WARN = 'warn'
    ERROR = 'error'


class OpenGlRenderer(str, Enum):
    """
    Enumeration of OpenGL renderer options.

    Attributes:
        SWANGLE: Represents the SWANGLE OpenGL renderer.
        ANGLE: Represents the ANGLE OpenGL renderer.
        EGL: Represents the EGL OpenGL renderer.
        SWIFTSHADER: Represents the SWIFTSHADER OpenGL renderer.
        VULKAN: Represents the VULKAN OpenGL renderer.
    """

    SWANGLE = 'swangle'
    ANGLE = 'angle'
    EGL = 'egl'
    SWIFTSHADER = 'swiftshader'
    VULKAN = 'vulkan'


@dataclass
class ChromiumOptions:
    """
    Options for configuring the Chromium browser.

    Attributes:
        ignore_certificate_errors (Optional[bool]): If True, ignores certificate errors.
        disable_web_security (Optional[bool]): If True, disables web security.
        gl (Optional[OpenGlRenderer]): Specifies the OpenGL renderer to use.
        headless (Optional[bool]): If True, runs Chromium in headless mode.
        user_agent (Optional[str]): Specifies a custom user agent.
        enable_multi_process_on_linux (Optional[bool]):
            If True, enables multi-process mode on Linux.
    """

    ignore_certificate_errors: Optional[bool] = None
    disable_web_security: Optional[bool] = None
    gl: Optional[OpenGlRenderer] = None
    headless: Optional[bool] = None
    user_agent: Optional[str] = None
    enable_multi_process_on_linux: Optional[bool] = None


@dataclass
class CustomCredentialsWithoutSensitiveData:
    """
    Represents credentials without sensitive data.

    Attributes:
        endpoint (str): The endpoint associated with the credentials.
    """

    endpoint: str


@dataclass
class CustomCredentials(CustomCredentialsWithoutSensitiveData):
    """
    Represents custom credentials, extending credentials without sensitive data.

    Attributes:
        access_key_id (Optional[str]): The access key ID.
        secret_access_key (Optional[str]): The secret access key.
    """

    access_key_id: Optional[str] = None
    secret_access_key: Optional[str] = None
    region: Optional[str] = None

    def serialize_params(self) -> Dict:
        """
        Serializes the credentials to a dictionary.
        """
        parameters = {
            'accessKeyId': self.access_key_id,
            'secretAccessKey': self.secret_access_key,
            'region': self.region,
            'endpoint': self.endpoint,
        }
        return parameters


@dataclass
class OutNameInputObject:
    """
    Defines output naming and storage options.

    Attributes:
        bucketName (str): The name of the S3 bucket for output storage.
        key (str): The key name within the S3 bucket.
        s3_output_provider (Optional[CustomCredentials]):
             Optional custom credentials for the S3 output provider.
    """

    bucketName: str
    key: str
    s3_output_provider: Optional[CustomCredentials] = None


class DeleteAfter(Enum):
    """
    Enumeration for specifying the time period after which an item should be deleted.

    Attributes:
        ONE_DAY: Represents deletion after one day.
        THREE_DAYS: Represents deletion after three days.
        SEVEN_DAYS: Represents deletion after seven days.
        THIRTY_DAYS: Represents deletion after thirty days.
    """

    ONE_DAY = '1-day'
    THREE_DAYS = '3-days'
    SEVEN_DAYS = '7-days'
    THIRTY_DAYS = '30-days'


@dataclass
class RenderMediaResponse:
    """
    Response data after rendering.
    """

    bucket_name: str
    render_id: str


@dataclass
class RenderProgressParams:
    """
    Parameters for checking the progress of video rendering.
    """

    render_id: str
    bucket_name: str
    function_name: str
    region: str
    log_level: str
    force_path_style: Optional[bool] = None
    s3_output_provider: Optional[CustomCredentials] = None

    def serialize_params(self) -> Dict:
        """
        Convert instance attributes to a dictionary for serialization.
        """
        parameters: dict[str, Any] = {
            'renderId': self.render_id,
            'bucketName': self.bucket_name,
            'type': 'status',
            "version": VERSION,
            "logLevel": self.log_level,
            "s3OutputProvider": self.s3_output_provider,
        }
        if self.force_path_style is not None:
            parameters['forcePathStyle'] = self.force_path_style
        else:
            parameters['forcePathStyle'] = False

        if self.s3_output_provider is not None:
            parameters['s3OutputProvider'] = self.s3_output_provider.serialize_params()
        else:
            parameters['s3OutputProvider'] = None
        return parameters


@dataclass
class CostsInfo:
    """
    Represents the cost-related information for a specific service or product.

    Attributes:
        accrued_so_far (float): The total cost that has been accrued so far.
        display_cost (str): The cost displayed to the user, possibly formatted as a string.
        currency (str): The type of currency used for the costs, e.g., 'USD', 'EUR'.
        disclaimer (str): Any disclaimer or additional information related to the costs.
    """

    accrued_so_far: float
    display_cost: str
    currency: str
    disclaimer: str


@dataclass
class PlayInBrowser:
    """
    The video should play in the browser when the link is clicked.
    """

    type: Literal['play-in-browser']
    # You can define additional fields as needed


@dataclass
class ShouldDownload:
    """
    The video should download when the link is clicked.
    """

    type: Literal['download']
    fileName: str  # Additional fields for this type


@dataclass
class Webhook:
    """
    Represents a webhook.
    """

    secret: str
    url: str
    customData: Optional[Dict] = None


@dataclass
class RenderMediaParams:
    """
    Parameters for video rendering.
    """

    input_props: Optional[Dict[str, Any]] = None
    bucket_name: Optional[str] = None
    region: Optional[str] = None
    out_name: Optional[Union[str, OutNameInputObject]] = None
    prefer_lossless: Optional[bool] = False
    composition: str = ""
    serve_url: str = ""
    frames_per_lambda: Optional[int] = None
    private_serialized_input_props: Optional[Dict] = None
    codec: str = 'h264'
    version: str = ""
    image_format: ValidStillImageFormats = ValidStillImageFormats.JPEG
    crf: Optional[int] = None
    env_variables: Optional[Dict] = None
    metadata: Optional[Dict] = None
    max_retries: int = 1
    jpeg_quality: int = 80
    privacy: Privacy = Privacy.PUBLIC
    color_space: Optional[str] = None
    log_level: Optional[LogLevel] = LogLevel.INFO
    frame_range: Optional[str] = None
    timeout_in_milliseconds: Optional[int] = 30000
    chromium_options: Optional[ChromiumOptions] = None
    scale: Optional[int] = 1
    every_nth_frame: Optional[int] = 1
    number_of_gif_loops: Optional[int] = None
    concurrency_per_lambda: Optional[int] = 1
    concurrency: Optional[int] = None
    download_behavior: Optional[Union[PlayInBrowser, ShouldDownload]] = field(
        default_factory=lambda: PlayInBrowser(type='play-in-browser')
    )
    muted: bool = False
    overwrite: bool = False
    force_path_style: Optional[bool] = None
    audio_bitrate: Optional[int] = None
    video_bitrate: Optional[int] = None
    webhook: Optional[Webhook] = None
    force_height: Optional[int] = None
    force_fps: Optional[int] = None
    force_duration_in_frames: Optional[int] = None
    offthreadvideo_cache_size_in_bytes: Optional[int] = None
    media_cache_size_in_bytes: Optional[int] = None
    offthreadvideo_threads: Optional[int] = None
    force_width: Optional[int] = None
    # **DEPRECATED** - Use `license_key` instead. Will be removed in a future version.
    api_key: Optional[str] = None
    license_key: Optional[str] = None
    audio_codec: Optional[str] = None
    storage_class: Optional[str] = None
    renderer_function_name: Optional[str] = None
    pro_res_profile: Optional[str] = None
    x264_preset: Optional[str] = None
    pixel_format: Optional[str] = None
    delete_after: Optional[str] = None
    encoding_buffer_size: Optional[str] = None
    encoding_max_rate: Optional[str] = None
    is_production: Optional[bool] = None

    def serialize_params(self) -> Dict:
        """
        Convert instance attributes to a dictionary for serialization.
        """
        parameters = {
            'rendererFunctionName': self.renderer_function_name,
            'framesPerLambda': self.frames_per_lambda,
            'concurrency': self.concurrency,
            'composition': self.composition,
            'serveUrl': self.serve_url,
            'inputProps': self.private_serialized_input_props,
            'codec': self.codec,
            'imageFormat': self.image_format,
            'maxRetries': self.max_retries,
            'jpegQuality': self.jpeg_quality,
            'storageClass': self.storage_class,
            'envVariables': self.env_variables,
            'metadata': self.metadata,
            'privacy': self.privacy,
            'colorSpace': self.color_space,
            'logLevel': self.log_level,
            'frameRange': self.frame_range,
            'outName': self.out_name,
            'preferLossless': self.prefer_lossless,
            'timeoutInMilliseconds': self.timeout_in_milliseconds,
            'chromiumOptions': (
                self.chromium_options if self.chromium_options is not None else {}
            ),
            'scale': self.scale,
            'everyNthFrame': self.every_nth_frame,
            'numberOfGifLoops': self.number_of_gif_loops,
            'concurrencyPerLambda': self.concurrency_per_lambda,
            'downloadBehavior': self.download_behavior,
            'muted': self.muted,
            'version': VERSION,
            'overwrite': self.overwrite,
            'audioBitrate': self.audio_bitrate,
            'videoBitrate': self.video_bitrate,
            'webhook': self.webhook,
            'forceHeight': self.force_height,
            'forceFps': self.force_fps,
            'forceDurationInFrames': self.force_duration_in_frames,
            'offthreadVideoCacheSizeInBytes': self.offthreadvideo_cache_size_in_bytes,
            'mediaCacheSizeInBytes': self.media_cache_size_in_bytes,
            'offthreadVideoThreads': self.offthreadvideo_threads,
            'forceWidth': self.force_width,
            'licenseKey': self.license_key if self.license_key is not None else self.api_key,
            'bucketName': self.bucket_name,
            'audioCodec': self.audio_codec,
            'x264Preset': self.x264_preset,
            'deleteAfter': self.delete_after,
            'encodingBufferSize': self.encoding_buffer_size,
            'encodingMaxRate': self.encoding_max_rate,
            'isProduction': self.is_production,
            'type': 'start',
        }

        if self.crf is not None:
            parameters['crf'] = self.crf
        else:
            parameters['crf'] = None

        if self.env_variables is None:
            parameters['envVariables'] = {}

        if self.metadata is None:
            parameters['metadata'] = {}

        if self.pixel_format is not None:
            parameters['pixelFormat'] = self.pixel_format
        else:
            parameters['pixelFormat'] = None

        if self.pro_res_profile is not None:
            parameters['proResProfile'] = self.pro_res_profile
        else:
            parameters['proResProfile'] = None

        if self.x264_preset is not None:
            parameters['x264Preset'] = self.x264_preset

        if self.force_path_style is not None:
            parameters['forcePathStyle'] = self.force_path_style
        else:
            parameters['forcePathStyle'] = False
        return parameters


@dataclass
class RenderStillParams:
    """
    Parameters for video rendering.
    """

    composition: str
    serve_url: str = ""
    input_props: Optional[Dict[str, Any]] = None
    private_serialized_input_props: Optional[Dict[str, Any]] = None
    image_format: ValidStillImageFormats = ValidStillImageFormats.JPEG
    privacy: Privacy = Privacy.PUBLIC
    max_retries: Optional[int] = 1
    env_variables: Optional[Dict[str, str]] = None
    jpeg_quality: Optional[int] = 80
    frame: Optional[int] = 0
    attempt: Optional[int] = 1
    log_level: Optional[LogLevel] = LogLevel.INFO
    out_name: Optional[Union[str, OutNameInputObject]] = None
    timeout_in_milliseconds: Optional[int] = 30000
    chromium_options: Optional[ChromiumOptions] = None
    scale: Optional[float] = 1
    download_behavior: Union[PlayInBrowser, ShouldDownload] = field(
        default_factory=lambda: PlayInBrowser(type='play-in-browser')
    )
    force_width: Optional[int] = None
    # **DEPRECATED** - Use `license_key` instead. Will be removed in a future version.
    api_key: Optional[int] = None
    license_key: Optional[str] = None
    storage_class: Optional[str] = None
    force_height: Optional[int] = None
    force_fps: Optional[int] = None
    force_duration_in_frames: Optional[int] = None
    force_bucket_name: Optional[str] = None
    dump_browser_logs: Optional[bool] = None
    delete_after: Optional[DeleteAfter] = None
    force_path_style: Optional[bool] = None
    offthreadvideo_cache_size_in_bytes: Optional[int] = None
    media_cache_size_in_bytes: Optional[int] = None
    offthreadvideo_threads: Optional[int] = None
    streamed: bool = False
    is_production: Optional[bool] = None

    def serialize_params(self) -> Dict:
        """
        Serializes the parameters of the current object into a dictionary.

        This method consolidates both mandatory and optional attributes of the object
        into a single dictionary. Mandatory attributes include region, functionName,
        serveUrl, composition, inputProps, imageFormat, and privacy. Optional attributes
        are added to the dictionary only if they are not None.

        The optional attributes include maxRetries, envVariables, jpegQuality, frame,
        logLevel, outName, timeoutInMilliseconds, chromiumOptions, scale, downloadBehavior,
        forceWidth, forceHeight, forceBucketName, and deleteAfter.
        Default values are provided for 'inputProps' (empty dictionary) and 'downloadBehavior'
        ('type': 'play-in-browser') if they are not explicitly set.

        Returns:
            Dict: A dictionary containing all the serialized parameters of the object.
        """
        parameters = {
            'type': 'still',
            'composition': self.composition,
            'inputProps': self.private_serialized_input_props or {},
            'imageFormat': self.image_format,
            'privacy': self.privacy,
            'serveUrl': self.serve_url,
            'version': VERSION,
            'timeoutInMilliseconds': self.timeout_in_milliseconds,
            'maxRetries': self.max_retries,
            'envVariables': (
                self.env_variables if self.env_variables is not None else {}
            ),
            'jpegQuality': self.jpeg_quality,
            'storageClass': self.storage_class,
            'frame': self.frame,
            'logLevel': self.log_level,
            'outName': self.out_name,
            'chromiumOptions': (
                self.chromium_options if self.chromium_options is not None else {}
            ),
            'scale': self.scale,
            'downloadBehavior': self.download_behavior
            or PlayInBrowser(type='play-in-browser'),
            'forceWidth': self.force_width,
            'licenseKey': self.license_key if self.license_key is not None else self.api_key,
            'forceHeight': self.force_height,
            'forceFps': self.force_fps,
            'forceDurationInFrames': self.force_duration_in_frames,
            'forceBucketName': self.force_bucket_name,
            'deleteAfter': self.delete_after,
            'attempt': self.attempt,
            'offthreadVideoCacheSizeInBytes': self.offthreadvideo_cache_size_in_bytes,
            'mediaCacheSizeInBytes': self.media_cache_size_in_bytes,
            'offthreadVideoThreads': self.offthreadvideo_threads,
            'streamed': self.streamed,
            'isProduction': self.is_production,
        }

        if self.force_path_style is not None:
            parameters['forcePathStyle'] = self.force_path_style
        else:
            parameters['forcePathStyle'] = False

        return parameters


@dataclass
class RenderStillResponse:
    """
    Represents the output information of a rendering operation performed on AWS Lambda.

    Attributes:
        estimated_price (CostsInfo):
            An object containing detailed cost information related to the rendering.
        url (str): The URL where the rendered image is stored or can be accessed.
        size_in_bytes (int): The size of the rendered image file in bytes.
        bucket_name (str): The name of the S3 bucket where the rendered image is stored.
        render_id (str): A unique identifier for the rendering operation.
        cloud_watch_logs (str): The CloudWatch logs associated with the rendering operation.
    """

    estimated_price: CostsInfo
    url: str
    size_in_bytes: int
    bucket_name: str
    render_id: str
    outKey: str


@dataclass
class RenderMediaProgress:
    """
    Progress of video rendering.
    """

    overallProgress: float = 0.0
    chunks: int = 0
    done: bool = False
    encodingStatus: Optional[Any] = None
    costs: Optional[Any] = None
    renderId: str = ""
    renderMetadata: Optional[Any] = None
    outputFile: Optional[Any] = None
    outKey: Optional[str] = None
    timeToFinish: Optional[float] = None
    errors: List[Any] = field(default_factory=list)
    fatalErrorEncountered: bool = False
    currentTime: int = 0
    renderSize: int = 0
    outputSizeInBytes: Optional[int] = None
    lambdasInvoked: int = 0
    framesRendered: Optional[int] = None