使用Antlr4的TokenStreamRewriter修改内容

498 阅读1分钟

需求:在css中使用变量控制部分属性的值

  • 项目中配置antlr版本
<dependency>
    <groupId>org.antlr</groupId>
    <artifactId>antlr4-runtime</artifactId>
    <version>4.9.3</version>
</dependency>
  • 使用官方css.g4 demo生成api
$ antlr4 -no-listener -visitor css.g4
  • 使用TokenStreamRewriter修改
package css;

import org.antlr.v4.runtime.CharStreams;
import org.antlr.v4.runtime.CommonTokenStream;
import org.antlr.v4.runtime.TokenStreamRewriter;
import org.antlr.v4.runtime.tree.ParseTree;

public class Css {
    public static void main(String[] args) {
        CssLexer lexer = new CssLexer(CharStreams.fromString(".b { color: red; } .c { color: blue; }"));
        CommonTokenStream stream = new CommonTokenStream(lexer);
        CssParser parser = new CssParser(stream);
        ParseTree tree = parser.stylesheet();
        EvalVisitor evalVisitor = new EvalVisitor(stream);
        evalVisitor.visit(tree);
        System.out.println(evalVisitor.rewriter.getText());
    }
}

class EvalVisitor extends CssBaseVisitor<String> {
    TokenStreamRewriter rewriter;
    public EvalVisitor(CommonTokenStream tokenStream) {
        rewriter = new TokenStreamRewriter(tokenStream);
    }

    @Override
    public String visitExpr(CssParser.ExprContext ctx) {
        String text = ctx.getText();
        if (text.equals("red")) {
            rewriter.replace(ctx.getStart(), ctx.getStop(), "#ff0000");
        }
        return super.visitExpr(ctx);
    }
}

完整代码:github.com/schizobulia…

目前我司在招高级前端、ios、android开发(完全远程)。大佬们有兴趣的发简历
邮箱:MjgzMzMyNDUyOEBxcS5jb20=

如果你喜欢我的作品,请考虑赞助,以保持它们的可持续性。