#include "common/util.h"
#include "sidebar.h"
#include "qt_window.h"
#include "selfdrive/hardware/hw.h"

StatusWidget::StatusWidget(bool has_substatus, QWidget *parent) : QFrame(parent) {
  layout = new QVBoxLayout();
  layout->setSpacing(0);

  status = new QLabel(this);

  if (has_substatus) {
    layout->setContentsMargins(50, 24, 16, 24);
    status->setAlignment(Qt::AlignLeft | Qt::AlignHCenter);
    status->setStyleSheet(R"(font-size: 65px; font-weight: 500;)");

    substatus = new QLabel(this);
    substatus->setAlignment(Qt::AlignLeft | Qt::AlignHCenter);
    substatus->setStyleSheet(R"(font-size: 30px; font-weight: 400;)");

    layout->addWidget(status, 0, Qt::AlignLeft);
    layout->addWidget(substatus, 0, Qt::AlignLeft);
  } else {
    layout->setContentsMargins(40, 24, 16, 24);

    status->setAlignment(Qt::AlignCenter);
    status->setStyleSheet(R"(font-size: 38px; font-weight: 500;)");
    layout->addWidget(status, 0, Qt::AlignCenter);
  }

  setMinimumHeight(124);
  setStyleSheet("background-color: transparent;");
  setLayout(layout);
}

void StatusWidget::paintEvent(QPaintEvent *e) {
  QPainter p(this);
  p.setRenderHint(QPainter::Antialiasing, true);
  p.setPen(QPen(QColor(0xb2b2b2), 3, Qt::SolidLine, Qt::FlatCap));
  // origin at 1.5,1.5 because qt issues with pixel perfect borders
  p.drawRoundedRect(QRectF(1.5, 1.5, size().width()-3, size().height()-3), 30, 30);

  p.setPen(Qt::NoPen);
  p.setBrush(color);
  p.setClipRect(0,0,25+6,size().height()-6,Qt::ClipOperation::ReplaceClip);
  p.drawRoundedRect(QRectF(6, 6, size().width()-12, size().height()-12), 25, 25);
}

void StatusWidget::update(const QString &label, const QColor &c, const QString &msg) {
  status->setText(label);
  if (substatus != nullptr) {
    substatus->setText(msg);
  }
  if (color != c) {
    color = c;
    repaint();
  }
  return;
}

SignalWidget::SignalWidget(QWidget *parent) : QFrame(parent), _strength(0) {
  layout = new QVBoxLayout();
  layout->setMargin(0);
  layout->setSpacing(0);
  layout->insertSpacing(0, 45);

  label = new QLabel(this);
  label->setStyleSheet(R"(font-size: 35px; font-weight: 400;)");
  layout->addWidget(label, 0, Qt::AlignLeft);

  setFixedWidth(177);
  setLayout(layout);
}

void SignalWidget::paintEvent(QPaintEvent *e) {
  QPainter p(this);
  p.setRenderHint(QPainter::Antialiasing, true);
  p.setPen(Qt::NoPen);
  p.setBrush(Qt::white);
  for (int i = 0; i < 5; i++) {
    if (i == _strength) {
      p.setPen(Qt::NoPen);
      p.setBrush(Qt::darkGray);
    }
    p.drawEllipse(QRectF(_dotspace * i, _top, _dia, _dia));
  }
}

void SignalWidget::update(const QString &text, int strength) {
  label->setText(text);
  if (_strength != strength) {
    _strength = strength;
    repaint();
  }
}

