#include "autocomplete.hpp"
#include "selection_dialog.hpp"

Autocomplete::Autocomplete(Source::BaseView *view, bool &interactive_completion, guint &last_keyval, bool pass_buffer_and_strip_word, bool use_thread)
    : view(view), interactive_completion(interactive_completion), pass_buffer_and_strip_word(pass_buffer_and_strip_word), use_thread(use_thread) {
  view->get_buffer()->signal_insert().connect([this, &last_keyval](const Gtk::TextIter & /*iter*/, const Glib::ustring & /*text*/, int /*bytes*/) {
    if(CompletionDialog::get() && CompletionDialog::get()->is_visible()) {
      cancel_reparse();
      return;
    }
    if(!this->view->has_focus())
      return;
    if(is_continue_key(last_keyval) && (this->interactive_completion || state != State::idle))
      run();
    else {
      stop();

      if(is_restart_key(last_keyval) && this->interactive_completion)
        run();
    }
  });
  view->get_buffer()->signal_erase().connect([this](const Gtk::TextIter & /*start*/, const Gtk::TextIter & /*end*/) {
    if(CompletionDialog::get() && CompletionDialog::get()->is_visible()) {
      cancel_reparse();
      return;
    }
    if(!this->view->has_focus())
      return;
    stop();
  });

  view->get_buffer()->signal_mark_set().connect([this](const Gtk::TextIter &iterator, const Glib::RefPtr<Gtk::TextBuffer::Mark> &mark) {
    if(mark->get_name() == "insert")
      stop();
  });

  view->signal_key_release_event().connect(
      [](GdkEventKey *event) {
        if(CompletionDialog::get() && CompletionDialog::get()->is_visible()) {
          if(CompletionDialog::get()->on_key_release(event))
            return true;
        }
        return false;
      },
      false);

  view->signal_focus_out_event().connect([this](GdkEventFocus *event) {
    stop();
    return false;
  });
}

void Autocomplete::run() {
  if(run_check()) {
    if(!is_processing())
      return;

    if(state == State::canceled)
      state = State::restarting;

    if(state != State::idle)
      return;

    state = State::starting;

    before_add_rows();

    if(use_thread && thread.joinable())
      thread.join();
    auto iter = view->get_buffer()->get_insert()->get_iter();
    auto line = iter.get_line();
    auto line_index = iter.get_line_index();
    Glib::ustring buffer;
    if(pass_buffer_and_strip_word) {
      auto pos = iter.get_offset() - 1;
      buffer = view->get_buffer()->get_text();
      while(pos >= 0 && view->is_token_char(buffer[pos])) {
        buffer.replace(pos, 1, " ");
        line_index--;
        pos--;
      }
    }

    auto func = [this, line, line_index, buffer = std::move(buffer)] {
      auto lock = get_parse_lock();
      if(!is_processing())
        return;
      stop_parse();

      rows.clear();
      auto &buffer_raw = const_cast<std::string &>(buffer.raw());
      bool success = add_rows(buffer_raw, line, line_index);
      if(!is_processing())
        return;

      if(success) {
        auto func = [this]() {
          after_add_rows();
          if(state == State::restarting) {
            state = State::idle;
            reparse();
            run();
          }
          else if(state == State::canceled || rows.empty()) {
            state = State::idle;
            reparse();
          }
          else {
            auto start_iter = view->get_buffer()->get_insert()->get_iter();
            std::size_t prefix_size;
            {
              LockGuard lock(prefix_mutex);
              prefix_size = prefix.size();
            }
            if(prefix_size > 0 && !start_iter.backward_chars(prefix_size)) {
              state = State::idle;
              reparse();
              return;
            }
            CompletionDialog::create(view, start_iter);
            setup_dialog();
            for(auto &row : rows) {
              CompletionDialog::get()->add_row(row);
              row.clear();
            }
            state = State::idle;

            view->get_buffer()->begin_user_action();
            CompletionDialog::get()->show();
          }
        };
        if(use_thread)
          dispatcher.post(std::move(func));
        else
          func();
      }
      else {
        auto func = [this] {
          state = State::canceled;
          on_add_rows_error();
        };
        if(use_thread)
          dispatcher.post(std::move(func));
        else
          func();
      }
    };
    if(use_thread)
      thread = std::thread(std::move(func));
    else
      func();
  }

  if(state != State::idle)
    cancel_reparse();
}

void Autocomplete::stop() {
  if(state == State::starting || state == State::restarting)
    state = State::canceled;
}

void Autocomplete::setup_dialog() {
  CompletionDialog::get()->on_show = [this] {
    on_show();
  };

  CompletionDialog::get()->on_hide = [this]() {
    view->get_buffer()->end_user_action();
    tooltips.hide();
    tooltips.clear();
    on_hide();
    reparse();
  };

  CompletionDialog::get()->on_change = [this](boost::optional<unsigned int> index, const std::string &text) {
    if(on_change)
      on_change(index, text);

    if(!index) {
      clear_tooltips();
      return;
    }

    auto set_buffer = set_tooltip_buffer(*index);
    if(!set_buffer)
      tooltips.hide();
    else {
      tooltips.clear();
      auto iter = CompletionDialog::get()->start_mark->get_iter();
      tooltips.emplace_back(view, iter, iter, [set_buffer = std::move(set_buffer)](Tooltip &tooltip) {
        set_buffer(tooltip);
      });
      tooltips.show(true);
    }
  };

  CompletionDialog::get()->on_select = [this](unsigned int index, const std::string &text, bool hide_window) {
    if(on_select)
      on_select(index, text, hide_window);
  };
}