#include "mainwindow.h"
#include "ui_mainwindow.h"
#include "blackgui/atcstationlistmodel.h"
#include "blackcore/dbus_server.h"
#include "blackcore/context_network.h"
#include "blackcore/context_application.h"
#include "blackcore/network.h"
#include "blackmisc/avaircraft.h"
#include <QMouseEvent>

using namespace BlackCore;
using namespace BlackMisc;
using namespace BlackGui;
using namespace BlackMisc::Network;
using namespace BlackMisc::Aviation;
using namespace BlackMisc::PhysicalQuantities;
using namespace BlackMisc::Geo;
using namespace BlackMisc::Settings;
using namespace BlackMisc::Audio;

/*
 * Constructor
 */
MainWindow::MainWindow(GuiModes::WindowMode windowMode, QWidget *parent) :
    QMainWindow(parent, windowMode == GuiModes::WindowFrameless ? (Qt::Window | Qt::FramelessWindowHint) : Qt::Tool),
    ui(new Ui::MainWindow),
    m_infoWindow(nullptr),
    m_init(false), m_windowMode(windowMode), m_audioTestRunning(NoAudioTest),
    // contexts and runtime
    m_coreAvailable(false), m_contextNetworkAvailable(false), m_contextAudioAvailable(false),

    // timers
    m_timerUpdateAtcStationsOnline(nullptr), m_timerUpdateAircraftsInRange(nullptr), m_timerUpdateUsers(nullptr),
    m_timerCollectedCockpitUpdates(nullptr), m_timerContextWatchdog(nullptr),
    m_timerStatusBar(nullptr), m_timerAudioTests(nullptr), m_timerSimulator(nullptr),
    // context menus
    m_contextMenuAudio(nullptr), m_contextMenuStatusMessageList(nullptr),
    // cockpit
    m_inputFocusedWidget(nullptr),
    // status bar
    m_statusBarIcon(nullptr), m_statusBarLabel(nullptr)
{
    if (windowMode == GuiModes::WindowFrameless)
    {
        // http://stackoverflow.com/questions/18316710/frameless-and-transparent-window-qt5
        this->setAttribute(Qt::WA_NoSystemBackground, true);
        this->setAttribute(Qt::WA_TranslucentBackground, true);
        // this->setAttribute(Qt::WA_PaintOnScreen);
    }
    ui->setupUi(this);
}

/*
 * Destructor
 */
MainWindow::~MainWindow() {}

/*
 * Graceful shutdown
 */
void MainWindow::gracefulShutdown()
{
    if (!this->m_init) return;
    this->m_init = false;
    if (this->m_rt->getIContextApplication())
        this->m_rt->getIContextApplication()->notifyAboutComponentChange(IContextApplication::ComponentGui, IContextApplication::ActionStops);

    // close info window
    if (this->m_infoWindow)
    {
        this->m_infoWindow->close();
        this->m_infoWindow = nullptr;
    }

    // shut down all timers
    this->stopUpdateTimers(true);
    if (this->m_timerContextWatchdog)
    {
        this->m_timerContextWatchdog->stop();
        this->m_timerContextWatchdog->disconnect(this);
    }
    if (this->m_timerCollectedCockpitUpdates)
    {
        this->m_timerCollectedCockpitUpdates->stop();
        this->m_timerCollectedCockpitUpdates->disconnect(this);
    }

    // if we have a context, we shut some things down
    if (this->m_contextNetworkAvailable)
    {
        if (this->m_rt->getIContextNetwork()->isConnected())
        {
            if (this->m_contextAudioAvailable)
            {
                this->m_rt->getIContextAudio()->leaveAllVoiceRooms();
                this->m_rt->getIContextAudio()->disconnect(this); // break down signal / slots
            }
            this->m_rt->getIContextNetwork()->disconnectFromNetwork();
            this->m_rt->getIContextNetwork()->disconnect(this); // avoid any status update signals, etc.
        }
    }
}

/*
 * Close event, window closes
 */
