/*******************************************************************************
 * Copyright (c) 2014 Jeff Martin.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Public License v3.0
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/gpl.html
 * 
 * Contributors:
 *     Jeff Martin - initial API and implementation
 ******************************************************************************/
package cuchaz.enigma.bytecode;

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

import javassist.CtBehavior;
import javassist.CtClass;
import javassist.CtConstructor;
import javassist.CtMethod;
import javassist.bytecode.Descriptor;
import cuchaz.enigma.mapping.ArgumentEntry;
import cuchaz.enigma.mapping.BehaviorEntry;
import cuchaz.enigma.mapping.ClassEntry;
import cuchaz.enigma.mapping.ConstructorEntry;
import cuchaz.enigma.mapping.MethodEntry;
import cuchaz.enigma.mapping.Translator;

public class MethodParameterWriter
{
	private Translator m_translator;
	
	public MethodParameterWriter( Translator translator )
	{
		m_translator = translator;
	}
	
	public void writeMethodArguments( CtClass c )
	{
		// Procyon will read method arguments from the "MethodParameters" attribute, so write those
		ClassEntry classEntry = new ClassEntry( Descriptor.toJvmName( c.getName() ) );
		for( CtBehavior behavior : c.getDeclaredBehaviors() )
		{
			int numParams = Descriptor.numOfParameters( behavior.getMethodInfo().getDescriptor() );
			if( numParams <= 0 )
			{
				continue;
			}
			
			// get the behavior entry
			BehaviorEntry behaviorEntry;
			if( behavior instanceof CtMethod )
			{
				behaviorEntry = new MethodEntry( classEntry, behavior.getMethodInfo().getName(), behavior.getSignature() );
			}
			else if( behavior instanceof CtConstructor )
			{
				behaviorEntry = new ConstructorEntry( classEntry, behavior.getSignature() );
			}
			else
			{
				throw new Error( "Unsupported behavior type: " + behavior.getClass().getName() );
			}
			
			// get the list of parameter names
			List<String> names = new ArrayList<String>( numParams );
			for( int i=0; i<numParams; i++ )
			{
				names.add( m_translator.translate( new ArgumentEntry( behaviorEntry, i, "" ) ) );
			}
			
			// save the mappings to the class
			MethodParametersAttribute.updateClass( behavior.getMethodInfo(), names );
		}
	}
}