// SPDX-License-Identifier: GPL-3.0-only

#[cfg(feature = "debug")]
use crate::state::Fps;

use crate::{
    backend::render,
    config::OutputConfig,
    shell::Shell,
    state::{BackendData, ClientState, Common, Data},
    utils::prelude::*,
};

use anyhow::{Context, Result};
use smithay::{
    backend::{
        allocator::{dmabuf::Dmabuf, gbm::GbmDevice, Format},
        drm::{DrmDevice, DrmEvent, DrmEventTime, DrmNode, GbmBufferedSurface, NodeType},
        egl::{EGLContext, EGLDevice, EGLDisplay},
        input::InputEvent,
        libinput::{LibinputInputBackend, LibinputSessionInterface},
        renderer::{
            gles2::Gles2Renderbuffer,
            multigpu::{egl::EglGlesBackend, GpuManager},
            Bind,
        },
        session::{auto::AutoSession, Session, Signal},
        udev::{all_gpus, primary_gpu, UdevBackend, UdevEvent},
    },
    reexports::{
        calloop::{
            timer::{TimeoutAction, Timer},
            Dispatcher, EventLoop, InsertError, LoopHandle, RegistrationToken,
        },
        drm::control::{connector, crtc, Device as ControlDevice, ModeTypeFlags},
        input::Libinput,
        nix::{fcntl::OFlag, sys::stat::dev_t},
        wayland_server::{
            protocol::{wl_output, wl_surface::WlSurface},
            DisplayHandle, Resource,
        },
    },
    utils::signaling::{Linkable, SignalToken, Signaler},
    wayland::{
        dmabuf::DmabufGlobal,
        output::{Mode as OutputMode, Output, PhysicalProperties},
    },
};

use std::{
    cell::RefCell,
    collections::{HashMap, HashSet},
    path::PathBuf,
    rc::Rc,
    time::Duration,
};

mod drm_helpers;
mod session_fd;
mod socket;
use session_fd::*;
use socket::*;

pub struct KmsState {
    devices: HashMap<DrmNode, Device>,
    api: GpuManager<EglGlesBackend>,
    pub primary: DrmNode,
    session: AutoSession,
    signaler: Signaler<Signal>,
    _restart_token: SignalToken,
    _tokens: Vec<RegistrationToken>,
}

pub struct Device {
    render_node: DrmNode,
    surfaces: HashMap<crtc::Handle, Surface>,
    allocator: Rc<RefCell<GbmDevice<SessionFd>>>,
    drm: Dispatcher<'static, DrmDevice<SessionFd>, Data>,
    formats: HashSet<Format>,
    supports_atomic: bool,
    event_token: Option<RegistrationToken>,
    socket: Option<Socket>,
}

pub struct Surface {
    surface: Option<GbmBufferedSurface<Rc<RefCell<GbmDevice<SessionFd>>>, SessionFd>>,
    connector: connector::Handle,
    output: Output,
    last_submit: Option<DrmEventTime>,
    refresh_rate: u32,
    vrr: bool,
    pending: bool,
    render_timer_token: Option<RegistrationToken>,
    #[cfg(feature = "debug")]
    fps: Fps,
}

