intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 89 insertions, 0 deletions

diff --git a/library/cpp/yson/json/json_writer.h b/library/cpp/yson/json/json_writer.h
new file mode 100644
index 0000000000..d84ac0de53
--- /dev/null
+++ b/library/cpp/yson/json/json_writer.h
@@ -0,0 +1,89 @@
+#pragma once
+
+#include <library/cpp/yson/public.h>
+#include <library/cpp/yson/consumer.h>
+
+#include <library/cpp/json/json_writer.h>
+
+#include <util/generic/vector.h>
+
+namespace NYT {
+    ////////////////////////////////////////////////////////////////////////////////
+
+    enum EJsonFormat {
+        JF_TEXT,
+        JF_PRETTY
+    };
+
+    enum EJsonAttributesMode {
+        JAM_NEVER,
+        JAM_ON_DEMAND,
+        JAM_ALWAYS
+    };
+
+    enum ESerializedBoolFormat {
+        SBF_BOOLEAN,
+        SBF_STRING
+    };
+
+    class TJsonWriter
+       : public ::NYson::TYsonConsumerBase {
+    public:
+        TJsonWriter(
+            IOutputStream* output,
+            ::NYson::EYsonType type = ::NYson::EYsonType::Node,
+            EJsonFormat format = JF_TEXT,
+            EJsonAttributesMode attributesMode = JAM_ON_DEMAND,
+            ESerializedBoolFormat booleanFormat = SBF_STRING);
+
+        TJsonWriter(
+            IOutputStream* output,
+            NJson::TJsonWriterConfig config,
+            ::NYson::EYsonType type = ::NYson::EYsonType::Node,
+            EJsonAttributesMode attributesMode = JAM_ON_DEMAND,
+            ESerializedBoolFormat booleanFormat = SBF_STRING);
+
+        void Flush();
+
+        void OnStringScalar(TStringBuf value) override;
+        void OnInt64Scalar(i64 value) override;
+        void OnUint64Scalar(ui64 value) override;
+        void OnDoubleScalar(double value) override;
+        void OnBooleanScalar(bool value) override;
+
+        void OnEntity() override;
+
+        void OnBeginList() override;
+        void OnListItem() override;
+        void OnEndList() override;
+
+        void OnBeginMap() override;
+        void OnKeyedItem(TStringBuf key) override;
+        void OnEndMap() override;
+
+        void OnBeginAttributes() override;
+        void OnEndAttributes() override;
+
+    private:
+        THolder<NJson::TJsonWriter> UnderlyingJsonWriter;
+        NJson::TJsonWriter* JsonWriter;
+        IOutputStream* Output;
+        ::NYson::EYsonType Type;
+        EJsonAttributesMode AttributesMode;
+        ESerializedBoolFormat BooleanFormat;
+
+        void WriteStringScalar(const TStringBuf& value);
+
+        void EnterNode();
+        void LeaveNode();
+        bool IsWriteAllowed();
+
+        TVector<bool> HasUnfoldedStructureStack;
+        int InAttributesBalance;
+        bool HasAttributes;
+        int Depth;
+    };
+
+    ////////////////////////////////////////////////////////////////////////////////
+
+}