com.prezi.spaghetti.javascript.JavaScriptEnumGeneratorVisitor Maven / Gradle / Ivy

Go to download
package com.prezi.spaghetti.javascript;

import com.google.common.base.Joiner;
import com.google.common.base.Strings;
import com.google.common.collect.Lists;
import com.prezi.spaghetti.ast.EnumNode;
import com.prezi.spaghetti.ast.EnumValueNode;

import java.util.List;

public class JavaScriptEnumGeneratorVisitor extends AbstractJavaScriptGeneratorVisitor {

	@Override
	public String visitEnumNode(EnumNode node) {
		return embedInPackageStructure(node, new PackagedGenerator() {
			@Override
			public String generate(EnumNode node, int indentLevel, String parent) {
				final String indent = Strings.repeat("\t", indentLevel);
				List values = Lists.newArrayList();
				int idx = 0;
				for (EnumValueNode value : node.getValues()) {
					values.add(indent + "\t\"" + value.getName() + "\": " + idx);
					idx++;
				}
				String entries = Joiner.on(",\n").join(values);
				return indent + parent + "." + node.getName() + " = {\n"
						+ entries + "\n"
						+ indent + "};\n";
			}
		});
	}
}