stclient.cpp

#include "stclient.h"

#include <cstring>

#include <wx/tokenzr.h>

#include "sdk.h" // Code::Blocks SDK
#include "cbstyledtextctrl.h"


const wxString stClient::MSG_END = wxT("<EOM>");
const wxString stClient::MSG_PING = wxT("ping");
const wxString stClient::MSG_CURSOR = wxT("moveCursor");
const wxString stClient::MSG_CREATE_CDB = wxT("createCDB");


stClient::stClient(unsigned int nServerPort, unsigned int nWritePort)
   : m_pServer(nullptr), m_bConnected(false), m_nServerPort(nServerPort), m_nWritePort(nWritePort)
{

}

stClient::~stClient()
{
    if(m_pServer)
    {
        m_pServer->Destroy();

    }
}

bool stClient::SendMessageToSourceTrail(const wxString& sMessage)
{

    wxIPV4address addr;
    addr.Hostname(wxT("localhost"));
    addr.Service(m_nWritePort);
// Create the socket
    wxSocketClient* pSocket = new wxSocketClient();

    // Wait for the connection event
    pSocket->Connect(addr, false);
    pSocket->WaitOnConnect(2);
    if(pSocket->IsConnected())
    {
        pSocket->SetFlags(wxSOCKET_BLOCK | wxSOCKET_WAITALL);
        Manager::Get()->GetLogManager()->Log(wxString::Format(wxT("SENT: '%s'"), sMessage.c_str()));
        // Protocol encoding is UTF-8
        const auto raw = sMessage.utf8_str();
        pSocket->Write(raw, std::strlen(raw));
        pSocket->Close();
        pSocket->Destroy();

        return true;
    }

    return false;
}


void stClient::HandleMessages(const wxString& sBuffer)
{
    Manager::Get()->GetLogManager()->Log(wxString::Format(wxT("HandleMessages: '%s'"), sBuffer.c_str()));
    m_sMessageBuffer << sBuffer;

    while(!m_sMessageBuffer.empty())
    {
        int nFind = m_sMessageBuffer.Find(wxT("<EOM>"));
        if(nFind == wxNOT_FOUND)
        {
            break;
        }
        else
        {
            wxString sMessage = m_sMessageBuffer.Left(nFind);
            m_sMessageBuffer = m_sMessageBuffer.Mid(nFind+5);
            DecodeMessage(sMessage);
        }
    }
}


void stClient::DecodeMessage(const wxString& sMessage)
{
    Manager::Get()->GetLogManager()->Log(wxString::Format(wxT("DecodeMessage: '%s'"), sMessage.c_str()));
    wxArrayString asDecode = wxStringTokenize(sMessage, wxT(">>"));
    for(size_t i = 0; i < asDecode.size(); ++i)
    {
        asDecode[i].Trim();
        Manager::Get()->GetLogManager()->Log(wxString::Format(
            wxT("%u = %s"),
            static_cast<unsigned int>(i),
            asDecode[i].c_str()
        ));
    }

    if(asDecode[0] == MSG_PING)
    {
        SendPing();
    }
    else if(asDecode[0] == MSG_CURSOR && asDecode.size() > 6)
    {
        unsigned long nLine;
        unsigned long nCol;
        if(asDecode[4].ToULong(&nLine) && asDecode[6].ToULong(&nCol))
        {
            MoveCursor(asDecode[2], nLine, nCol);
        }
    }
    else
    {
        Manager::Get()->GetLogManager()->Log(wxString::Format(
            wxT("'%s' != '%s'"),
            asDecode[0].c_str(),
            MSG_CURSOR.c_str()
        ));
    }
}

void stClient::MoveCursor(const wxString& sFile, unsigned long nLine, unsigned long nCol)
{
    Manager::Get()->GetLogManager()->Log(wxString::Format(
        wxT("MoveCursor: %s %u,%u"),
        sFile.c_str(),
        static_cast<unsigned int>(nLine),
        static_cast<unsigned int>(nCol)
    ));

    cbEditor* pEditor = (cbEditor*)Manager::Get()->GetEditorManager()->IsBuiltinOpen(sFile);
    if (!pEditor)
    {
        pEditor = Manager::Get()->GetEditorManager()->Open(sFile); //this will send a editor activated event
    }
    if (pEditor)
    {
        Manager::Get()->GetEditorManager()->SetActiveEditor(pEditor);
        pEditor->Activate(); //this does not run FillList, because m_Ignore is true here
        pEditor->SetFocus();           // ...and set focus to this editor
        cbEditor* pInbuilt = Manager::Get()->GetEditorManager()->GetBuiltinActiveEditor();
        pInbuilt->GotoLine(nLine, true);
        cbStyledTextCtrl* pControl = pInbuilt->GetControl();
        // These methods need the line and column 0-based, the protocol delivers them 1-based
        //TODO: Bug or feature, but the column seems to be 0-based while it is required 1-based on the sending side?
        nCol += pControl->PositionFromLine(nLine - 1);
        pControl->SetCurrentPos(nCol);
        pControl->SetSelection(nCol, nCol);
        pControl->EnsureCaretVisible();

    }
    else
    {
        Manager::Get()->GetLogManager()->Log(wxT("Could not create an editor for file"));
    }
}


