intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 68 insertions, 0 deletions

diff --git a/library/cpp/protobuf/json/proto2json_printer.h b/library/cpp/protobuf/json/proto2json_printer.h
new file mode 100644
index 0000000000..9dc5aa86c6
--- /dev/null
+++ b/library/cpp/protobuf/json/proto2json_printer.h
@@ -0,0 +1,68 @@
+#pragma once
+
+#include "json_output.h"
+
+#include <google/protobuf/descriptor.h>
+#include <google/protobuf/descriptor.pb.h>
+#include <google/protobuf/message.h>
+
+#include <util/generic/strbuf.h>
+#include <util/generic/string.h>
+
+namespace NProtobufJson {
+    struct TProto2JsonConfig;
+
+    class TProto2JsonPrinter {
+    public:
+        TProto2JsonPrinter(const TProto2JsonConfig& config);
+        virtual ~TProto2JsonPrinter();
+
+        virtual void Print(const NProtoBuf::Message& proto, IJsonOutput& json, bool closeMap = true);
+
+        virtual const TProto2JsonConfig& GetConfig() const {
+            return Config;
+        }
+
+    protected:
+        virtual TStringBuf MakeKey(const NProtoBuf::FieldDescriptor& field);
+
+        virtual void PrintField(const NProtoBuf::Message& proto,
+                                const NProtoBuf::FieldDescriptor& field,
+                                IJsonOutput& json,
+                                TStringBuf key = {});
+
+        void PrintRepeatedField(const NProtoBuf::Message& proto,
+                                const NProtoBuf::FieldDescriptor& field,
+                                IJsonOutput& json,
+                                TStringBuf key = {});
+
+        void PrintSingleField(const NProtoBuf::Message& proto,
+                              const NProtoBuf::FieldDescriptor& field,
+                              IJsonOutput& json,
+                              TStringBuf key = {});
+
+        void PrintKeyValue(const NProtoBuf::Message& proto,
+                           IJsonOutput& json);
+
+        TString MakeKey(const NProtoBuf::Message& proto,
+                        const NProtoBuf::FieldDescriptor& field);
+
+        template <bool InMapContext>
+        void PrintEnumValue(const TStringBuf& key,
+                            const NProtoBuf::EnumValueDescriptor* value,
+                            IJsonOutput& json);
+
+        template <bool InMapContext>
+        void PrintStringValue(const NProtoBuf::FieldDescriptor& field,
+                              const TStringBuf& key, const TString& value,
+                              IJsonOutput& json);
+
+        template <class T>
+        bool NeedStringifyNumber(T value) const;
+
+    protected:
+        const TProto2JsonConfig& Config;
+        TString TmpBuf;
+    };
+
+}