#include "mainwidget.h"
#include "actor.h"
#include "block.h"
#include "constants.h"
#include "audio.h"
#include "util.h"
#include "pacman.pb.h"

MainWidget::MainWidget(QWidget *parent)
: SceneHolder(parent), m_currentKey(0), m_running(false)
{
  m_color = connectToServer();
  if (m_color == Color::none)
  {
    QMessageBox::critical(this, "Error", "Failed to connect to server, falling back to local test mode");
    // TODO: quit application here or sth
    m_socket = NULL;
    m_color = Color::red;
    QTimer *timer = new QTimer(this);
    connect(timer, SIGNAL(timeout()), this, SLOT(tick()));
    timer->start(Constants::tick);
  }

  /* call updateMap after m_color ist set! */
  createGui();
  updateMap(Util::createDemoMap());

  if (m_socket != NULL)
    connect(m_socket, SIGNAL(readyRead()), this, SLOT(tick()));

  // TODO: use mycolor
  qDebug() << "mycolor=" << m_color;

  //TODO: play intro as soon as there are enough players
  //connect(AudioPlayer::self(), SIGNAL(finished()), this, SLOT(startGame()));
  //AudioPlayer::self()->play(AudioPlayer::Intro);
  startGame();
}

void MainWidget::createGui()
{
  setFocusPolicy(Qt::StrongFocus);

  /* first one is always the own score */
  QHBoxLayout *scoreLayout = new QHBoxLayout();
  for (unsigned int i = 0; Color::order[i] != Color::none ; ++i)
  {
    QGroupBox *scoreBox = new QGroupBox(QString("Spieler %1").arg(i + 1), this);
    scoreBox->setObjectName(QString("actor%1").arg(i + 1));
    scoreBox->setCheckable(true);
    connect(scoreBox, SIGNAL(clicked()), this, SLOT(playerScoreClicked()));

    QGridLayout *playerLayout = new QGridLayout();
    scoreBox->setLayout(playerLayout);

    playerLayout->addWidget(new QLabel("Rundenpunkte:", this), 0, 0);
    playerLayout->addWidget(new QLabel("Gesamtpunkte:", this), 1, 0);

    playerLayout->addWidget(new QLabel("", this), 0, 1);
    playerLayout->addWidget(new QLabel("", this), 1, 1);

    if (Color::order[i] == m_color)
      scoreLayout->insertWidget(0, scoreBox);
    else
      scoreLayout->addWidget(scoreBox);
    m_playerScoreLayouts.append(playerLayout);
  }

  QVBoxLayout *layout = new QVBoxLayout(this);
  layout->addLayout(scoreLayout);

  QGraphicsView *window = new QGraphicsView(m_scene, this);
  window->setFrameStyle(0);
  window->setAlignment(Qt::AlignLeft | Qt::AlignTop);
  window->setFixedSize(Constants::map_size_pixel.width, Constants::map_size_pixel.height);
  window->setWindowFlags(window->windowFlags() & ~Qt::WindowMaximizeButtonHint);
  window->setFocusPolicy(Qt::NoFocus);
  layout->addWidget(window);

  QFile css(":/stylesheet");
  css.open(QFile::ReadOnly);
  qApp->setStyleSheet(QLatin1String(css.readAll()));

  setLayout(layout);
}

void MainWidget::updateScore(const ProtoBuf::MapUpdate& packet)
{
  for(unsigned i = 0; Color::order[i] != Color::none; ++i)
  {
    QGridLayout *score = m_playerScoreLayouts.at(i);
    QLabel *turnPointsLbl = dynamic_cast<QLabel*>(score->itemAtPosition(0, 1)->widget());
    turnPointsLbl->setText(QString::number(packet.round_points(i)));

    QLabel *allPointsLbl = dynamic_cast<QLabel*>(score->itemAtPosition(1, 1)->widget());
    allPointsLbl->setText(QString::number(packet.round_points(i)));
  }
}

Transmission::field_t MainWidget::translateKey(int key)
{
  switch(key)
  {
    case Qt::Key_W:
    case Qt::Key_Up:
      return Transmission::direction_up;
      break;
    case Qt::Key_S:
    case Qt::Key_Down:
      return Transmission::direction_down;
      break;
    case Qt::Key_A:
    case Qt::Key_Left:
      return Transmission::direction_left;
      break;
    case Qt::Key_D:
    case Qt::Key_Right:
      return Transmission::direction_right;
      break;
    default:
      return Transmission::direction_none;
  }
}

