/* 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.IOException;

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.in.InputProcess.ClassInputProcess;
import org.xmlvm.proc.in.file.ClassFile;

import com.android.dx.dex.cf.CfOptions;
import com.android.dx.dex.cf.CfTranslator;
import com.android.dx.dex.file.ClassDefItem;
import com.android.dx.dex.file.DexFile;

/**
 * This process takes Java Bytecode and turns it into the DEX format.
 */
public class DexOutputProcess extends XmlvmProcessImpl
{

	private static final String DEX_ENDING= ".dex";

	public DexOutputProcess(Arguments arguments)
	{
		super(arguments);
		addSupportedInput(ClassInputProcess.class);
		addSupportedInput(JavaByteCodeOutputProcess.class);
	}

	public boolean processPhase1(BundlePhase1 bundle)
	{
		return true;
	}

	public boolean processPhase2(BundlePhase2 bundle)
	{
		for (OutputFile preOutputFile : bundle.getOutputFiles())
		{
			OutputFile outputFile= generateDexFile(preOutputFile);
			if (outputFile == null)
			{
				return false;
			}
			bundle.removeOutputFile(preOutputFile);
			bundle.addOutputFile(outputFile);
		}
		return true;
	}

	private OutputFile generateDexFile(OutputFile classFile)
	{
		if (!classFile.getFullPath().startsWith(arguments.option_out()))
		{
			Log.error("DexOutputProcess: Something is wrong with the class output path.");
			return null;
		}
		String relativePath= classFile.getFullPath().substring(arguments.option_out().length() + 1);

		// Remove a starting slash or backslash.
		if (relativePath.startsWith("/") || relativePath.startsWith("\\"))
		{
			relativePath= relativePath.substring(1);
		}
		Log.debug("DExing:" + relativePath);

		CfOptions options= new CfOptions();
		options.strictNameCheck= false;
		ClassDefItem item= CfTranslator.translate(relativePath, classFile.getDataAsBytes(), options);
		DexFile dexFile= new DexFile();
		dexFile.add(item);
		try
		{
			byte[] rawDex= dexFile.toDex(null, false);
			OutputFile result= new OutputFile(rawDex);
			result.setLocation(classFile.getLocation());
			result.setFileName(classFile.getFileName().replace(ClassFile.CLASS_ENDING, DEX_ENDING));
			return result;
		}
		catch (IOException e)
		{
			e.printStackTrace();
			Log.error("Could not generate DEX file.");
		}
		return null;
	}
}