fix: toString should stringify node (#74)

* fix: toString should stringify node

* refactor: rename, rework inherited nodes for toString

* chore: remove commented code

Author: shellscape

lib/ValuesParser.js

@@ -52,6 +52,7 @@ module.exports = class ValuesParser extends Parser {
     const inline = Comment.testInline(token);
     const node = this.lastNode;
     node.inline = inline;
+    Object.setPrototypeOf(node, Comment.prototype);
   }
 
   fromFirst(tokens, Constructor) {

lib/ValuesStringifier.js

@@ -1,6 +1,11 @@
 const Stringifier = require('postcss/lib/stringifier');
 
-module.exports = class LessStringifier extends Stringifier {
+module.exports = class ValuesStringifier extends Stringifier {
+  static stringify(node, builder) {
+    const stringifier = new ValuesStringifier(builder);
+    stringifier.stringify(node);
+  }
+
   basic(node, value) {
     const print = value || node.value;
     const after = node.raws.after ? this.raw(node, 'after') || '' : '';

lib/index.js

@@ -11,7 +11,7 @@
 const Input = require('postcss/lib/input');
 
 const Parser = require('./ValuesParser');
-const Stringifier = require('./ValuesStringifier');
+const { stringify } = require('./ValuesStringifier');
 
 module.exports = {
   parse(css, options) {
@@ -20,13 +20,19 @@ module.exports = {
 
     parser.parse();
 
+    const { root } = parser;
+    const ogToString = root.toString;
+
+    function toString(stringifier) {
+      return ogToString.bind(root)(stringifier || module.exports.stringify);
+    }
+
+    root.toString = toString.bind(root);
+
     return parser.root;
   },
 
-  stringify(node, builder) {
-    const stringifier = new Stringifier(builder);
-    stringifier.stringify(node);
-  },
+  stringify,
 
   nodeToString(node) {
     let result = '';

lib/nodes/AtWord.js

@@ -12,10 +12,11 @@ const AtRule = require('postcss/lib/node');
 
 const { registerWalker } = require('../walker');
 
+const { stringify } = require('../ValuesStringifier');
+
 class AtWord extends AtRule {
-  constructor(options) {
-    super(options);
-    this.type = 'atword';
+  toString(stringifier = stringify) {
+    return super.toString(stringifier);
   }
 }
 

lib/nodes/Comment.js

@@ -8,61 +8,71 @@
   The above copyright notice and this permission notice shall be
   included in all copies or substantial portions of this Source Code Form.
 */
-const Comment = require('postcss/lib/comment');
+const PostCssComment = require('postcss/lib/comment');
+
+const { stringify } = require('../ValuesStringifier');
 
 const inlineRegex = /(\/\/)/;
 
-Comment.testInline = (token) => inlineRegex.test(token[1]);
+class Comment extends PostCssComment {
+  static testInline(token) {
+    return inlineRegex.test(token[1]);
+  }
 
-Comment.tokenizeNext = (tokens, parser) => {
-  const [first] = tokens;
-  const newlineIndex = tokens.findIndex((t) => /\n/.test(t[1]));
-  let bits = tokens;
-  let rest = [];
+  static tokenizeNext(tokens, parser) {
+    const [first] = tokens;
+    const newlineIndex = tokens.findIndex((t) => /\n/.test(t[1]));
+    let bits = tokens;
+    let rest = [];
 
-  if (newlineIndex >= 0) {
-    bits = tokens.slice(0, newlineIndex);
-    rest = tokens.slice(newlineIndex);
-  }
+    if (newlineIndex >= 0) {
+      bits = tokens.slice(0, newlineIndex);
+      rest = tokens.slice(newlineIndex);
+    }
+
+    bits = bits.map((t) => t[1]);
 
-  bits = bits.map((t) => t[1]);
+    // see tilde comment in tokenizeInline
+    const text = bits.concat('~~').join('');
+    const last = bits[bits.length - 1];
+    const newToken = ['comment', text, first[2], first[3], last[2], last[3]];
 
-  // see tilde comment in tokenizeInline
-  const text = bits.concat('~~').join('');
-  const last = bits[bits.length - 1];
-  const newToken = ['comment', text, first[2], first[3], last[2], last[3]];
+    parser.back([newToken, ...rest]);
+  }
 
-  parser.back([newToken, ...rest]);
-};
+  static tokenizeInline(tokens, parser) {
+    const [first, ...rest] = tokens;
+    const bits = first[1].split(/(\/\/.+)/).filter((t) => !!t);
+    const newTokens = [];
+    const [, , startLine, , endLine] = first;
+    let [, , , startChar, , endChar] = first;
 
-Comment.tokenizeInline = (tokens, parser) => {
-  const [first, ...rest] = tokens;
-  const bits = first[1].split(/(\/\/.+)/).filter((t) => !!t);
-  const newTokens = [];
-  const [, , startLine, , endLine] = first;
-  let [, , , startChar, , endChar] = first;
+    for (let bit of bits) {
+      const comment = bit.slice(0, 2) === '//';
+      const type = comment ? 'comment' : 'word';
 
-  for (let bit of bits) {
-    const comment = bit.slice(0, 2) === '//';
-    const type = comment ? 'comment' : 'word';
+      if (comment) {
+        // the Parser base comment() method trims the last two characters when creating the node
+        // these tildes are added to counter that. it's hacky, but it works, and we don't have to
+        // re-implement the method
+        bit += '~~';
+      }
 
-    if (comment) {
-      // the Parser base comment() method trims the last two characters when creating the node
-      // these tildes are added to counter that. it's hacky, but it works, and we don't have to
-      // re-implement the method
-      bit += '~~';
-    }
+      if (bit !== bits[0]) {
+        startChar = endChar + 1;
+      }
 
-    if (bit !== bits[0]) {
-      startChar = endChar + 1;
-    }
+      endChar = startChar + bit.length - 1;
 
-    endChar = startChar + bit.length - 1;
+      newTokens.push([type, bit, startLine, startChar, endLine, endChar]);
+    }
 
-    newTokens.push([type, bit, startLine, startChar, endLine, endChar]);
+    parser.back(newTokens.concat(rest));
   }
 
-  parser.back(newTokens.concat(rest));
-};
+  toString(stringifier = stringify) {
+    return super.toString(stringifier);
+  }
+}
 
 module.exports = Comment;

lib/nodes/Container.js

@@ -0,0 +1,21 @@
+/*
+  Copyright © 2018 Andrew Powell
+
+  This Source Code Form is subject to the terms of the Mozilla Public
+  License, v. 2.0. If a copy of the MPL was not distributed with this
+  file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+  The above copyright notice and this permission notice shall be
+  included in all copies or substantial portions of this Source Code Form.
+*/
+const PostCssContainer = require('postcss/lib/container');
+
+const { stringify } = require('../ValuesStringifier');
+
+class Container extends PostCssContainer {
+  toString(stringifier = stringify) {
+    return super.toString(stringifier);
+  }
+}
+
+module.exports = Container;

lib/nodes/Func.js

@@ -8,10 +8,10 @@
   The above copyright notice and this permission notice shall be
   included in all copies or substantial portions of this Source Code Form.
 */
-const Container = require('postcss/lib/container');
-
 const { registerWalker } = require('../walker');
 
+const Container = require('./Container');
+
 const colorFunctions = ['hsl', 'hsla', 'rgb', 'rgba'];
 
 class Func extends Container {

lib/nodes/Interpolation.js

@@ -8,10 +8,10 @@
   The above copyright notice and this permission notice shall be
   included in all copies or substantial portions of this Source Code Form.
 */
-const Container = require('postcss/lib/container');
-
 const { registerWalker } = require('../walker');
 
+const Container = require('./Container');
+
 class Interpolation extends Container {
   constructor(options = {}) {
     super(options);

lib/nodes/Node.js

@@ -0,0 +1,21 @@
+/*
+  Copyright © 2018 Andrew Powell
+
+  This Source Code Form is subject to the terms of the Mozilla Public
+  License, v. 2.0. If a copy of the MPL was not distributed with this
+  file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+  The above copyright notice and this permission notice shall be
+  included in all copies or substantial portions of this Source Code Form.
+*/
+const PostCssNode = require('postcss/lib/node');
+
+const { stringify } = require('../ValuesStringifier');
+
+class Node extends PostCssNode {
+  toString(stringifier = stringify) {
+    return super.toString(stringifier || {});
+  }
+}
+
+module.exports = Node;

lib/nodes/Numeric.js

@@ -9,10 +9,11 @@
   included in all copies or substantial portions of this Source Code Form.
 */
 const isNumber = require('is-number');
-const Node = require('postcss/lib/node');
 
 const { registerWalker } = require('../walker');
 
+const Node = require('./Node');
+
 const unitRegex = /%|ch|cm|em|ex|in|mm|pc|pt|px|rem|vh|vmax|vmin|vw$/i;
 
 class Numeric extends Node {