void MainWindow::closeEvent(QCloseEvent *event)
{
    Q_UNUSED(event);
    this->gracefulShutdown();
    // if (this->sender() != this) QMainWindow::closeEvent(event);
    QApplication::exit();
}

void MainWindow::mouseMoveEvent(QMouseEvent *event)
{
    // this->ui->fr_PseudoWindowBar->geometry().contains(event->pos())
    if (this->m_windowMode == GuiModes::WindowFrameless && event->buttons() & Qt::LeftButton)
    {
        move(event->globalPos() - this->m_dragPosition);
        event->accept();
        return;
    }
    QWidget::mouseMoveEvent(event);
}

void MainWindow::mousePressEvent(QMouseEvent *event)
{
    if (this->m_windowMode == GuiModes::WindowFrameless && event->button() == Qt::LeftButton)
    {
        this->m_dragPosition = event->globalPos() - this->frameGeometry().topLeft();
        event->accept();
        return;
    }
    QWidget::mousePressEvent(event);
}

/*
 * Select correct main page
 */
void MainWindow::setMainPage(bool start)
{
    if (start)
    {
        this->ui->sw_MainMiddle->setCurrentIndex(0);
        return;
    }

    QObject *sender = QObject::sender();
    if (sender == this->ui->pb_MainConnect || sender == this->ui->pb_MainStatus)
        this->ui->sw_MainMiddle->setCurrentIndex(MainPageStatus);
    else if (sender == this->ui->pb_MainAtc)
        this->ui->sw_MainMiddle->setCurrentIndex(MainPageAtc);
    else if (sender == this->ui->pb_MainAircrafts)
        this->ui->sw_MainMiddle->setCurrentIndex(MainPageAircrafts);
    else if (sender == this->ui->pb_MainCockpit)
        this->ui->sw_MainMiddle->setCurrentIndex(MainPageCockpit);
    else if (sender == this->ui->pb_MainUsers)
        this->ui->sw_MainMiddle->setCurrentIndex(MainPageUsers);
    else if (sender == this->ui->pb_MainTextMessages)
        this->ui->sw_MainMiddle->setCurrentIndex(MainPageTextMessages);
    else if (sender == this->ui->pb_MainFlightplan)
        this->ui->sw_MainMiddle->setCurrentIndex(MainPageFlightplan);
    else if (sender == this->ui->pb_MainSettings)
        this->ui->sw_MainMiddle->setCurrentIndex(MainPageSettings);
    else if (sender == this->ui->pb_MainSimulator)
        this->ui->sw_MainMiddle->setCurrentIndex(MainPageSimulator);
}

/*
 * Set main page
 */
void MainWindow::setMainPage(MainWindow::MainPageIndex mainPage)
{
    this->ui->sw_MainMiddle->setCurrentIndex(mainPage);
}

/*
 * Given main page selected?
 */
bool MainWindow::isMainPageSelected(MainWindow::MainPageIndex mainPage) const
{
    return this->ui->sw_MainMiddle->currentIndex() == static_cast<int>(mainPage);
}

/*
 * Connect to Network
 */
