#include <fstream>
#include "notebook.h"
#include "logging.h"
#include "singletons.h"
#include <iostream> //TODO: remove
using namespace std; //TODO: remove

namespace sigc {
  SIGC_FUNCTORS_DEDUCE_RESULT_TYPE_WITH_DECLTYPE
}

Notebook::View::View() {
  pack2(notebook);
  set_position(120);
}

Notebook::Controller::Controller() :
  directories() {
  INFO("Create notebook");
  Gsv::init();
  clipboard = Gtk::Clipboard::get();
  view.pack1(directories.widget(), true, true);
  CreateKeybindings();
  entry_box.signal_hide().connect([this]() {
    if(CurrentPage()!=-1) {
      CurrentSourceView()->grab_focus();
    }
  });
  view.notebook.signal_switch_page().connect([this](Gtk::Widget* page, guint page_num) {
    if(search_entry_shown && entry_box.labels.size()>0 && CurrentPage()!=-1) {
      CurrentSourceView()->search_highlight(last_search, case_sensitive_search, regex_search);
      entry_box.labels.back().update(0, std::to_string(CurrentSourceView()->get_search_occurences()));
    }
  });
  INFO("Notebook Controller Success");
}  // Constructor


void Notebook::Controller::CreateKeybindings() {
  auto menu=Singleton::menu();
  INFO("Notebook create signal handlers");
  directories.m_TreeView.signal_row_activated().connect(sigc::mem_fun(*this, &Notebook::Controller::OnDirectoryNavigation));

  menu->action_group->add(Gtk::Action::create("FileMenu", Gtk::Stock::FILE));

  menu->action_group->add(Gtk::Action::create("FileNewFile", "New file"), Gtk::AccelKey(menu->key_map["new_file"]), [this]() {
    OnFileNewFile();
  });
  menu->action_group->add(Gtk::Action::create("WindowCloseTab", "Close tab"), Gtk::AccelKey(menu->key_map["close_tab"]), [this]() {
    OnCloseCurrentPage();
  });
  menu->action_group->add(Gtk::Action::create("EditFind", "Find"), Gtk::AccelKey(menu->key_map["edit_find"]), [this]() {
    show_search_and_replace();
  });
  menu->action_group->add(Gtk::Action::create("EditCopy", "Copy"), Gtk::AccelKey(menu->key_map["edit_copy"]), [this]() {
    if (Pages() != 0) {
      CurrentSourceView()->get_buffer()->copy_clipboard(clipboard);
    }
  });
  menu->action_group->add(Gtk::Action::create("EditCut", "Cut"), Gtk::AccelKey(menu->key_map["edit_cut"]), [this]() {
	  if (Pages() != 0) {
	    CurrentSourceView()->get_buffer()->cut_clipboard(clipboard);
	  }
  });
  menu->action_group->add(Gtk::Action::create("EditPaste", "Paste"), Gtk::AccelKey(menu->key_map["edit_paste"]), [this]() {
	  if (Pages() != 0) {
	    CurrentSourceView()->get_buffer()->paste_clipboard(clipboard);
	  }
  });

  menu->action_group->add(Gtk::Action::create("EditUndo", "Undo"), Gtk::AccelKey(menu->key_map["edit_undo"]), [this]() {
	  INFO("On undo");
    Glib::RefPtr<Gsv::UndoManager> undo_manager = CurrentSourceView()->get_source_buffer()->get_undo_manager();
    if (Pages() != 0 && undo_manager->can_undo()) {
      undo_manager->undo();
    }
    INFO("Done undo");
	});

  menu->action_group->add(Gtk::Action::create("EditRedo", "Redo"), Gtk::AccelKey(menu->key_map["edit_redo"]), [this]() {
    INFO("On Redo");
    Glib::RefPtr<Gsv::UndoManager> undo_manager =
    CurrentSourceView()->get_source_buffer()->get_undo_manager();
    if (Pages() != 0 && undo_manager->can_redo()) {
      undo_manager->redo();
    }
    INFO("Done Redo");
  });
  
  menu->action_group->add(Gtk::Action::create("SourceGotoDeclaration", "Go to declaration"), Gtk::AccelKey(menu->key_map["goto_declaration"]), [this]() {
    if(CurrentPage()!=-1) {
      if(CurrentSourceView()->get_declaration_location) {
        auto location=CurrentSourceView()->get_declaration_location();
        if(location.first.size()>0) {
          open_file(location.first);
          CurrentSourceView()->get_buffer()->place_cursor(CurrentSourceView()->get_buffer()->get_iter_at_offset(location.second));
          CurrentSourceView()->scroll_to(CurrentSourceView()->get_buffer()->get_insert(), 0.0, 1.0, 0.5);
        }
      }
    }
  });
  
  menu->action_group->add(Gtk::Action::create("SourceGotoMethod", "Go to method"), Gtk::AccelKey(menu->key_map["goto_method"]), [this]() {
    if(CurrentPage()!=-1) {
      if(CurrentSourceView()->goto_method) {
        CurrentSourceView()->goto_method();
      }
    }
  });

  INFO("Notebook signal handlers sucsess");
}