pub fn init_backend(
    dh: &DisplayHandle,
    event_loop: &mut EventLoop<'static, Data>,
    state: &mut State,
) -> Result<()> {
    let (session, notifier) = AutoSession::new(None).context("Failed to acquire session")?;
    let signaler = notifier.signaler();

    let udev_backend = UdevBackend::new(session.seat(), None)?;
    let mut libinput_context =
        Libinput::new_with_udev::<LibinputSessionInterface<AutoSession>>(session.clone().into());
    libinput_context
        .udev_assign_seat(&session.seat())
        .map_err(|_| anyhow::anyhow!("Failed to assign seat to libinput"))?;
    let mut libinput_backend = LibinputInputBackend::new(libinput_context, None);
    libinput_backend.link(signaler.clone());

    let libinput_event_source = event_loop
        .handle()
        .insert_source(libinput_backend, move |mut event, _, data| {
            if let &mut InputEvent::DeviceAdded { ref mut device } = &mut event {
                data.state.common.config.read_device(device);
            }
            data.state
                .process_input_event(&data.display.handle(), event);
            for output in data.state.common.shell.outputs() {
                if let Err(err) = data
                    .state
                    .backend
                    .kms()
                    .schedule_render(&data.state.common.event_loop_handle, output)
                {
                    slog_scope::crit!(
                        "Error scheduling event loop for output {}: {:?}",
                        output.name(),
                        err
                    );
                }
            }
        })
        .map_err(|err| err.error)
        .context("Failed to initialize libinput event source")?;
    let session_event_source = event_loop
        .handle()
        .insert_source(notifier, |(), &mut (), _state| {})
        .map_err(|err| err.error)
        .context("Failed to initialize session event source")?;

    let api = GpuManager::new(EglGlesBackend, None).context("Failed to initialize renderers")?;

    // TODO get this info from system76-power, if available and setup a watcher
    let primary = if let Some(path) = std::env::var("COSMIC_RENDER_DEVICE")
        .ok()
        .and_then(|x| DrmNode::from_path(x).ok())
    {
        path
    } else {
        primary_gpu(session.seat())
            .ok()
            .flatten()
            .and_then(|x| DrmNode::from_path(x).ok())
            .and_then(|x| x.node_with_type(NodeType::Render).and_then(Result::ok))
            .unwrap_or_else(|| {
                for dev in all_gpus(session.seat()).expect("No GPU found") {
                    if let Some(node) = DrmNode::from_path(dev)
                        .ok()
                        .and_then(|x| x.node_with_type(NodeType::Render).and_then(Result::ok))
                    {
                        return node;
                    }
                }
                panic!("Failed to initialize any GPU");
            })
    };
    slog_scope::info!("Using {} as primary gpu for rendering", primary);

    let udev_dispatcher = Dispatcher::new(udev_backend, move |event, _, data: &mut Data| {
        match match event {
            UdevEvent::Added { device_id, path } => data
                .state
                .device_added(device_id, path, &data.display.handle())
                .with_context(|| format!("Failed to add drm device: {}", device_id)),
            UdevEvent::Changed { device_id } => data
                .state
                .device_changed(device_id)
                .with_context(|| format!("Failed to update drm device: {}", device_id)),
            UdevEvent::Removed { device_id } => data
                .state
                .device_removed(device_id, &data.display.handle())
                .with_context(|| format!("Failed to remove drm device: {}", device_id)),
        } {
            Ok(()) => {
                slog_scope::debug!("Successfully handled udev event")
            }
            Err(err) => {
                slog_scope::error!("Error while handling udev event: {}", err)
            }
        }
    });
    let udev_event_source = event_loop
        .handle()
        .register_dispatcher(udev_dispatcher.clone())
        .unwrap();

    let handle = event_loop.handle();
    let loop_signal = state.common.event_loop_signal.clone();
    let dispatcher = udev_dispatcher.clone();
    let _restart_token = signaler.register(move |signal| {
        if let Signal::ActivateSession = signal {
            let dispatcher = dispatcher.clone();
            handle.insert_idle(move |data| {
                for (dev, path) in dispatcher.as_source_ref().device_list() {
                    let drm_node = match DrmNode::from_dev_id(dev) {
                        Ok(node) => node,
                        Err(err) => {
                            slog_scope::error!(
                                "Failed to read drm device {}: {}",
                                path.display(),
                                err
                            );
                            continue;
                        }
                    };
                    if data.state.backend.kms().devices.contains_key(&drm_node) {
                        if let Err(err) = data.state.device_changed(dev) {
                            slog_scope::error!(
                                "Failed to update drm device {}: {}",
                                path.display(),
                                err
                            );
                        }
                    } else {
                        if let Err(err) =
                            data.state
                                .device_added(dev, path.into(), &data.display.handle())
                        {
                            slog_scope::error!(
                                "Failed to add drm device {}: {}",
                                path.display(),
                                err
                            );
                        }
                    }
                }
                data.state.common.output_configuration_state.update();

                data.state.common.config.read_outputs(
                    data.state.common.output_configuration_state.outputs(),
                    &mut data.state.backend,
                    &mut data.state.common.shell,
                    &data.state.common.event_loop_handle,
                );
                data.state.common.shell.refresh_outputs();
                data.state
                    .common
                    .config
                    .write_outputs(data.state.common.output_configuration_state.outputs());

                for surface in data
                    .state
                    .backend
                    .kms()
                    .devices
                    .values_mut()
                    .flat_map(|d| d.surfaces.values_mut())
                {
                    surface.pending = false;
                }
                for output in data.state.common.shell.outputs() {
                    if let Err(err) = data
                        .state
                        .backend
                        .kms()
                        .schedule_render(&data.state.common.event_loop_handle, output)
                    {
                        slog_scope::crit!(
                            "Error scheduling event loop for output {}: {:?}",
                            output.name(),
                            err
                        );
                    }
                }
            });
            loop_signal.wakeup();
        }
    });

    state.backend = BackendData::Kms(KmsState {
        api,
        _tokens: vec![
            libinput_event_source,
            session_event_source,
            udev_event_source,
        ],
        primary,
        session,
        signaler,
        _restart_token,
        devices: HashMap::new(),
    });

    for (dev, path) in udev_dispatcher.as_source_ref().device_list() {
        state
            .device_added(dev, path.into(), dh)
            .with_context(|| format!("Failed to add drm device: {}", path.display()))?;
    }
    Ok(())
}