void MainWindow::toggleNetworkConnection()
{
    CStatusMessageList msgs;
    if (!this->isContextNetworkAvailableCheck()) return;

    this->ui->lbl_StatusNetworkConnectedIcon->setPixmap(this->m_resPixmapConnectionConnecting);
    if (!this->m_rt->getIContextNetwork()->isConnected())
    {
        if (this->m_ownAircraft.getCallsign().isEmpty())
        {
            this->displayStatusMessage(CStatusMessage::getValidationError("missing callsign"));
            return;
        }

        // send latest aircraft to network/voice
        this->m_rt->getIContextNetwork()->setOwnAircraft(this->m_ownAircraft);
        if (this->m_contextAudioAvailable) this->m_rt->getIContextAudio()->setOwnAircraft(this->m_ownAircraft);

        // Login is based on setting current server
        INetwork::LoginMode mode = INetwork::LoginNormal;
        if (this->ui->rb_SettingsLoginStealthMode->isChecked())
        {
            mode = INetwork::LoginStealth;
            this->displayStatusMessage(CStatusMessage::getInfoMessage("login in stealth mode"));
        }
        else if (this->ui->rb_SettingsLoginObserver->isChecked())
        {
            mode = INetwork::LoginAsObserver;
            this->displayStatusMessage(CStatusMessage::getInfoMessage("login in observer mode"));
        }
        msgs = this->m_rt->getIContextNetwork()->connectToNetwork(static_cast<uint>(mode));
    }
    else
    {
        // disconnect from network
        this->stopUpdateTimers(); // stop update timers, to avoid updates during disconnecting (a short time frame)
        if (this->m_contextAudioAvailable) this->m_rt->getIContextAudio()->leaveAllVoiceRooms();
        msgs = this->m_rt->getIContextNetwork()->disconnectFromNetwork();
    }
    if (!msgs.isEmpty()) this->displayStatusMessages(msgs);
}

/*
 * Is the network context available?
 */
bool MainWindow::isContextNetworkAvailableCheck()
{
    if (this->m_contextNetworkAvailable) return true;
    this->displayStatusMessage(CStatusMessage(CStatusMessage::TypeCore, CStatusMessage::SeverityError, "Network context not available, no updates this time"));
    return false;
}

/*
 * Is the audio context available?
 */
bool MainWindow::isContextAudioAvailableCheck()
{
    if (this->m_contextAudioAvailable) return true;
    this->displayStatusMessage(CStatusMessage(CStatusMessage::TypeCore, CStatusMessage::SeverityError, "Voice context not available"));
    return false;
}

/*
 * Display a status message
 */
void MainWindow::displayStatusMessage(const CStatusMessage &statusMessage)
{
    this->ui->sb_MainStatusBar->show();
    this->m_timerStatusBar->start(3000);
    this->m_statusBarIcon->setPixmap(statusMessage.toIcon());
    this->m_statusBarLabel->setText(statusMessage.getMessage());

    // list
    this->ui->tvp_StatusMessages->insert(statusMessage);
    if (statusMessage.getSeverity() == CStatusMessage::SeverityError) this->displayOverlayInfo(statusMessage);
}

/*
 * Display a status message
 */
void MainWindow::displayStatusMessages(const CStatusMessageList &messages)
{
    if (messages.isEmpty()) return;
    foreach(CStatusMessage msg, messages)
    {
        this->displayStatusMessage(msg);
    }
}

void MainWindow::displayRedirectedOutput(const CStatusMessage &statusMessage, qint64 contextId)
{
    if (!this->m_rt->getIContextApplication()) return;
    if (this->m_rt->getIContextApplication()->getUniqueId() == contextId) return; //self triggered
    this->ui->te_StatusPageConsole->appendHtml(statusMessage.toHtml());
}

/*
* Connection terminated
*/
void MainWindow::connectionTerminated()
{
    this->updateGuiStatusInformation();
}

/*
* Connection status changed
*/
void MainWindow::connectionStatusChanged(uint /** from **/, uint to, const QString & /* message */)
{
    this->updateGuiStatusInformation();
    INetwork::ConnectionStatus newStatus = static_cast<INetwork::ConnectionStatus>(to);
    if (newStatus == INetwork::Connected)
        this->startUpdateTimers();
    else if (newStatus == INetwork::Disconnecting || newStatus == INetwork::Disconnected || newStatus == INetwork::DisconnectedError)
        this->stopUpdateTimers();

    // sounds
    switch (newStatus)
    {
    case INetwork::Connected:
        this->playNotifcationSound(BlackSound::CSoundGenerator::NotificationLogin);
        break;
    case INetwork::Disconnected:
        this->playNotifcationSound(BlackSound::CSoundGenerator::NotificationLogoff);
        break;
    case INetwork::DisconnectedError:
        this->playNotifcationSound(BlackSound::CSoundGenerator::NotificationError);
        break;
    default:
        break;
    }
}

