use {
    crate::{
        format::XRGB8888,
        gfx_api::GfxError,
        scale::Scale,
        state::State,
        video::{
            drm::DrmError,
            gbm::{GbmBo, GbmError, GBM_BO_USE_LINEAR, GBM_BO_USE_RENDERING},
            INVALID_MODIFIER, LINEAR_MODIFIER,
        },
    },
    jay_config::video::Transform,
    std::{ops::Deref, rc::Rc},
    thiserror::Error,
    uapi::OwnedFd,
};

#[derive(Debug, Error)]
pub enum ScreenshooterError {
    #[error("There is no render context")]
    NoRenderContext,
    #[error("Display is empty")]
    EmptyDisplay,
    #[error(transparent)]
    GbmError(#[from] GbmError),
    #[error(transparent)]
    RenderError(#[from] GfxError),
    #[error(transparent)]
    DrmError(#[from] DrmError),
    #[error("Render context does not support XRGB8888")]
    XRGB8888,
    #[error("Render context supports neither linear nor invalid modifier for XRGB8888 rendering")]
    Linear,
}

pub struct Screenshot {
    pub drm: Rc<OwnedFd>,
    pub bo: GbmBo,
}

pub fn take_screenshot(
    state: &State,
    include_cursor: bool,
) -> Result<Screenshot, ScreenshooterError> {
    let ctx = match state.render_ctx.get() {
        Some(ctx) => ctx,
        _ => return Err(ScreenshooterError::NoRenderContext),
    };
    let extents = state.root.extents.get();
    if extents.is_empty() {
        return Err(ScreenshooterError::EmptyDisplay);
    }
    let formats = ctx.formats();
    let mut usage = GBM_BO_USE_RENDERING;
    let modifiers = match formats.get(&XRGB8888.drm) {
        None => return Err(ScreenshooterError::XRGB8888),
        Some(f) if f.write_modifiers.contains(&LINEAR_MODIFIER) => &[LINEAR_MODIFIER],
        Some(f) if f.write_modifiers.contains(&INVALID_MODIFIER) => {
            usage |= GBM_BO_USE_LINEAR;
            &[INVALID_MODIFIER]
        }
        Some(_) => return Err(ScreenshooterError::Linear),
    };
    let gbm = ctx.gbm();
    let bo = gbm.create_bo(
        &state.dma_buf_ids,
        extents.width(),
        extents.height(),
        XRGB8888,
        modifiers,
        usage,
    )?;
    let fb = ctx.clone().dmabuf_fb(bo.dmabuf())?;
    fb.render_node(
        state.root.deref(),
        state,
        include_cursor.then_some(state.root.extents.get()),
        None,
        Scale::from_int(1),
        true,
        false,
        Transform::None,
    )?;
    let drm = gbm.drm.dup_render()?.fd().clone();
    Ok(Screenshot { drm, bo })
}