{

public:

Private()

{

dfout_C = 0;

rawmode = 0;

console_in = 0;

console_out = 0;

ConsoleWindow = 0;

default_attributes = 0;

state = con_unclaimed;

in_batch = false;

raw_cursor = 0;

};

virtual ~Private()

{

//sync();

}

public:

void print(const char *data)

{

fputs(data, dfout_C);

}

void print_text(color_ostream::color_value clr, const std::string &chunk)

{

if(!in_batch && state == con_lineedit)

{

clearline();

color(clr);

print(chunk.c_str());

reset_color();

prompt_refresh();

}

else

{

color(clr);

print(chunk.c_str());

}

}

void begin_batch()

{

assert(!in_batch);

in_batch = true;

if (state == con_lineedit)

{

clearline();

}

}

void end_batch()

{

assert(in_batch);

flush();

in_batch = false;

if (state == con_lineedit)

{

reset_color();

prompt_refresh();

}

}

void flush()

{

fflush(dfout_C);

}

int get_columns(void)

{

CONSOLE_SCREEN_BUFFER_INFO inf = { 0 };

GetConsoleScreenBufferInfo(console_out, &inf);

return (size_t)inf.dwSize.X;

}

int get_rows(void)

{

CONSOLE_SCREEN_BUFFER_INFO inf = { 0 };

GetConsoleScreenBufferInfo(console_out, &inf);

return (size_t)inf.dwSize.Y;

}

void clear()

{

system("cls");

}

void clearline()

{

CONSOLE_SCREEN_BUFFER_INFO inf = { 0 };

GetConsoleScreenBufferInfo(console_out, &inf);

// Blank to EOL

char* tmp = (char*)malloc(inf.dwSize.X);

memset(tmp, ' ', inf.dwSize.X);

output(tmp, inf.dwSize.X, 0, inf.dwCursorPosition.Y);

free(tmp);

COORD coord = {0, inf.dwCursorPosition.Y}; // Windows uses 0-based coordinates

SetConsoleCursorPosition(GetStdHandle(STD_OUTPUT_HANDLE), coord);

}

void gotoxy(int x, int y)

{

COORD coord = {x-1, y-1}; // Windows uses 0-based coordinates

SetConsoleCursorPosition(GetStdHandle(STD_OUTPUT_HANDLE), coord);

}

void color(int index)

{

HANDLE hConsole = GetStdHandle(STD_OUTPUT_HANDLE);

SetConsoleTextAttribute(hConsole, index == COLOR_RESET ? default_attributes : index);

}

void reset_color( void )

{

HANDLE hConsole = GetStdHandle(STD_OUTPUT_HANDLE);

SetConsoleTextAttribute(hConsole, default_attributes);

}

void cursor(bool enable)

{

if(enable)

{

HANDLE hConsoleOutput;

CONSOLE_CURSOR_INFO structCursorInfo;

hConsoleOutput = GetStdHandle( STD_OUTPUT_HANDLE );

GetConsoleCursorInfo( hConsoleOutput, &structCursorInfo ); // Get current cursor size

structCursorInfo.bVisible = TRUE;

SetConsoleCursorInfo( hConsoleOutput, &structCursorInfo );

}

else

{

HANDLE hConsoleOutput;

CONSOLE_CURSOR_INFO structCursorInfo;

hConsoleOutput = GetStdHandle( STD_OUTPUT_HANDLE );

GetConsoleCursorInfo( hConsoleOutput, &structCursorInfo ); // Get current cursor size

structCursorInfo.bVisible = FALSE;

SetConsoleCursorInfo( hConsoleOutput, &structCursorInfo );

}

}

void output(const char* str, size_t len, int x, int y)

{

COORD pos = { (SHORT)x, (SHORT)y };

DWORD count = 0;

WriteConsoleOutputCharacterA(console_out, str, len, pos, &count);

}

void prompt_refresh()

{

size_t cols = get_columns();

size_t plen = prompt.size();

const char * buf = raw_buffer.c_str();

size_t len = raw_buffer.size();

int cooked_cursor = raw_cursor;

while ((plen + cooked_cursor) >= cols)

{

buf++;

len--;

cooked_cursor--;

}

while (plen + len > cols)

{

len--;

}

CONSOLE_SCREEN_BUFFER_INFO inf = { 0 };

GetConsoleScreenBufferInfo(console_out, &inf);

output(prompt.c_str(), plen, 0, inf.dwCursorPosition.Y);

output(buf, len, plen, inf.dwCursorPosition.Y);

if (plen + len < (size_t)inf.dwSize.X)

{

// Blank to EOL

char* tmp = (char*)malloc(inf.dwSize.X - (plen + len));

memset(tmp, ' ', inf.dwSize.X - (plen + len));

output(tmp, inf.dwSize.X - (plen + len), len + plen, inf.dwCursorPosition.Y);

free(tmp);

}

inf.dwCursorPosition.X = (SHORT)(cooked_cursor + plen);

SetConsoleCursorPosition(console_out, inf.dwCursorPosition);

}

int prompt_loop(recursive_mutex * lock, CommandHistory & history)

{

raw_buffer.clear(); // make sure the buffer is empty!

size_t plen = prompt.size();

raw_cursor = 0;

int history_index = 0;

// The latest history entry is always our current buffer, that

// initially is just an empty string.

const std::string empty;

history.add(empty);

CONSOLE_SCREEN_BUFFER_INFO inf = { 0 };

GetConsoleScreenBufferInfo(console_out, &inf);

size_t cols = inf.dwSize.X;

output(prompt.c_str(), plen, 0, inf.dwCursorPosition.Y);

inf.dwCursorPosition.X = (SHORT)plen;

SetConsoleCursorPosition(console_out, inf.dwCursorPosition);

while (1)

{

INPUT_RECORD rec;

DWORD count;

lock->unlock();

ReadConsoleInputA(console_in, &rec, 1, &count);

lock->lock();

if (rec.EventType != KEY_EVENT || !rec.Event.KeyEvent.bKeyDown)

continue;

switch (rec.Event.KeyEvent.wVirtualKeyCode)

{

case VK_RETURN: // enter

history.remove();

return raw_buffer.size();

case VK_BACK: // backspace

if (raw_cursor > 0 && raw_buffer.size() > 0)

{

raw_buffer.erase(raw_cursor-1,1);

raw_cursor--;

prompt_refresh();

}

break;

case VK_LEFT: // left arrow

if (raw_cursor > 0)

{

raw_cursor--;

prompt_refresh();

}

break;

case VK_RIGHT: // right arrow

if (raw_cursor != raw_buffer.size())

{

raw_cursor++;

prompt_refresh();

}

break;

case VK_UP:

case VK_DOWN:

// up and down arrow: history

if (history.size() > 1)

{

// Update the current history entry before to

// overwrite it with tne next one.

history[history_index] = raw_buffer;

// Show the new entry

history_index += (rec.Event.KeyEvent.wVirtualKeyCode == VK_UP) ? 1 : -1;

if (history_index < 0)

{

history_index = 0;

break;

}

else if (history_index >= history.size())

{

history_index = history.size()-1;

break;

}

raw_buffer = history[history_index];

raw_cursor = raw_buffer.size();

prompt_refresh();

}

break;

case VK_DELETE:

// delete

if (raw_buffer.size() > 0 && raw_cursor < raw_buffer.size())

{

raw_buffer.erase(raw_cursor,1);

prompt_refresh();

}

break;

case VK_HOME:

raw_cursor = 0;

prompt_refresh();

break;

case VK_END:

raw_cursor = raw_buffer.size();

prompt_refresh();

break;

default:

if (rec.Event.KeyEvent.uChar.AsciiChar < ' ' ||

rec.Event.KeyEvent.uChar.AsciiChar > '~')

continue;

if (raw_buffer.size() == raw_cursor)

raw_buffer.append(1,rec.Event.KeyEvent.uChar.AsciiChar);

else

raw_buffer.insert(raw_cursor,1,rec.Event.KeyEvent.uChar.AsciiChar);

raw_cursor++;

prompt_refresh();

break;

}

}

}

int lineedit(const std::string & prompt, std::string & output, recursive_mutex * lock, CommandHistory & ch)

{

output.clear();

reset_color();

int count;

state = con_lineedit;

this->prompt = prompt;

count = prompt_loop(lock, ch);

if(count != -1)

output = raw_buffer;

state = con_unclaimed;

print("\n");

return count;

}

FILE * dfout_C;

int rawmode;

HANDLE console_in;

HANDLE console_out;

HWND ConsoleWindow;

HWND MainWindow;

WORD default_attributes;

// current state

enum console_state

{

con_unclaimed,

con_lineedit

} state;

bool in_batch;

std::string prompt; // current prompt string

std::string raw_buffer; // current raw mode buffer

int raw_cursor; // cursor position in the buffer

{

d = 0;

wlock = 0;

inited = false;

{

d = new Private();

int hConHandle;

long lStdHandle;

CONSOLE_SCREEN_BUFFER_INFO coninfo;

FILE *fp;

DWORD oldMode, newMode;

DWORD dwTheardId;

HWND h = ::GetTopWindow(0 );

while ( h )

{

DWORD pid;

dwTheardId = ::GetWindowThreadProcessId( h,&pid);

if ( pid == GetCurrentProcessId() )

{

// here h is the handle to the window

break;

}

h = ::GetNextWindow( h , GW_HWNDNEXT);

}

d->MainWindow = h;

// Allocate a console!

AllocConsole();

d->ConsoleWindow = GetConsoleWindow();

wlock = new recursive_mutex();

HMENU hm = GetSystemMenu(d->ConsoleWindow,false);

DeleteMenu(hm, SC_CLOSE, MF_BYCOMMAND);

// set the screen buffer to be big enough to let us scroll text

GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &coninfo);

d->default_attributes = coninfo.wAttributes;

coninfo.dwSize.Y = MAX_CONSOLE_LINES; // How many lines do you want to have in the console buffer

SetConsoleScreenBufferSize(GetStdHandle(STD_OUTPUT_HANDLE), coninfo.dwSize);

// redirect unbuffered STDOUT to the console

d->console_out = GetStdHandle(STD_OUTPUT_HANDLE);

lStdHandle = (long)d->console_out;

hConHandle = _open_osfhandle(lStdHandle, _O_TEXT);

d->dfout_C = _fdopen( hConHandle, "w" );

setvbuf( d->dfout_C, NULL, _IONBF, 0 );

// redirect unbuffered STDIN to the console

d->console_in = GetStdHandle(STD_INPUT_HANDLE);

lStdHandle = (long)d->console_in;

hConHandle = _open_osfhandle(lStdHandle, _O_TEXT);

fp = _fdopen( hConHandle, "r" );

*stdin = *fp;

setvbuf( stdin, NULL, _IONBF, 0 );

GetConsoleMode(GetStdHandle(STD_INPUT_HANDLE),&oldMode);

newMode = oldMode | ENABLE_ECHO_INPUT | ENABLE_PROCESSED_INPUT | ENABLE_LINE_INPUT;

SetConsoleMode(GetStdHandle(STD_INPUT_HANDLE),newMode);

SetConsoleCtrlHandler(NULL,true);

std::ios::sync_with_stdio();

// make our own weird streams so our IO isn't redirected

std::cin.tie(this);

clear();

inited = true;

// DOESN'T WORK - locks up DF!

// ForceForegroundWindow(d->MainWindow);

return true;

{

lock_guard <recursive_mutex> g(*wlock);

FreeConsole();

inited = false;

return true;

{

//color_ostream::begin_batch();

wlock->lock();

if (inited)

d->begin_batch();

{

if (inited)

d->end_batch();

wlock->unlock();

{

lock_guard <recursive_mutex> g(*wlock);

if (inited)

d->flush();

{

lock_guard <recursive_mutex> g(*wlock);

if (inited)

d->print_text(color, text);

{

lock_guard <recursive_mutex> g(*wlock);

int ret = -1;

if(inited)

ret = d->get_columns();

return ret;

{

lock_guard <recursive_mutex> g(*wlock);

int ret = -1;

if(inited)

ret = d->get_rows();

return ret;

{

lock_guard <recursive_mutex> g(*wlock);

if(inited)

d->clear();

{

lock_guard <recursive_mutex> g(*wlock);

if(inited)

d->gotoxy(x,y);

{

lock_guard <recursive_mutex> g(*wlock);

if(inited)

d->cursor(enable);

{

wlock->lock();

int ret = -2;

if(inited)

ret = d->lineedit(prompt,output,wlock,ch);

wlock->unlock();

return ret;