void MainWidget::tick()
{
  if (m_socket == NULL)
  {
    // OLD TEST MODE
    Actor::Movement mov = Util::transmissionMovementToActor(m_currentKey, Actor::None);
    QMapIterator<Color::Color, Actor*> i(m_actors);
    while (i.hasNext())
    {
      i.next();
      i.value()->move(mov);
      QList<QGraphicsItem *> list(i.value()->collidingItems());
      for(int j = 0; j < list.count(); ++j)
      {
        if (list.at(j)->parentItem() == i.value())
          continue;
        list.at(j)->setOpacity(0.6);
      }
    }
  }
  else
  {
    std::string dataStr;
    Util::QByteArrayToStdString(m_socket->readAll(), dataStr);

    //qDebug() << "read str " << dataStr.length();
    ProtoBuf::MapUpdate packet;
    bool worked = packet.ParseFromString(dataStr);
    Q_ASSERT(worked);
    Transmission::map_t map = Util::createUninitialisedMap();
    Q_ASSERT(packet.field_size() == (int) (Constants::map_size.width * Constants::map_size.height));
    int i = 0;
    for (unsigned int x = 0; x < Constants::map_size.width; ++x)
    {
      for (unsigned int y = 0; y < Constants::map_size.height; ++y)
      {
        map[x][y] = packet.field(i);
        ++i;
      }
    }
    updateMap(map);
    updateScore(packet);
  }
}

void MainWidget::keyPressEvent(QKeyEvent* event)
{
  if (!m_running)
    return;

  QWidget::keyPressEvent(event);
  Transmission::field_t newKey = translateKey(event->key());
  if (m_currentKey == newKey || newKey == Transmission::direction_none)
    return;
  m_currentKey = newKey;

  if (m_socket != NULL)
  {
    // send to server
    ProtoBuf::KeyPressUpdate packet;
    packet.set_newkey(m_currentKey);
    Util::sendPacket(packet, m_socket);
    qDebug() << "send key: " << m_currentKey;
  }
  else
  {
    // test stuff
    Actor::Movement mov = Util::transmissionMovementToActor(m_currentKey, Actor::None);
    QMapIterator<Color::Color, Actor*> i(m_actors);
    while (i.hasNext())
    {
      i.next();
      i.value()->move(mov);
    }
  }
}

void MainWidget::keyReleaseEvent(QKeyEvent* event)
{
  return; // currently not needed
  if (!m_running)
    return;

  QWidget::keyReleaseEvent(event);
  Transmission::field_t releasedKey = translateKey(event->key());
  if (releasedKey == m_currentKey)
  {
    // current key got released
    // if this is false, a key got released which has already
    // been replaced by a different key, so this case is disregarded
    m_currentKey = Transmission::direction_none;
  }
}

void MainWidget::startGame()
{
  m_running = true;
}

void MainWidget::playerScoreClicked()
{
  QGroupBox *tmp = qobject_cast<QGroupBox *>(sender());
  tmp->setChecked(true);
  return;
}

Color::Color MainWidget::connectToServer()
{
  // check command line arguments for server port
  const QStringList &args = QCoreApplication::arguments();
  QString srv = args.value(1, "127.0.0.1");
  qDebug() << "srv: " << srv;

  // connect to server
  m_socket = new QTcpSocket(this);
  m_socket->connectToHost(srv, Constants::Networking::port);
  bool worked =  m_socket->waitForConnected(Constants::Networking::connection_timeout);
  if (worked)
  {
    // additional init
    // first packet is our color
    worked = m_socket->waitForReadyRead();
    if (worked)
    {
      // receive color
      std::string data;
      Util::QByteArrayToStdString(m_socket->readAll(), data);
      ProtoBuf::WhoAmI packet;
      Q_ASSERT(worked);
      bool worked = packet.ParseFromString(data);
      return static_cast<Color::Color>(packet.color() & Transmission::color_mask);
    }
  }
  return Color::none;
}