Sidebar::Sidebar(QWidget *parent) : QFrame(parent) {
  QVBoxLayout *layout = new QVBoxLayout();
  layout->setContentsMargins(25, 50, 25, 50);
  layout->setSpacing(35);
  setFixedSize(300, vwp_h);

  QPushButton *s_btn = new QPushButton;
  s_btn->setStyleSheet(R"(
    border-image: url(../assets/images/button_settings.png);
  )");
  s_btn->setFixedSize(200, 117);
  layout->addWidget(s_btn, 0, Qt::AlignHCenter);
  QObject::connect(s_btn, &QPushButton::pressed, this, &Sidebar::openSettings);

  signal = new SignalWidget(this);
  layout->addWidget(signal, 0, Qt::AlignTop | Qt::AlignHCenter);

  temp = new StatusWidget(true, this);
  layout->addWidget(temp, 0, Qt::AlignTop);

  panda = new StatusWidget(false, this);
  layout->addWidget(panda, 0, Qt::AlignTop);

  connect = new StatusWidget(false, this);
  layout->addWidget(connect, 0, Qt::AlignTop);

  QImage image = QImageReader("../assets/images/button_home.png").read();
  QLabel *comma = new QLabel(this);
  comma->setPixmap(QPixmap::fromImage(image));
  comma->setAlignment(Qt::AlignCenter);
  layout->addWidget(comma, 1, Qt::AlignHCenter | Qt::AlignVCenter);

  layout->addStretch(1);

  setStyleSheet(R"(
    Sidebar {
      background-color: #393939;
    }
    * {
      color: white;
    }
  )");
  setLayout(layout);
}

void Sidebar::update(const UIState &s) {
  static std::map<NetStatus, std::pair<QString, QColor>> connectivity_map = {
      {NET_ERROR, {"CONNECT\nERROR", COLOR_DANGER}},
      {NET_CONNECTED, {"CONNECT\nONLINE", COLOR_GOOD}},
      {NET_DISCONNECTED, {"CONNECT\nOFFLINE", COLOR_WARNING}},
  };
  auto net_params = connectivity_map[s.scene.athenaStatus];
  connect->update(net_params.first, net_params.second);

  static std::map<cereal::DeviceState::ThermalStatus, QColor> temp_severity_map = {
      {cereal::DeviceState::ThermalStatus::GREEN, COLOR_GOOD},
      {cereal::DeviceState::ThermalStatus::YELLOW, COLOR_WARNING},
      {cereal::DeviceState::ThermalStatus::RED, COLOR_DANGER},
      {cereal::DeviceState::ThermalStatus::DANGER, COLOR_DANGER}};
  QString temp_val = QString("%1 °C").arg((int)s.scene.deviceState.getAmbientTempC());
  temp->update(temp_val, temp_severity_map[s.scene.deviceState.getThermalStatus()], "TEMP");

  static std::map<cereal::DeviceState::NetworkType, const char *> network_type_map = {
      {cereal::DeviceState::NetworkType::NONE, "--"},
      {cereal::DeviceState::NetworkType::WIFI, "WiFi"},
      {cereal::DeviceState::NetworkType::CELL2_G, "2G"},
      {cereal::DeviceState::NetworkType::CELL3_G, "3G"},
      {cereal::DeviceState::NetworkType::CELL4_G, "4G"},
      {cereal::DeviceState::NetworkType::CELL5_G, "5G"}};
  const char *network_type = network_type_map[s.scene.deviceState.getNetworkType()];
  static std::map<cereal::DeviceState::NetworkStrength, int> network_strength_map = {
      {cereal::DeviceState::NetworkStrength::UNKNOWN, 1},
      {cereal::DeviceState::NetworkStrength::POOR, 2},
      {cereal::DeviceState::NetworkStrength::MODERATE, 3},
      {cereal::DeviceState::NetworkStrength::GOOD, 4},
      {cereal::DeviceState::NetworkStrength::GREAT, 5}};
  const int img_idx = s.scene.deviceState.getNetworkType() == cereal::DeviceState::NetworkType::NONE ? 0 : network_strength_map[s.scene.deviceState.getNetworkStrength()];
  signal->update(network_type, img_idx);

  QColor panda_color = COLOR_GOOD;
  QString panda_message = "VEHICLE\nONLINE";
  if (s.scene.pandaType == cereal::PandaState::PandaType::UNKNOWN) {
    panda_color = COLOR_DANGER;
    panda_message = "NO\nPANDA";
  }
  else if (Hardware::TICI() && s.scene.started) {
    panda_color = s.scene.gpsOK ? COLOR_GOOD : COLOR_WARNING;
    panda_message = QString("SAT CNT\n%1").arg(s.scene.satelliteCount);
  }
  panda->update(panda_message, panda_color);
}