extern crate drm;
#[macro_use]
extern crate glium;
extern crate rand;
extern crate libudev;
#[macro_use(define_roles)]
extern crate smithay;
extern crate wayland_server;

#[macro_use]
extern crate slog;
extern crate slog_async;
extern crate slog_term;

extern crate ctrlc;

mod helpers;

use drm::control::{Device as ControlDevice, ResourceInfo};
use drm::control::connector::{Info as ConnectorInfo, State as ConnectorState};
use drm::control::encoder::Info as EncoderInfo;
use drm::control::crtc;
use drm::result::Error as DrmError;
use glium::Surface;
use helpers::{init_shell, GliumDrawer, MyWindowMap, Roles, SurfaceData};
use slog::{Drain, Logger};
use smithay::backend::drm::{DrmBackend, DrmDevice, DrmHandler};
use smithay::backend::graphics::egl::EGLGraphicsBackend;
use smithay::backend::udev::{UdevBackend, UdevHandler, udev_backend_bind};
use smithay::backend::session::SessionNotifier;
use smithay::backend::session::direct::{direct_session_bind, DirectSession};
use smithay::wayland::compositor::{CompositorToken, SubsurfaceRole, TraversalAction};
use smithay::wayland::compositor::roles::Role;
use smithay::wayland::shell::ShellState;
use smithay::wayland::shm::init_shm_global;
use std::cell::RefCell;
use std::collections::HashSet;
use std::io::Error as IoError;
use std::rc::Rc;
use std::sync::atomic::{AtomicBool, Ordering};
use std::sync::Arc;
use std::time::Duration;
use wayland_server::{StateToken, StateProxy};

fn main() {
    // A logger facility, here we use the terminal for this example
    let log = Logger::root(
        slog_term::FullFormat::new(slog_term::PlainSyncDecorator::new(std::io::stdout())).build().fuse(),
        o!(),
    );

    // Initialize the wayland server
    let (mut display, mut event_loop) = wayland_server::create_display();

    /*
     * Initialize the compositor
     */
    init_shm_global(&mut event_loop, vec![], log.clone());

    let (compositor_token, shell_state_token, window_map) = init_shell(&mut event_loop, log.clone());

    /*
     * Initialize session on the current tty
     */
    let (session, mut notifier) = DirectSession::new(None, log.clone()).unwrap();
    let session_token = event_loop.state().insert(session);

    let running = Arc::new(AtomicBool::new(true));
    let r = running.clone();
    ctrlc::set_handler(move || {
        r.store(false, Ordering::SeqCst);
    }).expect("Error setting Ctrl-C handler");

    /*
     * Initialize the udev backend
     */
    let context = libudev::Context::new().unwrap();
    let udev
        = UdevBackend::new(&mut event_loop, &context, &session_token, UdevHandlerImpl {
            shell_state_token,
            compositor_token,
            window_map: window_map.clone(),
            logger: log.clone(),
        }, log.clone()).unwrap();

    let udev_token = event_loop.state().insert(udev);
    let udev_session_id = notifier.register(udev_token.clone());
    let session_event_source = direct_session_bind(notifier, &mut event_loop, log.clone()).unwrap();
    let udev_event_source = udev_backend_bind(&mut event_loop, udev_token).unwrap();
    /*
     * Add a listening socket:
     */
    let name = display.add_socket_auto().unwrap().into_string().unwrap();
    println!("Listening on socket: {}", name);

    while running.load(Ordering::SeqCst) {
        event_loop.dispatch(Some(16)).unwrap();
        display.flush_clients();
        window_map.borrow_mut().refresh();
    }

    let mut notifier = session_event_source.remove();
    notifier.unregister(udev_session_id);

    let udev_token = udev_event_source.remove();
    let udev = event_loop.state().remove(udev_token);
    udev.close(event_loop.state());

    event_loop.state().remove(session_token);
}

struct UdevHandlerImpl {
    shell_state_token: StateToken<ShellState<SurfaceData, Roles, (), ()>>,
    compositor_token: CompositorToken<SurfaceData, Roles, ()>,
    window_map: Rc<RefCell<MyWindowMap>>,
    logger: ::slog::Logger,
}

