com.reprezen.genflow.swagger.nswag.NSwagRapidClientGenTemplate Maven / Gradle / Ivy

Go to download
package com.reprezen.genflow.swagger.nswag;

import java.io.File;
import java.io.IOException;

import com.reprezen.genflow.api.GenerationException;
import com.reprezen.genflow.api.swagger.SwaggerSource;
import com.reprezen.genflow.api.template.GenTemplate;
import com.reprezen.genflow.api.template.GenTemplateContext;
import com.reprezen.genflow.api.template.GenTemplateProperty;
import com.reprezen.genflow.api.trace.GenTemplateTrace;
import com.reprezen.genflow.api.trace.GenTemplateTraceItem;
import com.reprezen.genflow.api.zenmodel.ZenModelGenTemplate;
import com.reprezen.genflow.rapidml.swagger.SwaggerContractGenTemplate;
import com.reprezen.genflow.swagger.nswag.Config.Output;
import com.reprezen.rapidml.ZenModel;

import io.swagger.models.Swagger;

public class NSwagRapidClientGenTemplate extends ZenModelGenTemplate {

	@Override
	public boolean isSuppressed() {
		return !NSwagGenerator.isPlatformSupported();
	}

	private static final String SWAGGER_PREREQ_NAME = "swagger";

	@Override
	public String getName() {
		return "C# Client by NSwag";
	}

	@Override
	protected void configure() throws GenerationException {
		alsoKnownAs("com.modelsolv.reprezen.gentemplates.nswag.NSwagRapidClientGenTemplate");
		defineZenModelSource();
		define(prerequisite() //
				.named(SWAGGER_PREREQ_NAME) //
				.withDescription("GenTarget to produce Swagger from RAPID model") //
				.on(SwaggerContractGenTemplate.class) //
				.required());
		Config.defineGenTemplateParameters(this, Config.Output.CS_CLIENT);
		define(GenTemplateProperty.nswagProvider());
	}

	@Override
	protected StaticGenerator getStaticGenerator() {
		return new Generator(this, context);
	}

	private static class Generator extends GenTemplate.StaticGenerator {

		public Generator(GenTemplate genTemplate, GenTemplateContext context) {
			super(genTemplate, context);
		}

		@Override
		public void generate(ZenModel model) throws GenerationException {
			Config config = Config.fromContext(context);
			File outputFile = new File(context.getOutputDirectory(), model.getName() + " Client.cs");
			Swagger swagger = getSwaggerFromPrereq();
			try {
				new NSwagGenerator(config).generate(config.getNSwagRunFile(swagger, Output.CS_CLIENT, outputFile));
			} catch (IOException e) {
				throw new GenerationException("Failed to run NSwag Generation", e);
			}
		}

		private Swagger getSwaggerFromPrereq() throws GenerationException {
			GenTemplateTrace trace = context.getPrerequisiteTrace(SWAGGER_PREREQ_NAME);
			for (GenTemplateTraceItem item : trace.getTraceItems()) {
				if (item.getType() == GenTemplateTrace.FILE_ITEM_TYPE) {
					return new SwaggerSource(item.getOutputFile()).load();
				}
			}
			throw new GenerationException("Failed to load Swagger model from prerequisite GenTarget");
		}
	}
}