refactor isFile to canPrint in type node

Angry-Potato · Jul 19, 2019 · 18b9b52 · 18b9b52
1 parent 6f548ce
commit 18b9b52
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/src/nodes.js b/src/nodes.js
@@ -38,8 +38,6 @@ const nodes = {
   [NODE_TYPES.FUNCTION_CALL]: require(`./nodes/${NODE_TYPES.FUNCTION_CALL}`)
 };
 
-const isInteger = node => node.form == "INT";
-
 const isOperator = node => node.token;
 
 const isFunctionCall = node => node.expr && node.typeArgs && node.args;
@@ -77,7 +75,7 @@ const assignType = node => {
     return { ...node, astType: NODE_TYPES.STRING };
   } else if (nodes[NODE_TYPES.TYPE].canPrint(node)) {
     return { ...node, astType: NODE_TYPES.TYPE };
-  } else if (isInteger(node)) {
+  } else if (nodes[NODE_TYPES.INTEGER].canPrint(node)) {
     return { ...node, astType: NODE_TYPES.INTEGER };
   } else if (isOperator(node)) {
     return { ...node, astType: NODE_TYPES.OPERATOR };

diff --git a/src/nodes/integer.js b/src/nodes/integer.js
@@ -4,7 +4,10 @@ const {
   }
 } = require("prettier");
 
-module.exports = (path, opts, print) => {
-  const node = path.getValue();
-  return node.value;
+module.exports = {
+  canPrint: node => node.form == "INT",
+  print: (path, opts, print) => {
+    const node = path.getValue();
+    return node.value;
+  }
 };