textOutputPlugin

Author: Pavel Siska

new-process-api/textOutputPlugin.hpp

@@ -0,0 +1,94 @@
+#pragma once
+
+#include "fieldDescriptor.hpp"
+#include "outputPlugin.hpp"
+
+namespace ipxp {
+
+inline std::ostream& operator<<(std::ostream& os, std::byte b)
+{
+	os << "0x" << std::hex << std::uppercase << static_cast<int>(b);
+	return os;
+}
+
+template<typename T>
+static void
+printScalar(const FieldDescriptor& field, const ScalarAccessor<T>& accessor, const void* data)
+{
+	std::cout << "[" << field.getGroup() << "] " << field.getName() << ": " << accessor(data)
+			  << "\n";
+}
+
+template<typename T>
+static void
+printVector(const FieldDescriptor& field, const VectorAccessor<T>& accessor, const void* data)
+{
+	std::cout << "[" << field.getGroup() << "] " << field.getName() << ": [";
+
+	bool first = true;
+	for (const auto& value : accessor(data)) {
+		if (!first)
+			std::cout << ", ";
+		std::cout << value;
+		first = false;
+	}
+
+	std::cout << "]\n";
+}
+
+static void
+printScalarVariant(const FieldDescriptor& field, const ScalarValueGetter& variant, const void* data)
+{
+	const auto visitor = [&](const auto& accessor) { printScalar(field, accessor, data); };
+	std::visit(visitor, variant);
+}
+
+static void
+printVectorVariant(const FieldDescriptor& field, const VectorValueGetter& variant, const void* data)
+{
+	const auto visitor = [&](const auto& accessor) { printVector(field, accessor, data); };
+	std::visit(visitor, variant);
+}
+
+class TextOutputPlugin : public OutputPlugin {
+public:
+	TextOutputPlugin(const FieldManager& manager, const std::vector<ProcessPluginEntry>& plugins)
+		: OutputPlugin(manager, plugins)
+	{
+	}
+
+	void processRecord(FlowRecordUniquePtr& flowRecord) override
+	{
+		auto outputFields = m_fieldManager.getBiflowFields();
+		auto fn = [&](ProcessPlugin* processPlugin) {
+			// const void* pluginExportData = processPlugin->getExportData();
+			const void* pluginExportData = flowRecord->getPluginContext(0);
+
+			for (const auto& outputField : outputFields) {
+				if (!outputField.isInRecord(*flowRecord.get())) {
+					continue;
+				}
+
+				const auto& getter = outputField.getValueGetter();
+
+				std::visit(
+					[&](const auto& variant) {
+						using GetterT = std::decay_t<decltype(variant)>;
+						if constexpr (std::is_same_v<GetterT, ScalarValueGetter>) {
+							printScalarVariant(outputField, variant, pluginExportData);
+						} else if constexpr (std::is_same_v<GetterT, VectorValueGetter>) {
+							printVectorVariant(outputField, variant, pluginExportData);
+						}
+					},
+					getter);
+			}
+		};
+
+		for (const auto& pluginEntry : m_plugins) {
+			std::cout << "Processing " << pluginEntry.name << "\n";
+			fn(reinterpret_cast<ProcessPlugin*>(pluginEntry.plugin.get()));
+		}
+	}
+};
+
+} // namespace ipxp