Merge pull request #24 from FuelLabs/silentcicero-unsafemaths

Allow for unsafeMaths

Author: SilentCicero

examples/example.js

@@ -52,4 +52,6 @@ const source = yulp.compile(`
   }
   `, fs, './examples');
 
-console.log(yulp.print(source.results), source);
+const yulOutput = yulp.print(source.results);
+
+console.log(yulOutput);

package.json

@@ -1,6 +1,6 @@
 {
   "name": "yulp",
-  "version": "0.2.2",
+  "version": "0.2.3",
   "description": "A low-level highly-efficient language for the Ethereum Virtual Machine.",
   "main": "src/index.js",
   "scripts": {

src/index.test.js

@@ -149,6 +149,18 @@ test('yulp should be yul', t => {
   t.equal(print(compile(` object "SimpleStore" { code { const nick := 0 } } `).results),
     `object \"SimpleStore\" { code {  } }`, 'const');
 
+  t.equal(print(compile(` object "SimpleStore" { code { const nick := unsafeAdd(1, 2) } } `).results),
+    `object \"SimpleStore\" { code { let nick := add(1, 2) } }`, 'unsafe add');
+
+  t.equal(print(compile(` object "SimpleStore" { code { const nick := unsafeDiv(1, 2) } } `).results),
+    `object \"SimpleStore\" { code { let nick := div(1, 2) } }`, 'unsafe div');
+
+  t.equal(print(compile(` object "SimpleStore" { code { const nick := unsafeMul(1, 2) } } `).results),
+    `object \"SimpleStore\" { code { let nick := mul(1, 2) } }`, 'unsafe mul');
+
+    t.equal(print(compile(` object "SimpleStore" { code { const nick := unsafeSub(1, 2) } } `).results),
+    `object \"SimpleStore\" { code { let nick := sub(1, 2) } }`, 'unsafe sub');
+
   t.equal(print(compile(` object "SimpleStore" { code { require(0, error"nick") require(1, error"john") } } `).results),
     `object \"SimpleStore\" { code {\nfunction require(arg, message) {\n  if lt(arg, 1) {\n    mstore(0, message)\n    revert(0, 32)\n  }\n}\n require(0, 0x01) require(1, 0x02) } }`, 'const');
 

src/yulplus.ne

@@ -768,6 +768,27 @@ Block -> "{" _ Statement (_ Statement):* _ "}" {% function(d, l) {
       v.toString = () => 'add';
     }
 
+    if (v.type === 'FunctionCallIdentifier'
+      && v.name === 'unsafeMul') {
+      v.text = 'mul';
+      v.value = 'mul';
+      v.toString = () => 'mul';
+    }
+
+    if (v.type === 'FunctionCallIdentifier'
+      && v.name === 'unsafeDiv') {
+      v.text = 'div';
+      v.value = 'div';
+      v.toString = () => 'div';
+    }
+
+    if (v.type === 'FunctionCallIdentifier'
+      && v.name === 'unsafeSub') {
+      v.text = 'sub';
+      v.value = 'sub';
+      v.toString = () => 'sub';
+    }
+
     if (v.type === 'FunctionCallIdentifier'
       && v.name === 'add'
       && !v.noSafeMath) {