JsonGetKeys

zenustech · Oct 29, 2024 · f367bf6 · f367bf6
1 parent 9b71388
commit f367bf6
Showing 1 changed file with 24 additions and 1 deletion.
diff --git a/zeno/src/nodes/JsonProcess.cpp b/zeno/src/nodes/JsonProcess.cpp
@@ -13,7 +13,7 @@
 #include <tinygltf/json.hpp>
 #include <zeno/zeno.h>
 
-using Json = nlohmann::json;
+using Json = nlohmann::ordered_json;
 
 namespace zeno {
 struct JsonObject : IObjectClone<JsonObject> {
@@ -415,6 +415,29 @@ ZENDEFNODE(JsonGetString, {
         "deprecated"
     },
 });
+
+struct JsonGetKeys : zeno::INode {
+    virtual void apply() override {
+        auto json = get_input<JsonObject>("json");
+        auto list = std::make_shared<ListObject>();
+        for (auto& [key, _] : json->json.items()) {
+            list->arr.emplace_back(std::make_shared<zeno::StringObject>(key));
+        }
+        set_output2("keys", list);
+    }
+};
+ZENDEFNODE(JsonGetKeys, {
+    {
+        {"json"},
+    },
+    {
+        "keys",
+    },
+    {},
+    {
+        "json"
+    },
+});
 struct JsonGetTypeName : zeno::INode {
     virtual void apply() override {
         auto json = get_input<JsonObject>("json");