impl UdevHandlerImpl {
    pub fn scan_connectors<'a, S: Into<StateProxy<'a>>>(&self, state: S, device: &mut DrmDevice<GliumDrawer<DrmBackend>>) {
        // Get a set of all modesetting resource handles (excluding planes):
        let res_handles = device.resource_handles().unwrap();

        // Use first connected connector
        let connector_infos: Vec<ConnectorInfo> = res_handles
            .connectors()
            .iter()
            .map(|conn| {
                ConnectorInfo::load_from_device(device, *conn).unwrap()
            })
            .filter(|conn| conn.connection_state() == ConnectorState::Connected)
            .inspect(|conn| info!(self.logger, "Connected: {:?}", conn.connector_type()))
            .collect();

        let mut used_crtcs: HashSet<crtc::Handle> = HashSet::new();

        let mut state = state.into();

        // very naive way of finding good crtc/encoder/connector combinations. This problem is np-complete
        for connector_info in connector_infos {
            let encoder_infos = connector_info.encoders().iter().flat_map(|encoder_handle| EncoderInfo::load_from_device(device, *encoder_handle)).collect::<Vec<EncoderInfo>>();
            for encoder_info in encoder_infos {
                for crtc in res_handles.filter_crtcs(encoder_info.possible_crtcs()) {
                    if !used_crtcs.contains(&crtc) {
                        let mode = connector_info.modes()[0]; // Use first mode (usually highest resoltion, but in reality you should filter and sort and check and match with other connectors, if you use more then one.)
                        // create a backend
                        let renderer_token = device.create_backend(&mut state, crtc, mode, vec![connector_info.handle()]).unwrap();

                        // render first frame
                        {
                            let renderer = state.get_mut(renderer_token);
                            let mut frame = renderer.draw();
                            frame.clear_color(0.8, 0.8, 0.9, 1.0);
                            frame.finish().unwrap();
                        }

                        used_crtcs.insert(crtc);
                        break;
                    }
                }
            }
        }
    }
}

impl UdevHandler<GliumDrawer<DrmBackend>, DrmHandlerImpl> for UdevHandlerImpl {
    fn device_added<'a, S: Into<StateProxy<'a>>>(&mut self, state: S, device: &mut DrmDevice<GliumDrawer<DrmBackend>>) -> Option<DrmHandlerImpl>
    {
        self.scan_connectors(state, device);

        Some(DrmHandlerImpl {
            shell_state_token: self.shell_state_token.clone(),
            compositor_token: self.compositor_token.clone(),
            window_map: self.window_map.clone(),
            logger: self.logger.clone(),
        })
    }

    fn device_changed<'a, S: Into<StateProxy<'a>>>(&mut self, state: S, device: &StateToken<DrmDevice<GliumDrawer<DrmBackend>>>) {
        //quick and dirty
        let mut state = state.into();
        self.device_removed(&mut state, device);
        state.with_value(device, |state, device| self.scan_connectors(state, device));
    }

    fn device_removed<'a, S: Into<StateProxy<'a>>>(&mut self, state: S, device: &StateToken<DrmDevice<GliumDrawer<DrmBackend>>>) {
        state.into().with_value(device, |state, device| {
            let crtcs = device.current_backends().into_iter().map(|backend| state.get(backend).crtc()).collect::<Vec<crtc::Handle>>();
            let mut state: StateProxy = state.into();
            for crtc in crtcs {
                device.destroy_backend(&mut state, &crtc);
            }
        });
    }

    fn error<'a, S: Into<StateProxy<'a>>>(&mut self, _state: S, error: IoError) {
        error!(self.logger, "{:?}", error);
    }
}

pub struct DrmHandlerImpl {
    shell_state_token: StateToken<ShellState<SurfaceData, Roles, (), ()>>,
    compositor_token: CompositorToken<SurfaceData, Roles, ()>,
    window_map: Rc<RefCell<MyWindowMap>>,
    logger: ::slog::Logger,
}

impl DrmHandler<GliumDrawer<DrmBackend>> for DrmHandlerImpl {
    fn ready<'a, S: Into<StateProxy<'a>>>(&mut self, state: S, _device: &mut DrmDevice<GliumDrawer<DrmBackend>>,
             backend: &StateToken<GliumDrawer<DrmBackend>>, _crtc: crtc::Handle, _frame: u32, _duration: Duration) {
        let state = state.into();
        let drawer = state.get(backend);
        let mut frame = drawer.draw();
        frame.clear_color(0.8, 0.8, 0.9, 1.0);
        // redraw the frame, in a simple but inneficient way
        {
            let screen_dimensions = drawer.get_framebuffer_dimensions();
            self.window_map
                .borrow()
                .with_windows_from_bottom_to_top(|toplevel_surface, initial_place| {
                    if let Some(wl_surface) = toplevel_surface.get_surface() {
                        // this surface is a root of a subsurface tree that needs to be drawn
                        self.compositor_token
                            .with_surface_tree_upward(
                                wl_surface,
                                initial_place,
                                |_surface, attributes, role, &(mut x, mut y)| {
                                    if let Some((ref contents, (w, h))) = attributes.user_data.buffer {
                                        // there is actually something to draw !
                                        if let Ok(subdata) = Role::<SubsurfaceRole>::data(role) {
                                            x += subdata.x;
                                            y += subdata.y;
                                        }
                                        drawer.render(
                                            &mut frame,
                                            contents,
                                            (w, h),
                                            (x, y),
                                            screen_dimensions,
                                        );
                                        TraversalAction::DoChildren((x, y))
                                    } else {
                                        // we are not display, so our children are neither
                                        TraversalAction::SkipChildren
                                    }
                                },
                            )
                            .unwrap();
                    }
                });
        }
        if let Err(err) = frame.finish() {
            error!(self.logger, "Error during rendering: {:?}", err);
        }
    }

    fn error<'a, S: Into<StateProxy<'a>>>(&mut self, _state: S, _device: &mut DrmDevice<GliumDrawer<DrmBackend>>,
             error: DrmError) {
        error!(self.logger, "{:?}", error);
    }
}