lefty/src/join.js

A JSX transformer for migi

import homunculus from 'homunculus';

let JsNode = homunculus.getClass('Node', 'es6');
let Token = homunculus.getClass('Token');

module.exports = function(node, word) {
  let res = recursion(node, { 's': '', 'word': word });
  return res.s;
};

function recursion(node, res) {
  let isToken = node.name() === JsNode.TOKEN;
  let isVirtual = isToken && node.token().type() === Token.VIRTUAL;
  if(isToken) {
    let token = node.token();
    if(!isVirtual) {
      if(res.word && [Token.ID, Token.NUMBER, Token.KEYWORD].indexOf(token.type()) > -1) {
        res.s += ' ';
      }
      if(token.content() === '}' && res.s.charAt(res.s.length - 1) === ';') {
        res.s = res.s.replace(/;$/, '');
      }
      res.s += token.content();
      res.word = [Token.ID, Token.NUMBER, Token.KEYWORD].indexOf(token.type()) > -1;
    }
    else if(token.content() === ';') {
      res.s += ';';
      res.word = false;
    }
  }
  else {
    node.leaves().forEach(function(leaf) {
      recursion(leaf, res);
    });
  }
  return res;
}