// ============================================================================
//
// Copyright (C) 2006-2012 Talend Inc. - www.talend.com
//
// This source code is available under agreement available at
// %InstallDIR%\features\org.talend.rcp.branding.%PRODUCTNAME%\%PRODUCTNAME%license.txt
//
// You should have received a copy of the agreement
// along with this program; if not, write to Talend SA
// 9 rue Pages 92150 Suresnes, France
//
// ============================================================================
package org.talend.core.model.components;

import java.util.ArrayList;
import java.util.List;

import org.talend.core.model.metadata.IMetadataTable;
import org.talend.core.model.process.IConnection;

/**
 * DOC smallet class global comment. Detailled comment <br/>
 * 
 * $Id: IODataComponentContainer.java 38013 2010-03-05 14:21:59Z mhirt $
 * 
 */
public class IODataComponentContainer {

    private List<IODataComponent> inputs;

    private List<IODataComponent> ouputs;

    public IODataComponentContainer() {
        this.inputs = new ArrayList<IODataComponent>();
        this.ouputs = new ArrayList<IODataComponent>();
    }

    /**
     * Getter for inputs.
     * 
     * @return the inputs
     */
    public List<IODataComponent> getInputs() {
        return this.inputs;
    }

    /**
     * Getter for ouputs.
     * 
     * @return the ouputs
     */
    public List<IODataComponent> getOuputs() {
        return this.ouputs;
    }

    public IODataComponent getDataComponent(IConnection connection) {
        for (IODataComponent current : inputs) {
            if (current.getUniqueName().equals(connection.getUniqueName())) {
                return current;
            }
        }
        for (IODataComponent current : ouputs) {
            if (current.getUniqueName().equals(connection.getUniqueName())) {
                return current;
            }
        }
        return null;
    }

    public IMetadataTable getTable(IConnection connection) {
        IODataComponent current = getDataComponent(connection);
        if (current == null) {
            return null;
        } else {
            return current.getTable();
        }
    }
}