Add support for user-defined operators

prettier-solidity · Mar 1, 2023 · 838e1d9 · 838e1d9
1 parent 7875618
commit 838e1d9
Show file tree

Hide file tree

Showing 3 changed files with 51 additions and 15 deletions.
diff --git a/src/nodes/UsingForDeclaration.js b/src/nodes/UsingForDeclaration.js
@@ -7,21 +7,39 @@ const {
 const { printSeparatedList } = require('../common/printer-helpers');
 
 const UsingForDeclaration = {
-  print: ({ node, path, print, options }) => [
-    'using ',
-    node.functions && node.functions.length
-      ? [
-          '{',
-          printSeparatedList(node.functions, {
-            firstSeparator: options.bracketSpacing ? line : softline
-          }),
-          '}'
-        ]
-      : node.libraryName,
-    ' for ',
-    node.typeName ? path.call(print, 'typeName') : '*',
-    node.isGlobal ? ' global;' : ';'
-  ]
+  print: ({ node, path, print, options }) => {
+    const parts = ['using '];
+
+    if (node.functions && node.functions.length) {
+      const importedFunctions = [];
+      for (let i = 0; i < node.functions.length; i += 1) {
+        const fun = node.functions[i];
+        const operator = node.operators[i];
+
+        if (operator) {
+          importedFunctions.push(`${fun} as ${operator}`);
+        } else {
+          importedFunctions.push(fun);
+        }
+      }
+
+      parts.push('{');
+      parts.push(
+        printSeparatedList(importedFunctions, {
+          firstSeparator: options.bracketSpacing ? line : softline
+        })
+      );
+      parts.push('}');
+    } else {
+      parts.push(node.libraryName);
+    }
+
+    parts.push(' for ');
+    parts.push(node.typeName ? path.call(print, 'typeName') : '*');
+    parts.push(node.isGlobal ? ' global;' : ';');
+
+    return parts;
+  }
 };
 
 module.exports = UsingForDeclaration;
diff --git a/tests/format/AllSolidityFeatures/AllSolidityFeatures.sol b/tests/format/AllSolidityFeatures/AllSolidityFeatures.sol
@@ -534,3 +534,9 @@ contract WithUncheckedBlock {
     unchecked { return x * y; }
   }
 }
+
+// user-defined operators
+using {  add as +   }   for   Fixed18   global  ;
+using {  add   as +  , sub as - } for   Fixed18 global  ;
+using {  add   , sub as -   } for Fixed18   global  ;
+using {    add as   +  , sub   } for   Fixed18 global  ;
diff --git a/tests/format/AllSolidityFeatures/__snapshots__/jsfmt.spec.js.snap b/tests/format/AllSolidityFeatures/__snapshots__/jsfmt.spec.js.snap
@@ -543,6 +543,12 @@ contract WithUncheckedBlock {
   }
 }
 
+// user-defined operators
+using {  add as +   }   for   Fixed18   global  ;
+using {  add   as +  , sub as - } for   Fixed18 global  ;
+using {  add   , sub as -   } for Fixed18   global  ;
+using {    add as   +  , sub   } for   Fixed18 global  ;
+
 =====================================output=====================================
 // Examples taken from the Solidity documentation online.
 
@@ -1136,5 +1142,11 @@ contract WithUncheckedBlock {
     }
 }
 
+// user-defined operators
+using {add as +} for Fixed18 global;
+using {add as +, sub as -} for Fixed18 global;
+using {add, sub as -} for Fixed18 global;
+using {add as +, sub} for Fixed18 global;
+
 ================================================================================
 `;