/*
 * Copyright (c) 2017 TOYOTA MOTOR CORPORATION
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef WM_WAYLAND_HPP
#define WM_WAYLAND_HPP

#include "controller_hooks.hpp"
#include "ivi-controller-client-protocol.h"
#include "util.hpp"

#include <functional>
#include <memory>
#include <unordered_map>
#include <vector>

/**
 * @struct wayland_proxy
 */
template <typename ProxyT>
struct wayland_proxy {
   std::unique_ptr<ProxyT, std::function<void(ProxyT *)>> proxy;
   wayland_proxy(wayland_proxy const &) = delete;
   wayland_proxy &operator=(wayland_proxy const &) = delete;
   wayland_proxy(void *p)
      : wayland_proxy(p,
                      reinterpret_cast<void (*)(ProxyT *)>(wl_proxy_destroy)) {}
   wayland_proxy(void *p, std::function<void(ProxyT *)> &&p_del)
      : proxy(std::unique_ptr<ProxyT, std::function<void(ProxyT *)>>(
           static_cast<ProxyT *>(p), p_del)) {}
};

/**
 * namespace wl
 */
namespace wl {

/**
 * @struct registry
 */
struct registry : public wayland_proxy<struct wl_registry> {
   typedef std::function<void(struct wl_registry *, uint32_t, uint32_t)> binder;
   std::unordered_map<std::string, binder> bindings;

   registry(registry const &) = delete;
   registry &operator=(registry const &) = delete;
   registry(struct wl_display *d);

   void add_global_handler(char const *iface, binder bind);

   // Events
   void global(uint32_t name, char const *iface, uint32_t v);
   void global_remove(uint32_t name);
};

/**
 * @struct display
 */
struct display {
   std::unique_ptr<struct wl_display, void (*)(struct wl_display *)> d;
   struct registry r;

   display(display const &) = delete;
   display &operator=(display const &) = delete;
   display();
   bool ok() const;
   void roundtrip();
   int dispatch();
   int dispatch_pending();
   int read_events();
   void flush();
   int get_fd() const;
   int get_error();

   // Lets just proxy this for the registry
   inline void add_global_handler(char const *iface, registry::binder bind) {
      this->r.add_global_handler(iface, bind);
   }
};

/**
 * @struct output
 */
struct output : wayland_proxy<struct wl_output> {
   int width{};
   int height{};
   int refresh{};
   int transform{};

   output(output const &) = delete;
   output &operator=(output const &) = delete;
   output(struct wl_registry *r, uint32_t name, uint32_t v);

   // Events
   void geometry(int32_t x, int32_t y, int32_t pw, int32_t ph, int32_t subpel,
                 char const *make, char const *model, int32_t tx);
   void mode(uint32_t flags, int32_t w, int32_t h, int32_t r);
   void done();
   void scale(int32_t factor);
};
}  // namespace wl

/**
 * namespace compositor
 */
namespace compositor {

struct size {
   uint32_t w, h;
};

struct rect {
   int32_t w, h;
   int32_t x, y;
};

static const constexpr rect full_rect = rect{-1, -1, 0, 0};

inline bool operator==(struct rect a, struct rect b) {
   return a.w == b.w && a.h == b.h && a.x == b.x && a.y == b.y;
}

struct controller;

struct controller_child {
   struct controller *parent;
   uint32_t id;

   controller_child(controller_child const &) = delete;
   controller_child &operator=(controller_child const &) = delete;
   controller_child(struct controller *c, uint32_t i) : parent(c), id(i) {}
   virtual ~controller_child() {}
};

struct surface_properties {
   uint32_t id;  // let's just save an ID here too
   struct rect dst_rect;
   struct rect src_rect;
   struct size size;
   int32_t orientation;
   int32_t visibility;
   float opacity;
};

/**
 * @struct surface
 */
struct surface : public wayland_proxy<struct ivi_controller_surface>,
                 controller_child {
   surface(surface const &) = delete;
   surface &operator=(surface const &) = delete;
   surface(uint32_t i, struct controller *c);

   // Requests
   void set_visibility(uint32_t visibility);
   void set_opacity(wl_fixed_t opacity);
   void set_source_rectangle(int32_t x, int32_t y, int32_t width,
                             int32_t height);
   void set_destination_rectangle(int32_t x, int32_t y, int32_t width,
                                  int32_t height);
   void set_configuration(int32_t width, int32_t height);
   void set_orientation(int32_t orientation);
   void screenshot(const char *filename);
   void send_stats();
   void destroy(int32_t destroy_scene_object);
};

/**
 * @struct layer
 */
struct layer : public wayland_proxy<struct ivi_controller_layer>,
               controller_child {
   layer(layer const &) = delete;
   layer &operator=(layer const &) = delete;
   layer(uint32_t i, struct controller *c);
   layer(uint32_t i, int32_t w, int32_t h, struct controller *c);

   // Requests
   void set_visibility(uint32_t visibility);
   void set_opacity(wl_fixed_t opacity);
   void set_source_rectangle(int32_t x, int32_t y, int32_t width,
                             int32_t height);
   void set_destination_rectangle(int32_t x, int32_t y, int32_t width,
                                  int32_t height);
   void set_configuration(int32_t width, int32_t height);
   void set_orientation(int32_t orientation);
   void screenshot(const char *filename);
   void clear_surfaces();
   void add_surface(struct surface *surface);
   void remove_surface(struct surface *surface);
   void set_render_order(std::vector<uint32_t> const &ro);
};

/**
 * @struct screen
 */
struct screen : public wayland_proxy<struct ivi_controller_screen>,
                controller_child {
   screen(screen const &) = delete;
   screen &operator=(screen const &) = delete;
   screen(uint32_t i, struct controller *c, struct ivi_controller_screen *p);
   void clear();
   void add_layer(layer *l);
   void set_render_order(std::vector<uint32_t> const &ro);
};

/**
 * @struct controller
 */
struct controller : public wayland_proxy<struct ivi_controller> {
   // This controller is still missing ivi-input

