#pragma once

#include <xcb/xcb.h>
#include <xcb/xcb_icccm.h>

#include <i3ipc++/ipc.hpp>

#include "common.hpp"
#include "components/x11/connection.hpp"
#include "components/x11/randr.hpp"
#include "config.hpp"
#include "utils/socket.hpp"
#include "utils/string.hpp"

LEMONBUDDY_NS

namespace i3_util {
  using connection_t = i3ipc::connection;

  /**
   * Get all i3 root windows
   */
  auto root_windows(connection& conn, string output_name = "") {
    vector<xcb_window_t> roots;
    auto children = conn.query_tree(conn.screen()->root).children();

    for (auto it = children.begin(); it != children.end(); it++) {
      auto cookie = xcb_icccm_get_wm_name(conn, *it);
      xcb_icccm_get_text_property_reply_t reply;

      if (xcb_icccm_get_wm_name_reply(conn, cookie, &reply, nullptr) == 0)
        continue;

      if (("[i3 con] output " + output_name).compare(0, 16 + output_name.length(), reply.name) != 0)
        continue;

      roots.emplace_back(*it);
    }

    return roots;
  }

  /**
   * Restack given window above the i3 root window
   * defined for the given monitor
   *
   * Fixes the issue with always-on-top window's
   */
  bool restack_above_root(connection& conn, const monitor_t& mon, const xcb_window_t win) {
    for (auto&& root : root_windows(conn, mon->name)) {
      const uint32_t value_mask = XCB_CONFIG_WINDOW_SIBLING | XCB_CONFIG_WINDOW_STACK_MODE;
      const uint32_t value_list[2]{root, XCB_STACK_MODE_ABOVE};

      conn.configure_window_checked(win, value_mask, value_list);
      conn.flush();

      return true;
    }

    return false;
  }
}

LEMONBUDDY_NS_END