#pragma once

#include <string>
#include <vector>

#include "imgui.h"
#include "tools/cabana/dbc/dbcmanager.h"


class SignalSelector {
public:
  struct ListItem {
    ListItem(const MessageId &msg_id, const cabana::Signal *sig) : msg_id(msg_id), sig(sig) {}
    MessageId msg_id;
    const cabana::Signal *sig;
  };

  SignalSelector(std::string title);
  const std::vector<ListItem> &selectedItems() const { return selected_list_; }
  inline void addSelected(const MessageId &id, const cabana::Signal *sig) { selected_list_.emplace_back(id, sig); }
  void open() { open_ = true; show_ = false; accepted_ = false; }
  bool draw();
  bool accepted() const { return accepted_; }

private:
  void updateAvailableList(int index);
  void add(int row);
  void remove(int row);
  void drawList(const char *id, std::vector<ListItem> &list, int *current_row, bool show_msg_name, bool *double_clicked, const ImVec2 &size);

  struct ComboItem {
    std::string text;
    MessageId id;
  };
  std::string title_;
  std::vector<ComboItem> msgs_combo_;
  int msgs_combo_index_ = -1;
  std::string msgs_combo_filter_;
  std::vector<ListItem> available_list_;
  std::vector<ListItem> selected_list_;
  int available_row_ = -1;
  int selected_row_ = -1;
  bool accepted_ = false;
  bool open_ = false;
  bool show_ = false;
};