impl State {
    fn device_added(&mut self, dev: dev_t, path: PathBuf, dh: &DisplayHandle) -> Result<()> {
        if !self.backend.kms().session.is_active() {
            return Ok(());
        }

        let fd = SessionFd::new(
            self.backend
                .kms()
                .session
                .open(
                    &path,
                    OFlag::O_RDWR | OFlag::O_CLOEXEC | OFlag::O_NOCTTY | OFlag::O_NONBLOCK,
                )
                .with_context(|| {
                    format!(
                        "Failed to optain file descriptor for drm device: {}",
                        path.display()
                    )
                })?,
        );
        let mut drm = DrmDevice::new(fd.clone(), false, None)
            .with_context(|| format!("Failed to initialize drm device for: {}", path.display()))?;
        let drm_node = DrmNode::from_dev_id(dev)?;
        let supports_atomic = drm.is_atomic();

        let gbm = GbmDevice::new(fd)
            .with_context(|| format!("Failed to initialize GBM device for {}", path.display()))?;
        let egl_display = EGLDisplay::new(&gbm, None).with_context(|| {
            format!("Failed to create EGLDisplay for device: {}", path.display())
        })?;
        let egl_device = EGLDevice::device_for_display(&egl_display).with_context(|| {
            format!("Unable to find matching egl device for {}", path.display())
        })?;
        let render_node = egl_device
            .try_get_render_node()
            .ok()
            .and_then(std::convert::identity)
            .with_context(|| {
                format!(
                    "Failed to determine path of egl device for {}",
                    path.display()
                )
            })?;
        let egl_context = EGLContext::new(&egl_display, None).with_context(|| {
            format!(
                "Failed to create EGLContext for device {:?}:{}",
                egl_device,
                path.display()
            )
        })?;
        let formats = egl_context.dmabuf_render_formats().clone();

        drm.link(self.backend.kms().signaler.clone());
        let dispatcher =
            Dispatcher::new(drm, move |event, metadata, data: &mut Data| match event {
                DrmEvent::VBlank(crtc) => {
                    if let Some(device) = data.state.backend.kms().devices.get_mut(&drm_node) {
                        if let Some(surface) = device.surfaces.get_mut(&crtc) {
                            match surface.surface.as_mut().map(|x| x.frame_submitted()) {
                                Some(Ok(_)) => {
                                    surface.last_submit = metadata.take().map(|data| data.time);
                                    surface.pending = false;
                                    data.state
                                        .common
                                        .shell
                                        .active_space_mut(&surface.output)
                                        .space
                                        .send_frames(
                                            data.state.common.start_time.elapsed().as_millis()
                                                as u32,
                                        );
                                }
                                Some(Err(err)) => {
                                    slog_scope::warn!("Failed to submit frame: {}", err)
                                }
                                None => {} // got disabled
                            };
                        }
                    }
                }
                DrmEvent::Error(err) => {
                    slog_scope::warn!("Failed to read events of device {:?}: {}", dev, err);
                }
            });
        let token = self
            .common
            .event_loop_handle
            .register_dispatcher(dispatcher.clone())
            .with_context(|| format!("Failed to add drm device to event loop: {}", dev))?;

        let socket = match self.create_socket(dh, render_node, formats.clone().into_iter()) {
            Ok(socket) => Some(socket),
            Err(err) => {
                slog_scope::warn!(
                    "Failed to initialize hardware-acceleration for clients on {}: {}",
                    render_node,
                    err
                );
                None
            }
        };

        let mut device = Device {
            render_node,
            surfaces: HashMap::new(),
            allocator: Rc::new(RefCell::new(gbm)),
            drm: dispatcher,
            formats,
            supports_atomic,
            event_token: Some(token),
            socket,
        };

        let outputs = device.enumerate_surfaces()?.added; // There are no removed outputs on newly added devices
        let mut wl_outputs = Vec::new();
        let mut w = self.common.shell.global_space().size.w;
        for (crtc, conn) in outputs {
            match device.setup_surface(crtc, conn, (w, 0)) {
                Ok(output) => {
                    w += output
                        .user_data()
                        .get::<RefCell<OutputConfig>>()
                        .unwrap()
                        .borrow()
                        .mode_size()
                        .w;
                    wl_outputs.push(output);
                }
                Err(err) => slog_scope::warn!("Failed to initialize output: {}", err),
            };
        }
        self.backend.kms().devices.insert(drm_node, device);

        self.common
            .output_configuration_state
            .add_heads(wl_outputs.iter());
        self.common.output_configuration_state.update();
        for output in wl_outputs {
            if let Err(err) = self.backend.kms().apply_config_for_output(
                &output,
                &mut self.common.shell,
                false,
                &self.common.event_loop_handle,
            ) {
                slog_scope::warn!("Failed to initialize output: {}", err);
            }
        }
        self.common.config.read_outputs(
            self.common.output_configuration_state.outputs(),
            &mut self.backend,
            &mut self.common.shell,
            &self.common.event_loop_handle,
        );
        self.common.shell.refresh_outputs();
        self.common
            .config
            .write_outputs(self.common.output_configuration_state.outputs());

        Ok(())
    }

