org.stjs.generator.writer.CommentWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of generator Show documentation

There is a newer version: 3.3.2.2

package org.stjs.generator.writer;

import javax.lang.model.element.Element;

import org.stjs.generator.GenerationContext;
import org.stjs.generator.javac.InternalUtils;
import org.stjs.generator.visitor.VisitorContributor;

import com.sun.source.tree.Tree;

/**
 * add the javadoc comment for the processed nodes.
 * 
 * @author acraciun
 * 
 * @param 
 */
public class CommentWriter implements WriterFilter {

	@Override
	public JS visit(WriterVisitor visitor, Tree tree, GenerationContext context,
			VisitorContributor, WriterVisitor> chain) {
		JS jsNode = chain.visit(visitor, tree, context);
		if (jsNode == null) {
			return null;
		}

		Element element = InternalUtils.symbol(tree);
		if (element != null) {
			String comment = context.getElements().getDocComment(element);
			if (comment != null) {
				return context.js().comment(jsNode, comment);
			}
		}
		return jsNode;
	}

}