Console.cc

Go to the documentation of this file.

// **************************************************************************
// **************************************************************************
#include "Console.h"

#include <unistd.h>

#include <sstream>
#include <iostream>

#include "tools.h"

#include "ReadlineColor.h"

using namespace std;

// --------------------------------------------------------------------------
//
Console::Console(const char *name) : Readline(name), fContinous(false)
{
    fLogO.SetNullOutput();
    fLogI.SetNullOutput();
    fLogO.SetBacklog(true);
    fLogI.SetBacklog(true);
}

// --------------------------------------------------------------------------
//
//
Console::~Console()
{
    // flush buffer to display before it is destroyed in its destructor
    fLogO.Display();
    fLogI.Display();
}

void Console::PrintReadlineError(const std::string &str)
{
    fLogI << kRed << str << endl;
}

// --------------------------------------------------------------------------
//
//
bool Console::PrintGeneralHelp()
{
    return ReadlineColor::PrintGeneralHelp(fLogI, GetName());
}

// --------------------------------------------------------------------------
//
//
bool Console::PrintCommands()
{
    return ReadlineColor::PrintCommands(fLogI);
}

// --------------------------------------------------------------------------
//
//
bool Console::PrintKeyBindings()
{
    return ReadlineColor::PrintKeyBindings(fLogI);
}

void Console::Lock()
{
    // FIXME: Check missing
    fLogO.Display(true);
    fLogO.SetBacklog(false);
    fLogO.SetNullOutput(false);
}

void Console::Unlock()
{
    // FIXME: Check missing
    fLogO.SetNullOutput(true);
    fLogO.SetBacklog(true);
}

// --------------------------------------------------------------------------
//
//
bool Console::Process(const string &str)
{
    if (ReadlineColor::Process(fLogI, str))
        return true;

    if (str.substr(0, 3)==".w ")
    {
        Lock();
        usleep(stoul(str.substr(3))*1000);
        Unlock();
        return true;
    }

    if (Readline::Process(str))
        return true;

    return false;
}

// --------------------------------------------------------------------------
//
//
void Console::Startup()
{
    // Call readline's startup (just in case, it is empty)
    Readline::Startup();

    // First flush the buffer of the stream which is synchronous
    // with the prompt
    fLogI.Display(true);

    // Now flush the stream which is asychronous
    fLogO.Display(true);

    // The order has the advantage that output initiated by the prompt
    // is not interrupter by the synchronous stream
}

// --------------------------------------------------------------------------
//
//
void Console::EventHook(bool)
{
    // If the output is continous and we are going to output something
    // first jump back to the beginning of the line (well, that
    // doesn't work well if the input line is already two lines)
    // and then flush the buffer.
    const bool newline = fContinous && fLogO.GetSizeBacklog()>0;
    if (newline)
    {
        // Clear the line we are going to overwrite
        std::cout << "\r\033[0K";
        fLogO.Display(true);
    }

    // Call Readline's EventHook to update the prompt
    // and signal readline so that a new prompt is displayed
    Readline::EventHook(newline);
}

string Console::GetLinePrompt() const
{
    const string siz = fLogO.GetSizeStr();

    ostringstream str;
    str << '[' << GetLine();
    return fContinous ? str.str()+']' : str.str()+':'+siz+']';
}

// --------------------------------------------------------------------------
//
//
void Console::Run(const char *)
{
    // Flush the buffer before we print the boot message
    fLogO.Display(true);

    ReadlineColor::PrintBootMsg(fLogI, GetName());

    // Flush the buffer before we start out readline loop
    fLogI.Display(true);
    fLogO.Display(true);

    // Now run readlines main loop
    Readline::Run();

    // flush buffer to display
    fLogI.Display(true);
    fLogO.Display(true);
}

// **************************************************************************
// **************************************************************************

ConsoleStream::~ConsoleStream()
{
    fLogO.Display();
}

// --------------------------------------------------------------------------
//
ConsoleStream::ConsoleStream(const char *name) : Readline(name)
{
    fLogO.SetBacklog(false);
    fLogO.SetNullOutput(false);
    ReadlineColor::PrintBootMsg(fLogO, GetName(), false);
}

void ConsoleStream::PrintReadlineError(const std::string &str)
{
    fLogO << kRed << str << endl;
}

// --------------------------------------------------------------------------
//
//
void ConsoleStream::Run(const char *)
{
    while (!IsStopped())
    {
        const string buf = GetExternalInput();
        SetExternalInput("");
        if (!buf.empty())
            ProcessLine(buf);

        usleep(100000);
    }
}