mirror of
https://github.com/BeamMP/BeamMP-Server.git
synced 2025-07-03 16:25:35 +00:00
90 lines
2.6 KiB
C++
90 lines
2.6 KiB
C++
#include "TConsole.h"
|
|
#include "Common.h"
|
|
#include "Compat.h"
|
|
|
|
#include "TLuaEngine.h"
|
|
|
|
#include <ctime>
|
|
#include <sstream>
|
|
|
|
std::string GetDate() {
|
|
std::chrono::system_clock::time_point now = std::chrono::system_clock::now();
|
|
time_t tt = std::chrono::system_clock::to_time_t(now);
|
|
tm local_tm {};
|
|
#ifdef WIN32
|
|
localtime_s(&local_tm, &tt);
|
|
#else // unix
|
|
localtime_r(&tt, &local_tm);
|
|
#endif // WIN32
|
|
std::stringstream date;
|
|
int S = local_tm.tm_sec;
|
|
int M = local_tm.tm_min;
|
|
int H = local_tm.tm_hour;
|
|
std::string Secs = (S > 9 ? std::to_string(S) : "0" + std::to_string(S));
|
|
std::string Min = (M > 9 ? std::to_string(M) : "0" + std::to_string(M));
|
|
std::string Hour = (H > 9 ? std::to_string(H) : "0" + std::to_string(H));
|
|
date
|
|
<< "["
|
|
<< local_tm.tm_mday << "/"
|
|
<< local_tm.tm_mon + 1 << "/"
|
|
<< local_tm.tm_year + 1900 << " "
|
|
<< Hour << ":"
|
|
<< Min << ":"
|
|
<< Secs
|
|
<< "] ";
|
|
/* TODO
|
|
if (Debug) {
|
|
date << ThreadName()
|
|
<< " ";
|
|
}
|
|
*/
|
|
return date.str();
|
|
}
|
|
|
|
TConsole::TConsole() {
|
|
mCommandline.enable_history();
|
|
mCommandline.set_history_limit(20);
|
|
mCommandline.set_prompt("> ");
|
|
bool success = mCommandline.enable_write_to_file("Server.log");
|
|
if (!success) {
|
|
beammp_error("unable to open file for writing: \"Server.log\"");
|
|
}
|
|
mCommandline.on_command = [this](Commandline& c) {
|
|
auto cmd = c.get_command();
|
|
mCommandline.write("> " + cmd);
|
|
if (cmd == "exit") {
|
|
beammp_info("gracefully shutting down");
|
|
Application::GracefullyShutdown();
|
|
} else if (cmd == "clear" || cmd == "cls") {
|
|
// TODO: clear screen
|
|
} else {
|
|
if (!mLuaEngine) {
|
|
beammp_info("Lua not started yet, please try again in a second");
|
|
} else {
|
|
auto Future = mLuaEngine->EnqueueScript(mStateId, std::make_shared<std::string>(cmd));
|
|
while (!Future->Ready) {
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(1));
|
|
}
|
|
if (Future->Error) {
|
|
beammp_lua_error(Future->ErrorMessage);
|
|
}
|
|
}
|
|
}
|
|
};
|
|
}
|
|
|
|
void TConsole::Write(const std::string& str) {
|
|
auto ToWrite = GetDate() + str;
|
|
mCommandline.write(ToWrite);
|
|
// TODO write to logfile, too
|
|
}
|
|
|
|
void TConsole::WriteRaw(const std::string& str) {
|
|
mCommandline.write(str);
|
|
}
|
|
|
|
void TConsole::InitializeLuaConsole(TLuaEngine& Engine) {
|
|
mLuaEngine = &Engine;
|
|
Engine.EnsureStateExists(mStateId, "Console");
|
|
}
|