com.mageddo.rawstringliterals.MultilineTreePathScanner Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rsl-compiler Show documentation
String multiline support for Java without all concatenation and stuff
The newest version!
package com.mageddo.rawstringliterals;

import com.mageddo.rawstringliterals.javac.Method;
import com.sun.source.tree.ClassTree;
import com.sun.source.tree.CompilationUnitTree;
import com.sun.source.util.TreePath;
import com.sun.source.util.TreePathScanner;
import com.sun.source.util.Trees;
import com.sun.tools.javac.code.Symbol.ClassSymbol;
import com.sun.tools.javac.tree.JCTree;
import com.sun.tools.javac.tree.JCTree.*;
import com.sun.tools.javac.tree.TreeMaker;
import com.sun.tools.javac.tree.TreeTranslator;

import javax.tools.JavaFileObject;
import java.util.Collections;
import java.util.List;

public class MultilineTreePathScanner extends TreePathScanner {

	private final TreeMaker maker;
	private final Trees trees;
	private final ClassSymbol classSymbol;
	private final String annotationName;
	private ClassScanner classScanner;

	public MultilineTreePathScanner(
		TreeMaker maker, Trees trees, ClassSymbol classSymbol, ClassScanner classScanner
	) {
		this.maker = maker;
		this.trees = trees;
		this.classSymbol = classSymbol;
		this.classScanner = classScanner;
		this.annotationName = References.MULTILINE_ANNOTATION.getSimpleName();
	}

	@Override
	public Trees visitClass(final ClassTree classTree, final CompilationUnitTree unitTree) {

		if (unitTree instanceof JCCompilationUnit) {

			final JCCompilationUnit compilationUnit = (JCCompilationUnit) unitTree;
			if (compilationUnit.sourcefile.getKind() != JavaFileObject.Kind.SOURCE) {
				return trees;
			}
			compilationUnit.accept(new TreeTranslator() {
				@Override
				public void visitMethodDef(JCTree.JCMethodDecl jcMethodDecl) {
					super.visitMethodDef(jcMethodDecl);
					processStatement(jcMethodDecl, jcMethodDecl.getBody());
				}
			});
		}
		return trees;
	}

	private void processStatement(JCMethodDecl jcMethodDecl, JCStatement statement) {
		if(statement instanceof JCVariableDecl){
			setupVar(jcMethodDecl, (JCVariableDecl) statement);
		} else {
			for (final JCStatement jcStatement : getStatements(statement)) {
				processStatement(jcMethodDecl, jcStatement);
			}
		}
	}

	private List getStatements(JCStatement statement) {
		if(statement instanceof JCTry){
			return ((JCTry) statement).getBlock().getStatements();
		}
		if(statement instanceof JCBlock){
			return ((JCBlock) statement).getStatements();
		}
		return Collections.emptyList();
	}

	private void setupVar(JCMethodDecl jcMethodDecl, JCVariableDecl variableDecl) {
		for (final JCAnnotation annotation : variableDecl.mods.annotations) {
			if(annotation.getAnnotationType() instanceof JCIdent) {
				final JCIdent annotationType = (JCIdent) annotation.getAnnotationType();
				if(annotationType.getName().toString().equals(annotationName)){
					final String varValue = classScanner.findMultilineVar(
						classSymbol, new Method(jcMethodDecl), variableDecl.getName().toString(), annotationName
					);
					try {
						variableDecl.init = maker.Literal(varValue);
					} catch (Throwable e){
						throw new IllegalStateException(String.format(
							"can't define variable %s with value %s on method %s.%s",
							variableDecl.getName(), varValue, classSymbol.getSimpleName(), jcMethodDecl.getName()
						), e);
					}
				}
			}
		}
	}

	public void scan() {
		final TreePath path = trees.getPath(classSymbol);
		this.scan(path, path.getCompilationUnit());
	}
}