// Decompile selected function recursively (until a given number of new functions is reached)
// @category _Reman3
// @menupath Reman3.Dump N Functions

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

import ghidra.app.script.GhidraScript;
import ghidra.pcodeCPort.address.Address;
import ghidra.program.model.listing.Function;
import re3lib.FunctionDumper;
import re3lib.GlobalDumper;
import re3lib.PCallTracer;
import re3lib.RecompileConfig;
import re3lib.TypeDumper;

public class DumpCurrentFunctionN extends GhidraScript {
  final int NumFunctions = 8;

  // class Entry {
  //   Function function;
  // }
  // class QueueEntry {
  //   Function function;
  //   List<Function> callees;
  // }

  // HashSet<Address> visited = new HashSet<>();

  // QueueEntry enter(Function function) {
  //   if (visited.contains(function.getEntryPoint()))
  //     return null;

  //   visited.add(function.getEntryPoint());

  //   QueueEntry entry = new QueueEntry();
  //   entry.function = function;

  //   function.getCalledFunctions(monitor);
    
  // }

  @Override
  public void run() throws Exception {
    RecompileConfig.INSTANCE = new RecompileConfig(this);
    RecompileConfig.INSTANCE.createDirectories();

    GlobalDumper globalDumper = new GlobalDumper(this);
    globalDumper.loadGlobalManifest();

    FunctionDumper functionDumper = new FunctionDumper(this, globalDumper);

    PCallTracer tracer = new PCallTracer();
    tracer.setBlacklist(functionDumper.functionAddrBlackList);
    tracer.traceCalls(getFunctionContaining(currentAddress));

    List<Address> queue = new ArrayList<>();

    // List<Function> functionsToDump = new ArrayList<>();
    // List<Function> functionsToDumpNew = new ArrayList<>();
    // for (Function func : tracer.out) {
    //   if (FunctionDumper.isDumpedFix(func))
    //     continue;

    //   println("Dump: " + func.getName());
    //   functionsToDump.add(func);

    //   if (!FunctionDumper.isDumpedAuto(func))
    //     functionsToDumpNew.add(func);
    // }

    // if (!functionsToDump.isEmpty()) {
    //   String newOpt = "Only new (" + functionsToDumpNew.size() + ")";
    //   String okOpt = "Yes (" + functionsToDump.size() + ")";
    //   String choice = askChoice("Confirmation", "About to generate " + functionsToDump.size() + " functions ("
    //       + functionsToDumpNew.size() + " new), continue?",
    //       new ArrayList<String>() {
    //         {
    //           add(okOpt);
    //           add(newOpt);
    //           add("No");
    //         }
    //       }, okOpt);
    //   if (choice == okOpt) {
    //   } else if (choice == newOpt) {
    //     functionsToDump = functionsToDumpNew;
    //   } else {
    //     return;
    //   }

    //   for (Function func : functionsToDump) {
    //     functionDumper.dump(func);
    //   }

    //   if (functionDumper.createdFile)
    //     RecompileConfig.INSTANCE.touchCMakeTimestamp();

    //   globalDumper.dumpGlobals();
    //   globalDumper.saveGlobalManifest();
    // }

    // // Dump types
    // TypeDumper dumper = new TypeDumper(this);
    // dumper.run();
  }
}