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

import javassist.CtClass;
import javassist.CtMethod;
import javassist.bytecode.ConstPool;
import javassist.bytecode.Descriptor;
import cuchaz.enigma.bytecode.ConstPoolEditor;
import cuchaz.enigma.mapping.ClassEntry;
import cuchaz.enigma.mapping.MethodEntry;

public class BridgeFixer
{
	private JarIndex m_index;
	
	public BridgeFixer( JarIndex index )
	{
		m_index = index;
	}

	public void fixBridges( CtClass c )
	{
		// rename declared methods
		for( CtMethod method : c.getDeclaredMethods() )
		{
			// get the method entry
			MethodEntry methodEntry = new MethodEntry(
				new ClassEntry( Descriptor.toJvmName( c.getName() ) ),
				method.getName(),
				method.getSignature()
			);
			MethodEntry bridgeMethodEntry = m_index.getBridgeMethod( methodEntry );
			if( bridgeMethodEntry != null )
			{
				// fix this bridged method
				method.setName( bridgeMethodEntry.getName() );
			}
		}
		
		// rename method references
		// translate all the field and method references in the code by editing the constant pool
		ConstPool constants = c.getClassFile().getConstPool();
		ConstPoolEditor editor = new ConstPoolEditor( constants );
		for( int i=1; i<constants.getSize(); i++ )
		{
			switch( constants.getTag( i ) )
			{
				case ConstPool.CONST_Methodref:
				case ConstPool.CONST_InterfaceMethodref:
				{
					// translate the name and type
					MethodEntry methodEntry = new MethodEntry(
						new ClassEntry( Descriptor.toJvmName( editor.getMemberrefClassname( i ) ) ),
						editor.getMemberrefName( i ),
						editor.getMemberrefType( i )
					);
					MethodEntry bridgeMethodEntry = m_index.getBridgeMethod( methodEntry );
					if( bridgeMethodEntry != null )
					{
						// FIXIT FIXIT FIXIT FIXIT FIXIT FIXIT FIXIT
						editor.changeMemberrefNameAndType( i, bridgeMethodEntry.getName(), bridgeMethodEntry.getSignature() );
					}
				}
				break;
			}
		}
	}
}