61 lines
1.4 KiB
C++
61 lines
1.4 KiB
C++
#include "TerminalHelper.hpp"
|
|
#include <VTUtils.hpp>
|
|
#include <VTEncoding.hpp>
|
|
#include <cstdarg>
|
|
#include <iostream>
|
|
#include <cstdio>
|
|
|
|
#if defined(LIBCMO_OS_WIN32)
|
|
#include <Windows.h>
|
|
#include <cstdio>
|
|
#include <io.h>
|
|
#include <fcntl.h>
|
|
#endif
|
|
|
|
namespace Unvirt {
|
|
namespace TerminalHelper {
|
|
|
|
// all of these functions only works on Windows platform
|
|
// due to shitty Windows implementations.
|
|
|
|
bool EnsureTerminalColor(void) {
|
|
#if defined(LIBCMO_OS_WIN32)
|
|
if (_isatty(_fileno(stdout))) {
|
|
HANDLE h_output;
|
|
DWORD dw_mode;
|
|
|
|
h_output = (HANDLE)_get_osfhandle(_fileno(stdout));
|
|
if (!GetConsoleMode(h_output, &dw_mode)) return false;
|
|
if (!SetConsoleMode(h_output, dw_mode | ENABLE_VIRTUAL_TERMINAL_PROCESSING)) return false;
|
|
|
|
return true;
|
|
} else return false;
|
|
#else
|
|
return true;
|
|
#endif
|
|
}
|
|
|
|
bool EnsureTerminalEncoding(void) {
|
|
#if defined(LIBCMO_OS_WIN32)
|
|
if (!SetConsoleCP(CP_UTF8)) return false;
|
|
if (!SetConsoleOutputCP(CP_UTF8)) return false;
|
|
|
|
/*_setmode(_fileno(stdout), _O_U8TEXT);*/
|
|
_setmode(_fileno(stdin), _O_U16TEXT);
|
|
#endif
|
|
return true;
|
|
}
|
|
|
|
void GetCmdLine(std::string& u8cmd) {
|
|
fputs(UNVIRT_TERMCOL_LIGHT_GREEN(("Unvirt> ")), stdout);
|
|
#if defined(LIBCMO_OS_WIN32)
|
|
std::wstring wcmd;
|
|
std::getline(std::wcin, wcmd);
|
|
LibCmo::EncodingHelper::WcharToChar(wcmd, u8cmd, CP_UTF8);
|
|
#else
|
|
std::getline(std::cin, u8cmd);
|
|
#endif
|
|
}
|
|
|
|
}
|
|
} |