IntenseFinance/gnucash
3
0
Fork 0
mirror of https://github.com/Gnucash/gnucash.git synced 2025-02-25 18:55:30 -06:00
Code Issues Packages Projects Releases Wiki Activity

[gnucash-commands.cpp] colorize cpp get_line on all platforms

Browse Source
This commit is contained in:
Christopher Lam 2023-10-23 15:51:38 +08:00
parent 47e6f09993
commit 7f7e61b4fa
1 changed files with 34 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
gnucash/gnucash-commands.cpp
View File

@ -60,6 +60,38 @@ static std::string empty_string{};
/* This static indicates the debugging module that this .o belongs to. */ /* This static indicates the debugging module that this .o belongs to. */
static QofLogModule log_module = GNC_MOD_GUI; static QofLogModule log_module = GNC_MOD_GUI;
#ifdef _WIN32
struct ConsoleStruct
{
public:
bool has_ansi () { return m_has_ansi; };
private:
HANDLE m_stdoutHandle = INVALID_HANDLE_VALUE;
DWORD m_outModeInit = 0;
bool m_has_ansi = false;
ConsoleStruct () : m_stdoutHandle {GetStdHandle(STD_OUTPUT_HANDLE)}
{
if (m_stdoutHandle != INVALID_HANDLE_VALUE && GetConsoleMode(m_stdoutHandle, &m_outModeInit))
{
SetConsoleMode (m_stdoutHandle, m_outModeInit | ENABLE_VIRTUAL_TERMINAL_PROCESSING);
m_has_ansi = true;
}
}
~ConsoleStruct ()
{
if (m_stdoutHandle == INVALID_HANDLE_VALUE)
return;
printf ("\x1b[0m");
SetConsoleMode (m_stdoutHandle, m_outModeInit);
}
} console_ansi;
#else
struct ConsoleStruct
{
bool has_ansi () { return true; };
} console_ansi;
#endif
static int static int
cleanup_and_exit_with_failure (QofSession *session) cleanup_and_exit_with_failure (QofSession *session)
{ {
@ -110,7 +142,8 @@ static std::string
get_line (const char* prompt) get_line (const char* prompt)
{ {
std::string rv; std::string rv;
std::cout << prompt; std::cout << (console_ansi.has_ansi() ? "\x1b[1;33m" : "") << prompt
<< (console_ansi.has_ansi() ? "\x1b[m" : "");
if (!std::getline (std::cin, rv)) if (!std::getline (std::cin, rv))
gnc_shutdown_cli (1); gnc_shutdown_cli (1);
return rv; return rv;