Add stats generation (#177)

* Add stats generation

* Parameters and inner classes too

* Fixes

1 files changed, 27 insertions, 2 deletions

diff --git a/src/main/java/cuchaz/enigma/gui/GuiController.java b/src/main/java/cuchaz/enigma/gui/GuiController.java
index 54b5c92..69f12e2 100644
--- a/src/main/java/cuchaz/enigma/gui/GuiController.java
+++ b/src/main/java/cuchaz/enigma/gui/GuiController.java
@@ -20,6 +20,8 @@ import cuchaz.enigma.api.service.ObfuscationTestService;
 import cuchaz.enigma.bytecode.translators.SourceFixVisitor;
 import cuchaz.enigma.config.Config;
 import cuchaz.enigma.gui.dialog.ProgressDialog;
+import cuchaz.enigma.gui.stats.StatsGenerator;
+import cuchaz.enigma.gui.stats.StatsMember;
 import cuchaz.enigma.gui.util.History;
 import cuchaz.enigma.throwables.MappingParseException;
 import cuchaz.enigma.translation.Translator;
@@ -31,17 +33,19 @@ import cuchaz.enigma.translation.representation.entry.Entry;
 import cuchaz.enigma.translation.representation.entry.FieldEntry;
 import cuchaz.enigma.translation.representation.entry.MethodEntry;
 import cuchaz.enigma.utils.ReadableToken;
+import cuchaz.enigma.utils.Utils;
 import org.objectweb.asm.Opcodes;
 
 import javax.annotation.Nullable;
 import javax.swing.*;
+import java.awt.*;
 import java.awt.event.ItemEvent;
-import java.io.PrintWriter;
-import java.io.StringWriter;
+import java.io.*;
 import java.nio.file.Path;
 import java.util.Collection;
 import java.util.List;
 import java.util.Optional;
+import java.util.Set;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
@@ -527,4 +531,25 @@ public class GuiController {
 
                 refreshCurrentClass(reference);
         }
+
+        public void openStats(Set<StatsMember> includedMembers) {
+                ProgressDialog.runOffThread(gui.getFrame(), progress -> {
+                        String data = new StatsGenerator(project).generate(progress, includedMembers);
+
+                        try {
+                                File statsFile = File.createTempFile("stats", ".html");
+
+                                try (FileWriter w = new FileWriter(statsFile)) {
+                                        w.write(
+                                                        Utils.readResourceToString("/stats.html")
+                                                                 .replace("/*data*/", data)
+                                        );
+                                }
+
+                                Desktop.getDesktop().open(statsFile);
+                        } catch (IOException e) {
+                                throw new Error(e);
+                        }
+                });
+        }
 }