/*
 * Copyright (c) 2022, Sam Atkins <atkinssj@serenityos.org>
 * Copyright (c) 2025, Jelle Raaijmakers <jelle@ladybird.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#include <LibGfx/Bitmap.h>
#include <LibGfx/DecodedImageFrame.h>
#include <LibWeb/HTML/Canvas/CanvasDrawImage.h>
#include <LibWeb/HTML/DecodedImageData.h>
#include <LibWeb/HTML/ImageBitmap.h>

namespace Web::HTML {

Gfx::IntSize canvas_image_source_dimensions(CanvasImageSource const& image)
{
    return image.visit(
        [](GC::Ref<HTMLImageElement> source) -> Gfx::IntSize {
            if (auto frame = source->current_image_frame(); frame.has_value())
                return frame->size();

            // FIXME: This is very janky and not correct.
            return { source->width(), source->height() };
        },
        [](GC::Ref<SVG::SVGImageElement> source) -> Gfx::IntSize {
            if (auto decoded_image_frame = source->current_image_frame(); decoded_image_frame.has_value())
                return decoded_image_frame->size();

            // FIXME: This is very janky and not correct.
            return { source->width()->anim_val()->value(), source->height()->anim_val()->value() };
        },
        [](GC::Ref<HTMLCanvasElement> source) -> Gfx::IntSize {
            return { source->width(), source->height() };
        },
        [](GC::Ref<ImageBitmap> source) -> Gfx::IntSize {
            if (auto* bitmap = source->bitmap())
                return bitmap->size();
            return { source->width(), source->height() };
        },
        [](GC::Ref<OffscreenCanvas> source) -> Gfx::IntSize {
            if (auto bitmap = source->bitmap())
                return bitmap->size();
            return {};
        },
        [](GC::Ref<HTMLVideoElement> source) -> Gfx::IntSize {
            return { source->video_width(), source->video_height() };
        });
}

Optional<Gfx::DecodedImageFrame> canvas_image_source_frame(CanvasImageSource const& image)
{
    return image.visit(
        [](OneOf<GC::Ref<HTMLImageElement>, GC::Ref<SVG::SVGImageElement>> auto const& element) -> Optional<Gfx::DecodedImageFrame> {
            auto image_data = element->decoded_image_data();
            if (!image_data)
                return {};

            Gfx::IntSize size;
            auto intrinsic_width = element->intrinsic_width();
            auto intrinsic_height = element->intrinsic_height();
            if (intrinsic_width.has_value() && intrinsic_height.has_value())
                size = { intrinsic_width->to_int(), intrinsic_height->to_int() };

            return image_data->default_frame(size);
        },
        [](GC::Ref<HTMLCanvasElement> const& canvas) -> Optional<Gfx::DecodedImageFrame> {
            canvas->prepare_for_compositing();
            auto bitmap = canvas->get_bitmap_from_surface();
            if (!bitmap)
                return {};
            return Gfx::DecodedImageFrame { *bitmap };
        },
        [](OneOf<GC::Ref<ImageBitmap>, GC::Ref<OffscreenCanvas>> auto const& source) -> Optional<Gfx::DecodedImageFrame> {
            auto bitmap = source->bitmap();
            if (!bitmap)
                return {};
            return Gfx::DecodedImageFrame { *bitmap };
        },
        [](GC::Ref<HTMLVideoElement> const& source) -> Optional<Gfx::DecodedImageFrame> {
            return source->current_decoded_image_frame();
        });
}

WebIDL::ExceptionOr<void> CanvasDrawImage::draw_image(CanvasImageSource const& image, float destination_x, float destination_y)
{
    // If not specified, the dw and dh arguments must default to the values of sw and sh, interpreted such that one CSS pixel in the image is treated as one unit in the output bitmap's coordinate space.
    // If the sx, sy, sw, and sh arguments are omitted, then they must default to 0, 0, the image's intrinsic width in image pixels, and the image's intrinsic height in image pixels, respectively.
    // If the image has no intrinsic dimensions, then the concrete object size must be used instead, as determined using the CSS "Concrete Object Size Resolution" algorithm, with the specified size having
    // neither a definite width nor height, nor any additional constraints, the object's intrinsic properties being those of the image argument, and the default object size being the size of the output bitmap.
    auto size = canvas_image_source_dimensions(image);
    return draw_image_internal(image, 0, 0, size.width(), size.height(), destination_x, destination_y, size.width(), size.height());
}

WebIDL::ExceptionOr<void> CanvasDrawImage::draw_image(CanvasImageSource const& image, float destination_x, float destination_y, float destination_width, float destination_height)
{
    // If the sx, sy, sw, and sh arguments are omitted, then they must default to 0, 0, the image's intrinsic width in image pixels, and the image's intrinsic height in image pixels, respectively.
    // If the image has no intrinsic dimensions, then the concrete object size must be used instead, as determined using the CSS "Concrete Object Size Resolution" algorithm, with the specified size having
    // neither a definite width nor height, nor any additional constraints, the object's intrinsic properties being those of the image argument, and the default object size being the size of the output bitmap.
    auto size = canvas_image_source_dimensions(image);
    return draw_image_internal(image, 0, 0, size.width(), size.height(), destination_x, destination_y, destination_width, destination_height);
}

WebIDL::ExceptionOr<void> CanvasDrawImage::draw_image(CanvasImageSource const& image, float source_x, float source_y, float source_width, float source_height, float destination_x, float destination_y, float destination_width, float destination_height)
{
    return draw_image_internal(image, source_x, source_y, source_width, source_height, destination_x, destination_y, destination_width, destination_height);
}

}
