information-system-model-documentation-generator/src/main/java/org/gcube/informationsystem/utils/documentation/generator/ClassesDiscoveryGenerator.java

package org.gcube.informationsystem.utils.documentation.generator;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.ServiceLoader;

import org.gcube.informationsystem.base.reference.AccessType;
import org.gcube.informationsystem.base.reference.Element;
import org.gcube.informationsystem.types.TypeMapper;
import org.gcube.informationsystem.types.reference.Type;
import org.gcube.informationsystem.utils.discovery.ElementSpecilizationDiscovery;
import org.gcube.informationsystem.utils.discovery.RegistrationProvider;

/**
 * @author Luca Frosini (ISTI - CNR)
 */
public class ClassesDiscoveryGenerator extends Generator {
	
	protected TreeGenerator treeGenerator;
	
	public ClassesDiscoveryGenerator() {
		
	}
	
	public List<Package> getPackages(){
		
		List<Package> packages = new ArrayList<Package>();
			
		AccessType[] accessTypes = new AccessType[] {
			AccessType.PROPERTY, 
			AccessType.ENTITY,
			AccessType.RELATION
		};
		
		for(AccessType accessType : accessTypes) {
			Class<Element> clz = accessType.getTypeClass();
			packages.add(clz.getPackage());
		}
		
		return packages;
	}
	
	public List<Package> getExtensionsPackages(){
		List<Package> packages = new ArrayList<Package>();
		ServiceLoader<? extends RegistrationProvider> regsitrationProviders = ServiceLoader
				.load(RegistrationProvider.class);
		for(RegistrationProvider registrationProvider : regsitrationProviders) {
			packages.addAll(registrationProvider.getPackagesToRegister());
		}
		return packages;
	}
	
	public void generate() throws Exception {
		List<Package> packages = getPackages();
		packages.addAll(getExtensionsPackages());
		generate(packages);
	}
	
	protected <E extends Element> void  discover(AccessType at, List<Package> packages) throws Exception {
		ElementSpecilizationDiscovery<E> discovery = new ElementSpecilizationDiscovery<>(at.getTypeClass(), false);
		if(Objects.nonNull(packages)) {
			for(Package p : packages) {
				discovery.addPackage(p);
			}
		}
		discovery.discover();
		
		for(Class<? extends E> clz : discovery.getDiscovered()) {
			Type type = TypeMapper.createTypeDefinition(clz);
			treeGenerator.addType(type);
		}
	}
	
	public void generate(List<Package> packages) throws Exception {
		treeGenerator = new TreeGenerator();
		AccessType[] types = new AccessType[] {
				AccessType.PROPERTY, 
				AccessType.RESOURCE, AccessType.FACET, 
				AccessType.CONSISTS_OF, AccessType.IS_RELATED_TO};
		for(AccessType at : types) {
			discover(at, packages);
		}
		treeGenerator.generate();
	}
}