Adding the option to standardise the type declarations (#171)

Author: Janther

src/options.js

@@ -1,5 +1,6 @@
 const CATEGORY_GLOBAL = 'Global';
 const CATEGORY_COMMON = 'Common';
+const CATEGORY_SOLIDITY = 'Solidity';
 
 const options = {
   bracketSpacing: {
@@ -48,6 +49,27 @@ const options = {
     type: 'boolean',
     default: false,
     description: 'Indent with tabs instead of spaces.'
+  },
+  explicitTypes: {
+    category: CATEGORY_SOLIDITY,
+    type: 'choice',
+    default: 'always',
+    description: 'Change when type aliases are used.',
+    choices: [
+      {
+        value: 'always',
+        description:
+          'Prefer the explicit types `uint256`, `int256`, and `bytes1`.'
+      },
+      {
+        value: 'never',
+        description: 'Prefer the type aliases `uint`, `int`, and `byte`.'
+      },
+      {
+        value: 'preserve',
+        description: 'Respect the type used by the developer.'
+      }
+    ]
   }
 };
 

src/parser.js

@@ -2,7 +2,7 @@ const extract = require('extract-comments');
 // https://prettier.io/docs/en/plugins.html#parsers
 const parser = require('solidity-parser-antlr');
 
-function parse(text) {
+function parse(text, parsers, options) {
   const parsed = parser.parse(text, { loc: true, range: true });
   parsed.comments = extract(text);
 
@@ -14,6 +14,17 @@ function parse(text) {
       if (ctx.loopExpression) {
         ctx.loopExpression.omitSemicolon = true;
       }
+    },
+    ElementaryTypeName(ctx) {
+      if (options.explicitTypes === 'always') {
+        if (ctx.name === 'uint') ctx.name = 'uint256';
+        if (ctx.name === 'int') ctx.name = 'int256';
+        if (ctx.name === 'byte') ctx.name = 'bytes1';
+      } else if (options.explicitTypes === 'never') {
+        if (ctx.name === 'uint256') ctx.name = 'uint';
+        if (ctx.name === 'int256') ctx.name = 'int';
+        if (ctx.name === 'bytes1') ctx.name = 'byte';
+      }
     }
   });