From e0116672eddd6ae5fcb579eeaffd6d008e735fa7 Mon Sep 17 00:00:00 2001
From: Jaredlll08 <jaredlll08@gmail.com>
Date: Sun, 28 May 2017 23:43:54 +0200
Subject: [PATCH] Added the document annotation

---
 src/main/java/com/blamejared/ModTweaker.java  | 64 +++++++++++++++++--
 .../blamejared/api/annotations/Document.java  |  9 +++
 2 files changed, 68 insertions(+), 5 deletions(-)
 create mode 100644 src/main/java/com/blamejared/api/annotations/Document.java

diff --git a/src/main/java/com/blamejared/ModTweaker.java b/src/main/java/com/blamejared/ModTweaker.java
index 52dd1c8..dc93678 100644
--- a/src/main/java/com/blamejared/ModTweaker.java
+++ b/src/main/java/com/blamejared/ModTweaker.java
@@ -1,9 +1,15 @@
 package com.blamejared;
 
-import com.blamejared.api.annotations.Handler;
+import com.blamejared.api.annotations.*;
 import minetweaker.MineTweakerAPI;
 import net.minecraftforge.fml.common.*;
 import net.minecraftforge.fml.common.event.*;
+import stanhebben.zenscript.annotations.Optional;
+import stanhebben.zenscript.annotations.*;
+
+import java.io.*;
+import java.lang.reflect.*;
+import java.util.*;
 
 import static com.blamejared.reference.Reference.*;
 
@@ -11,23 +17,71 @@ import static com.blamejared.reference.Reference.*;
 public class ModTweaker {
     
     @Mod.EventHandler
-    public void preInit(FMLPreInitializationEvent e) {
+    public void preInit(FMLPreInitializationEvent e) throws IOException {
+        File output = new File("output.txt");
+        BufferedWriter writer = new BufferedWriter(new FileWriter(output));
+        
         e.getAsmData().getAll(Handler.class.getCanonicalName()).forEach(data -> {
             try {
                 Class clazz = Class.forName(data.getClassName());
                 Handler handl = (Handler) clazz.getAnnotation(Handler.class);
-                if(Loader.isModLoaded(handl.modid())) {
+                if(Loader.isModLoaded(handl.value())) {
                     MineTweakerAPI.registerClass(clazz);
+                    boolean document = false;
+                    for(Method method : clazz.getDeclaredMethods()) {
+                        if(method.isAnnotationPresent(Document.class)) {
+                            document = true;
+                        }
+                    }
+                    if(document) {
+                        ZenClass zen = (ZenClass) clazz.getAnnotation(ZenClass.class);
+                        writer.write("# " + handl.value() + "\n");
+                        writer.write("## " + zen.value() + "\n\n");
+                        
+                        for(Method method : clazz.getDeclaredMethods()) {
+                            if(method.isAnnotationPresent(Document.class)) {
+                                StringBuilder builder = new StringBuilder();
+                                
+                                Document doc = method.getAnnotation(Document.class);
+                                Map<String, Parameter> paramMap = new LinkedHashMap<>();
+                                for(int i = 0; i < method.getParameters().length; i++) {
+                                    paramMap.put(doc.value()[i], method.getParameters()[i]);
+                                }
+                                builder.append("//");
+                                paramMap.forEach((key, val) -> {
+                                    if(val.isAnnotationPresent(Optional.class)) {
+                                        builder.append("<span style=\"color:red\">");
+                                    }
+                                    builder.append(val.getType().getSimpleName()).append(" ").append(key);
+                                    if(val.isAnnotationPresent(Optional.class)) {
+                                        builder.append("</span>");
+                                    }
+                                    builder.append(", ");
+                                });
+                                builder.reverse().delete(0, 2).reverse();
+                                builder.append("\n```");
+                                builder.append(zen.value()).append(".");
+                                builder.append(method.getName()).append("(");
+                                paramMap.forEach((key, val) -> {
+                                    builder.append(val.getType().getSimpleName()).append(" ").append(key).append(", ");
+                                });
+                                builder.reverse().delete(0, 2).reverse();
+                                builder.append(");");
+                                builder.append("```\n");
+                                writer.write(builder.toString());
+                            }
+                        }
+                    }
                 }
-            } catch(ClassNotFoundException e1) {
+            } catch(ClassNotFoundException | IOException e1) {
                 e1.printStackTrace();
             }
         });
+        writer.close();
     }
     
     @Mod.EventHandler
     public void init(FMLInitializationEvent e) {
-    
     }
     
     @Mod.EventHandler
diff --git a/src/main/java/com/blamejared/api/annotations/Document.java b/src/main/java/com/blamejared/api/annotations/Document.java
new file mode 100644
index 0000000..ddb4068
--- /dev/null
+++ b/src/main/java/com/blamejared/api/annotations/Document.java
@@ -0,0 +1,9 @@
+package com.blamejared.api.annotations;
+
+import java.lang.annotation.*;
+
+@Retention(RetentionPolicy.RUNTIME)
+@Target(ElementType.METHOD)
+public @interface Document {
+    String[] value();
+}