#pragma once

#include <lua.hpp>
#include <optional>

#include "LuaRuntime.hpp"
#include "WindowActionRouter.hpp"
#include "events/Event.hpp"
#include "widgets/BrowserWindow.hpp"
#include "widgets/Decorations.hpp"
#include "widgets/WebView.hpp"

int lua_api_schedule_fn(lua_State *state) {
  lua_pushvalue(state, 1);
  const int function_ref = luaL_ref(state, LUA_REGISTRYINDEX);
  auto action = [state, function_ref]() {
    preserve_top(state, {
      lua_rawgeti(state, LUA_REGISTRYINDEX, function_ref);
      if (lua_pcall(state, 0, 0, 0) != LUA_OK) {
        const char *error = lua_tostring(state, -1);
        qDebug() << "Error calling Lua function:" << error;
      }
      luaL_unref(state, LUA_REGISTRYINDEX, function_ref);
    })
  };

  auto &runtime = LuaRuntime::instance();
  emit runtime.schedule_for_next_tick(action);

  lua_pushnil(state);
  return 1;
}

int lua_api_view_set_url(lua_State *state) {
  const char *url = lua_tostring(state, 1);
  WebViewId view_id = lua_isnoneornil(state, 2) ? 0 : lua_tointeger(state, 2);
  auto &runtime = LuaRuntime::instance();
  emit runtime.url_opened(url, OpenType::OpenUrl, view_id);
  return 1;
}

int lua_api_view_set_html(lua_State *state) {
  const char *html = lua_tostring(state, 1);
  WebViewId view_id = lua_isnoneornil(state, 2) ? 0 : lua_tointeger(state, 2);
  auto &runtime = LuaRuntime::instance();
  emit runtime.webview_html_set_requested(html, view_id);
  return 1;
}

int lua_api_view_run_js(lua_State *state) {
  const char *js_code = lua_tostring(state, 1);

  lua_getfield(state, 2, "view");
  WebViewId view_id = lua_isnoneornil(state, 3) ? 0 : lua_tointeger(state, 3);
  lua_pop(state, 1);

  lua_getfield(state, 2, "on_result");
  JsOnResultFunc action = [](auto &) {};
  if (lua_isfunction(state, 3)) {
    lua_pushvalue(state, 3);
    const int function_ref = luaL_ref(state, LUA_REGISTRYINDEX);
    action = [state, function_ref](const std::optional<QVariant> &value) {
      preserve_top(state, {
        lua_rawgeti(state, LUA_REGISTRYINDEX, function_ref);
        LuaRuntime::push_qvariant(state, value);
        lua_pcall(state, 1, 0, 0);
        luaL_unref(state, LUA_REGISTRYINDEX, function_ref);
      })
    };
  }
  lua_pop(state, 1);

  auto &runtime = LuaRuntime::instance();
  emit runtime.webview_js_eval_requested(js_code, view_id, action);

  return 1;
}

int lua_api_view_create(lua_State *state) {
  const char *url = luaL_optstring(state, 1, "");
  auto &runtime = LuaRuntime::instance();
  emit runtime.url_opened(url, OpenType::OpenUrlInView, 0);
  return 1;
}

int lua_api_config_set(lua_State *state) {
  const char *key = lua_tostring(state, 1);
  auto &runtime = LuaRuntime::instance();
  QVariant value = LuaRuntime::get_lua_value(runtime.get_state(), 2, "");
  emit runtime.config_updated(key, value);
  return 1;
}

int lua_api_config_get(lua_State *state) {
  const char *key = lua_tostring(state, 1);
  auto &router = WindowActionRouter::instance();
  auto value = router.fetch_config_value(key);

  if (value.typeId() == QMetaType::QString)
    lua_pushstring(state, value.toString().toStdString().c_str());
  else if (value.typeId() == QMetaType::Int)
    lua_pushinteger(state, value.toInt());
  else if (value.typeId() == QMetaType::Double)
    lua_pushnumber(state, value.toDouble());
  else
    lua_pushnil(state);

  return 1;
}

