AntlrParser.java

/*-
 * #%L
 * io.earcam.instrumental.module.osgi
 * %%
 * Copyright (C) 2018 earcam
 * %%
 * SPDX-License-Identifier: (BSD-3-Clause OR EPL-1.0 OR Apache-2.0 OR MIT)
 * 
 * You <b>must</b> choose to accept, in full - any individual or combination of 
 * the following licenses:
 * <ul>
 * 	<li><a href="https://opensource.org/licenses/BSD-3-Clause">BSD-3-Clause</a></li>
 * 	<li><a href="https://www.eclipse.org/legal/epl-v10.html">EPL-1.0</a></li>
 * 	<li><a href="https://www.apache.org/licenses/LICENSE-2.0">Apache-2.0</a></li>
 * 	<li><a href="https://opensource.org/licenses/MIT">MIT</a></li>
 * </ul>
 * #L%
 */
package io.earcam.instrumental.module.jpms.parser;

import static java.util.Locale.ROOT;
import static java.util.stream.Collectors.toSet;

import java.util.Set;
import java.util.TreeSet;
import java.util.stream.Stream;

import io.earcam.instrumental.module.jpms.ModuleInfo;
import io.earcam.instrumental.module.jpms.ModuleInfoBuilder;
import io.earcam.instrumental.module.jpms.RequireModifier;
import io.earcam.instrumental.module.jpms.parser.Java9BaseListener;
import io.earcam.instrumental.module.jpms.parser.Java9Parser.ExportsDirectiveContext;
import io.earcam.instrumental.module.jpms.parser.Java9Parser.ModuleDeclarationContext;
import io.earcam.instrumental.module.jpms.parser.Java9Parser.ModuleNameContext;
import io.earcam.instrumental.module.jpms.parser.Java9Parser.OpensDirectiveContext;
import io.earcam.instrumental.module.jpms.parser.Java9Parser.ProvidesDirectiveContext;
import io.earcam.instrumental.module.jpms.parser.Java9Parser.RequiresDirectiveContext;
import io.earcam.instrumental.module.jpms.parser.Java9Parser.RequiresModifierContext;
import io.earcam.instrumental.module.jpms.parser.Java9Parser.UsesDirectiveContext;

// TODO annotations, and possibly comments?
/**
 * <p>
 * ModuleInfoParser class.
 * </p>
 *
 */
class AntlrParser extends Java9BaseListener {

	private final ModuleInfoBuilder builder = ModuleInfo.moduleInfo();


	/**
	 * <p>
	 * builder.
	 * </p>
	 *
	 * @return a {@link io.earcam.instrumental.module.jpms.ModuleInfoBuilder} object.
	 */
	public ModuleInfoBuilder builder()
	{
		return builder;
	}


	@Override
	public void enterModuleDeclaration(ModuleDeclarationContext ctx)
	{
		super.enterModuleDeclaration(ctx);
		builder.named(ctx.moduleName().getText());
	}


	@Override
	public void enterRequiresDirective(RequiresDirectiveContext ctx)
	{
		super.enterRequiresDirective(ctx);
		Set<RequireModifier> modifiers = ctx.requiresModifier().stream()
				.map(RequiresModifierContext::getText)
				.map(m -> m.toUpperCase(ROOT))
				.map(RequireModifier::valueOf)
				.collect(toSet());

		builder.requiring(ctx.moduleName().getText(), modifiers, null);
	}


	@Override
	public void enterExportsDirective(ExportsDirectiveContext ctx)
	{
		super.enterExportsDirective(ctx);
		String[] to = toModuleNames(ctx.moduleName().stream());
		builder.exporting(ctx.packageName().getText(), to);
	}


	private String[] toModuleNames(Stream<ModuleNameContext> stream)
	{
		return stream
				.map(ModuleNameContext::getText)
				.toArray(s -> new String[s]);
	}


	@Override
	public void enterOpensDirective(OpensDirectiveContext ctx)
	{
		super.enterOpensDirective(ctx);
		String[] to = toModuleNames(ctx.moduleName().stream());
		builder.opening(ctx.packageName().getText(), to);
	}


	@Override
	public void enterUsesDirective(UsesDirectiveContext ctx)
	{
		super.enterUsesDirective(ctx);
		builder.using(ctx.getChild(1).getText());
	}


	@Override
	public void enterProvidesDirective(ProvidesDirectiveContext ctx)
	{
		super.enterProvidesDirective(ctx);

		TreeSet<String> concretes = new TreeSet<>();
		for(int i = 3; i < ctx.getChildCount(); i += 2) {
			concretes.add(ctx.getChild(i).getText());
		}

		builder.providing(ctx.getChild(1).getText(), concretes);
	}
}