@nirguna/plugin-fasm 1.1.0 → 1.2.0

package/README.md

@@ -46,4 +46,4 @@ Transforms:
 - ✅[`remove-useless-braces`](https://putout.cloudcmd.io/#/gist/140689014540d48d501c1fa50b3694ca/e1521db4f951c38c17a6c3af46456c47f1f3a1a8);
 - ✅[`remove-useless-promise`](https://putout.cloudcmd.io/#/gist/d21eb87bc77b53e54c28b1ae8442a403/1b35064142068e623a564f578517c2aace55ceba);
 - ✅[`switch-cmp-operands`](https://putout.cloudcmd.io/#/gist/cdf71331f1024c9c966d4756c6159684/9d3f2ebeb41cd096de7919f0c39bc75cbaf5ce0c);
-- ✅[`insert-target`](https://putout.cloudcmd.io/#/gist/e7e3f495fdfde1b065117fc0ae41873c/b6fe3dc81d33c81d332ce5d551b8a2fb40b63d70);
+- ✅[`insert-target`](https://putout.cloudcmd.io/#/gist/e7e3f495fdfde1b065117fc0ae41873c/b6fe3dc81d33c81d332ce5d551b8a2fb40b63d70);

package/lib/add-label-prefix/index.js

@@ -41,8 +41,15 @@ function getIds(path, name) {
             if (isMemberExpression(path.parentPath))
                 return;
             
-            if (name === path.node.name)
-                ids.push(path);
+            if (name !== path.node.name)
+                return;
+            
+            const calleePath = path.parentPath.get('callee');
+            
+            if (calleePath === path)
+                return;
+            
+            ids.push(path);
         },
     });
     

package/lib/convert-args-to-regs/index.js

@@ -80,6 +80,21 @@ export const replace = () => ({
     },
 });
 
+const TARGET = {
+    linux: {
+        size: 8,
+        bits: 'i64',
+    },
+    kernel: {
+        size: 2,
+        bits: 'i16',
+    },
+    boot: {
+        size: 2,
+        bits: 'i16',
+    },
+};
+
 const REG = {
     i16: {
         eax: 'ax',
@@ -104,7 +119,19 @@ const BYTES = {
     i64: 8,
 };
 
+function getTarget({scope}) {
+    const programPath = scope.getProgramParent().path;
+    const {target} = programPath.node.extra;
+    
+    return target;
+}
+
 function getBytes(path) {
+    const target = getTarget(path);
+    
+    if (target)
+        return TARGET[target].size;
+    
     const {returnType} = path.node;
     
     if (!returnType)
@@ -116,6 +143,14 @@ function getBytes(path) {
 }
 
 function getRegister(path, reg) {
+    const target = getTarget(path);
+    
+    if (target) {
+        const {bits} = TARGET[target];
+        
+        return REG[bits][reg];
+    }
+    
     const {returnType} = path.node;
     
     if (!returnType)

package/lib/insert-target/index.js

@@ -15,9 +15,14 @@ const TARGET = {
     }`,
 };
 
-export const report = (path, {target}) => `Insert target: '${target}'`;
+export const report = (path, {options}) => {
+    const {target} = options;
+    return `Insert target: '${target}'`;
+};
 
-export const fix = (path, {target = 'boot'}) => {
+export const fix = (path, {options}) => {
+    const {target} = options;
+    
     path.node.extra.target = target;
     path.node.body.unshift(template.ast(TARGET[target]));
 };

package/lib/split-binary-expression/index.js

@@ -24,8 +24,8 @@ export const replace = () => ({
             return '__a += __c';
         
         return `{
-    	__a = __b;
-        __a += __c
-    }`;
+            __a = __b;
+            __a += __c
+        }`;
     },
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nirguna/plugin-fasm",
-  "version": "1.1.0",
+  "version": "1.2.0",
   "type": "module",
   "author": "coderaiser <mnemonic.enemy@gmail.com> (https://github.com/coderaiser)",
   "description": "🐊nirguna plugin adds ability to optimize fasm",