use super::AnonymousObject;
use wayland_commons::utils::UserData;
use wayland_commons::Interface;

use wayland_sys::client::*;

use event_queue::QueueToken;

use imp::{NewProxyInner, ProxyInner};

use wayland_commons::MessageGroup;
use ProxyMap;

/// An handle to a wayland proxy
///
/// This represents a wayland object instantiated in your client
/// session. Several handles to the same object can exist at a given
/// time, and cloning them won't create a new protocol object, only
/// clone the handle. The lifetime of the protocol object is **not**
/// tied to the lifetime of these handles, but rather to sending or
/// receiving destroying messages.
///
/// These handles are notably used to send requests to the server. To do this
/// you need to convert them to the corresponding Rust object (using `.into()`)
/// and use methods on the Rust object.
pub struct Proxy<I: Interface> {
    _i: ::std::marker::PhantomData<&'static I>,
    pub(crate) inner: ProxyInner,
}

impl<I: Interface> Clone for Proxy<I> {
    fn clone(&self) -> Proxy<I> {
        Proxy {
            _i: ::std::marker::PhantomData,
            inner: self.inner.clone(),
        }
    }
}

impl<I: Interface> PartialEq for Proxy<I> {
    fn eq(&self, other: &Proxy<I>) -> bool {
        self.equals(other)
    }
}

impl<I: Interface> Eq for Proxy<I> {}

impl<I: Interface> Proxy<I> {
    pub(crate) fn wrap(inner: ProxyInner) -> Proxy<I> {
        Proxy {
            _i: ::std::marker::PhantomData,
            inner,
        }
    }

    /// Send a request through this object
    ///
    /// **Warning:** This method is mostly intended to be used by code generated
    /// by `wayland-scanner`, and you should probably never need to use it directly,
    /// but rather use the appropriate methods on the Rust object.
    ///
    /// This is the generic method to send requests.
    ///
    /// If your request needs to create an object, use `send_constructor`.
    pub fn send(&self, msg: I::Request) {
        #[cfg(feature = "native_lib")]
        {
            if !self.is_external() && !self.is_alive() {
                return;
            }
        }
        #[cfg(not(feature = "native_lib"))]
        {
            if !self.is_alive() {
                return;
            }
        }
        if msg.since() > self.version() {
            let opcode = msg.opcode() as usize;
            panic!(
                "Cannot send request {} which requires version >= {} on proxy {}@{} which is version {}.",
                I::Request::MESSAGES[opcode].name,
                msg.since(),
                I::NAME,
                self.id(),
                self.version()
            );
        }
        self.inner.send::<I>(msg)
    }

    /// Send a request creating an object through this object
    ///
    /// **Warning:** This method is mostly intended to be used by code generated
    /// by `wayland-scanner`, and you should probably never need to use it directly,
    /// but rather use the appropriate methods on the Rust object.
    ///
    /// This is the generic method to send requests that create objects
    ///
    /// The slot in the message corresponding with the newly created object must have
    /// been filled by a placeholder object (see `child_placeholder`).
    pub fn send_constructor<J, F>(
        &self,
        msg: I::Request,
        implementor: F,
        version: Option<u32>,
    ) -> Result<J, ()>
    where
        J: Interface + From<Proxy<J>>,
        F: FnOnce(NewProxy<J>) -> J,
    {
        if !self.is_alive() {
            return Err(());
        }
        if msg.since() > self.version() {
            let opcode = msg.opcode() as usize;
            panic!(
                "Cannot send request {} which requires version >= {} on proxy {}@{} which is version {}.",
                I::Request::MESSAGES[opcode].name,
                msg.since(),
                I::NAME,
                self.id(),
                self.version()
            );
        }
        self.inner
            .send_constructor::<I, J>(msg, version)
            .map(NewProxy::wrap)
            .map(implementor)
            .map(Into::into)
    }

    /// Check if the object associated with this proxy is still alive
    ///
    /// Will return `false` if the object has been destroyed.
    ///
    /// If the object is not managed by this library, this will always
    /// returns `true`.
    pub fn is_alive(&self) -> bool {
        self.inner.is_alive()
    }

    /// Retrieve the interface version of this wayland object instance
    ///
    /// Returns 0 on dead objects
    pub fn version(&self) -> u32 {
        self.inner.version()
    }

    /// Retrieve the object id of this wayland object
    pub fn id(&self) -> u32 {
        self.inner.id()
    }