int lua_api_keymap_set(lua_State *state) {
  const char *mode = lua_tostring(state, 1);
  const char *keyseq = lua_tostring(state, 2);

  lua_pushvalue(state, 3);
  const int function_ref = luaL_ref(state, LUA_REGISTRYINDEX);
  auto action = [state, function_ref]() {
    preserve_top(state, {
      lua_rawgeti(state, LUA_REGISTRYINDEX, function_ref);
      if (lua_pcall(state, 0, 0, 0) != LUA_OK) {
        const char *error = lua_tostring(state, -1);
        qDebug() << "Error calling Lua function:" << error;
      }
    })
  };
  // TODO: Cleanup function ref on after keymap clear

  auto &runtime = LuaRuntime::instance();
  emit runtime.keymap_add_requested(mode, keyseq, action);

  lua_pushnil(state);
  return 1;
}

int lua_api_view_current(lua_State *state) {
  auto &router = WindowActionRouter::instance();
  auto view_id = router.fetch_current_view_id();
  lua_pushinteger(state, view_id);
  return 1;
}

int lua_view_list(lua_State *state) {
  auto &router = WindowActionRouter::instance();
  auto views = router.fetch_webview_data_list();
  lua_newtable(state);

  int index = 1; // 1-indexed
  for (auto &view : views) {
    lua_newtable(state);

    lua_pushstring(state, "id");
    lua_pushinteger(state, view.id);
    lua_settable(state, -3);

    lua_pushstring(state, "url");
    lua_pushstring(state, view.url.toStdString().c_str());
    lua_settable(state, -3);

    lua_pushstring(state, "title");
    lua_pushstring(state, view.title.toStdString().c_str());
    lua_settable(state, -3);

    lua_rawseti(state, -2, index);
    index++;
  }

  return 1;
}

int lua_history_back(lua_State *state) {
  auto &runtime = LuaRuntime::instance();

  qsizetype history_index = lua_isnoneornil(state, 1) ? 1 : lua_tointeger(state, 1);

  WebViewId view_id = lua_isnoneornil(state, 2) ? 0 : lua_tointeger(state, 2);

  emit runtime.history_back_requested(view_id, history_index);
  return 1;
}

int lua_history_forward(lua_State *state) {
  auto &runtime = LuaRuntime::instance();

  qsizetype history_index = lua_isnoneornil(state, 1) ? 1 : lua_tointeger(state, 1);

  WebViewId view_id = lua_isnoneornil(state, 2) ? 0 : lua_tointeger(state, 2);

  emit runtime.history_forward_requested(view_id, history_index);
  return 1;
}

int lua_view_close(lua_State *state) {
  auto &runtime = LuaRuntime::instance();

  WebViewId view_id = lua_isnoneornil(state, 1) ? 0 : lua_tointeger(state, 1);

  emit runtime.webview_closed(view_id);
  return 1;
}

int lua_view_select(lua_State *state) {
  if (lua_isnoneornil(state, 1))
    return 1; // TODO: return nil (for others too)

  auto &runtime = LuaRuntime::instance();
  WebViewId view_id = lua_tointeger(state, 1);
  emit runtime.webview_selected(view_id);
  return 1;
}

int lua_event_register(lua_State *state) {
  EventHandlerRequest event;
  auto top = lua_gettop(state);

  lua_getfield(state, 1, "events");
  auto event_names = LuaRuntime::lua_tostringlist(state);

  event.event_names.swap(event_names);
  if (event.event_names.size() == 0) {
    lua_settop(state, top);
    lua_pushboolean(state, false);
    return 1;
  }

  lua_getfield(state, 1, "patterns");
  auto patterns = LuaRuntime::lua_tostringlist(state);
  event.patterns.swap(patterns);

  lua_getfield(state, 1, "callback");
  if (!lua_isfunction(state, -1)) {
    lua_settop(state, top);
    lua_pushboolean(state, false);
    return 1;
  }

  const int function_ref = luaL_ref(state, LUA_REGISTRYINDEX);
  event.function_ref = function_ref;
  // TODO: Delete ref on clear callback
  event.handler = [state, function_ref](Event *event) {
    preserve_top(state, {
      lua_rawgeti(state, LUA_REGISTRYINDEX, function_ref);
      event->lua_push(state);
      lua_pcall(state, 1, 0, 0);
    })
  };

  WindowActionRouter::instance().register_event(event);

  lua_settop(state, top);
  lua_pushboolean(state, true);
  return 1;
}

int lua_api_search_set_text(lua_State *state) {
  const char *text = lua_tostring(state, 1);
  WebViewId view_id = lua_isnoneornil(state, 2) ? 0 : lua_tointeger(state, 2);
  auto &runtime = LuaRuntime::instance();
  emit runtime.search_requested(text, view_id);
  return 1;
}