bool stClient::StartServer()
{
    Manager::Get()->GetLogManager()->Log(wxT("stClient::StartServer"));
    // Create the address - defaults to localhost:0 initially
    wxIPV4address addr;
    addr.Service(m_nServerPort);
    // Create the socket. We maintain a class pointer so we can
    // sut it down
    m_pServer = new wxSocketServer(addr);
    // We use Ok() here to see if the server is really listening
    if (! m_pServer->Ok())
    {
        Manager::Get()->GetLogManager()->Log(wxT("stClient Failed to start server"));
        return false;
    }
    Manager::Get()->GetLogManager()->Log(wxT("stClient Server started"));
    // Set up the event handler and subscribe to connection events
    int nId = wxNewId();
    m_pServer->SetEventHandler(*this, nId);
    m_pServer->SetNotify(wxSOCKET_CONNECTION_FLAG);
    m_pServer->Notify(true);

    Connect(nId, wxEVT_SOCKET, (wxObjectEventFunction)&stClient::OnServerSocketEvent);
    return true;
}


void stClient::OnServerSocketEvent(wxSocketEvent& event)
{
    Manager::Get()->GetLogManager()->Log(wxT("OnServerSocketEvent: Connection"));
    // Accept the new connection and get the socket pointer
    wxSocketBase* pSock = m_pServer->Accept(false);
    // Tell the new socket how and where to process its events
    int nId = wxNewId();
    pSock->SetEventHandler(*this, nId);
    pSock->SetNotify(wxSOCKET_INPUT_FLAG | wxSOCKET_LOST_FLAG);
    pSock->Notify(true);
    Connect(nId, wxEVT_SOCKET, (wxObjectEventFunction)&stClient::OnSocketEvent);
}
void stClient::OnSocketEvent(wxSocketEvent& event)
{
    Manager::Get()->GetLogManager()->Log(wxT("OnSocketEvent"));
    wxSocketBase * pSock = event.GetSocket();
    wxString message;
    unsigned int count = 0;
    char buf[512];
    switch(event.GetSocketEvent())
    {
        case wxSOCKET_INPUT:
            Manager::Get()->GetLogManager()->Log(wxT("OnSocketEvent: Input"));
            pSock->Read(buf, sizeof(buf));
            count = pSock->LastCount();
            // Protocol encoding is UTF-8
            message = wxString::FromUTF8(buf, count);
            Manager::Get()->GetLogManager()->Log(wxString::Format(
                wxT("OnSocketEvent: Read %u: %s"),
                static_cast<unsigned int>(count),
                message.c_str()
            ));
            HandleMessages(message);
            break;
        // The server hangs up after sending the data
        case wxSOCKET_LOST:
            Manager::Get()->GetLogManager()->Log(wxT("OnSocketEvent: Closed"));
            pSock->Destroy();
            break;

    }
}


void stClient::SendPing()
{
    SendMessageToSourceTrail(wxT("ping>>CodeBlocks<EOM>"));
}

void stClient::SendLocation()
{

    //get current active editor
    cbEditor* pEditor = Manager::Get()->GetEditorManager()->GetBuiltinActiveEditor();
    if(pEditor)
    {
        cbStyledTextCtrl* pControl = pEditor->GetControl();
        wxString sMessage;
        int nPosition = pControl->GetCurrentPos();
        int nLine = pControl->GetCurrentLine();
        wxString sFilename = pEditor->GetFilename();

        sFilename.Replace(wxT("\\"), wxT("/"));

        // The methods deliver the line and column 0-based, the protocol needs them 1-based
        sMessage.Printf(wxT("setActiveToken>>%s>>%d>>%d<EOM>"), sFilename.c_str(), nLine + 1, pControl->GetColumn(nPosition) + 1);
        SendMessageToSourceTrail(sMessage);
    }
}