    /// Access the arbitrary payload associated to this object
    ///
    /// You need to specify the expected type of this payload, and this
    /// function will return `None` if either the types don't match or
    /// you are attempting to access a non `Send + Sync` user data from the
    /// wrong thread.
    ///
    /// This value is associated to the Proxy when you implement it, and you
    /// cannot access it mutably afterwards. If you need interior mutability,
    /// you are responsible for using a `Mutex` or similar type to achieve it.
    pub fn user_data<UD: 'static>(&self) -> Option<&UD> {
        self.inner.get_user_data()
    }

    /// Check if the other proxy refers to the same underlying wayland object
    pub fn equals(&self, other: &Proxy<I>) -> bool {
        self.inner.equals(&other.inner)
    }

    /// Create a new child object
    ///
    /// **Warning:** This method is mostly intended to be used by code generated
    /// by `wayland-scanner`, and you should probably never need to use it directly,
    /// but rather use the appropriate methods on the Rust object.
    ///
    /// This creates a new wayland object, considered as a
    /// child of this object. It will notably inherit its interface
    /// version.
    ///
    /// The created object should immediately be implemented and sent
    /// in a request to the server, to keep the object list properly
    /// synchronized. Failure to do so will likely cause a protocol
    /// error.
    pub fn child<C: Interface>(&self) -> NewProxy<C> {
        NewProxy {
            _i: ::std::marker::PhantomData,
            inner: self.inner.child::<C>(),
        }
    }

    /// Creates a handle of this proxy with its actual type erased
    pub fn anonymize(&self) -> Proxy<AnonymousObject> {
        Proxy {
            _i: ::std::marker::PhantomData,
            inner: self.inner.clone(),
        }
    }

    /// Create a wrapper for this object for queue management
    ///
    /// As assigning a proxy to an event queue can be a racy operation
    /// in contexts involving multiple thread, this provides a facility
    /// to do this safely.
    ///
    /// The wrapper object created behaves like a regular `Proxy`, except that
    /// all objects created as the result of its requests will be assigned to
    /// the queue associated to the provided token, rather than the queue of
    /// their parent. This does not change the queue of the proxy itself.
    pub fn make_wrapper(&self, queue: &QueueToken) -> Result<I, ()>
    where
        I: From<Proxy<I>>,
    {
        let inner = self.inner.make_wrapper(&queue.inner)?;

        Ok(Proxy {
            _i: ::std::marker::PhantomData,
            inner,
        }
        .into())
    }

    /// Create a placeholder object, to be used with `send_constructor`
    ///
    /// **Warning:** This method is mostly intended to be used by code generated
    /// by `wayland-scanner`, and you should probably never need to use it directly,
    /// but rather use the appropriate methods on the Rust object.
    pub fn child_placeholder<J: Interface + From<Proxy<J>>>(&self) -> J {
        Proxy::wrap(self.inner.child_placeholder()).into()
    }
}

impl<I: Interface> Proxy<I> {
    /// Check whether this proxy is managed by the library or not
    ///
    /// See `from_c_ptr` for details.
    ///
    /// NOTE: This method will panic if called while the `native_lib` feature is
    /// not activated.
    pub fn is_external(&self) -> bool {
        #[cfg(feature = "native_lib")]
        {
            self.inner.is_external()
        }
        #[cfg(not(feature = "native_lib"))]
        {
            panic!("[wayland-client] C interfacing methods can only be used with the `native_lib` cargo feature.")
        }
    }

    /// Get a raw pointer to the underlying wayland object
    ///
    /// Retrieve a pointer to the object from the `libwayland-client.so` library.
    /// You will mostly need it to interface with C libraries needing access
    /// to wayland objects (to initialize an opengl context for example).
    ///
    /// NOTE: This method will panic if called while the `native_lib` feature is
    /// not activated.
    pub fn c_ptr(&self) -> *mut wl_proxy {
        #[cfg(feature = "native_lib")]
        {
            self.inner.c_ptr()
        }
        #[cfg(not(feature = "native_lib"))]
        {
            panic!("[wayland-client] C interfacing methods can only be used with the `native_lib` cargo feature.")
        }
    }

    /// Create a `Proxy` instance from a C pointer
    ///
    /// Create a `Proxy` from a raw pointer to a wayland object from the
    /// C library.
    ///
    /// If the pointer was previously obtained by the `c_ptr()` method, this
    /// constructs a new proxy for the same object just like the `clone()`
    /// method would have.
    ///
    /// If the object was created by some other C library you are interfacing
    /// with, it will be created in an "unmanaged" state: wayland-client will
    /// treat it as foreign, and as such most of the safeties will be absent.
    /// Notably the lifetime of the object can't be tracked, so the `alive()`
    /// method will always return `false` and you are responsible of not using
    /// an object past its destruction (as this would cause a protocol error).
    /// You will also be unable to associate any user data pointer to this object.
    ///
    /// In order to handle protocol races, invoking it with a NULL pointer will
    /// create an already-dead object.
    ///
    /// NOTE: This method will panic if called while the `native_lib` feature is
    /// not activated.
    pub unsafe fn from_c_ptr(_ptr: *mut wl_proxy) -> Proxy<I>
    where
        I: From<Proxy<I>>,
    {
        #[cfg(feature = "native_lib")]
        {
            Proxy {
                _i: ::std::marker::PhantomData,
                inner: ProxyInner::from_c_ptr::<I>(_ptr),
            }
        }
        #[cfg(not(feature = "native_lib"))]
        {
            panic!("[wayland-client] C interfacing methods can only be used with the `native_lib` cargo feature.")
        }
    }
}

