/* Copyright (c) 2002-2011 by XMLVM.org
 *
 * Project Info:  http://www.xmlvm.org
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 2.1 of the License, or
 * (at your option) any later version.
 *
 * This library 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 Lesser General Public
 * License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301,
 * USA.
 */

package org.xmlvm.proc.out;

import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.StringReader;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import org.xmlvm.Log;
import org.xmlvm.main.Arguments;
import org.xmlvm.proc.BundlePhase1;
import org.xmlvm.proc.BundlePhase2;
import org.xmlvm.proc.XmlvmProcessImpl;
import org.xmlvm.proc.XmlvmResource;
import org.xmlvm.proc.lib.LibraryLoader;
import org.xmlvm.util.ClassListLoader;
import org.xmlvm.util.universalfile.UniversalFile;
import org.xmlvm.util.universalfile.UniversalFileCreator;

/**
 * Makes sure that all referenced resources from the input resources are loaded.
 * This is done recursively.
 */
public class RecursiveResourceLoadingProcess extends XmlvmProcessImpl
{

	public RecursiveResourceLoadingProcess(Arguments arguments)
	{
		super(arguments);
		addAllXmlvmEmittingProcessesAsInput();
	}

	public boolean processPhase1(BundlePhase1 bundle)
	{
		// We create a map that maps type name to the resource.
		Map<String, XmlvmResource> xmlvmResources= new HashMap<String, XmlvmResource>();
		for (XmlvmResource resource : bundle.getResources())
		{
			xmlvmResources.put(resource.getFullName(), resource);
		}
		if (arguments.option_load_dependencies() && !arguments.option_disable_load_dependencies())
		{
			if (arguments.option_greenlist() != null)
			{ // Force loading of all greenlist classes
				preloadGreenlistFiles(arguments.option_greenlist(), xmlvmResources);
			}

			LibraryLoader libraryLoader= new LibraryLoader(arguments);

			// Add all required resources, that will not be referenced to the
			// usual way.
			for (XmlvmResource requiredResource : libraryLoader.loadMonolithicLibraries())
			{
				xmlvmResources.put(requiredResource.getFullName(), requiredResource);
			}

			// Make sure we have all types that are referenced loaded.
			libraryLoader.loadAllReferencedTypes(xmlvmResources);
		}
		bundle.addResources(xmlvmResources.values());
		return true;
	}

	public boolean processPhase2(BundlePhase2 bundle)
	{
		return true;
	}

	private void preloadGreenlistFiles(String greenlistFile, Map<String, XmlvmResource> resources)
	{
		UniversalFile greenList= UniversalFileCreator.createFile(new File(greenlistFile));
		LibraryLoader libraryLoader= new LibraryLoader(arguments);
		UniversalFile defaultGreenList= UniversalFileCreator.createFile("/lib/greenlist.txt", "lib/greenlist.txt");
		for (UniversalFile file : new UniversalFile[] { greenList, defaultGreenList })
		{
			Set<String> classes= ClassListLoader.loadGreenlist(file);
			if (classes != null)
			{ // If any green list exists, force inclusion of its contents
				for (String className : classes)
				{
					if (!resources.containsKey(className))
					{
						XmlvmResource resource= libraryLoader.load(className);
						if (resource != null)
						{
							resources.put(resource.getFullName(), resource);
						}
					}
				}
			}
		}
	}
}