@nirguna/plugin-fasm 1.2.0 → 1.4.0

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

@@ -1,6 +1,10 @@
 import {types} from 'putout';
 
-const {isMemberExpression} = types;
+const {
+    isMemberExpression,
+    isAwaitExpression,
+} = types;
+
 const PREFIX = '__nirguna_';
 
 export const report = ({name, newName}) => `Add prefix to label: '${name}' -> '${newName}'`;
@@ -45,8 +49,9 @@ function getIds(path, name) {
                 return;
             
             const calleePath = path.parentPath.get('callee');
+            const parentAwait = isAwaitExpression(path.parentPath.parentPath);
             
-            if (calleePath === path)
+            if (!parentAwait && calleePath === path)
                 return;
             
             ids.push(path);

package/lib/apply-registers/index.js

@@ -2,19 +2,7 @@ import {types} from 'putout';
 
 const {isArrayExpression} = types;
 
-function getType(arg) {
-    return arg.__nirguna_type || 'i16';
-}
-
-export const report = (path) => {
-    const arg1 = path.get('left.0');
-    const arg2 = path.get('right.0');
-    const type1 = getType(arg1);
-    const type2 = getType(arg2);
-    
-    if (type1 !== type2)
-        return `Types mismatch: '${arg1}:${type1} = ${arg2}:${type2}'`;
-    
+export const report = () => {
     return `Use registers to address memory`;
 };
 
@@ -25,16 +13,8 @@ export const match = () => ({
 });
 
 export const replace = () => ({
-    '[__a] = [__b]': (vars, path) => {
-        const arg1 = path.get('left.0');
-        const arg2 = path.get('right.0');
-        const type1 = getType(arg1);
-        const type2 = getType(arg2);
-        
-        if (type1 !== type2)
-            return path;
-        
-        const reg = type1 === 'i8' ? 'al' : 'ax';
+    '[__a] = [__b]': () => {
+        const reg = 'ax';
         
         return `{
             mov(${reg}, [__b]);

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

@@ -3,6 +3,7 @@ import {
     types,
     operator,
 } from 'putout';
+import {TARGETS} from '#targets';
 
 const {
     replaceWithMultiple,
@@ -80,21 +81,6 @@ 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',
@@ -130,7 +116,7 @@ function getBytes(path) {
     const target = getTarget(path);
     
     if (target)
-        return TARGET[target].size;
+        return TARGETS[target].size;
     
     const {returnType} = path.node;
     
@@ -146,7 +132,7 @@ function getRegister(path, reg) {
     const target = getTarget(path);
     
     if (target) {
-        const {bits} = TARGET[target];
+        const {bits} = TARGETS[target];
         
         return REG[bits][reg];
     }

package/lib/insert-target/index.js

@@ -1,19 +1,5 @@
 import {template} from 'putout';
-
-const createBinary = (address) => `{
-    org(${address});
-    use16();
-}`;
-
-const TARGET = {
-    boot: createBinary('0x7c00'),
-    kernel: createBinary('0x7e00'),
-    nemesis: createBinary('0x500'),
-    linux: `{
-        format.ELF64.executable;
-        entry.$;
-    }`,
-};
+import {TARGETS} from '#targets';
 
 export const report = (path, {options}) => {
     const {target} = options;
@@ -24,7 +10,9 @@ export const fix = (path, {options}) => {
     const {target} = options;
     
     path.node.extra.target = target;
-    path.node.body.unshift(template.ast(TARGET[target]));
+    const {code} = TARGETS[target];
+    
+    path.node.body.unshift(template.ast(code));
 };
 
 export const include = () => ['Program'];
@@ -38,5 +26,5 @@ export const filter = (path, {options}) => {
     if (path.node.extra.target)
         return;
     
-    return TARGET[target];
+    return TARGETS[target];
 };

package/lib/targets.js

@@ -0,0 +1,30 @@
+const createBinary = (address) => `{
+    org(${address});
+    use16();
+}`;
+
+export const TARGETS = {
+    linux: {
+        size: 8,
+        bits: 'i64',
+        code: `{
+            format.ELF64.executable;
+            entry.$;
+        }`,
+    },
+    kernel: {
+        size: 2,
+        bits: 'i16',
+        code: createBinary('0x7e00'),
+    },
+    boot: {
+        size: 2,
+        bits: 'i16',
+        code: createBinary('0x7c00'),
+    },
+    nemesis: {
+        size: 2,
+        bits: 'i16',
+        code: createBinary('0x500'),
+    },
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nirguna/plugin-fasm",
-  "version": "1.2.0",
+  "version": "1.4.0",
   "type": "module",
   "author": "coderaiser <mnemonic.enemy@gmail.com> (https://github.com/coderaiser)",
   "description": "🐊nirguna plugin adds ability to optimize fasm",
@@ -13,6 +13,9 @@
     "type": "git",
     "url": "git+https://github.com/coderaiser/putout.git"
   },
+  "imports": {
+    "#targets": "./lib/targets.js"
+  },
   "scripts": {
     "test": "madrun test",
     "watch:test": "madrun watch:test",
@@ -35,7 +38,7 @@
   "devDependencies": {
     "@putout/plugin-remove-nested-blocks": "^9.1.0",
     "@putout/test": "^15.1.1",
-    "c8": "^10.0.0",
+    "c8": "^11.0.0",
     "eslint": "^10.0.1",
     "eslint-plugin-n": "^17.0.0",
     "eslint-plugin-putout": "^31.0.1",