From a9a9a6f71fc77be894e4db081e2c4c3796378f6e Mon Sep 17 00:00:00 2001
From: Red-Asuka <liyd0623@gmail.com>
Date: Mon, 6 Jan 2025 11:43:49 +0800
Subject: [PATCH] feat(core): add script execution functionality with uvm
 integration

---
 packages/core/package.json                    |  3 ++
 packages/core/src/utils/index.ts              |  1 +
 packages/core/src/utils/sandbox.ts            | 49 +++++++++++++++++++
 packages/types/base.ts                        |  4 ++
 .../ui/src/components/script/TabFunction.vue  | 32 ++++++++++--
 pnpm-lock.yaml                                | 16 ++++++
 6 files changed, 100 insertions(+), 5 deletions(-)
 create mode 100644 packages/core/src/utils/sandbox.ts

diff --git a/packages/core/package.json b/packages/core/package.json
index 261e3cfa5..a31beca4b 100644
--- a/packages/core/package.json
+++ b/packages/core/package.json
@@ -9,10 +9,13 @@
     "test:dev": "vitest"
   },
   "dependencies": {
+    "buffer": "^6.0.3",
     "uuid": "^11.0.3",
+    "uvm": "^4.0.0",
     "vitest": "^2.1.4"
   },
   "devDependencies": {
+    "@types/mqttx": "workspace:^",
     "@types/uuid": "^10.0.0",
     "@vitest/coverage-istanbul": "^2.1.4",
     "typescript": "~5.6.3"
diff --git a/packages/core/src/utils/index.ts b/packages/core/src/utils/index.ts
index 9a99b4c54..cc3182052 100644
--- a/packages/core/src/utils/index.ts
+++ b/packages/core/src/utils/index.ts
@@ -1 +1,2 @@
 export * from './getMessageId'
+export * from './sandbox'
diff --git a/packages/core/src/utils/sandbox.ts b/packages/core/src/utils/sandbox.ts
new file mode 100644
index 000000000..3805a5eb6
--- /dev/null
+++ b/packages/core/src/utils/sandbox.ts
@@ -0,0 +1,49 @@
+import type { MessageType } from 'mqttx'
+// eslint-disable-next-line unicorn/prefer-node-protocol
+import { Buffer } from 'buffer'
+// @ts-expect-error uvm is not defined
+import uvm from 'uvm'
+
+export function executeScript(opts: {
+  script: string
+  payload: string
+  messageType: MessageType
+  index?: number
+}): Promise<Buffer> {
+  return new Promise((resolve, reject) => {
+    uvm.spawn(
+      {
+        bootCode: `
+          bridge.on('runScript', function ({ script, payload, messageType, index }) {
+            try {
+              function execute(callback) {
+                return callback(payload, messageType, index);
+              };
+              bridge.dispatch(
+                'scriptResult', 
+                Function('execute', 'payload', 'messageType', 'index', script)(execute, payload, messageType, index)
+              );
+            } catch (error) {
+              bridge.dispatch('scriptError', error.message);
+            }
+          });
+        `,
+        bootTimeout: 1000,
+        dispatchTimeout: 1000,
+      },
+      (err: any, context: any) => {
+        if (err) reject(err)
+
+        context.on('scriptResult', (result: any) => {
+          resolve(Buffer.from(result?.toString() ?? ''))
+        })
+
+        context.on('scriptError', (error?: string) => {
+          reject(new Error(error))
+        })
+
+        context.dispatch('runScript', opts)
+      },
+    )
+  })
+}
diff --git a/packages/types/base.ts b/packages/types/base.ts
index 84e0d1b2c..93def56f1 100644
--- a/packages/types/base.ts
+++ b/packages/types/base.ts
@@ -13,6 +13,10 @@ export type Protocol = 'mqtt' | 'mqtts'
 
 export type QoS = 0 | 1 | 2
 
+export type PayloadType = 'Plaintext' | 'Base64' | 'JSON' | 'Hex' | 'CBOR' | 'MsgPack'
+
+export type MessageType = 'all' | 'received' | 'publish'
+
 export type Lang = 'en' | 'zh' | 'ja' | 'hu' | 'tr'
 
 export type Theme = 'light' | 'dark' | 'night'
diff --git a/packages/ui/src/components/script/TabFunction.vue b/packages/ui/src/components/script/TabFunction.vue
index ebb496e73..3552ef685 100644
--- a/packages/ui/src/components/script/TabFunction.vue
+++ b/packages/ui/src/components/script/TabFunction.vue
@@ -1,5 +1,6 @@
 <script setup lang="ts">
 import type { ScriptFunction } from 'mqttx'
+import { executeScript } from '@mqttx/core'
 import { useScriptFunctionStore } from '../../stores'
 
 const { scriptFunctions } = storeToRefs(useScriptFunctionStore())
@@ -31,18 +32,39 @@ const defaultFunction: Record<string, { input: string, content: string }> = {
     input: JSON.stringify({ msg: 'hello' }, null, 2),
     content: `/**
 * @description: default script
-* @param {any} value - Payload
-* @param {string} msgType - Message type, value is 'received' or 'publish'
+* @param {string} message - Message payload
+* @param {string} messageType - Message type, value is 'received' or 'publish'
 * @param {number} index - Index of the message, valid only when script is used in the publish message and timed message is enabled
 * @return {any} - Payload after script processing
 */
-function handlePayload(value, msgType, index) {
-  return value.msg
+function handlePayload(message, messageType, index) {
+  const payload = JSON.parse(message)
+  return payload.msg
 }
 
-execute(handlePayload)`,
+return execute(handlePayload)`,
   },
 }
+onMounted(async () => {
+  await executeScript({
+    script: defaultFunction[currentLang.value].content,
+    payload: defaultFunction[currentLang.value].input,
+    messageType: 'received',
+    index: 0,
+  })
+    .then((result) => {
+      console.log(result)
+    })
+    .catch((error) => {
+      if (error instanceof Error) {
+        ElMessage({
+          message: error.toString(),
+          type: 'error',
+          plain: true,
+        })
+      }
+    })
+})
 const functionContent = ref('')
 watch([currentLang, currentFunction], () => {
   functionContent.value = currentFunction.value ? currentFunction.value.content : defaultFunction[currentLang.value].content
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
index 801fc4f53..05eed663e 100644
--- a/pnpm-lock.yaml
+++ b/pnpm-lock.yaml
@@ -392,13 +392,22 @@ importers:
 
   packages/core:
     dependencies:
+      buffer:
+        specifier: ^6.0.3
+        version: 6.0.3
       uuid:
         specifier: ^11.0.3
         version: 11.0.3
+      uvm:
+        specifier: ^4.0.0
+        version: 4.0.0
       vitest:
         specifier: ^2.1.4
         version: 2.1.4(@types/node@22.9.0)(jsdom@25.0.1)(sass@1.80.6)
     devDependencies:
+      '@types/mqttx':
+        specifier: workspace:^
+        version: link:../types
       '@types/uuid':
         specifier: ^10.0.0
         version: 10.0.0
@@ -10856,6 +10865,13 @@ packages:
     hasBin: true
     dev: false
 
+  /uvm@4.0.0:
+    resolution: {integrity: sha512-eD52y5MkTftMnfVd0TXZFNRfetFsNnFziv+JsRjfCkvWDc5BTPwccihsk2kVpUFoVnxlkjOVAsB1SakPYsaQRA==}
+    engines: {node: '>=18'}
+    dependencies:
+      flatted: 3.3.1
+    dev: false
+
   /validate-npm-package-license@3.0.4:
     resolution: {integrity: sha512-DpKm2Ui/xN7/HQKCtpZxoRWBhZ9Z0kqtygG8XCgNQ8ZlDnxuQmWhj566j8fN4Cu3/JmbhsDo7fcAJq4s9h27Ew==}
     dependencies: