D-Net

package eu.dnetlib.dhp.common.java.jsonworkflownodes;

import java.util.HashMap;

import java.util.Map;

import org.apache.avro.Schema;

import eu.dnetlib.dhp.common.java.jsonworkflownodes.StringPortSpecificationExtractor.PortSpecification;

import eu.dnetlib.dhp.common.java.porttype.AvroPortType;

import eu.dnetlib.dhp.common.java.porttype.PortType;

import eu.dnetlib.dhp.common.utils.AvroUtils;

/**

 * @author Mateusz Kobos

 */

public class PortSpecifications {

	private static final String[] propertyRegexps =

			new String[]{"[\\w\\.]+", "[\\w\\./_\\-]+"};

	private final Map<String, SpecificationValues> specs;

    public static class SpecificationValues {

        private final Schema schema;

        private final String jsonFilePath;

        public SpecificationValues(Schema schema, String jsonFilePath) {

            this.schema = schema;

            this.jsonFilePath = jsonFilePath;

        }

        public Schema getSchema() {

            return schema;

        }

        public String getJsonFilePath() {

            return jsonFilePath;

        }

    }

	public PortSpecifications(String[] portSpecifications){

		StringPortSpecificationExtractor portSpecExtractor =

				new StringPortSpecificationExtractor(propertyRegexps);

		specs = new HashMap<String, SpecificationValues>();

		for(int i = 0; i < portSpecifications.length; i++){

			PortSpecification portSpec = portSpecExtractor.getSpecification(portSpecifications[i]);

			Schema schema = AvroUtils.toSchema(portSpec.getProperties()[0]);

			String jsonPath = portSpec.getProperties()[1];

			specs.put(portSpec.getName(), new SpecificationValues(schema, jsonPath));

		}

	}

	public SpecificationValues get(String portName){

		return specs.get(portName);

	}

	public Map<String, PortType> getPortTypes(){

		Map<String, PortType> ports = new HashMap<String, PortType>();

		for(Map.Entry<String, SpecificationValues> e: specs.entrySet()){

			Schema schema = e.getValue().schema;

			ports.put(e.getKey(), new AvroPortType(schema));

		}

		return ports;

	}

}