com.regnosys.testing.GeneratedCode Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rosetta-testing Show documentation

Rune Testing is a java library that is utilised by Rosetta Code Generators and models expressed in the Rosetta DSL.

There is a newer version: 11.31.0

package com.regnosys.testing;

import org.eclipse.xtext.xbase.testing.JavaSource;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

public class GeneratedCode {

	private final Map generated;

	public GeneratedCode(Map generated) {
		this.generated = generated;
	}

	public Map getGenerated() {
		return generated;
	}

	public Set getGeneratedClassNames() {
		return generated.keySet();
	}

	public List getJavaSource() {
		return generated.entrySet().stream()
			.map(e -> new JavaSource(toJavaFile(e.getKey()), e.getValue())).collect(Collectors.toList());

	}

	public void writeClasses(String directory) throws IOException {
		for (Map.Entry entry : generated.entrySet()) {
			var className = entry.getKey();
			var path = Paths.get("target/" + directory + "/java", toJavaFile(className));
			Files.createDirectories(path.getParent());
			Files.write(path, entry.getValue().getBytes());
		}
	}

	private String toJavaFile(String className) {
		return className.replace(".", File.separator) + ".java";
	}

}