From f85912fd255cb0e2f60b43060f17c3ac91c193b6 Mon Sep 17 00:00:00 2001
From: Robbert van der Helm <mail@robbertvanderhelm.nl>
Date: Tue, 29 Dec 2020 21:51:44 +0100
Subject: [PATCH] Add logging for IUnitData structs

---
 src/common/logging/vst3.cpp                  | 35 ++++++++++++++++++++
 src/common/logging/vst3.h                    |  4 +++
 src/common/serialization/vst3/plugin-proxy.h |  1 +
 3 files changed, 40 insertions(+)

diff --git a/src/common/logging/vst3.cpp b/src/common/logging/vst3.cpp
index 435ab1f7..70388205 100644
--- a/src/common/logging/vst3.cpp
+++ b/src/common/logging/vst3.cpp
@@ -461,6 +461,30 @@ bool Vst3Logger::log_request(bool is_host_vst,
     });
 }
 
+bool Vst3Logger::log_request(bool is_host_vst,
+                             const YaUnitData::UnitDataSupported&) {
+    return log_request_base(is_host_vst, [&](auto& message) {
+        message << "IUnitData::unitDataSupported()";
+    });
+}
+
+bool Vst3Logger::log_request(bool is_host_vst,
+                             const YaUnitData::GetUnitData& request) {
+    return log_request_base(is_host_vst, [&](auto& message) {
+        message << "IUnitData::getUnitData(listId = " << request.unit_id
+                << ", &data)";
+    });
+}
+
+bool Vst3Logger::log_request(bool is_host_vst,
+                             const YaUnitData::SetUnitData& request) {
+    return log_request_base(is_host_vst, [&](auto& message) {
+        message << "IUnitData::setUnitData(listId = " << request.unit_id
+                << ", data = <IBStream* containing " << request.data.size()
+                << "bytes>)";
+    });
+}
+
 bool Vst3Logger::log_request(bool is_host_vst,
                              const YaUnitInfo::GetUnitCount& request) {
     return log_request_base(is_host_vst, [&](auto& message) {
@@ -1002,6 +1026,17 @@ void Vst3Logger::log_response(
     });
 }
 
+void Vst3Logger::log_response(bool is_host_vst,
+                              const YaUnitData::GetUnitDataResponse& response) {
+    log_response_base(is_host_vst, [&](auto& message) {
+        message << response.result.string();
+        if (response.result == Steinberg::kResultOk) {
+            message << ", <IBStream* containing " << response.data.size()
+                    << " bytes>";
+        }
+    });
+}
+
 void Vst3Logger::log_response(bool is_host_vst,
                               const YaUnitInfo::GetUnitInfoResponse& response) {
     log_response_base(is_host_vst, [&](auto& message) {
diff --git a/src/common/logging/vst3.h b/src/common/logging/vst3.h
index bf4604a5..7cf87565 100644
--- a/src/common/logging/vst3.h
+++ b/src/common/logging/vst3.h
@@ -114,6 +114,9 @@ class Vst3Logger {
                      const YaProgramListData::GetProgramData&);
     bool log_request(bool is_host_vst,
                      const YaProgramListData::SetProgramData&);
+    bool log_request(bool is_host_vst, const YaUnitData::UnitDataSupported&);
+    bool log_request(bool is_host_vst, const YaUnitData::GetUnitData&);
+    bool log_request(bool is_host_vst, const YaUnitData::SetUnitData&);
     bool log_request(bool is_host_vst, const YaUnitInfo::GetUnitCount&);
     bool log_request(bool is_host_vst, const YaUnitInfo::GetUnitInfo&);
     bool log_request(bool is_host_vst, const YaUnitInfo::GetProgramListCount&);
@@ -181,6 +184,7 @@ class Vst3Logger {
     void log_response(bool is_host_vst, const Configuration&);
     void log_response(bool is_host_vst,
                       const YaProgramListData::GetProgramDataResponse&);
+    void log_response(bool is_host_vst, const YaUnitData::GetUnitDataResponse&);
     void log_response(bool is_host_vst, const YaUnitInfo::GetUnitInfoResponse&);
     void log_response(bool is_host_vst,
                       const YaUnitInfo::GetProgramListInfoResponse&);
diff --git a/src/common/serialization/vst3/plugin-proxy.h b/src/common/serialization/vst3/plugin-proxy.h
index e988445d..05b3a489 100644
--- a/src/common/serialization/vst3/plugin-proxy.h
+++ b/src/common/serialization/vst3/plugin-proxy.h
@@ -26,6 +26,7 @@
 #include "plugin/edit-controller.h"
 #include "plugin/plugin-base.h"
 #include "plugin/program-list-data.h"
+#include "plugin/unit-data.h"
 #include "plugin/unit-info.h"
 
 #pragma GCC diagnostic push