/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package org.apache.taverna.scufl2.cwl;

import java.util.*;

import org.apache.taverna.scufl2.api.core.Workflow;
import org.apache.taverna.scufl2.api.core.Processor;

import org.apache.taverna.scufl2.api.port.InputWorkflowPort;
import org.apache.taverna.scufl2.api.port.OutputWorkflowPort;
import org.apache.taverna.scufl2.api.port.InputProcessorPort;
import org.apache.taverna.scufl2.api.port.OutputProcessorPort;

import com.fasterxml.jackson.databind.JsonNode;

public class CWLParser {

    private JsonNode cwlFile;
    private YAMLHelper yamlHelper;

    public CWLParser(JsonNode cwlFile) {
        this.cwlFile = cwlFile;
        this.yamlHelper = new YAMLHelper();
    }

    public Set<Step> parseSteps() {
        return yamlHelper.processSteps(cwlFile);
    }

    public Set<PortDetail> parseInputs() {
        Map<String, PortDetail> inputs = yamlHelper.processInputDetails(cwlFile);
        Map<String, Integer> inputDepths = yamlHelper.processInputDepths(cwlFile);

        if(inputs == null || inputDepths == null) {
            return null;
        }
        Set<PortDetail> result = new HashSet<PortDetail>();
        for(String id: inputs.keySet()) {
            PortDetail port = inputs.get(id);
            port.setId(id);
            int depth = inputDepths.get(id);
            port.setDepth(depth);
            result.add(port);
        }

        return result;
    }

    public Set<PortDetail> parseOutputs() {
        Map<String, PortDetail> inputs = yamlHelper.processOutputDetails(cwlFile);

        if(inputs == null) {
            return null;
        }
        Set<PortDetail> result = new HashSet<PortDetail>();
        for(String id: inputs.keySet()) {
            PortDetail port = inputs.get(id);
            result.add(port);
        }

        return result;
    }

}
