2021-06-27 21:25:29 +00:00
|
|
|
/***
|
|
|
|
Copyright (C) 2021 J Reece Wilson (a/k/a "Reece"). All rights reserved.
|
|
|
|
|
|
|
|
File: Hooks.cpp
|
|
|
|
Date: 2021-6-12
|
|
|
|
Author: Reece
|
|
|
|
***/
|
2021-09-30 14:57:41 +00:00
|
|
|
#include <Source/RuntimeInternal.hpp>
|
2021-06-27 21:25:29 +00:00
|
|
|
#include "Hooks.hpp"
|
|
|
|
|
|
|
|
namespace Aurora::Console::Hooks
|
|
|
|
{
|
2021-10-24 10:19:47 +00:00
|
|
|
static AuThreadPrimitives::MutexUnique_t gMutex;
|
|
|
|
static AuList<LineHook_cb> gLineFunctionalCallbacks;
|
|
|
|
static AuList<AuSPtr<IConsoleSubscriber>> gLineSubscribers;
|
2021-06-27 21:25:29 +00:00
|
|
|
|
2021-09-06 10:58:08 +00:00
|
|
|
AUKN_SYM void AddSubscription(const AuSPtr<IConsoleSubscriber> &subscriber)
|
2021-06-27 21:25:29 +00:00
|
|
|
{
|
2021-09-06 10:58:08 +00:00
|
|
|
AU_LOCK_GUARD(gMutex);
|
|
|
|
AuTryInsert(gLineSubscribers, subscriber);
|
|
|
|
}
|
|
|
|
|
|
|
|
AUKN_SYM void RemoveSubscription(const AuSPtr<IConsoleSubscriber> &subscriber)
|
|
|
|
{
|
|
|
|
AU_LOCK_GUARD(gMutex);
|
|
|
|
AuTryDeleteList(gLineSubscribers, subscriber);
|
|
|
|
}
|
|
|
|
|
|
|
|
AUKN_SYM void AddFunctionalHook(LineHook_cb hook)
|
|
|
|
{
|
|
|
|
AU_LOCK_GUARD(gMutex);
|
|
|
|
AuTryInsert(gLineFunctionalCallbacks, hook);
|
2021-06-27 21:25:29 +00:00
|
|
|
}
|
|
|
|
|
2021-11-05 17:34:23 +00:00
|
|
|
AUKN_SYM void SetCallbackAndDisableCmdProcessing(const AuSPtr<Hooks::ITextLineSubscriber> &subscriber)
|
|
|
|
{
|
|
|
|
gExternalLineProcessor = subscriber;
|
|
|
|
}
|
|
|
|
|
2021-12-24 16:25:12 +00:00
|
|
|
static void WriteLoggerFailedWarning()
|
2021-06-27 21:25:29 +00:00
|
|
|
{
|
2022-01-19 17:08:13 +00:00
|
|
|
static AuString kLoggerError = "Something went from while dispatching a log line\n";
|
2021-12-24 16:25:12 +00:00
|
|
|
Console::WriteStdOut(kLoggerError.data(), kLoggerError.size());
|
|
|
|
|
|
|
|
#if defined(AURORA_IS_MODERNNT_DERIVED)
|
|
|
|
OutputDebugStringA(kLoggerError.c_str());
|
|
|
|
#endif
|
|
|
|
}
|
2021-06-27 21:25:29 +00:00
|
|
|
|
2021-12-24 16:25:12 +00:00
|
|
|
static void TryWrite(const ConsoleMessage &msg)
|
|
|
|
{
|
|
|
|
for (const auto &callback : gLineFunctionalCallbacks)
|
2021-06-27 21:25:29 +00:00
|
|
|
{
|
2021-12-24 16:25:12 +00:00
|
|
|
try
|
2021-06-27 21:25:29 +00:00
|
|
|
{
|
|
|
|
callback(msg);
|
|
|
|
}
|
2021-12-24 16:25:12 +00:00
|
|
|
catch (...)
|
|
|
|
{
|
|
|
|
WriteLoggerFailedWarning();
|
|
|
|
}
|
|
|
|
}
|
2021-09-06 10:58:08 +00:00
|
|
|
|
2021-12-24 16:25:12 +00:00
|
|
|
for (const auto &sub : gLineSubscribers)
|
|
|
|
{
|
|
|
|
try
|
2021-09-06 10:58:08 +00:00
|
|
|
{
|
|
|
|
sub->OnMessage(msg);
|
|
|
|
}
|
2021-12-24 16:25:12 +00:00
|
|
|
catch (...)
|
|
|
|
{
|
|
|
|
WriteLoggerFailedWarning();
|
|
|
|
}
|
2021-06-27 21:25:29 +00:00
|
|
|
}
|
2021-12-24 16:25:12 +00:00
|
|
|
}
|
2021-09-06 10:58:08 +00:00
|
|
|
|
2021-12-24 16:25:12 +00:00
|
|
|
void WriteLine(const ConsoleMessage &msg)
|
|
|
|
{
|
|
|
|
AU_LOCK_GUARD(gMutex);
|
2021-09-06 10:58:08 +00:00
|
|
|
|
2021-12-24 16:25:12 +00:00
|
|
|
try
|
|
|
|
{
|
|
|
|
if (msg.line.find('\n') == AuString::npos) [[likely]]
|
|
|
|
{
|
|
|
|
TryWrite(msg);
|
|
|
|
}
|
|
|
|
else [[unlikely]]
|
|
|
|
{
|
|
|
|
Parse::SplitNewlines(msg.line,
|
|
|
|
[&](const AuString &line)
|
2021-09-06 10:58:08 +00:00
|
|
|
{
|
2021-12-24 16:25:12 +00:00
|
|
|
ConsoleMessage dup = msg;
|
|
|
|
dup.line = line;
|
|
|
|
|
|
|
|
TryWrite(msg);
|
|
|
|
});
|
|
|
|
}
|
|
|
|
}
|
|
|
|
catch (...)
|
|
|
|
{
|
|
|
|
// Loggers experiencing something fucky is a liablity
|
|
|
|
WriteLoggerFailedWarning();
|
2021-06-27 21:25:29 +00:00
|
|
|
}
|
|
|
|
}
|
2021-09-06 10:58:08 +00:00
|
|
|
|
|
|
|
void Init()
|
|
|
|
{
|
2021-10-24 10:19:47 +00:00
|
|
|
gMutex = AuThreadPrimitives::MutexUnique();
|
2021-09-06 10:58:08 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void Deinit()
|
|
|
|
{
|
|
|
|
gLineFunctionalCallbacks.clear();
|
|
|
|
gLineSubscribers.clear();
|
|
|
|
gMutex.reset();
|
|
|
|
}
|
2021-06-27 21:25:29 +00:00
|
|
|
}
|