void Notebook::Controller::show_search_and_replace() {
  entry_box.clear();
  entry_box.labels.emplace_back();
  auto label_it=entry_box.labels.begin();
  label_it->update=[this, label_it](int state, const std::string& message){
    if(state==0) {
      int number=stoi(message);
      if(number<1)
        label_it->set_text("");
      else if(number==1)
        label_it->set_text("1 result found");
      else
        label_it->set_text(std::to_string(number)+" results found");
    }
  };
  entry_box.entries.emplace_back(last_search, [this](const std::string& content){
    if(CurrentPage()!=-1)
      CurrentSourceView()->search_forward();
  });
  auto search_entry_it=entry_box.entries.begin();
  search_entry_it->set_placeholder_text("Find");
  if(CurrentPage()!=-1) {
    CurrentSourceView()->search_highlight(search_entry_it->get_text(), case_sensitive_search, regex_search);
    delayed_search_label_update.disconnect();
    auto this_view=CurrentSourceView();
    delayed_search_label_update=Glib::signal_timeout().connect([this, this_view, label_it]() {
      if(this_view==CurrentSourceView() && search_entry_shown && entry_box.labels.size()>0)
        label_it->update(0, std::to_string(this_view->get_search_occurences()));
      return false;
    }, 500);
  }
  search_entry_it->signal_key_press_event().connect([this](GdkEventKey* event){
    if(event->keyval==GDK_KEY_Return && event->state==GDK_SHIFT_MASK) {
      if(CurrentPage()!=-1)
        CurrentSourceView()->search_backward();
    }
    return false;
  });
  search_entry_it->signal_changed().connect([this, search_entry_it, label_it](){
    last_search=search_entry_it->get_text();
    if(CurrentPage()!=-1) {
      CurrentSourceView()->search_highlight(search_entry_it->get_text(), case_sensitive_search, regex_search);
      delayed_search_label_update.disconnect();
      auto this_view=CurrentSourceView();
      delayed_search_label_update=Glib::signal_timeout().connect([this, this_view, label_it]() {
        if(this_view==CurrentSourceView() && search_entry_shown && entry_box.labels.size()>0)
          label_it->update(0, std::to_string(this_view->get_search_occurences()));
        return false;
      }, 500);
    }
  });
  
  entry_box.entries.emplace_back(last_replace, [this, label_it](const std::string &content){
    if(CurrentPage()!=-1)
      CurrentSourceView()->replace_forward(content);
  });
  auto replace_entry_it=entry_box.entries.begin();
  replace_entry_it++;
  replace_entry_it->set_placeholder_text("Replace");
  replace_entry_it->signal_key_press_event().connect([this, replace_entry_it](GdkEventKey* event){
    if(event->keyval==GDK_KEY_Return && event->state==GDK_SHIFT_MASK) {
      if(CurrentPage()!=-1)
        CurrentSourceView()->replace_backward(replace_entry_it->get_text());
    }
    return false;
  });
  replace_entry_it->signal_changed().connect([this, replace_entry_it](){
    last_replace=replace_entry_it->get_text();
  });
  
  entry_box.buttons.emplace_back("Find", [this](){
    if(CurrentPage()!=-1)
      CurrentSourceView()->search_forward();
  });
  entry_box.buttons.emplace_back("Replace", [this, replace_entry_it](){
    if(CurrentPage()!=-1)
      CurrentSourceView()->replace_forward(replace_entry_it->get_text());
  });
  entry_box.buttons.emplace_back("Replace all", [this, replace_entry_it, label_it](){
    if(CurrentPage()!=-1) {
      CurrentSourceView()->replace_all(replace_entry_it->get_text());
      label_it->update(0, std::to_string(CurrentSourceView()->get_search_occurences()));
    }
  });
  entry_box.toggle_buttons.emplace_back("Match case");
  entry_box.toggle_buttons.back().set_active(case_sensitive_search);
  entry_box.toggle_buttons.back().on_activate=[this, search_entry_it, label_it](){
    case_sensitive_search=!case_sensitive_search;
    if(CurrentPage()!=-1) {
      CurrentSourceView()->search_highlight(search_entry_it->get_text(), case_sensitive_search, regex_search);
      label_it->update(0, std::to_string(CurrentSourceView()->get_search_occurences()));
    }
  };
  entry_box.toggle_buttons.emplace_back("Use regex", [this, search_entry_it, label_it](){
    regex_search=!regex_search;
    if(CurrentPage()!=-1) {
      CurrentSourceView()->search_highlight(search_entry_it->get_text(), case_sensitive_search, regex_search);
      label_it->update(0, std::to_string(CurrentSourceView()->get_search_occurences()));
    }
  });
  entry_box.signal_hide().connect([this]() {
    for(int c=0;c<Pages();c++)
      source_views.at(c)->view->search_highlight("", case_sensitive_search, regex_search);
    search_entry_shown=false;
  });
  search_entry_shown=true;
  entry_box.show();
}

void Notebook::Controller::open_file(std::string path) {
  INFO("Notebook open file");
  INFO("Notebook create page");
  for(int c=0;c<Pages();c++) {
    if(path==source_views.at(c)->view->file_path) {
      view.notebook.set_current_page(c);
      return;
    }
  }
  source_views.emplace_back(new Source(path, project_path));
  scrolled_windows.emplace_back(new Gtk::ScrolledWindow());
  hboxes.emplace_back(new Gtk::HBox());
  scrolled_windows.back()->add(*source_views.back()->view);
  hboxes.back()->pack_start(*scrolled_windows.back(), true, true);
  
  boost::filesystem::path file_path(source_views.back()->view->file_path);
  std::string title=file_path.filename().string();
  view.notebook.append_page(*hboxes.back(), title);
  view.notebook.show_all_children();
  view.notebook.set_current_page(Pages()-1);
  view.notebook.set_focus_child(*source_views.back()->view);
  CurrentSourceView()->get_buffer()->set_modified(false);
  //Add star on tab label when the page is not saved:
  CurrentSourceView()->get_buffer()->signal_modified_changed().connect([this]() {
    boost::filesystem::path file_path(CurrentSourceView()->file_path);
    std::string title=file_path.filename().string();
    if(CurrentSourceView()->get_buffer()->get_modified())
      title+="*";
    view.notebook.set_tab_label_text(*(view.notebook.get_nth_page(CurrentPage())), title);
  });
  auto this_view=CurrentSourceView();
  CurrentSourceView()->get_buffer()->signal_end_user_action().connect([this, this_view](){
    if(this_view==CurrentSourceView() && search_entry_shown && entry_box.labels.size()>0) {
      entry_box.labels.back().update(0, std::to_string(this_view->get_search_occurences()));
    }
  });
}

void Notebook::Controller::OnCloseCurrentPage() {
  INFO("Notebook close page");
  if (Pages() != 0) {
    if(CurrentSourceView()->get_buffer()->get_modified()){
      AskToSaveDialog();
    }
    int page = CurrentPage();
    view.notebook.remove_page(page);
    source_views.erase(source_views.begin()+ page);
    scrolled_windows.erase(scrolled_windows.begin()+page);
    hboxes.erase(hboxes.begin()+page);
  }
}
void Notebook::Controller::OnFileNewFile() {
  entry_box.clear();
  entry_box.entries.emplace_back("untitled", [this](const std::string& content){
    std::string filename=content;
    if(filename!="") {
      if(project_path!="" && !boost::filesystem::path(filename).is_absolute())
        filename=project_path+"/"+filename;
      boost::filesystem::path p(filename);
      if(boost::filesystem::exists(p)) {
        Singleton::terminal()->print("Error: "+p.string()+" already exists.\n");
      }
      else {
        std::ofstream f(p.string().c_str());
        if(f) {
          open_file(boost::filesystem::canonical(p).string());
          Singleton::terminal()->print("New file "+p.string()+" created.\n");
          if(project_path!="")
            directories.open_folder(project_path); //TODO: Do refresh instead
        }
        else {
          Singleton::terminal()->print("Error: could not create new file "+p.string()+".\n");
        }
        f.close();
      }
    }
    entry_box.hide();
  });
  auto entry_it=entry_box.entries.begin();
  entry_box.buttons.emplace_back("Create file", [this, entry_it](){
    entry_it->activate();
  });
  entry_box.show();
}