int lua_api_search_next(lua_State *state) {
  WebViewId view_id = lua_isnoneornil(state, 1) ? 0 : lua_tointeger(state, 1);
  auto &runtime = LuaRuntime::instance();
  emit runtime.search_next_requested(view_id);
  return 1;
}

int lua_api_search_previous(lua_State *state) {
  WebViewId view_id = lua_isnoneornil(state, 1) ? 0 : lua_tointeger(state, 1);
  auto &runtime = LuaRuntime::instance();
  emit runtime.search_previous_requested(view_id);
  return 1;
}

int lua_api_view_open_devtools(lua_State *state) {
  WebViewId view_id = lua_isnoneornil(state, 1) ? 0 : lua_tointeger(state, 1);
  auto &runtime = LuaRuntime::instance();
  emit runtime.devtools_requested(view_id);
  return 1;
}

int lua_api_search_get_text(lua_State *state) {
  auto &router = WindowActionRouter::instance();
  auto search_text = router.fetch_current_search_text();
  lua_pushstring(state, search_text.toStdString().c_str());
  return 1;
}

int lua_api_keymap_get_mode(lua_State *state) {
  auto &router = WindowActionRouter::instance();
  auto search_text = router.fetch_current_mode();
  lua_pushstring(state, search_text.toStdString().c_str());
  return 1;
}

int lua_api_keymap_set_mode(lua_State *state) {
  const auto *mode = lua_tostring(state, 1);
  auto &runtime = LuaRuntime::instance();
  emit runtime.keymap_mode_update_requested(mode);
  lua_pushnil(state);
  return 1;
}

int lua_api_view_scroll(lua_State *state) {
  long deltax = lua_tointeger(state, 1);
  long deltay = lua_tointeger(state, 2);
  WebViewId view_id = lua_isnoneornil(state, 1) ? 0 : lua_tointeger(state, 1);
  auto &runtime = LuaRuntime::instance();
  emit runtime.webview_scroll_requested(view_id, (int)deltax, (int)deltay);

  lua_pushnil(state);
  return 1;
}

int lua_api_view_scroll_top(lua_State *state) {
  WebViewId view_id = lua_isnoneornil(state, 1) ? 0 : lua_tointeger(state, 1);
  auto &runtime = LuaRuntime::instance();
  emit runtime.webview_scroll_top_requested(view_id);
  lua_pushnil(state);
  return 1;
}
int lua_api_view_scroll_bottom(lua_State *state) {
  WebViewId view_id = lua_isnoneornil(state, 1) ? 0 : lua_tointeger(state, 1);
  auto &runtime = LuaRuntime::instance();
  emit runtime.webview_scroll_bottom_requested(view_id);
  lua_pushnil(state);
  return 1;
}

int lua_api_decorations_set_enabled(lua_State *state) {
  auto type = (DecorationType)lua_tointeger(state, 1);
  bool enabled = lua_toboolean(state, 2);
  std::optional<WindowId> win_id =
      lua_isnoneornil(state, 3) ? std::nullopt : std::make_optional(lua_tointeger(state, 3));
  auto &runtime = LuaRuntime::instance();
  emit runtime.decoration_set_enabled(type, enabled, win_id);
  lua_pushnil(state);
  return 1;
}

int lua_api_decorations_set_size(lua_State *state) {
  auto type = (DecorationType)lua_tointeger(state, 1);
  uint16_t size = lua_tointeger(state, 2);
  qDebug() << "::api" << type << size;
  std::optional<WindowId> win_id =
      lua_isnoneornil(state, 3) ? std::nullopt : std::make_optional(lua_tointeger(state, 3));
  auto &runtime = LuaRuntime::instance();
  emit runtime.decoration_set_size(type, size, win_id);
  lua_pushnil(state);
  return 1;
}

int lua_api_decorations_get_enabled(lua_State *state) {
  auto type = (DecorationType)lua_tointeger(state, 1);
  WindowId win_id = lua_isnoneornil(state, 2) ? lua_tointeger(state, 2) : 0;
  auto &router = WindowActionRouter::instance();
  auto value = router.fetch_is_decoration_enabled(type, win_id);
  lua_pushboolean(state, value);
  return 1;
}