#[cfg(feature = "native_lib")]
impl Proxy<::protocol::wl_display::WlDisplay> {
    pub(crate) unsafe fn from_c_display_wrapper(
        ptr: *mut wl_proxy,
    ) -> Proxy<::protocol::wl_display::WlDisplay> {
        Proxy {
            _i: ::std::marker::PhantomData,
            inner: ProxyInner::from_c_display_wrapper(ptr),
        }
    }
}

/// A newly-created proxy that needs implementation
///
/// Whenever a new wayland object is created, you will
/// receive it as a `NewProxy`. You then have to provide an
/// implementation for it, in order to process the incoming
/// events it may receive. Once this done you will be able
/// to use it as a regular Rust object.
///
/// Implementations are structs implementing the appropriate
/// variant of the `Implementation` trait. They can also be
/// closures.
pub struct NewProxy<I: Interface> {
    _i: ::std::marker::PhantomData<*const I>,
    pub(crate) inner: NewProxyInner,
}

impl<I: Interface + 'static> NewProxy<I> {
    #[allow(dead_code)]
    pub(crate) fn wrap(inner: NewProxyInner) -> NewProxy<I> {
        NewProxy {
            _i: ::std::marker::PhantomData,
            inner,
        }
    }

    /// Implement this proxy using the given handler and implementation data.
    ///
    /// The handler must be a struct implementing the `EventHandler` trait of the `I` interface.
    ///
    /// This must be called from the same thread as the one owning the event queue this
    /// new proxy is attached to and will panic otherwise. A proxy by default inherits
    /// the event queue of its parent object.
    ///
    /// If you don't want an object to inherits its parent queue, see `Proxy::make_wrapper`.
    ///
    /// If you want the user data to be accessed from other threads,
    /// see `implement_user_data_threadsafe`.
    pub fn implement<T, UD>(self, mut handler: T, user_data: UD) -> I
    where
        T: 'static,
        UD: 'static,
        I: HandledBy<T> + From<Proxy<I>>,
        I::Event: MessageGroup<Map = ProxyMap>,
    {
        let implementation = move |event, proxy: I| I::handle(&mut handler, event, proxy);

        self.implement_closure(implementation, user_data)
    }

    /// Implement this proxy using the given implementation closure and implementation data.
    ///
    /// This must be called from the same thread as the one owning the event queue this
    /// new proxy is attached to and will panic otherwise. A proxy by default inherits
    /// the event queue of its parent object.
    ///
    /// If you don't want an object to inherits its parent queue, see `Proxy::make_wrapper`.
    ///
    /// If you want the user data to be accessed from other threads,
    /// see `implement_closure_user_data_threadsafe`.
    pub fn implement_closure<F, UD>(self, implementation: F, user_data: UD) -> I
    where
        F: FnMut(I::Event, I) + 'static,
        UD: 'static,
        I: From<Proxy<I>>,
        I::Event: MessageGroup<Map = ProxyMap>,
    {
        if !self.inner.is_queue_on_current_thread() {
            panic!("Trying to implement a proxy with a non-Send implementation from an other thread than the one of its event queue.");
        }
        let inner = unsafe {
            self.inner
                .implement::<I, _>(implementation, UserData::new(user_data))
        };
        Proxy {
            _i: ::std::marker::PhantomData,
            inner,
        }
        .into()
    }

    /// Implement this proxy using a dummy handler which does nothing.
    pub fn implement_dummy(self) -> I
    where
        I: From<Proxy<I>>,
        I::Event: MessageGroup<Map = ProxyMap>,
    {
        self.implement_closure_threadsafe(|_, _| (), ())
    }

    /// Implement this proxy using the given handler and implementation data.
    ///
    /// The handler must be a struct implementing the `EventHandler` trait of the `I` interface.
    ///
    /// This function allows you to implement a proxy from an other thread than the one where its
    /// event queue is attach, but the implementation thus must be `Send`.
    pub fn implement_threadsafe<T, UD>(self, mut handler: T, user_data: UD) -> I
    where
        T: Send + 'static,
        UD: Send + Sync + 'static,
        I: HandledBy<T> + From<Proxy<I>>,
        I::Event: MessageGroup<Map = ProxyMap>,
    {
        let implementation = move |event, proxy: I| I::handle(&mut handler, event, proxy);

        self.implement_closure_threadsafe(implementation, user_data)
    }

    /// Implement this proxy using given closure and implementation data from any thread.
    ///
    /// This function allows you to implement a proxy from an other thread than the one where its
    /// event queue is attach, but the implementation thus must be `Send`.
    pub fn implement_closure_threadsafe<F, UD>(self, implementation: F, user_data: UD) -> I
    where
        F: FnMut(I::Event, I) + Send + 'static,
        UD: Send + Sync + 'static,
        I: From<Proxy<I>>,
        I::Event: MessageGroup<Map = ProxyMap>,
    {
        let inner = unsafe {
            self.inner
                .implement::<I, _>(implementation, UserData::new_threadsafe(user_data))
        };
        Proxy {
            _i: ::std::marker::PhantomData,
            inner,
        }
        .into()
    }

    /// Implement this proxy using the given handler and implementation data.
    ///
    /// This must be called from the same thread as the one owning the event queue this
    /// new proxy is attached to and will panic otherwise. However the user data
    /// may be accessed from other threads.
    pub fn implement_user_data_threadsafe<T, UD>(self, mut handler: T, user_data: UD) -> I
    where
        T: 'static,
        UD: Send + Sync + 'static,
        I: HandledBy<T> + From<Proxy<I>>,
        I::Event: MessageGroup<Map = ProxyMap>,
    {
        let implementation = move |event, proxy: I| I::handle(&mut handler, event, proxy);

        self.implement_closure_user_data_threadsafe(implementation, user_data)
    }

    /// Implement this proxy using the given closure and implementation data.
    ///
    /// This must be called from the same thread as the one owning the event queue this
    /// new proxy is attached to and will panic otherwise. However the user data
    /// may be accessed from other threads.
    pub fn implement_closure_user_data_threadsafe<F, UD>(self, implementation: F, user_data: UD) -> I
    where
        F: FnMut(I::Event, I) + 'static,
        UD: Send + Sync + 'static,
        I: From<Proxy<I>>,
        I::Event: MessageGroup<Map = ProxyMap>,
    {
        if !self.inner.is_queue_on_current_thread() {
            panic!("Trying to implement a proxy with a non-Send implementation from an other thread than the one of its event queue.");
        }
        let inner = unsafe {
            self.inner
                .implement::<I, _>(implementation, UserData::new_threadsafe(user_data))
        };
        Proxy {
            _i: ::std::marker::PhantomData,
            inner,
        }
        .into()
    }
}

