Method visitNextNameAndArgs

src/FormatVisitor.cpp:1300–1355

visit next NameAndArgsContext of VarSuffixContext

// visit next NameAndArgsContext of VarSuffixContext
void FormatVisitor::visitNextNameAndArgs(LuaParser::VarSuffixContext* ctx) {
    // find args of NAME
    auto* varCtx = dynamic_cast<LuaParser::VarContext*>(ctx->parent);
    const std::vector<LuaParser::VarSuffixContext*>& arr = varCtx->varSuffix();
    unsigned int index = find(arr.begin(), arr.end(), ctx) - arr.begin();
    if (index + 1 < arr.size()) {
        for (auto* na : arr[index + 1]->nameAndArgs()) {
            // if next NameAndArgs's COLON is not null
            // then formatter can place a break before COLON.
            // So do not need to calc column length for current NAME, break the loop.
            // example:
            // self.xxx.yyy:foo()
            // the 'xxx' and 'yyy' is a 'Variable', and 'foo' is a 'Method'
            // It better break before 'foo' than break before 'yyy'
            if (na->COLON() != nullptr) {
                break;
            }
            visitNameAndArgs(na);
            if (na != arr[index + 1]->nameAndArgs().back()) {
                cur_writer() << commentAfter(na, "");
            }
        }
    } else {
        auto* varOrExpCtx = dynamic_cast<LuaParser::VarOrExpContext*>(varCtx->parent);
        // parent maybe VarlistContext
        if (varOrExpCtx != nullptr) {
            tree::ParseTree* parent = varOrExpCtx->parent;

            auto* peCtx = dynamic_cast<LuaParser::PrefixexpContext*>(parent);
            if (peCtx != nullptr) {
                for (auto* na : peCtx->nameAndArgs()) {
                    if (na->COLON() != nullptr) {
                        break;
                    }
                    visitNameAndArgs(na);
                    if (na != peCtx->nameAndArgs().back()) {
                        cur_writer() << commentAfter(na, "");
                    }
                }
            }

            auto* fcCtx = dynamic_cast<LuaParser::FunctioncallContext*>(parent);
            if (fcCtx != nullptr) {
                for (auto* na : fcCtx->nameAndArgs()) {
                    if (na->COLON() != nullptr) {
                        break;
                    }
                    visitNameAndArgs(na);
                    if (na != fcCtx->nameAndArgs().back()) {
                        cur_writer() << commentAfter(na, "");
                    }
                }
            }
        }
    }
}

// (COLON NAME)? args;