SingleTypeProjectContributor.java

package com.example.custominitializr.generator.contributor;

import java.io.IOException;
import java.io.PrintWriter;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.Collections;

import io.spring.initializr.generator.io.template.MustacheTemplateRenderer;
import io.spring.initializr.generator.project.ResolvedProjectDescription;
import io.spring.initializr.generator.project.contributor.ProjectContributor;

public class SingleTypeProjectContributor implements ProjectContributor {

	private final MustacheTemplateRenderer templateRenderer;

	private final ResolvedProjectDescription projectDescription;

	private final String templateName;

	public SingleTypeProjectContributor(MustacheTemplateRenderer templateRenderer,
			ResolvedProjectDescription projectDescription, String templateName) {
		this.templateRenderer = templateRenderer;
		this.projectDescription = projectDescription;
		this.templateName = templateName;
	}

	@Override
	public void contribute(Path projectRoot) throws IOException {
		Path srcDir = this.projectDescription.getBuildSystem().getMainDirectory(projectRoot,
				this.projectDescription.getLanguage());
		Path typeDir = srcDir.resolve(this.projectDescription.getPackageName().replace('.', '/'));
		if (!Files.exists(typeDir)) {
			Files.createDirectories(typeDir);
		}
		Path file = Files.createFile(typeDir.resolve(this.templateName));
		try (PrintWriter writer = new PrintWriter(Files.newBufferedWriter(file))) {
			writer.println(this.templateRenderer.render(this.templateName,
					Collections.singletonMap("packageName", this.projectDescription.getPackageName())));
		}
	}

}

Test.java.mustache

package {{packageName}};

public class Test {

    public void doIt() throws Exception {
        //
    }

}