/*
* Timer event
*/
void MainWindow::timerBasedUpdates()
{
    QObject *sender = QObject::sender();
    if (sender == this->m_timerUpdateAtcStationsOnline)
    {
        int t = this->ui->hs_SettingsGuiAtcRefreshTime->value() * 1000;
        this->m_timerUpdateAtcStationsOnline->start(t);
        this->reloadAtcStationsOnline();
    }
    else if (sender == this->m_timerUpdateAircraftsInRange)
    {
        int t = this->ui->hs_SettingsGuiAircraftRefreshTime->value() * 1000;
        this->m_timerUpdateAircraftsInRange->start(t);
        this->reloadAircraftsInRange();
    }
    else if (sender == this->m_timerUpdateUsers)
    {
        int t = this->ui->hs_SettingsGuiUserRefreshTime->value() * 1000;
        this->m_timerUpdateUsers->start(t);
        this->reloadAllUsers();
    }
    else if (sender == this->m_timerContextWatchdog)
    {
        this->setContextAvailability();
        this->updateGuiStatusInformation();
    }
    else if (sender == this->m_timerSimulator)
    {
        this->updateSimulatorData();
    }

    // own aircraft
    if (sender == this->m_timerUpdateAircraftsInRange || sender == this->m_timerUpdateAtcStationsOnline)
    {
        this->reloadOwnAircraft(); // regular updates
    }
}

/*
* Context availability
*/
void MainWindow::setContextAvailability()
{
    qint64 t = QDateTime::currentMSecsSinceEpoch();
    this->m_coreAvailable = this->m_rt->getIContextApplication()->ping(t) == t;
    this->m_contextNetworkAvailable = this->m_coreAvailable || this->m_rt->getIContextNetwork()->usingLocalObjects();
    this->m_contextAudioAvailable = this->m_coreAvailable || this->m_rt->getIContextAudio()->usingLocalObjects();
}

/*
* Middle panel changed
*/
void MainWindow::middlePanelChanged(int /* index */)
{
    if (this->isContextNetworkAvailableCheck())
    {
        // remark, ATC stations is handled by tab changed

        if (this->ui->sw_MainMiddle->currentWidget() == this->ui->pg_AircraftsInRange)
        {
            if (this->ui->tvp_AircraftsInRange->rowCount() < 1)
                this->reloadAircraftsInRange();
        }
    }
}

/*
* Update GUI
*/
void MainWindow::updateGuiStatusInformation()
{
    const QString now = QDateTime::currentDateTimeUtc().toString("yyyy-MM-dd HH:mm:ss");
    QString network("unavailable");
    if (this->m_contextNetworkAvailable)
    {
        bool dbus = !this->m_rt->getIContextNetwork()->usingLocalObjects();
        network =  dbus ? now : "local";
        this->ui->cb_StatusWithDBus->setChecked(dbus);
    }

    // handle voice, mute
    QString voice("unavailable");
    if (this->m_contextAudioAvailable)
    {
        voice = this->m_rt->getIContextAudio()->usingLocalObjects() ? "local" : now;
        this->ui->pb_SoundMute->setEnabled(true);
    }
    else
    {
        // voice not available
        this->ui->pb_SoundMute->setEnabled(false);
    }

    // update status fields
    this->ui->le_StatusNetworkContext->setText(network);
    this->ui->le_StatusAudioContext->setText(voice);

    // Connected button
    if (this->m_contextNetworkAvailable && this->m_rt->getIContextNetwork()->isConnected())
    {
        if (this->ui->lbl_StatusNetworkConnectedIcon->toolTip().startsWith("dis", Qt::CaseInsensitive))
            this->ui->lbl_StatusNetworkConnectedIcon->setToolTip(now);
        this->ui->pb_MainConnect->setText("Disconnect");
        this->ui->pb_MainConnect->setStyleSheet("background-color: green");
        this->ui->lbl_StatusNetworkConnectedIcon->setPixmap(this->m_resPixmapConnectionConnected);
    }
    else
    {
        this->ui->lbl_StatusNetworkConnectedIcon->setToolTip("disconnected");
        this->ui->pb_MainConnect->setText("Connect");
        this->ui->pb_MainConnect->setStyleSheet("background-color: ");
        this->ui->lbl_StatusNetworkConnectedIcon->setPixmap(this->m_resPixmapConnectionDisconnected);
    }
}