    fn device_changed(&mut self, dev: dev_t) -> Result<()> {
        if !self.backend.kms().session.is_active() {
            return Ok(());
        }

        let drm_node = DrmNode::from_dev_id(dev)?;
        let mut outputs_removed = Vec::new();
        let mut outputs_added = Vec::new();
        if let Some(device) = self.backend.kms().devices.get_mut(&drm_node) {
            let changes = device.enumerate_surfaces()?;
            let mut w = self.common.shell.global_space().size.w;
            for crtc in changes.removed {
                if let Some(surface) = device.surfaces.remove(&crtc) {
                    if let Some(token) = surface.render_timer_token {
                        self.common.event_loop_handle.remove(token);
                    }
                    w -= surface.output.current_mode().map(|m| m.size.w).unwrap_or(0);
                    outputs_removed.push(surface.output.clone());
                }
            }
            for (crtc, conn) in changes.added {
                match device.setup_surface(crtc, conn, (w, 0)) {
                    Ok(output) => {
                        w += output
                            .user_data()
                            .get::<RefCell<OutputConfig>>()
                            .unwrap()
                            .borrow()
                            .mode_size()
                            .w;
                        outputs_added.push(output);
                    }
                    Err(err) => slog_scope::warn!("Failed to initialize output: {}", err),
                };
            }
        }

        self.common
            .output_configuration_state
            .remove_heads(outputs_removed.iter());
        self.common
            .output_configuration_state
            .add_heads(outputs_added.iter());
        for output in outputs_added {
            if let Err(err) = self.backend.kms().apply_config_for_output(
                &output,
                &mut self.common.shell,
                false,
                &self.common.event_loop_handle,
            ) {
                slog_scope::warn!("Failed to initialize output: {}", err);
            }
        }
        self.common.output_configuration_state.update();
        self.common.config.read_outputs(
            self.common.output_configuration_state.outputs(),
            &mut self.backend,
            &mut self.common.shell,
            &self.common.event_loop_handle,
        );
        self.common.shell.refresh_outputs();
        self.common
            .config
            .write_outputs(self.common.output_configuration_state.outputs());

        Ok(())
    }

