/*
 * ProGuard -- shrinking, optimization, obfuscation, and preverification
 *             of Java bytecode.
 *
 * Copyright (c) 2002-2017 Eric Lafortune @ GuardSquare
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU 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 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
 */
package proguard.classfile.editor;

import proguard.classfile.*;
import proguard.classfile.attribute.*;
import proguard.classfile.attribute.visitor.AttributeVisitor;
import proguard.classfile.constant.Utf8Constant;
import proguard.classfile.util.*;
import proguard.classfile.visitor.ClassVisitor;

import java.util.Arrays;

/**
 * This ClassVisitor sorts the interfaces of the program classes that it visits.
 *
 * @author Eric Lafortune
 */
public class InterfaceSorter
extends      SimplifiedVisitor
implements   ClassVisitor,
             AttributeVisitor
{
    private static final boolean DEBUG = false;


    // Implementations for ClassVisitor.

    public void visitProgramClass(ProgramClass programClass)
    {
        int[] interfaces      = programClass.u2interfaces;
        int   interfacesCount = programClass.u2interfacesCount;

        if (interfacesCount > 1)
        {
            // Sort the interfaces.
            Arrays.sort(interfaces, 0, interfacesCount);

            // Update the signature.
            programClass.attributesAccept(this);

            // Remove any duplicate entries.
            boolean[] delete = null;
            for (int index = 1; index < interfacesCount; index++)
            {
                if (interfaces[index] == interfaces[index - 1])
                {
                    // Lazily create the array.
                    if (delete == null)
                    {
                        delete = new boolean[interfacesCount];
                    }

                    delete[index] = true;
                }
            }

            if (delete != null)
            {
                new InterfaceDeleter(delete).visitProgramClass(programClass);
            }
        }
    }


    // Implementations for AttributeVisitor.

    public void visitAnyAttribute(Clazz clazz, Attribute attribute) {}


    public void visitSignatureAttribute(Clazz clazz, SignatureAttribute signatureAttribute)
    {
        Clazz[] referencedClasses    = signatureAttribute.referencedClasses;
        Clazz[] newReferencedClasses = referencedClasses == null ? null :
            new Clazz[referencedClasses.length];

        // Recompose the signature types in a string buffer.
        StringBuffer newSignatureBuffer = new StringBuffer();

        // Also update the array with referenced classes.
        int referencedClassIndex    = 0;
        int newReferencedClassIndex = 0;

        // Process the generic definitions and superclass.
        InternalTypeEnumeration internalTypeEnumeration =
            new InternalTypeEnumeration(signatureAttribute.getSignature(clazz));

        // Copy the variable type declarations.
        if (internalTypeEnumeration.hasFormalTypeParameters())
        {
            String type = internalTypeEnumeration.formalTypeParameters();

            // Append the type.
            newSignatureBuffer.append(type);

            // Copy any referenced classes.
            if (newReferencedClasses != null)
            {
                int classCount =
                    new DescriptorClassEnumeration(type).classCount();

                for (int counter = 0; counter < classCount; counter++)
                {
                    newReferencedClasses[newReferencedClassIndex++] =
                        referencedClasses[referencedClassIndex++];
                }
            }

            if (DEBUG)
            {
                System.out.println("InterfaceDeleter:   type parameters = " + type);
            }
        }

        // Copy the super class type.
        if (internalTypeEnumeration.hasMoreTypes())
        {
            String type = internalTypeEnumeration.nextType();

            // Append the type.
            newSignatureBuffer.append(type);

            // Copy any referenced classes.
            if (newReferencedClasses != null)
            {
                int classCount =
                    new DescriptorClassEnumeration(type).classCount();

                for (int counter = 0; counter < classCount; counter++)
                {
                    newReferencedClasses[newReferencedClassIndex++] =
                        referencedClasses[referencedClassIndex++];
                }
            }

            if (DEBUG)
            {
                System.out.println("InterfaceSorter:   super class type = " + type);
            }
        }

        int firstReferencedInterfaceIndex = referencedClassIndex;

        // Copy the interface types, based on the sorted interface classes.
        // This has the advantage that we will disregard any interface types
        // that are not in the interface classes, like in some versions of
        // the Scala runtime library.
        for (int interfaceIndex = 0; interfaceIndex < clazz.getInterfaceCount(); interfaceIndex++)
        {
            // Consider the interface class name.
            String interfaceName = clazz.getInterfaceName(interfaceIndex);

            referencedClassIndex = firstReferencedInterfaceIndex;

            // Find the corresponding interface type.
            InternalTypeEnumeration internalInterfaceTypeEnumeration =
                new InternalTypeEnumeration(signatureAttribute.getSignature(clazz));

            // Skip the superclass type.
            internalInterfaceTypeEnumeration.nextType();

            while (internalInterfaceTypeEnumeration.hasMoreTypes())
            {
                String type = internalInterfaceTypeEnumeration.nextType();

                DescriptorClassEnumeration classEnumeration =
                    new DescriptorClassEnumeration(type);

                int classCount =
                    classEnumeration.classCount();

                classEnumeration.nextFluff();

                if (interfaceName.equals(classEnumeration.nextClassName()))
                {
                    // Append the type.
                    newSignatureBuffer.append(type);

                    // Copy any referenced classes.
                    if (newReferencedClasses != null)
                    {
                        for (int counter = 0; counter < classCount; counter++)
                        {
                            newReferencedClasses[newReferencedClassIndex++] =
                                referencedClasses[referencedClassIndex++];
                        }
                    }

                    if (DEBUG)
                    {
                        System.out.println("InterfaceSorter:   interface type = " + type);
                    }
                }
                else
                {
                    // Skip all referenced classes.
                    referencedClassIndex += classCount;
                }
            }
        }

        String newSignature = newSignatureBuffer.toString();

        // Did the signature change?
        if (!newSignature.equals(signatureAttribute.getSignature(clazz)))
        {
            // Update the signature.
            ((Utf8Constant)((ProgramClass)clazz).constantPool[signatureAttribute.u2signatureIndex]).setString(newSignatureBuffer.toString());

            // Update the referenced classes.
            signatureAttribute.referencedClasses = newReferencedClasses;

            if (DEBUG)
            {
                System.out.println("InterfaceSorter: result = "+newSignature);
                System.out.println("InterfaceSorter: referenced classes:");

                if (newReferencedClasses != null)
                {
                    for (int index = 0; index < newReferencedClasses.length; index++)
                    {
                        System.out.println("  #"+index+" "+newReferencedClasses[index]);
                    }
                }
            }
        }
    }
}