/*
 * Opacity 0-100
 */
void MainWindow::changeWindowOpacity(int opacity)
{
    if (opacity < 0)
    {
        QObject *sender = QObject::sender();
        if (sender == this->ui->pb_MainKeypadOpacity050)
            opacity = 50;
        else if (sender == this->ui->pb_MainKeypadOpacity100)
            opacity = 100;
        else
            return;
    }
    qreal o = opacity / 100.0;
    o = o < 0.3 ? 0.3 : o;
    QWidget::setWindowOpacity(o);
    this->ui->hs_SettingsGuiOpacity->setValue(o * 100.0);
}

/*
 * Display the info window
 */
void MainWindow::displayOverlayInfo(const QString &message)
{
    if (!this->m_infoWindow)
    {
        this->m_infoWindow = new CInfoWindow(this);
    }

    // display window
    if (message.isEmpty())
    {
        this->m_infoWindow->hide();
    }
    else
    {
        this->m_infoWindow->setInfoMessage(message);
    }
}

/*
 * Info window by
 */
void MainWindow::displayOverlayInfo(const CStatusMessage &message)
{
    this->displayOverlayInfo(message.getMessage());
    // further code goes here, such as marking errors as red ...
}

/*
 * Read users
 */
void MainWindow::reloadAllUsers()
{
    if (!this->isContextNetworkAvailableCheck()) return;
    this->ui->tvp_AllUsers->update(this->m_rt->getIContextNetwork()->getUsers());
}

void MainWindow::updateSimulatorData()
{
    if (this->m_rt->getIContextSimulator()->isConnected())
        ui->le_SimulatorStatus->setText("Connected");
    else
        ui->le_SimulatorStatus->setText("Not connected");

    CAircraft ownAircraft = this->m_rt->getIContextSimulator()->getOwnAircraft();
    ui->le_SimulatorLatitude->setText(ownAircraft.getSituation().latitude().toFormattedQString());
    ui->le_SimulatorLongitude->setText(ownAircraft.getSituation().longitude().toFormattedQString());
    ui->le_SimulatorAltitude->setText(ownAircraft.getSituation().getAltitude().toFormattedQString());
    ui->le_SimulatorPitch->setText(ownAircraft.getSituation().getPitch().toFormattedQString());
    ui->le_SimulatorBank->setText(ownAircraft.getSituation().getBank().toFormattedQString());
    ui->le_SimulatorHeading->setText(ownAircraft.getSituation().getHeading().toFormattedQString());
    ui->le_SimulatorGroundSpeed->setText(ownAircraft.getSituation().getGroundSpeed().toFormattedQString());

    ui->le_SimulatorCom1Active->setText(ownAircraft.getCom1System().getFrequencyActive().toFormattedQString());
    ui->le_SimulatorCom1Standby->setText(ownAircraft.getCom1System().getFrequencyStandby().toFormattedQString());
    ui->le_SimulatorCom2Active->setText(ownAircraft.getCom2System().getFrequencyActive().toFormattedQString());
    ui->le_SimulatorCom2Standby->setText(ownAircraft.getCom2System().getFrequencyStandby().toFormattedQString());
    ui->le_SimulatorTransponder->setText(ownAircraft.getTransponderCodeFormatted());
}

void MainWindow::simulatorConnectionChanged(bool isAvailable)
{
    // Simulator timer, TODO remove later
    if (isAvailable)
        this->m_timerSimulator->start(500);
    else
        this->m_timerSimulator->stop();
}