Support function classes, arg variable

Author: ceifa

bin/demoon

@@ -3,6 +3,6 @@
 const { start } = require('../src/index');
 
 const snippets = process.argv.splice(2);
-const [entryFile] = snippets
+const [entryFile, ...arg] = snippets
 
-start(entryFile)
+start(entryFile, arg)

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "demoon",
-  "version": "0.0.6",
+  "version": "0.0.7",
   "description": "Lua + Node",
   "main": "src/index.js",
   "bin": {
@@ -13,6 +13,6 @@
     "lua"
   ],
   "dependencies": {
-    "wasmoon": "1.8.2"
+    "wasmoon": "1.11.0"
   }
 }

package.json

@@ -1,8 +1,9 @@
-const { LuaFactory } = require('wasmoon')
+const { LuaFactory, LuaMultiReturn } = require('wasmoon')
 const path = require('path')
-const fs = require('fs').promises
+const FunctionClassTypeExtension = require('./legacyclasses')
+const fs = require('fs/promises')
 
-const start = async (entryFile) => {
+const start = async (entryFile, arg) => {
     const factory = new LuaFactory(undefined, process.env)
 
     const fullEntryFile = path.resolve(process.cwd(), entryFile)
@@ -13,6 +14,8 @@ const start = async (entryFile) => {
 
     const engine = await factory.createEngine({ injectObjects: true })
 
+    engine.global.registerTypeExtension(10, new FunctionClassTypeExtension)
+    engine.global.set('arg', arg)
     engine.global.set('typeof', value => typeof value)
     engine.global.set('instanceof', (value, type) => value instanceof type)
     engine.global.set('new', (constructor, ...args) => new constructor(...args))

src/index.js

@@ -0,0 +1,29 @@
+const { decorateProxy, LuaTypeExtension } = require("wasmoon");
+
+const functionClasses = [
+    Buffer,
+    Object,
+    Array,
+    String,
+    Number,
+]
+
+module.exports = class extends LuaTypeExtension {
+    constructor(thread, injectObject) {
+        super(thread, 'js_functionclass')
+    }
+
+    pushValue(thread, { target, options }) {
+        // If is a function not bounded yet
+        if (typeof target === 'function' && !options?.proxy) {
+            const isLegacyNativeClass = functionClasses.includes(target)
+            // If it has a prototype, probably is something important
+            const hasPrototype = target.prototype && Object.keys(target.prototype).length > 0
+
+            if (isLegacyNativeClass || hasPrototype) {
+                thread.pushValue(decorateProxy(target, { proxy: true }))
+                return true
+            }
+        }
+    }
+}