#ifndef JUCI_SOURCE_H_
#define JUCI_SOURCE_H_
#include <iostream>
#include <unordered_map>
#include <vector>
#include "gtkmm.h"
#include "clangmm.h"
#include <thread>
#include <mutex>
#include <string>
#include <atomic>
#include "gtksourceviewmm.h"
#include "terminal.h"
#include "tooltips.h"
#include "selectiondialog.h"
#include <set>

class Source {
public:
  class Config {
  public:
    bool legal_extension(std::string e) const ;
    unsigned tab_size;
    bool show_line_numbers, highlight_current_line;
    std::string tab, background, background_selected, background_tooltips, font;
    char tab_char=' ';
    std::vector<std::string> extensions;
    std::unordered_map<std::string, std::string> tags, types;
  };  // class Config

  class Range {
  public:
    Range(unsigned start_offset, unsigned end_offset, int kind):
      start_offset(start_offset), end_offset(end_offset), kind(kind) {}
    unsigned start_offset;
    unsigned end_offset;
    int kind;
  };

  class AutoCompleteData {
  public:
    explicit AutoCompleteData(const std::vector<clang::CompletionChunk> &chunks) :
      chunks(chunks) { }
    std::vector<clang::CompletionChunk> chunks;
    std::string brief_comments;
  };

  class View : public Gsv::View {
  public:
    View(const std::string& file_path, const std::string& project_path);
    ~View();
    
    bool save();
    
    void search_highlight(const std::string &text, bool case_sensitive, bool regex);
    std::function<void(int number)> update_search_occurrences;
    void search_forward();
    void search_backward();
    void replace_forward(const std::string &replacement);
    void replace_backward(const std::string &replacement);
    void replace_all(const std::string &replacement);
    
    std::string get_line(size_t line_number);
    std::string get_line_before_insert();
    
    std::string file_path;
    std::string project_path;
    
    std::function<std::pair<std::string, unsigned>()> get_declaration_location;
    std::function<void()> goto_method;
    std::function<std::string()> get_token;
    std::function<std::string()> get_token_name;
    std::function<void(const std::string &token)> tag_similar_tokens;
    std::function<size_t(const std::string &token, const std::string &text)> rename_similar_tokens;
  protected:
    bool on_key_press_event(GdkEventKey* key);
  private:
    GtkSourceSearchContext *search_context;
    GtkSourceSearchSettings *search_settings;
    static void search_occurrences_updated(GtkWidget* widget, GParamSpec* property, gpointer data);
  };  // class View
  
  class GenericView : public View {
  public:
    GenericView(const std::string& file_path, const std::string& project_path);
  };
  
  class ClangViewParse : public View {
  public:
    ClangViewParse(const std::string& file_path, const std::string& project_path);
    ~ClangViewParse();
  protected:
    void start_reparse();
    bool on_key_press_event(GdkEventKey* key);
    bool on_focus_out_event(GdkEventFocus* event);
    std::unique_ptr<clang::TranslationUnit> clang_tu;
    std::mutex parsing_mutex;
    std::unique_ptr<clang::Tokens> clang_tokens;
    bool clang_readable=false;
    sigc::connection delayed_reparse_connection;
    sigc::connection delayed_tooltips_connection;
  private:
    std::map<std::string, std::string> get_buffer_map() const;
    // inits the syntax highligthing on file open
    void init_syntax_highlighting(const std::map<std::string, std::string>
                                &buffers,
                                int start_offset,
                                int end_offset);
    int reparse(const std::map<std::string, std::string> &buffers);
    void update_syntax();
    std::set<std::string> last_syntax_tags;
    void update_diagnostics();
    void update_types();
    Tooltips diagnostic_tooltips;
    Tooltips type_tooltips;
    bool on_motion_notify_event(GdkEventMotion* event);
    void on_mark_set(const Gtk::TextBuffer::iterator& iterator, const Glib::RefPtr<Gtk::TextBuffer::Mark>& mark);
    
    bool on_scroll_event(GdkEventScroll* event);
    static clang::Index clang_index;
    std::vector<std::string> get_compilation_commands();
        
    std::shared_ptr<Terminal::InProgress> parsing_in_progress;
    Glib::Dispatcher parse_done;
    Glib::Dispatcher parse_start;
    std::thread parse_thread;
    std::map<std::string, std::string> parse_thread_buffer_map;
    std::mutex parse_thread_buffer_map_mutex;
    std::atomic<bool> parse_thread_go;
    std::atomic<bool> parse_thread_mapped;
    std::atomic<bool> parse_thread_stop;
  };
    
  class ClangViewAutocomplete : public ClangViewParse {
  public:
    ClangViewAutocomplete(const std::string& file_path, const std::string& project_path);
  protected:
    bool on_key_press_event(GdkEventKey* key);
    bool on_focus_out_event(GdkEventFocus* event);
  private:
    void start_autocomplete();
    void autocomplete();
    std::unique_ptr<CompletionDialog> completion_dialog;
    bool completion_dialog_shown=false;
    std::vector<Source::AutoCompleteData> get_autocomplete_suggestions(int line_number, int column, std::map<std::string, std::string>& buffer_map);
    Glib::Dispatcher autocomplete_done;
    sigc::connection autocomplete_done_connection;
    bool autocomplete_starting=false;
    std::atomic<bool> autocomplete_cancel_starting;
    guint last_keyval=0;
    std::string prefix;
    std::mutex prefix_mutex;
  };

  class ClangViewRefactor : public ClangViewAutocomplete {
  public:
    ClangViewRefactor(const std::string& file_path, const std::string& project_path);
  private:
    Glib::RefPtr<Gtk::TextTag> similar_tokens_tag;
    std::string last_similar_tokens_tagged;
    std::unique_ptr<SelectionDialog> selection_dialog;
    bool renaming=false;
  };
  
  class ClangView : public ClangViewRefactor {
  public:
    ClangView(const std::string& file_path, const std::string& project_path):
      ClangViewRefactor(file_path, project_path) {}
  };

  Source(const std::string& file_path, std::string project_path);
  
  std::unique_ptr<Source::View> view;
};  // class Source
#endif  // JUCI_SOURCE_H_