int lua_api_decorations_get_view(lua_State *state) {
  auto type = (DecorationType)lua_tointeger(state, 1);
  WindowId win_id = lua_isnoneornil(state, 2) ? lua_tointeger(state, 2) : 0;
  auto &router = WindowActionRouter::instance();
  auto view_id = router.fetch_get_decoration_view_id(type, win_id);
  if (view_id.has_value())
    lua_pushinteger(state, view_id.value());
  else
    lua_pushnil(state);
  return 1;
}

int lua_api_view_reload(lua_State *state) {
  WebViewId webview_id = lua_isnoneornil(state, 1) ? 0 : lua_tointeger(state, 1);
  auto &runtime = LuaRuntime::instance();
  emit runtime.webview_reload_requested(webview_id);
  lua_pushnil(state);
  return 1;
}

// :: string -> (table -> nil) -> WebViewId -> nil
int lua_api_view_expose(lua_State *state) {
  const char *name = lua_tostring(state, 1);
  WebViewId view_id = lua_isnoneornil(state, 3) ? 0 : lua_tointeger(state, 3);

  lua_pushvalue(state, 2);
  const int function_ref = luaL_ref(state, LUA_REGISTRYINDEX);
  auto action = [state, function_ref](const RpcArgs &args) {
    preserve_top(state, {
      lua_rawgeti(state, LUA_REGISTRYINDEX, function_ref);
      lua_newtable(state);
      for (auto &pair : args) {
        lua_pushstring(state, pair.first.toStdString().c_str());
        lua_pushstring(state, pair.second.toString().toStdString().c_str());
        lua_settable(state, -3);
      }
      if (lua_pcall(state, 1, 0, 0) != LUA_OK) {
        const char *error = lua_tostring(state, -1);
        qDebug() << "Error calling Lua function:" << error;
      }
    })
  };
  // TODO: Check if we need to cleanup function ref after webview done?

  auto &runtime = LuaRuntime::instance();
  emit runtime.webview_rpc_action_defined(name, action, view_id);

  lua_pushnil(state);
  return 1;
}

// NOLINTNEXTLINE
static luaL_Reg internals_api[] = {
    luaL_Reg{"event_add_listener", &lua_event_register},
    luaL_Reg{"config_set", &lua_api_config_set},
    luaL_Reg{"config_get", &lua_api_config_get},
    luaL_Reg{"keymap_set", &lua_api_keymap_set},
    luaL_Reg{"keymap_get_mode", &lua_api_keymap_get_mode},
    luaL_Reg{"keymap_set_mode", &lua_api_keymap_set_mode},
    luaL_Reg{"view_close", &lua_view_close},
    luaL_Reg{"view_create", &lua_api_view_create},
    luaL_Reg{"view_current", &lua_api_view_current},
    luaL_Reg{"view_list", &lua_view_list},
    luaL_Reg{"view_select", &lua_view_select},
    luaL_Reg{"view_set_url", &lua_api_view_set_url},
    luaL_Reg{"view_set_html", &lua_api_view_set_html},
    luaL_Reg{"view_run_js", &lua_api_view_run_js},
    luaL_Reg{"view_open_devtools", &lua_api_view_open_devtools},
    luaL_Reg{"view_scroll", &lua_api_view_scroll},
    luaL_Reg{"view_scroll_to_top", &lua_api_view_scroll_top},
    luaL_Reg{"view_scroll_to_bottom", &lua_api_view_scroll_bottom},
    luaL_Reg{"view_reload", &lua_api_view_reload},
    luaL_Reg{"view_expose", &lua_api_view_expose},
    luaL_Reg{"history_back", &lua_history_back},
    luaL_Reg{"history_forward", &lua_history_forward},
    luaL_Reg{"search_get_text", &lua_api_search_get_text},
    luaL_Reg{"search_set_text", &lua_api_search_set_text},
    luaL_Reg{"search_previous", &lua_api_search_previous},
    luaL_Reg{"search_next", &lua_api_search_next},
    luaL_Reg{"decorations_set_enabled", &lua_api_decorations_set_enabled},
    luaL_Reg{"decorations_get_enabled", &lua_api_decorations_get_enabled},
    luaL_Reg{"decorations_set_size", &lua_api_decorations_set_size},
    luaL_Reg{"decorations_get_view", &lua_api_decorations_get_view},
    luaL_Reg{"schedule", &lua_api_schedule_fn},
    luaL_Reg{nullptr, nullptr},
};