#include "modules/sway/window.hpp"

waybar::modules::sway::Window::Window(const std::string& id, const Bar& bar,
                                      const Json::Value& config)
    : ALabel(config, "{}"), bar_(bar), windowId_(-1) {
  label_.set_name("window");
  if (!id.empty()) {
    label_.get_style_context()->add_class(id);
  }
  if (label_.get_max_width_chars() == -1) {
    label_.set_hexpand(true);
    label_.set_ellipsize(Pango::EllipsizeMode::ELLIPSIZE_END);
  }
  ipc_.subscribe("[\"window\",\"workspace\"]");
  getFocusedWindow();
  // Launch worker
  worker();
}

void waybar::modules::sway::Window::worker() {
  thread_ = [this] {
    try {
      auto res = ipc_.handleEvent();
      auto parsed = parser_.parse(res.payload);
      // Check for waybar prevents flicker when hovering window module
      if ((parsed["change"] == "focus" || parsed["change"] == "title") &&
          parsed["container"]["focused"].asBool() &&
          parsed["container"]["name"].asString() != "waybar") {
        window_ = Glib::Markup::escape_text(parsed["container"]["name"].asString());
        windowId_ = parsed["container"]["id"].asInt();
        dp.emit();
      } else if ((parsed["change"] == "close" && parsed["container"]["focused"].asBool() &&
                  windowId_ == parsed["container"]["id"].asInt()) ||
                 (parsed["change"] == "focus" && parsed["current"]["focus"].isArray() &&
                  parsed["current"]["focus"].empty())) {
        window_.clear();
        windowId_ = -1;
        dp.emit();
      }
    } catch (const std::exception& e) {
      std::cerr << "Window: " << e.what() << std::endl;
    }
  };
}

auto waybar::modules::sway::Window::update() -> void {
  label_.set_markup(fmt::format(format_, window_));
  if (tooltipEnabled()) {
    label_.set_tooltip_text(window_);
  }
}

std::tuple<int, std::string> waybar::modules::sway::Window::getFocusedNode(Json::Value nodes) {
  for (auto const& node : nodes) {
    if (node["focused"].asBool() && node["type"] == "con") {
      return {node["id"].asInt(), node["name"].asString()};
    }
    auto [id, name] = getFocusedNode(node["nodes"]);
    if (id > -1 && !name.empty()) {
      return {id, name};
    }
  }
  return {-1, std::string()};
}

void waybar::modules::sway::Window::getFocusedWindow() {
  try {
    auto res = ipc_.sendCmd(IPC_GET_TREE);
    auto parsed = parser_.parse(res.payload);
    auto [id, name] = getFocusedNode(parsed["nodes"]);
    windowId_ = id;
    window_ = name;
    Glib::signal_idle().connect_once(sigc::mem_fun(*this, &Window::update));
  } catch (const std::exception& e) {
    std::cerr << e.what() << std::endl;
  }
}