com.regnosys.rosetta.generator.jsonschema.JsonSchemaCodeGenerator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of json-schema Show documentation

There is a newer version: 11.27.1

package com.regnosys.rosetta.generator.jsonschema;

import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.ResourceSet;

import com.google.inject.Inject;
import com.regnosys.rosetta.generator.external.AbstractExternalGenerator;
import com.regnosys.rosetta.rosetta.RosettaEnumeration;
import com.regnosys.rosetta.rosetta.RosettaMetaType;
import com.regnosys.rosetta.rosetta.RosettaModel;
import com.regnosys.rosetta.rosetta.simple.Data;

public class JsonSchemaCodeGenerator extends AbstractExternalGenerator {

	@Inject
	private JsonSchemaTypeGenerator typeSchemaGenerator;

	@Inject 
	private JsonSchemaMetaFieldGenerator metaFieldGenerator;

	@Inject
	private JsonSchemaEnumGenerator enumGenerator;
	
	public JsonSchemaCodeGenerator() {
		super("JsonSchema");
	}

	@Override
	public Map generate(Resource resource, RosettaModel model, String version) {
		return Collections.emptyMap();
	}

	@Override
	public Map afterAllGenerate(ResourceSet set,
			Collection models, String version) {

		List rosettaData = models.stream().flatMap(m -> m.getElements().stream()).filter((e) -> e instanceof Data)
				.map(Data.class::cast).collect(Collectors.toList());

		List metaTypes = models.stream().flatMap(m -> m.getElements().stream())
				.filter(RosettaMetaType.class::isInstance).map(RosettaMetaType.class::cast)
				.collect(Collectors.toList());

		List rosettaEnums = models.stream().flatMap(m -> m.getElements().stream())
				.filter(RosettaEnumeration.class::isInstance).map(RosettaEnumeration.class::cast)
				.collect(Collectors.toList());

		Map result = new HashMap<>();

		result.putAll(typeSchemaGenerator.generateTypeDefinitions(rosettaData));
		result.putAll(metaFieldGenerator.generateMetaFields(rosettaData, metaTypes));
		result.putAll(enumGenerator.generateEnumDefinitions(rosettaEnums));

		return result;
	}

}