    fn device_removed(&mut self, dev: dev_t, dh: &DisplayHandle) -> Result<()> {
        let drm_node = DrmNode::from_dev_id(dev)?;
        let mut outputs_removed = Vec::new();
        if let Some(mut device) = self.backend.kms().devices.remove(&drm_node) {
            for surface in device.surfaces.values_mut() {
                if let Some(token) = surface.render_timer_token.take() {
                    self.common.event_loop_handle.remove(token);
                }
                outputs_removed.push(surface.output.clone());
            }
            if let Some(token) = device.event_token.take() {
                self.common.event_loop_handle.remove(token);
            }
            if let Some(socket) = device.socket.take() {
                self.common.event_loop_handle.remove(socket.token);
                self.common
                    .dmabuf_state
                    .destroy_global::<State>(dh, socket.dmabuf_global);
                dh.remove_global(socket.drm_global);
            }
        }
        self.common
            .output_configuration_state
            .remove_heads(outputs_removed.iter());
        self.common.output_configuration_state.update();

        if self.backend.kms().session.is_active() {
            self.common.config.read_outputs(
                self.common.output_configuration_state.outputs(),
                &mut self.backend,
                &mut self.common.shell,
                &self.common.event_loop_handle,
            );
            self.common.shell.refresh_outputs();
            self.common
                .config
                .write_outputs(self.common.output_configuration_state.outputs());
        }

        Ok(())
    }
}

pub struct OutputChanges {
    pub added: Vec<(crtc::Handle, connector::Handle)>,
    pub removed: Vec<crtc::Handle>,
}

impl Device {
    pub fn enumerate_surfaces(&mut self) -> Result<OutputChanges> {
        let drm = &mut *self.drm.as_source_mut();

        // enumerate our outputs
        let config = drm_helpers::display_configuration(drm, self.supports_atomic)?;

        let surfaces = self
            .surfaces
            .iter()
            .map(|(c, s)| (*c, s.connector))
            .collect::<HashMap<crtc::Handle, connector::Handle>>();

        let added = config
            .iter()
            .filter(|(conn, crtc)| surfaces.get(&crtc).map(|c| c != *conn).unwrap_or(true))
            .map(|(conn, crtc)| (crtc, conn))
            .map(|(crtc, conn)| (*crtc, *conn))
            .collect::<Vec<_>>();
        let removed = surfaces
            .iter()
            .filter(|(crtc, conn)| config.get(conn).map(|c| c != *crtc).unwrap_or(true))
            .map(|(crtc, _)| *crtc)
            .collect::<Vec<_>>();

        Ok(OutputChanges { added, removed })
    }

