#pragma once

#include <memory>
#include <set>
#include <string>
#include <utility>
#include <vector>

#include "tools/cabana/ui/widgets/binaryview.h"
#include "tools/cabana/ui/chart/chartswidget.h"
#include "tools/cabana/ui/widgets/historylog.h"
#include "tools/cabana/ui/widgets/signalview.h"
#include "tools/cabana/ui/widgets/tabbar.h"


class ElidedLabel {
public:
  explicit ElidedLabel(const std::string &text = {});
  void setText(const std::string &text) { text_ = text; }
  void setToolTip(const std::string &tip) { tooltip_ = tip; }
  void draw(float width);

  Observable<> clicked;

private:
  std::string text_, tooltip_;
};


class EditMessageDialog {
public:
  struct Result {
    MessageId msg_id;
    std::string name, node, comment;
    int size;
  };

  EditMessageDialog(const MessageId &msg_id, const std::string &title, int size, float parent_width);
  bool draw();
  bool accepted() const { return accepted_; }
  Result result() const;

private:
  void validateName(const std::string &text);

  MessageId msg_id_;
  std::string original_name_;
  std::string name_edit_;
  std::string node_;
  std::string comment_edit_;
  std::string error_label_;
  int size_spin_;
  bool ok_enabled_ = true;
  std::string window_title_;
  float width_;
  bool opened_ = false;
  bool accepted_ = false;
  bool closed_ = false;
};

class DetailWidget {
public:
  DetailWidget(ChartsWidget *charts);
  void setMessage(const MessageId &message_id);
  void refresh();
  void draw();
  std::pair<std::string, std::vector<std::string>> serializeMessageIds() const;
  void restoreTabs(const std::string &active_msg_id, const std::vector<std::string> &msg_ids);
  std::vector<std::pair<std::string, ImRect>> helpRects() const;

private:
  void drawToolBar();
  void drawTabWidget();
  int findOrAddTab(const MessageId& message_id);
  void showTabBarContextMenu(int index);
  void editMsg();
  void updateState(const std::set<MessageId> *msgs = nullptr);

  MessageId msg_id_;
  const char *warning_icon_ = nullptr;
  std::string warning_label_;
  ElidedLabel name_label_;
  bool warning_widget_visible_ = false;
  TabBar tabbar_;
  int tab_widget_index_ = 0;
  bool action_remove_msg_enabled_ = false;
  bool heatmap_live_ = true;
  std::string heatmap_all_text_ = "All";
  ImRect binary_view_rect_, signal_view_rect_;
  std::unique_ptr<LogsWidget> history_log_;
  std::unique_ptr<BinaryView> binary_view_;
  std::unique_ptr<SignalView> signal_view_;
  ChartsWidget *charts_;
  std::unique_ptr<EditMessageDialog> edit_dlg_;
  Connections connections_;
};

class CenterWidget {
public:
  CenterWidget() = default;
  void setChartsWidget(ChartsWidget *charts) { charts_ = charts; }
  void setMessage(const MessageId &message_id) { ensureDetailWidget()->setMessage(message_id); }
  DetailWidget* getDetailWidget() { return detail_widget.get(); }
  DetailWidget* ensureDetailWidget();
  void clear();
  void draw();

private:
  void drawWelcomeWidget();
  std::unique_ptr<DetailWidget> detail_widget;
  ChartsWidget *charts_ = nullptr;
};