ConnectionPanel.cpp

Go to the documentation of this file.

#include "Panels/ConnectionPanel.h"
#include "Network/TCPServer.h"
 
namespace Ember {
namespace Monitor {
 
enum { ID_START_STOP = wxID_HIGHEST + 1, ID_DISCONNECT };
 
wxBEGIN_EVENT_TABLE(ConnectionPanel, EmberUI::Panel) EVT_BUTTON(ID_START_STOP, ConnectionPanel::OnStartStop)
    EVT_BUTTON(ID_DISCONNECT, ConnectionPanel::OnDisconnect) wxEND_EVENT_TABLE()
 
        ConnectionPanel::ConnectionPanel(wxWindow *parent, TCPServer *server)
    : EmberUI::Panel(parent, "ConnectionPanel"), m_server(server) {
    CreateUI();
    UpdateStatus();
}
 
void ConnectionPanel::CreateUI() {
    wxBoxSizer *mainSizer = new wxBoxSizer(wxVERTICAL);
 
    wxStaticBox *serverBox = new wxStaticBox(this, wxID_ANY, "Server");
    wxStaticBoxSizer *serverSizer = new wxStaticBoxSizer(serverBox, wxVERTICAL);
 
    wxBoxSizer *statusRow = new wxBoxSizer(wxHORIZONTAL);
    statusRow->Add(new wxStaticText(this, wxID_ANY, "Status:"), 0, wxALIGN_CENTER_VERTICAL | wxRIGHT, 5);
    m_serverStatusLabel = new wxStaticText(this, wxID_ANY, "Stopped");
    m_serverStatusLabel->SetForegroundColour(*wxRED);
    statusRow->Add(m_serverStatusLabel, 1, wxALIGN_CENTER_VERTICAL);
    serverSizer->Add(statusRow, 0, wxEXPAND | wxALL, 5);
 
    wxBoxSizer *portRow = new wxBoxSizer(wxHORIZONTAL);
    portRow->Add(new wxStaticText(this, wxID_ANY, "Port:"), 0, wxALIGN_CENTER_VERTICAL | wxRIGHT, 5);
    m_portInput = new wxTextCtrl(this, wxID_ANY, "12345", wxDefaultPosition, wxSize(70, -1));
    portRow->Add(m_portInput, 0, wxALIGN_CENTER_VERTICAL | wxRIGHT, 10);
    m_startStopButton = new wxButton(this, ID_START_STOP, "Start");
    portRow->Add(m_startStopButton, 0, wxALIGN_CENTER_VERTICAL);
    serverSizer->Add(portRow, 0, wxEXPAND | wxALL, 5);
 
    mainSizer->Add(serverSizer, 0, wxEXPAND | wxALL, 5);
 
    wxStaticBox *clientBox = new wxStaticBox(this, wxID_ANY, "Client");
    wxStaticBoxSizer *clientSizer = new wxStaticBoxSizer(clientBox, wxVERTICAL);
 
    wxBoxSizer *clientStatusRow = new wxBoxSizer(wxHORIZONTAL);
    clientStatusRow->Add(new wxStaticText(this, wxID_ANY, "Status:"), 0, wxALIGN_CENTER_VERTICAL | wxRIGHT, 5);
    m_clientStatusLabel = new wxStaticText(this, wxID_ANY, "Not Connected");
    m_clientStatusLabel->SetForegroundColour(wxColour(128, 128, 128));
    clientStatusRow->Add(m_clientStatusLabel, 1, wxALIGN_CENTER_VERTICAL);
    clientSizer->Add(clientStatusRow, 0, wxEXPAND | wxALL, 5);
 
    wxBoxSizer *addrRow = new wxBoxSizer(wxHORIZONTAL);
    addrRow->Add(new wxStaticText(this, wxID_ANY, "Address:"), 0, wxALIGN_CENTER_VERTICAL | wxRIGHT, 5);
    m_clientAddressLabel = new wxStaticText(this, wxID_ANY, "-");
    addrRow->Add(m_clientAddressLabel, 1, wxALIGN_CENTER_VERTICAL);
    clientSizer->Add(addrRow, 0, wxEXPAND | wxALL, 5);
 
    wxBoxSizer *nameRow = new wxBoxSizer(wxHORIZONTAL);
    nameRow->Add(new wxStaticText(this, wxID_ANY, "Name:"), 0, wxALIGN_CENTER_VERTICAL | wxRIGHT, 5);
    m_clientNameLabel = new wxStaticText(this, wxID_ANY, "-");
    nameRow->Add(m_clientNameLabel, 1, wxALIGN_CENTER_VERTICAL);
    clientSizer->Add(nameRow, 0, wxEXPAND | wxALL, 5);
 
    wxBoxSizer *treeRow = new wxBoxSizer(wxHORIZONTAL);
    treeRow->Add(new wxStaticText(this, wxID_ANY, "Tree:"), 0, wxALIGN_CENTER_VERTICAL | wxRIGHT, 5);
    m_treeIdLabel = new wxStaticText(this, wxID_ANY, "-");
    treeRow->Add(m_treeIdLabel, 1, wxALIGN_CENTER_VERTICAL);
    clientSizer->Add(treeRow, 0, wxEXPAND | wxALL, 5);
 
    m_disconnectButton = new wxButton(this, ID_DISCONNECT, "Disconnect");
    m_disconnectButton->Enable(false);
    clientSizer->Add(m_disconnectButton, 0, wxALIGN_RIGHT | wxALL, 5);
 
    mainSizer->Add(clientSizer, 0, wxEXPAND | wxALL, 5);
 
    mainSizer->AddStretchSpacer(1);
 
    SetSizer(mainSizer);
}
 
void ConnectionPanel::UpdateStatus() {
    if (m_server->IsRunning()) {
        m_serverStatusLabel->SetLabel(wxString::Format("Listening on %d", m_server->GetPort()));
        m_serverStatusLabel->SetForegroundColour(wxColour(0, 128, 0));
        m_startStopButton->SetLabel("Stop");
        m_portInput->Enable(false);
    } else {
        m_serverStatusLabel->SetLabel("Stopped");
        m_serverStatusLabel->SetForegroundColour(*wxRED);
        m_startStopButton->SetLabel("Start");
        m_portInput->Enable(true);
    }
 
    if (m_server->IsClientConnected()) {
        m_clientStatusLabel->SetLabel("Connected");
        m_clientStatusLabel->SetForegroundColour(wxColour(0, 128, 0));
        m_clientAddressLabel->SetLabel(m_server->GetClientAddress());
 
        wxString clientName = m_server->GetClientName();
        m_clientNameLabel->SetLabel(clientName.empty() ? "(pending handshake)" : clientName);
 
        m_disconnectButton->Enable(true);
    } else {
        m_clientStatusLabel->SetLabel("Not Connected");
        m_clientStatusLabel->SetForegroundColour(wxColour(128, 128, 128));
        m_clientAddressLabel->SetLabel("-");
        m_clientNameLabel->SetLabel("-");
        m_treeIdLabel->SetLabel("-");
        m_disconnectButton->Enable(false);
    }
 
    wxWindow::Refresh();
}
 
void ConnectionPanel::OnStartStop(wxCommandEvent &event) {
    if (m_server->IsRunning()) {
        m_server->Stop();
    } else {
        long port = 12345;
        m_portInput->GetValue().ToLong(&port);
        if (port < 1 || port > 65535) {
            wxMessageBox("Invalid port number. Please enter a value between 1 and 65535.", "Error",
                         wxOK | wxICON_ERROR);
            return;
        }
        if (!m_server->Start(static_cast<uint16_t>(port))) {
            wxMessageBox("Failed to start server. Port may be in use.", "Error", wxOK | wxICON_ERROR);
            return;
        }
    }
    UpdateStatus();
}
 
void ConnectionPanel::OnDisconnect(wxCommandEvent &event) {
    m_server->DisconnectClient();
    UpdateStatus();
}
 
void ConnectionPanel::SetTreeStatus(const wxString &status) {
    m_treeIdLabel->SetLabel(status);
    m_treeIdLabel->SetForegroundColour(wxColour(255, 165, 0));
    wxWindow::Refresh();
}
 
void ConnectionPanel::SetTreeId(const wxString &treeId) {
    m_treeIdLabel->SetLabel(treeId);
    m_treeIdLabel->SetForegroundColour(wxColour(0, 128, 0));
    wxWindow::Refresh();
}
 
} // namespace Monitor
} // namespace Ember