    fn setup_surface(
        &mut self,
        crtc: crtc::Handle,
        conn: connector::Handle,
        position: (i32, i32),
    ) -> Result<Output> {
        let drm = &mut *self.drm.as_source_mut();
        let crtc_info = drm.get_crtc(crtc)?;
        let conn_info = drm.get_connector(conn)?;
        let vrr = drm_helpers::set_vrr(drm, crtc, conn, true).unwrap_or(false);
        let interface = drm_helpers::interface_name(drm, conn)?;
        let edid_info = drm_helpers::edid_info(drm, conn)?;
        let mode = crtc_info.mode().unwrap_or_else(|| {
            conn_info
                .modes()
                .iter()
                .find(|mode| mode.mode_type().contains(ModeTypeFlags::PREFERRED))
                .copied()
                .unwrap_or(conn_info.modes()[0])
        });
        let refresh_rate = drm_helpers::calculate_refresh_rate(mode);
        let output_mode = OutputMode {
            size: (mode.size().0 as i32, mode.size().1 as i32).into(),
            refresh: refresh_rate as i32,
        };
        let (phys_w, phys_h) = conn_info.size().unwrap_or((0, 0));
        let output = Output::new(
            interface,
            PhysicalProperties {
                size: (phys_w as i32, phys_h as i32).into(),
                // TODO: We need to read that from the connector properties
                subpixel: wl_output::Subpixel::Unknown,
                make: edid_info.manufacturer,
                model: edid_info.model,
            },
            None,
        );
        for mode in conn_info.modes() {
            let refresh_rate = drm_helpers::calculate_refresh_rate(*mode);
            let mode = OutputMode {
                size: (mode.size().0 as i32, mode.size().1 as i32).into(),
                refresh: refresh_rate as i32,
            };
            output.add_mode(mode);
        }
        output.set_preferred(output_mode);
        output.change_current_state(
            Some(output_mode),
            // TODO: Readout property for monitor rotation
            Some(wl_output::Transform::Normal),
            None,
            Some(position.into()),
        );
        output.user_data().insert_if_missing(|| {
            RefCell::new(OutputConfig {
                mode: ((output_mode.size.w, output_mode.size.h), Some(refresh_rate)),
                vrr,
                position,
                ..Default::default()
            })
        });

        let data = Surface {
            output: output.clone(),
            surface: None,
            connector: conn,
            vrr,
            refresh_rate,
            last_submit: None,
            pending: false,
            render_timer_token: None,
            #[cfg(feature = "debug")]
            fps: Fps::default(),
        };
        self.surfaces.insert(crtc, data);

        Ok(output)
    }
}

const MAX_CPU_COPIES: usize = 3;

fn render_node_for_output(
    dh: &DisplayHandle,
    output: &Output,
    target_node: DrmNode,
    shell: &Shell,
) -> DrmNode {
    let workspace = shell.active_space(output);
    let nodes = workspace
        .get_fullscreen(output)
        .map(|w| vec![w])
        .unwrap_or_else(|| workspace.space.windows().collect::<Vec<_>>())
        .into_iter()
        .flat_map(|w| {
            dh.get_client(w.toplevel().wl_surface().id())
                .ok()?
                .get_data::<ClientState>()
                .unwrap()
                .drm_node
                .clone()
        })
        .collect::<Vec<_>>();
    if nodes.contains(&target_node) || nodes.len() < MAX_CPU_COPIES {
        target_node
    } else {
        nodes
            .iter()
            .fold(HashMap::new(), |mut count_map, node| {
                let count = count_map.entry(node).or_insert(0);
                *count += 1;
                count_map
            })
            .into_iter()
            .reduce(|a, b| if a.1 > b.1 { a } else { b })
            .map(|(node, _)| *node)
            .unwrap_or(target_node)
    }
}

