From 5b15e754798d6d2552faf2f68413734cf1ebd505 Mon Sep 17 00:00:00 2001
From: Jon King <jon@sibipro.com>
Date: Sat, 27 May 2023 18:17:31 -0700
Subject: [PATCH] feat: add jsonata query operation

---
 package-lock.json               |  9 +++++
 package.json                    |  1 +
 src/core/operations/Jsonata.mjs | 65 +++++++++++++++++++++++++++++++++
 3 files changed, 75 insertions(+)
 create mode 100644 src/core/operations/Jsonata.mjs

diff --git a/package-lock.json b/package-lock.json
index 50639ea8..9cdad930 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -53,6 +53,7 @@
         "js-sha3": "^0.9.3",
         "jsesc": "^3.0.2",
         "json5": "^2.2.3",
+        "jsonata": "^2.0.3",
         "jsonpath-plus": "^9.0.0",
         "jsonwebtoken": "8.5.1",
         "jsqr": "^1.4.0",
@@ -12447,6 +12448,14 @@
         "node": ">=6"
       }
     },
+    "node_modules/jsonata": {
+      "version": "2.0.6",
+      "resolved": "https://registry.npmjs.org/jsonata/-/jsonata-2.0.6.tgz",
+      "integrity": "sha512-WhQB5tXQ32qjkx2GYHFw2XbL90u+LLzjofAYwi+86g6SyZeXHz9F1Q0amy3dWRYczshOC3Haok9J4pOCgHtwyQ==",
+      "engines": {
+        "node": ">= 8"
+      }
+    },
     "node_modules/jsonpath-plus": {
       "version": "9.0.0",
       "resolved": "https://registry.npmjs.org/jsonpath-plus/-/jsonpath-plus-9.0.0.tgz",
diff --git a/package.json b/package.json
index 20180e05..fd89c425 100644
--- a/package.json
+++ b/package.json
@@ -139,6 +139,7 @@
     "js-sha3": "^0.9.3",
     "jsesc": "^3.0.2",
     "json5": "^2.2.3",
+    "jsonata": "^2.0.3",
     "jsonpath-plus": "^9.0.0",
     "jsonwebtoken": "8.5.1",
     "jsqr": "^1.4.0",
diff --git a/src/core/operations/Jsonata.mjs b/src/core/operations/Jsonata.mjs
new file mode 100644
index 00000000..41cf2751
--- /dev/null
+++ b/src/core/operations/Jsonata.mjs
@@ -0,0 +1,65 @@
+/**
+ * @author Jon K (jon@ajarsoftware.com)
+ * @copyright Crown Copyright 2016
+ * @license Apache-2.0
+ */
+
+import jsonata from "jsonata";
+import Operation from "../Operation.mjs";
+import OperationError from "../errors/OperationError.mjs";
+
+/**
+ * Jsonata Query operation
+ */
+class JsonataQuery extends Operation {
+    /**
+     * JsonataQuery constructor
+     */
+    constructor() {
+        super();
+
+        this.name = "Jsonata Query";
+        this.module = "Code";
+        this.description =
+            "Query and transform JSON data with a jsonata query.";
+        this.infoURL = "https://docs.jsonata.org/overview.html";
+        this.inputType = "string";
+        this.outputType = "string";
+        this.args = [
+            {
+                name: "Query",
+                type: "text",
+                value: "string",
+            },
+        ];
+    }
+
+    /**
+     * @param {string} input
+     * @param {Object[]} args
+     * @returns {JSON}
+     */
+    async run(input, args) {
+        const [query] = args;
+        let result, jsonObj;
+
+        try {
+            jsonObj = JSON.parse(input);
+        } catch (err) {
+            throw new OperationError(`Invalid input JSON: ${err.message}`);
+        }
+
+        try {
+            const expression = jsonata(query);
+            result = await expression.evaluate(jsonObj);
+        } catch (err) {
+            throw new OperationError(
+                `Invalid Jsonata Expression: ${err.message}`
+            );
+        }
+
+        return JSON.stringify(result);
+    }
+}
+
+export default JsonataQuery;