void Notebook::Controller
::OnDirectoryNavigation(const Gtk::TreeModel::Path& path,
                        Gtk::TreeViewColumn* column) {
    INFO("Notebook directory navigation");
  Gtk::TreeModel::iterator iter = directories.m_refTreeModel->get_iter(path);
  if (iter) {
    Gtk::TreeModel::Row row = *iter;
    std::string upath = Glib::ustring(row[directories.view().m_col_path]);
    boost::filesystem::path fs_path(upath);
    if (boost::filesystem::is_directory(fs_path)) {
      directories.m_TreeView.row_expanded(path) ?
        directories.m_TreeView.collapse_row(path) :
        directories.m_TreeView.expand_row(path, false);
    } else {
      std::stringstream sstm;
      sstm << row[directories.view().m_col_path];
      std::string file = sstm.str();
      open_file(file);
    }
  }
}

Source::View* Notebook::Controller::CurrentSourceView() {
  INFO("Getting sourceview");
  return source_views.at(CurrentPage())->view.get();
}

int Notebook::Controller::CurrentPage() {
  return view.notebook.get_current_page();
}

int Notebook::Controller::Pages() {
  return view.notebook.get_n_pages();
}

bool Notebook::Controller:: OnSaveFile() {
  std::string path=CurrentSourceView()->file_path;
  return OnSaveFile(path);
}
bool Notebook::Controller:: OnSaveFile(std::string path) {
    INFO("Notebook save file with path");
    if (path != "" && CurrentSourceView()->get_buffer()->get_modified()) {
      std::ofstream file;
      file.open (path);
      file << CurrentSourceView()->get_buffer()->get_text();
      file.close();
      boost::filesystem::path path(CurrentSourceView()->file_path);
      std::string title=path.filename().string();
      view.notebook.set_tab_label_text(*view.notebook.get_nth_page(CurrentPage()), title);
      CurrentSourceView()->get_buffer()->set_modified(false);
      return true;
    }
  return false;
}


std::string Notebook::Controller::OnSaveFileAs(){
  INFO("Notebook save as");
  Gtk::FileChooserDialog dialog((Gtk::Window&)(*view.get_toplevel()), "Please choose a file", 
				Gtk::FILE_CHOOSER_ACTION_SAVE);
  DEBUG("SET TRANSISTEN FPR");
  dialog.set_position(Gtk::WindowPosition::WIN_POS_CENTER_ALWAYS);
  dialog.add_button("_Cancel", Gtk::RESPONSE_CANCEL);
  dialog.add_button("_Save", Gtk::RESPONSE_OK);
  //dialog.set_current_name("Untitled");
  DEBUG("RUN DIALOG");
  int result = dialog.run();
  DEBUG("DIALOG RUNNING");
  switch (result) {
  case(Gtk::RESPONSE_OK): {
    DEBUG("get_filename()");
    std::string path = dialog.get_filename();
    return path;
  }
  case(Gtk::RESPONSE_CANCEL): {
    break;
  }
  default: {
    DEBUG("Unexpected button clicked."); 
    break;
  }
  }
  return "";
}

void Notebook::Controller::AskToSaveDialog() {
  INFO("AskToSaveDialog");
  DEBUG("AskToSaveDialog: Finding file path");
  Gtk::MessageDialog dialog((Gtk::Window&)(*view.get_toplevel()), "Save file!",
			    false, Gtk::MESSAGE_QUESTION, Gtk::BUTTONS_YES_NO);
  dialog.set_secondary_text(
          "Do you want to save: " +
			    CurrentSourceView()->file_path+" ?");
  DEBUG("AskToSaveDialog: run dialog");
  int result = dialog.run();

  //Handle the response:
  DEBUG("AskToSaveDialog: switch response");
  switch(result)
  {
    case(Gtk::RESPONSE_YES):
    {
      DEBUG("AskToSaveDialog: save file: yes, trying to save file");
      OnSaveFile();
      DEBUG("AskToSaveDialog: save file: yes, saved sucess");
      break;
    }
    case(Gtk::RESPONSE_NO):
    {
      DEBUG("AskToSaveDialog: save file: no");
      break;
    }
    default:
    {
           DEBUG("AskToSaveDialog: unexpected action: Default switch");
      break;
    }
  }
}