@makano/rew 1.3.4 → 1.3.5

package/lib/rew/modules/context.js

@@ -76,7 +76,12 @@ module.exports.prepareContext = function (
       if(!context.module.exports) context.module.exports = {};
       context.module.exports[name] = object;
       context[name] = object;
-    }, Main: (cb) => (['main', cb?.main ?? cb]), attach: (object) => {
+    }, Main: (cb) => {
+      if(cb?.main){
+        cb.main._class = cb;
+      }
+      return (['main', cb?.main ?? cb]);
+    }, attach: (object) => {
       for(let i in object){
         if(!context[i]) context[i] = object[i];
       }

package/lib/rew/modules/runtime.js

@@ -52,9 +52,16 @@ module.exports.runPath = function runPath(filepath, options = {}, custom_context
 
 		if(context.module.main && (context.module.exports?.main || (typeof context.module.exports == "function" && context.module.exports.name == 'main'))){
 			const mainFn = context.module.exports.main ?? context.module.exports;
+			let ctx = context;
+			if(mainFn._class){
+				ctx = mainFn._class;
+				for(let i in context){
+					ctx[i] = context[i];
+				}
+			}
 			return {
 				context,
-				returns: mainFn.call(context, context.process.argv)
+				returns: mainFn.call(ctx, context.process.argv)
 			}
 		} else {
 			return {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@makano/rew",
-  "version": "1.3.4",
+  "version": "1.3.5",
   "description": "A simple coffescript runtime and app manager",
   "main": "main.js",
   "directories": {