impl Surface {
    pub fn render_output(
        &mut self,
        dh: &DisplayHandle,
        api: &mut GpuManager<EglGlesBackend>,
        target_node: &DrmNode,
        state: &mut Common,
    ) -> Result<()> {
        if self.surface.is_none() {
            return Ok(());
        }

        if render::needs_buffer_reset(&self.output, state) {
            self.surface.as_mut().unwrap().reset_buffers();
        }

        let render_node = render_node_for_output(dh, &self.output, *target_node, &state.shell);
        let mut renderer = api.renderer(&render_node, &target_node).unwrap();

        let surface = self.surface.as_mut().unwrap();
        let (buffer, age) = surface
            .next_buffer()
            .with_context(|| "Failed to allocate buffer")?;

        renderer
            .bind(buffer)
            .with_context(|| "Failed to bind buffer")?;

        match render::render_output(
            Some(&render_node),
            &mut renderer,
            age,
            state,
            &self.output,
            false,
            #[cfg(feature = "debug")]
            &mut self.fps,
        ) {
            Ok(_) => {
                surface
                    .queue_buffer()
                    .with_context(|| "Failed to submit buffer for display")?;
            }
            Err(err) => {
                surface.reset_buffers();
                anyhow::bail!("Rendering failed: {}", err);
            }
        };
        Ok(())
    }
}

impl KmsState {
    pub fn switch_vt(&mut self, num: i32) -> Result<(), anyhow::Error> {
        self.session.change_vt(num).map_err(Into::into)
    }

    pub fn apply_config_for_output(
        &mut self,
        output: &Output,
        shell: &mut Shell,
        test_only: bool,
        loop_handle: &LoopHandle<'_, Data>,
    ) -> Result<(), anyhow::Error> {
        let recreated = if let Some(device) = self
            .devices
            .values_mut()
            .find(|dev| dev.surfaces.values().any(|s| s.output == *output))
        {
            let (crtc, mut surface) = device
                .surfaces
                .iter_mut()
                .find(|(_, s)| s.output == *output)
                .unwrap();
            let output_config = output
                .user_data()
                .get::<RefCell<OutputConfig>>()
                .unwrap()
                .borrow();

            if !output_config.enabled {
                if !test_only {
                    if surface.surface.take().is_some() {
                        // just drop it
                        shell.remove_output(output);
                    }
                }
                false
            } else {
                let drm = &mut *device.drm.as_source_mut();
                let conn = surface.connector;
                let conn_info = drm.get_connector(conn)?;
                let mode = conn_info
                    .modes()
                    .iter()
                    .min_by_key(|mode| {
                        let refresh_rate = drm_helpers::calculate_refresh_rate(**mode);
                        (output_config.mode.1.unwrap() as i32 - refresh_rate as i32).abs()
                    })
                    .ok_or(anyhow::anyhow!("Unknown mode"))?;

                if !test_only {
                    if let Some(gbm_surface) = surface.surface.as_mut() {
                        if output_config.vrr != surface.vrr {
                            surface.vrr = drm_helpers::set_vrr(
                                drm,
                                *crtc,
                                conn_info.handle(),
                                output_config.vrr,
                            )?;
                        }
                        gbm_surface.use_mode(*mode).unwrap();
                        false
                    } else {
                        surface.vrr = drm_helpers::set_vrr(drm, *crtc, conn, output_config.vrr)
                            .unwrap_or(false);
                        surface.refresh_rate = drm_helpers::calculate_refresh_rate(*mode);
                        let mut drm_surface = drm.create_surface(*crtc, *mode, &[conn])?;
                        drm_surface.link(self.signaler.clone());

                        let target = GbmBufferedSurface::new(
                            drm_surface,
                            device.allocator.clone(),
                            device.formats.clone(),
                            None,
                        )
                        .with_context(|| {
                            format!(
                                "Failed to initialize Gbm surface for {}",
                                drm_helpers::interface_name(drm, conn)
                                    .unwrap_or_else(|_| String::from("Unknown"))
                            )
                        })?;
                        surface.surface = Some(target);
                        shell.add_output(output);
                        true
                    }
                } else {
                    false
                }
            }
        } else {
            false
        };

        shell.refresh_outputs();
        if recreated {
            if let Err(err) = self.schedule_render(loop_handle, output) {
                slog_scope::crit!(
                    "Error scheduling event loop for output {}: {:?}",
                    output.name(),
                    err
                );
            }
        }
        Ok(())
    }
    pub fn target_node_for_output(&self, output: &Output) -> Option<DrmNode> {
        self.devices
            .iter()
            .find(|(_, dev)| dev.surfaces.values().any(|s| s.output == *output))
            .map(|(target, _)| target)
            .copied()
    }

