import org.objectweb.asm.tree.ClassNode
import org.objectweb.asm.tree.FieldNode
import the.bytecode.club.bytecodeviewer.api.BCV
import the.bytecode.club.bytecodeviewer.api.Plugin
import the.bytecode.club.bytecodeviewer.api.PluginConsole
import the.bytecode.club.bytecodeviewer.gui.components.MultipleChoiceDialog

import java.lang.reflect.Field

import static the.bytecode.club.bytecodeviewer.Constants.nl

/**
 * This is an example of a string decrypter plugin
 */
class ExampleStringDecrypter extends Plugin {

    @Override
    void execute(List<ClassNode> classNodesList) {
        PluginConsole gui = new PluginConsole("Example String Decrypter")

        MultipleChoiceDialog dialog = new MultipleChoiceDialog("Bytecode Viewer - WARNING",
                "WARNING: This will load the classes into the JVM and execute the initialize function"
                        + nl + "for each class. IF THE FILE YOU'RE LOADING IS MALICIOUS, DO NOT CONTINUE.",
                new String[]{"Continue", "Cancel"})

        if(dialog.promptChoice() == 0)
        {
            for(ClassNode cn : classNodesList)
            {
                BCV.getClassNodeLoader().addClass(cn)

                for(Object o : cn.fields.toArray())
                {
                    FieldNode f = (FieldNode) o
                    if(f.name == "z") {// && f.desc.equals("([Ljava/lang/String;)V")) {
                        try
                        {
                            for(Field f2 : BCV.getClassNodeLoader().nodeToClass(cn).getFields())
                            {
                                String s = f2.get(null)
                                if(s != null && !s.empty)
                                    gui.appendText(cn + ":" + s)
                            }
                        } catch(Exception | StackOverflowError ignored) {}
                    }
                }

            }

            gui.setVisible(true)
        }
    }

}