/***************************************************************************
 *   Copyright (C) 2010 by Fabrizio Montesi <famontesi@gmail.com>          *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU Library General Public License as       *
 *   published by the Free Software Foundation; either version 2 of the    *
 *   License, or (at your option) any later version.                       *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU Library General Public     *
 *   License along with this program; if not, write to the                 *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 *                                                                         *
 *   For details about the authors of this software, see the AUTHORS file. *
 ***************************************************************************/

package jolie.lang.parse.util.impl;

import java.net.URI;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import jolie.lang.parse.ast.InputPortInfo;
import jolie.lang.parse.ast.InterfaceDefinition;
import jolie.lang.parse.ast.OutputPortInfo;
import jolie.lang.parse.ast.types.TypeDefinition;
import jolie.lang.parse.util.ProgramInspector;

/**
 * Implementation of {@link jolie.lang.parse.util.ProgramInspector}.
 * @author Fabrizio Montesi
 */
public class ProgramInspectorImpl implements ProgramInspector
{
	private final URI[] sources;
	private final Map< URI, List< TypeDefinition > > types;
	private final Map< URI, List< InterfaceDefinition > > interfaces;
	private final Map< URI, List< InputPortInfo > > inputPorts;
	private final Map< URI, List< OutputPortInfo > > outputPorts;

	public ProgramInspectorImpl(
		URI[] sources,
		Map< URI, List< TypeDefinition > > types,
		Map< URI, List< InterfaceDefinition > > interfaces,
		Map< URI, List< InputPortInfo > > inputPorts,
		Map< URI, List< OutputPortInfo > > outputPorts
	) {
		this.sources = sources;
		this.interfaces = interfaces;
		this.inputPorts = inputPorts;
		this.types = types;
		this.outputPorts = outputPorts;
	}

	public URI[] getSources()
	{
		return sources;
	}

	public TypeDefinition[] getTypes()
	{
		List< TypeDefinition > result = new ArrayList< TypeDefinition >();
		List< TypeDefinition > list;
		for( URI source : sources ) {
			list = types.get( source );
			if ( list != null ) {
				result.addAll( list );
			}
		}
		return result.toArray( new TypeDefinition[ 0 ] );
	}

	public TypeDefinition[] getTypes( URI source )
	{
		List< TypeDefinition > list = types.get( source );
		if ( list == null ) {
			return new TypeDefinition[ 0 ];
		}
		return list.toArray( new TypeDefinition[ 0 ] );
	}

	public InterfaceDefinition[] getInterfaces()
	{
		List< InterfaceDefinition > result = new ArrayList< InterfaceDefinition >();
		List< InterfaceDefinition > list;
		for( URI source : sources ) {
			list = interfaces.get( source );
			if ( list != null ) {
				result.addAll( list );
			}
		}
		return result.toArray( new InterfaceDefinition[ 0 ] );
	}

	public InterfaceDefinition[] getInterfaces( URI source )
	{
		List< InterfaceDefinition > list = interfaces.get( source );
		if ( list == null ) {
			return new InterfaceDefinition[ 0 ];
		}
		return list.toArray( new InterfaceDefinition[ 0 ] );
	}

	public InputPortInfo[] getInputPorts()
	{
		List< InputPortInfo > result = new ArrayList< InputPortInfo >();
		List< InputPortInfo > list;
		for( URI source : sources ) {
			list = inputPorts.get( source );
			if ( list != null ) {
				result.addAll( list );
			}
		}
		return result.toArray( new InputPortInfo[ 0 ] );
	}

	public InputPortInfo[] getInputPorts( URI source )
	{
		List< InputPortInfo > list = inputPorts.get( source );
		if ( list == null ) {
			return new InputPortInfo[ 0 ];
		}
		return list.toArray( new InputPortInfo[ 0 ] );
	}

	public OutputPortInfo[] getOutputPorts()
	{
		List< OutputPortInfo > result = new ArrayList< OutputPortInfo >();
		List< OutputPortInfo > list;
		for( URI source : sources ) {
			list = outputPorts.get( source );
			if ( list != null ) {
				result.addAll( list );
			}
		}
		return result.toArray( new OutputPortInfo[ 0 ] );
	}

	public OutputPortInfo[] getOutputPorts( URI source )
	{
		List< OutputPortInfo > list = outputPorts.get( source );
		if ( list == null ) {
			return new OutputPortInfo[ 0 ];
		}
		return list.toArray( new OutputPortInfo[ 0 ] );
	}
}