    pub fn try_early_import(
        &mut self,
        dh: &DisplayHandle,
        surface: &WlSurface,
        output: &Output,
        target: DrmNode,
        shell: &Shell,
    ) {
        let render = render_node_for_output(dh, &output, target, &shell);
        if let Err(err) = self.api.early_import(
            dh.get_client(surface.id())
                .ok()
                .and_then(|c| c.get_data::<ClientState>().unwrap().drm_node.clone()),
            render,
            surface,
        ) {
            slog_scope::debug!("Early import failed: {}", err);
        }
    }

    pub fn dmabuf_imported(
        &mut self,
        _dh: &DisplayHandle,
        global: &DmabufGlobal,
        dmabuf: Dmabuf,
    ) -> Result<()> {
        use smithay::backend::renderer::ImportDma;

        for device in self.devices.values() {
            if device
                .socket
                .as_ref()
                .map(|s| &s.dmabuf_global == global)
                .unwrap_or(false)
            {
                return self
                    .api
                    .renderer::<Gles2Renderbuffer>(&device.render_node, &device.render_node)?
                    .import_dmabuf(&dmabuf, None)
                    .map(|_| ())
                    .map_err(Into::into);
            }
        }
        unreachable!()
    }

    pub fn schedule_render(
        &mut self,
        loop_handle: &LoopHandle<'_, Data>,
        output: &Output,
    ) -> Result<(), InsertError<Timer>> {
        if let Some((device, crtc, surface)) = self
            .devices
            .iter_mut()
            .flat_map(|(node, d)| d.surfaces.iter_mut().map(move |(c, s)| (node, c, s)))
            .find(|(_, _, s)| s.output == *output)
        {
            if surface.surface.is_none() {
                return Ok(());
            }
            if !surface.pending {
                surface.pending = true;
                /*
                let instant = surface
                    .last_submit
                    .as_ref()
                    .and_then(|x| match x {
                        DrmEventTime::Monotonic(instant) => Some(instant),
                        DrmEventTime::Realtime(_) => None,
                    })
                    .map(|i| {
                        *i + Duration::from_secs_f64(1.0 / surface.refresh_rate as f64)
                            - Duration::from_millis(20) // render budget
                    });
                */

                let device = *device;
                let crtc = *crtc;
                surface.render_timer_token = Some(loop_handle.insert_source(
                    //if surface.vrr || instant.is_none() {
                    Timer::immediate(), /*} else {
                                            Timer::from_deadline(instant.unwrap())
                                        }*/
                    move |_time, _, data| {
                        let backend = data.state.backend.kms();
                        if let Some(device) = backend.devices.get_mut(&device) {
                            if let Some(surface) = device.surfaces.get_mut(&crtc) {
                                if let Err(err) = surface.render_output(
                                    &data.display.handle(),
                                    &mut backend.api,
                                    &device.render_node,
                                    &mut data.state.common,
                                ) {
                                    slog_scope::error!("Error rendering: {}", err);
                                    return TimeoutAction::ToDuration(Duration::from_secs_f64(
                                        1.0 / surface.refresh_rate as f64,
                                    ));
                                }
                            }
                        }
                        TimeoutAction::Drop
                    },
                )?);
            }
        }
        Ok(())
    }
}