From 5068d34c08f951a7ea6257d305a1627b09a95817 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Sat, 4 May 2024 19:44:55 +0200
Subject: Adding upstream version 0.11.1.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 src/yaml-extension-functions.cc | 103 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 103 insertions(+)
 create mode 100644 src/yaml-extension-functions.cc

(limited to 'src/yaml-extension-functions.cc')

diff --git a/src/yaml-extension-functions.cc b/src/yaml-extension-functions.cc
new file mode 100644
index 0000000..0a58652
--- /dev/null
+++ b/src/yaml-extension-functions.cc
@@ -0,0 +1,103 @@
+/**
+ * Copyright (c) 2022, Timothy Stack
+ *
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * * Redistributions of source code must retain the above copyright notice, this
+ * list of conditions and the following disclaimer.
+ * * Redistributions in binary form must reproduce the above copyright notice,
+ * this list of conditions and the following disclaimer in the documentation
+ * and/or other materials provided with the distribution.
+ * * Neither the name of Timothy Stack nor the names of its contributors
+ * may be used to endorse or promote products derived from this software
+ * without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ''AS IS'' AND ANY
+ * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR ANY
+ * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ * @file yaml-extension-functions.cc
+ */
+
+#include <string>
+
+#define RYML_SINGLE_HDR_DEFINE_NOW
+
+#include "ryml_all.hpp"
+#include "sqlite-extension-func.hh"
+#include "vtab_module.hh"
+#include "yajlpp/yajlpp.hh"
+
+using namespace lnav::roles::literals;
+
+static void
+ryml_error_to_um(const char* msg, size_t len, ryml::Location loc, void* ud)
+{
+    intern_string_t src = intern_string::lookup(
+        string_fragment::from_bytes(loc.name.data(), loc.name.size()));
+    auto& sf = *(static_cast<string_fragment*>(ud));
+    auto msg_str = string_fragment::from_bytes(msg, len).trim().to_string();
+
+    if (loc.offset == sf.length()) {
+        loc.line -= 1;
+    }
+    throw lnav::console::user_message::error("failed to parse YAML content")
+        .with_reason(msg_str)
+        .with_snippet(lnav::console::snippet::from(
+            source_location{src, (int32_t) loc.line}, ""));
+}
+
+static text_auto_buffer
+yaml_to_json(string_fragment in)
+{
+    ryml::Callbacks callbacks(&in, nullptr, nullptr, ryml_error_to_um);
+
+    ryml::set_callbacks(callbacks);
+    auto tree = ryml::parse_in_arena(
+        "input", ryml::csubstr{in.data(), (size_t) in.length()});
+
+    auto output = ryml::emit_json(
+        tree, tree.root_id(), ryml::substr{}, /*error_on_excess*/ false);
+    auto buf = auto_buffer::alloc(output.len);
+    buf.resize(output.len);
+    output = ryml::emit_json(tree,
+                             tree.root_id(),
+                             ryml::substr(buf.in(), buf.size()),
+                             /*error_on_excess*/ true);
+
+    return {std::move(buf)};
+}
+
+int
+yaml_extension_functions(struct FuncDef** basic_funcs,
+                         struct FuncDefAgg** agg_funcs)
+{
+    static struct FuncDef yaml_funcs[] = {
+        sqlite_func_adapter<decltype(&yaml_to_json), yaml_to_json>::builder(
+            help_text("yaml_to_json",
+                      "Convert a YAML document to a JSON-encoded string")
+                .sql_function()
+                .with_parameter({"yaml", "The YAML value to convert to JSON."})
+                .with_tags({"json", "yaml"})
+                .with_example({
+                    "To convert the document \"abc: def\"",
+                    "SELECT yaml_to_json('abc: def')",
+                })),
+
+        {nullptr},
+    };
+
+    *basic_funcs = yaml_funcs;
+
+    return SQLITE_OK;
+}
-- 
cgit v1.2.3