   typedef std::unordered_map<uintptr_t, uint32_t> proxy_to_id_map_type;
   typedef std::unordered_map<uint32_t, std::unique_ptr<struct surface>>
      surface_map_type;
   typedef std::unordered_map<uint32_t, std::unique_ptr<struct layer>>
      layer_map_type;
   typedef std::unordered_map<uint32_t, std::unique_ptr<struct screen>>
      screen_map_type;

   typedef std::unordered_map<uint32_t, struct surface_properties> props_map;

   // HACK:
   // The order of these member is mandatory, as when objects are destroyed
   // they will call their parent (that's us right here!) and remove their
   // proxy-to-id mapping. I.e. the *_proxy_to_id members need to be valid
   // when the surfaces/layers/screens maps are destroyed. This sucks, but
   // I cannot see a better solution w/o globals or some other horrible
   // call-our-parent construct.
   proxy_to_id_map_type surface_proxy_to_id;
   proxy_to_id_map_type layer_proxy_to_id;
   proxy_to_id_map_type screen_proxy_to_id;

   props_map sprops;
   props_map lprops;

   surface_map_type surfaces;
   layer_map_type layers;
   screen_map_type screens;

   size output_size;

   wm::controller_hooks *chooks;

   bool is_configured;

   void add_proxy_to_id_mapping(struct ivi_controller_surface *p, uint32_t id);
   void remove_proxy_to_id_mapping(struct ivi_controller_surface *p);
   void add_proxy_to_id_mapping(struct ivi_controller_layer *p, uint32_t id);
   void remove_proxy_to_id_mapping(struct ivi_controller_layer *p);
   void add_proxy_to_id_mapping(struct wl_output *p, uint32_t id);
   void remove_proxy_to_id_mapping(struct wl_output *p);

   bool surface_exists(uint32_t id) const {
      return this->surfaces.find(id) != this->surfaces.end();
   }

   bool layer_exists(uint32_t id) const {
      return this->layers.find(id) != this->layers.end();
   }

   controller(struct wl_registry *r, uint32_t name, uint32_t version);

   // Requests
   void commit_changes() const {
      ivi_controller_commit_changes(this->proxy.get());
   }
   void layer_create(uint32_t id, int32_t w, int32_t h);
   void surface_create(uint32_t id);

   // Events
   // controller
   void controller_screen(uint32_t id, struct ivi_controller_screen *screen);
   void controller_layer(uint32_t id);
   void controller_surface(uint32_t id);
   void controller_error(int32_t object_id, int32_t object_type,
                         int32_t error_code, char const *error_text);

   // surface
   void surface_visibility(struct surface *s, int32_t visibility);
   void surface_opacity(struct surface *s, float opacity);
   void surface_source_rectangle(struct surface *s, int32_t x, int32_t y,
                                 int32_t width, int32_t height);
   void surface_destination_rectangle(struct surface *s, int32_t x, int32_t y,
                                      int32_t width, int32_t height);
   void surface_configuration(struct surface *s, int32_t width, int32_t height);
   void surface_orientation(struct surface *s, int32_t orientation);
   void surface_pixelformat(struct surface *s, int32_t pixelformat);
   void surface_layer(struct surface *s, struct ivi_controller_layer *layer);
   void surface_stats(struct surface *s, uint32_t redraw_count,
                      uint32_t frame_count, uint32_t update_count, uint32_t pid,
                      const char *process_name);
   void surface_destroyed(struct surface *s);
   void surface_content(struct surface *s, int32_t content_state);

   // layer
   void layer_visibility(struct layer *l, int32_t visibility);
   void layer_opacity(struct layer *l, float opacity);
   void layer_source_rectangle(struct layer *l, int32_t x, int32_t y,
                               int32_t width, int32_t height);
   void layer_destination_rectangle(struct layer *l, int32_t x, int32_t y,
                                    int32_t width, int32_t height);
   void layer_configuration(struct layer *l, int32_t width, int32_t height);
   void layer_orientation(struct layer *l, int32_t orientation);
   void layer_screen(struct layer *l, struct wl_output *screen);
   void layer_destroyed(struct layer *l);
};
}  // namespace compositor

#endif  // !WM_WAYLAND_HPP