Method commentAfter

src/FormatVisitor.cpp:93–126

commentAfter returns comments between a ParseTreeNode and the next Node. expect: return expect string if no comments found

// commentAfter returns comments between a ParseTreeNode and the next Node.
// expect: return expect string if no comments found
std::string FormatVisitor::commentAfter(tree::ParseTree* node, const std::string& expect) {
    int l = lastNodeIndex(node);
    std::stringstream ss;
    // No space on left of first comment
    bool lastComment = node != nullptr;
    for (unsigned int i = l + 1; i < tokens_.size(); i++) {
        auto* token = tokens_[i];
        if (token->getType() == LuaLexer::COMMENT) {
            if (lastComment) {
                ss << " ";
            }
            ss << token->getText();
            lastComment = true;
        } else if (token->getType() == LuaLexer::LINE_COMMENT) {
            if (lastComment) {
                ss << " ";
            }
            ss << formatLineComment(token);
            ss << indent();
            lastComment = false;
        } else if (token->getType() == LuaLexer::SHEBANG) {
            ss << token->getText() << "\n\n";
        } else if (token->getType() == LuaLexer::WS) {
        } else {
            break;
        }
    }
    if (ss.tellp() == 0) {
        ss << expect;
    } else if (lastComment) {
        ss << " ";
    }
    return ss.str();
}

// commentAfterNewLine returns comments between a ParseTreeNode and the next Node.
// This function always expect a line break.