impl<I: Interface + 'static> NewProxy<I> {
    /// Get a raw pointer to the underlying wayland object
    ///
    /// Retrieve a pointer to the object from the `libwayland-client.so` library.
    /// You will mostly need it to interface with C libraries needing access
    /// to wayland objects (to initialize an opengl context for example).
    ///
    /// Use this if you need to pass an unimplemented object to the C library
    /// you are interfacing with.
    ///
    /// NOTE: This method will panic if called while the `native_lib` feature is
    /// not activated.
    pub fn c_ptr(&self) -> *mut wl_proxy {
        #[cfg(feature = "native_lib")]
        {
            self.inner.c_ptr()
        }
        #[cfg(not(feature = "native_lib"))]
        {
            panic!("[wayland-client] C interfacing methods can only be used with the `native_lib` cargo feature.")
        }
    }

    /// Create a `NewProxy` instance from a C pointer.
    ///
    /// By doing so, you assert that this wayland object was newly created and
    /// can be safely implemented. As implementing it will overwrite any previously
    /// associated data or implementation, this can cause weird errors akin to
    /// memory corruption if it was not the case.
    ///
    /// NOTE: This method will panic if called while the `native_lib` feature is
    /// not activated.
    pub unsafe fn from_c_ptr(_ptr: *mut wl_proxy) -> Self {
        #[cfg(feature = "native_lib")]
        {
            NewProxy {
                _i: ::std::marker::PhantomData,
                inner: NewProxyInner::from_c_ptr(_ptr),
            }
        }
        #[cfg(not(feature = "native_lib"))]
        {
            panic!("[wayland-client] C interfacing methods can only be used with the `native_lib` cargo feature.")
        }
    }
}

/// Provides a callback function to handle events of the implementing interface via `T`.
///
/// This trait is meant to be implemented automatically by code generated with `wayland-scanner`.
pub trait HandledBy<T>: Interface + Sized {
    /// Handles an event.
    fn handle(handler: &mut T, event: Self::Event, proxy: Self);
}