forked from quickshell/quickshell
io/ipchandler: add IpcHandler and qs msg
Also reworks the whole ipc system to use serialized variants.
This commit is contained in:
parent
3690812919
commit
5e2fb14551
GPG key ID:
4C88A185FB89301E
14 changed files with 1428 additions and 27 deletions
236
src/io/ipccomm.cpp
Normal file
236
src/io/ipccomm.cpp
Normal file
|
|
@ -0,0 +1,236 @@
|
|||
#include "ipccomm.hpp"
|
||||
#include <cstdio>
|
||||
#include <variant>
|
||||
|
||||
#include <qcontainerfwd.h>
|
||||
#include <qlogging.h>
|
||||
#include <qloggingcategory.h>
|
||||
#include <qtextstream.h>
|
||||
#include <qtypes.h>
|
||||
|
||||
#include "../core/generation.hpp"
|
||||
#include "../core/ipc.hpp"
|
||||
#include "../core/ipccommand.hpp"
|
||||
#include "../core/logging.hpp"
|
||||
#include "ipc.hpp"
|
||||
#include "ipchandler.hpp"
|
||||
|
||||
using namespace qs::ipc;
|
||||
|
||||
namespace qs::io::ipc::comm {
|
||||
|
||||
struct NoCurrentGeneration: std::monostate {};
|
||||
struct TargetNotFound: std::monostate {};
|
||||
struct FunctionNotFound: std::monostate {};
|
||||
|
||||
using QueryResponse = std::variant<
|
||||
std::monostate,
|
||||
NoCurrentGeneration,
|
||||
TargetNotFound,
|
||||
FunctionNotFound,
|
||||
QVector<WireTargetDefinition>,
|
||||
WireTargetDefinition,
|
||||
WireFunctionDefinition>;
|
||||
|
||||
void QueryMetadataCommand::exec(qs::ipc::IpcServerConnection* conn) const {
|
||||
auto resp = conn->responseStream<QueryResponse>();
|
||||
|
||||
if (auto* generation = EngineGeneration::currentGeneration()) {
|
||||
auto* registry = IpcHandlerRegistry::forGeneration(generation);
|
||||
|
||||
if (this->target.isEmpty()) {
|
||||
resp << registry->wireTargets();
|
||||
} else {
|
||||
auto* handler = registry->findHandler(this->target);
|
||||
|
||||
if (handler) {
|
||||
if (this->function.isEmpty()) {
|
||||
resp << handler->wireDef();
|
||||
} else {
|
||||
auto* func = handler->findFunction(this->function);
|
||||
|
||||
if (func) {
|
||||
resp << func->wireDef();
|
||||
} else {
|
||||
resp << FunctionNotFound();
|
||||
}
|
||||
}
|
||||
} else {
|
||||
resp << TargetNotFound();
|
||||
}
|
||||
}
|
||||
} else {
|
||||
resp << NoCurrentGeneration();
|
||||
}
|
||||
}
|
||||
|
||||
int queryMetadata(IpcClient* client, const QString& target, const QString& function) {
|
||||
client->sendMessage(IpcCommand(QueryMetadataCommand {.target = target, .function = function}));
|
||||
|
||||
QueryResponse slot;
|
||||
if (!client->waitForResponse(slot)) return -1;
|
||||
|
||||
if (std::holds_alternative<QVector<WireTargetDefinition>>(slot)) {
|
||||
const auto& targets = std::get<QVector<WireTargetDefinition>>(slot);
|
||||
|
||||
for (const auto& target: targets) {
|
||||
qCInfo(logBare).noquote() << target.toString();
|
||||
}
|
||||
|
||||
return 0;
|
||||
} else if (std::holds_alternative<WireTargetDefinition>(slot)) {
|
||||
qCInfo(logBare).noquote() << std::get<WireTargetDefinition>(slot).toString();
|
||||
} else if (std::holds_alternative<WireFunctionDefinition>(slot)) {
|
||||
qCInfo(logBare).noquote() << std::get<WireFunctionDefinition>(slot).toString();
|
||||
} else if (std::holds_alternative<TargetNotFound>(slot)) {
|
||||
qCCritical(logBare) << "Target not found.";
|
||||
} else if (std::holds_alternative<FunctionNotFound>(slot)) {
|
||||
qCCritical(logBare) << "Function not found.";
|
||||
} else if (std::holds_alternative<NoCurrentGeneration>(slot)) {
|
||||
qCCritical(logBare) << "Not ready to accept queries yet.";
|
||||
} else {
|
||||
qCCritical(logIpc) << "Received invalid IPC response from" << client;
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
|
||||
struct ArgParseFailed {
|
||||
WireFunctionDefinition definition;
|
||||
bool isCountMismatch = false;
|
||||
quint8 paramIndex = 0;
|
||||
};
|
||||
|
||||
DEFINE_SIMPLE_DATASTREAM_OPS(
|
||||
ArgParseFailed,
|
||||
data.definition,
|
||||
data.isCountMismatch,
|
||||
data.paramIndex
|
||||
);
|
||||
|
||||
struct Completed {
|
||||
bool isVoid = false;
|
||||
QString returnValue;
|
||||
};
|
||||
|
||||
DEFINE_SIMPLE_DATASTREAM_OPS(Completed, data.isVoid, data.returnValue);
|
||||
|
||||
using StringCallResponse = std::variant<
|
||||
std::monostate,
|
||||
NoCurrentGeneration,
|
||||
TargetNotFound,
|
||||
FunctionNotFound,
|
||||
ArgParseFailed,
|
||||
Completed>;
|
||||
|
||||
void StringCallCommand::exec(qs::ipc::IpcServerConnection* conn) const {
|
||||
auto resp = conn->responseStream<StringCallResponse>();
|
||||
|
||||
if (auto* generation = EngineGeneration::currentGeneration()) {
|
||||
auto* registry = IpcHandlerRegistry::forGeneration(generation);
|
||||
|
||||
auto* handler = registry->findHandler(this->target);
|
||||
if (!handler) {
|
||||
resp << TargetNotFound();
|
||||
return;
|
||||
}
|
||||
|
||||
auto* func = handler->findFunction(this->function);
|
||||
if (!func) {
|
||||
resp << FunctionNotFound();
|
||||
return;
|
||||
}
|
||||
|
||||
if (func->argumentTypes.length() != this->arguments.length()) {
|
||||
resp << ArgParseFailed {
|
||||
.definition = func->wireDef(),
|
||||
.isCountMismatch = true,
|
||||
};
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
auto storage = IpcCallStorage(*func);
|
||||
for (auto i = 0; i < this->arguments.length(); i++) {
|
||||
if (!storage.setArgumentStr(i, this->arguments.value(i))) {
|
||||
resp << ArgParseFailed {
|
||||
.definition = func->wireDef(),
|
||||
.paramIndex = static_cast<quint8>(i),
|
||||
};
|
||||
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
func->invoke(handler, storage);
|
||||
|
||||
resp << Completed {
|
||||
.isVoid = func->returnType == &VoidIpcType::INSTANCE,
|
||||
.returnValue = storage.getReturnStr(),
|
||||
};
|
||||
} else {
|
||||
conn->respond(StringCallResponse(NoCurrentGeneration()));
|
||||
}
|
||||
}
|
||||
|
||||
int callFunction(
|
||||
IpcClient* client,
|
||||
const QString& target,
|
||||
const QString& function,
|
||||
const QVector<QString>& arguments
|
||||
) {
|
||||
if (target.isEmpty()) {
|
||||
qCCritical(logBare) << "Target required to send message.";
|
||||
return -1;
|
||||
} else if (function.isEmpty()) {
|
||||
qCCritical(logBare) << "Function required to send message.";
|
||||
return -1;
|
||||
}
|
||||
|
||||
client->sendMessage(
|
||||
IpcCommand(StringCallCommand {.target = target, .function = function, .arguments = arguments})
|
||||
);
|
||||
|
||||
StringCallResponse slot;
|
||||
if (!client->waitForResponse(slot)) return -1;
|
||||
|
||||
if (std::holds_alternative<Completed>(slot)) {
|
||||
auto& result = std::get<Completed>(slot);
|
||||
if (!result.isVoid) {
|
||||
QTextStream(stdout) << result.returnValue << Qt::endl;
|
||||
}
|
||||
|
||||
return 0;
|
||||
} else if (std::holds_alternative<ArgParseFailed>(slot)) {
|
||||
auto& error = std::get<ArgParseFailed>(slot);
|
||||
|
||||
if (error.isCountMismatch) {
|
||||
auto correctCount = error.definition.arguments.length();
|
||||
|
||||
qCCritical(logBare).nospace()
|
||||
<< "Too " << (correctCount < arguments.length() ? "many" : "few")
|
||||
<< " arguments provided (" << correctCount << " required but " << arguments.length()
|
||||
<< " were provided.)";
|
||||
} else {
|
||||
const auto& provided = arguments.at(error.paramIndex);
|
||||
const auto& definition = error.definition.arguments.at(error.paramIndex);
|
||||
|
||||
qCCritical(logBare).nospace()
|
||||
<< "Unable to parse argument " << (error.paramIndex + 1) << " as " << definition.second
|
||||
<< ". Provided argument: " << provided;
|
||||
}
|
||||
|
||||
qCCritical(logBare).noquote() << "Function definition:" << error.definition.toString();
|
||||
} else if (std::holds_alternative<TargetNotFound>(slot)) {
|
||||
qCCritical(logBare) << "Target not found.";
|
||||
} else if (std::holds_alternative<FunctionNotFound>(slot)) {
|
||||
qCCritical(logBare) << "Function not found.";
|
||||
} else if (std::holds_alternative<NoCurrentGeneration>(slot)) {
|
||||
qCCritical(logBare) << "Not ready to accept queries yet.";
|
||||
} else {
|
||||
qCCritical(logIpc) << "Received invalid IPC response from" << client;
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
} // namespace qs::io::ipc::comm
|
||||
Loading…
Add table
Add a link
Reference in a new issue