添加生成合约doc文档类，修改生成方法签名文件格式

zhjgit · zhjgit · commit 4820a484d449 · 2018-11-21T22:39:25.000+08:00
diff --git a/src/main/java/com/sumingk/core/ContractDoc.java b/src/main/java/com/sumingk/core/ContractDoc.java
@@ -0,0 +1,90 @@
+package com.sumingk.core;
+
+import java.io.File;
+import java.io.FileNotFoundException;
+import java.io.FileOutputStream;
+import java.io.PrintStream;
+import java.util.List;
+
+/**
+ * create by sumingk 2018/11/20
+ * weixin_public_number : blockchain_do/刻意链习
+ * weibo: 刻意链习
+ */
+public class ContractDoc {
+    /**
+     * 生成的方法签名文件
+     */
+    public static final String CONTRACT_FILENAME = "/contract_DOC.html";
+
+    public static void main(String[] args) {
+        File file = new File(System.getProperty("user.dir") + MethodSign.OUTPUT_FILENAME);
+        if (file.exists()) file.delete();
+        MethodSign.anlyContractABI();
+
+        if (!file.exists()) {
+            System.err.println("File " + MethodSign.OUTPUT_FILENAME.substring(1) + " does not exist!");
+            return;
+        }
+        printDocHtml(file);
+    }
+
+    /**
+     * 生成contract_doc.html 文档
+     *
+     * @param file
+     */
+    public static void printDocHtml(File file) {
+        StringBuilder sb = new StringBuilder();
+        PrintStream printStream = null;
+
+        File apiFile = new File(System.getProperty("user.dir") + CONTRACT_FILENAME);
+        if (apiFile.exists()) apiFile.delete();
+
+        try {
+            printStream = new PrintStream(new FileOutputStream(apiFile));
+            sb.append("<html>");
+            sb.append("<head>");
+            sb.append("<title>Contract Method API</title>");
+            sb.append("<meta http-equiv=\"Content-Type\" content=\"text/html; charset=utf-8\" />");
+            sb.append("<style type=\"text/css\">");
+            sb.append("TABLE{border-collapse:collapse;border-left:solid 1 #000000; border-top:solid 1 #000000;padding:5px;}");
+            sb.append("TH{border-right:solid 1 #000000;border-bottom:solid 1 #000000;}");
+            sb.append("ol{list-style-type:none;counter-reset:sectioncounter;}");
+            sb.append("TD{font:normal;border-right:solid 1 #000000;border-bottom:solid 1 #000000;}");
+            sb.append("</style></head>");
+            sb.append("<body>");
+            sb.append("<div align=\"left\">");
+            sb.append("<ol>");
+
+            List<String> lists = FileOperate.readFileIntoStringArrList(file);
+            if (lists.size() > 0) {
+                for (int k = 0; k < lists.size(); k++) {
+                    String str = lists.get(k).replaceAll("\\s*", "");
+                    //排除 空格 空行
+                    if (str == null || "".equals(str)) continue;
+                    if (str.indexOf(".sol") > -1) {
+                        if (k < lists.size() - 1) {
+                            String nextStr = lists.get(k + 1);
+                            nextStr = nextStr.replaceAll("\\s*", "");
+                            if (nextStr == null || "".equals(nextStr)) {
+                                continue;
+                            }
+                            sb.append("<li>" + str + "</li>").append("<br/>");
+                        }
+                    } else {
+                        if (str.indexOf(":") > -1) {
+                            sb.append("<li>" + str.split(":")[1] + "</li>").append("<br/>");
+                        }
+                    }
+                }
+            }
+            sb.append("</ol>");
+            sb.append("</div></body></html>");
+            printStream.println(sb.toString());
+        } catch (FileNotFoundException e) {
+            e.printStackTrace();
+        }
+    }
+
+}
diff --git a/src/main/java/com/sumingk/core/FileOperate.java b/src/main/java/com/sumingk/core/FileOperate.java
@@ -111,6 +111,29 @@ public static void writeMethodSign(String filePath, String content) {
         }
     }
 
+    /**
+     * 逐行读取文件，返回集合
+     *
+     * @param file
+     * @return
+     */
+    public static List<String> readFileIntoStringArrList(File file) {
+        List<String> list = new ArrayList<>();
+        try {
+            InputStreamReader read = new InputStreamReader(new FileInputStream(file), "UTF-8");
+            BufferedReader bufferedReader = new BufferedReader(read);
+            String lineTxt = null;
+            while ((lineTxt = bufferedReader.readLine()) != null) {
+                list.add(lineTxt);
+            }
+            bufferedReader.close();
+            read.close();
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        return list;
+    }
+
     /**
      * 遍历目录下的所有文件
      *
@@ -140,9 +163,6 @@ public static List<File> getFile(String fileDir, int way) {
                 }
             }
         }
-        /*for (int k = 0; k < fileList.size(); k++) {
-            System.out.println(k + " :" + fileList.get(k).getName());
-        }*/
         return fileList;
     }
 
diff --git a/src/main/java/com/sumingk/core/MethodSign.java b/src/main/java/com/sumingk/core/MethodSign.java
@@ -26,7 +26,6 @@ public static void main(String[] args) {
         anlyContractABI();
     }
 
-
     /**
      * abi 解析
      */
@@ -68,7 +67,7 @@ public static void anlyContractABI() {
      */
     public static void truffleGenerate(File f1, Keccak.Digest256 digest256) {
         String fileName = f1.getName();
-        String printContractName = "--------------------------------" + "ContractName: " + fileName.substring(0, fileName.lastIndexOf("json")) + "sol" + "------------------------------------";
+        String printContractName = "--------------------------------" + fileName.substring(0, fileName.lastIndexOf("json")) + "sol" + "------------------------------------";
         System.out.println(printContractName);
         FileOperate.writeMethodSign(System.getProperty("user.dir") + OUTPUT_FILENAME, "\r\n" + printContractName + "\r\n");
         if (f1.canRead()) {
@@ -89,10 +88,9 @@ public static void truffleGenerate(File f1, Keccak.Digest256 digest256) {
     public static void solcGenerate(File f1, Keccak.Digest256 digest256) {
         String fileName = f1.getName();
         if (fileName.endsWith(".abi")) {//以.abi结尾的文件
-            String printContractName = "--------------------------------" + "ContractName: " + fileName.substring(0, fileName.lastIndexOf("abi")) + "sol" + "------------------------------------";
+            String printContractName = "--------------------------------" + fileName.substring(0, fileName.lastIndexOf("abi")) + "sol" + "------------------------------------";
             System.out.println(printContractName);
             FileOperate.writeMethodSign(System.getProperty("user.dir") + OUTPUT_FILENAME, "\r\n" + printContractName + "\r\n");
-
             if (f1.canRead()) {
                 JSONArray jsonArray = FileOperate.getDatafromFileForArray(f1.getPath());//abi文件
                 if (jsonArray != null && jsonArray.size() > 0) {
@@ -125,15 +123,16 @@ public static void digestMethodSign(JSONArray jsonArray, Keccak.Digest256 digest
                     sbf.append(")");
                     String finalFunc = funcName + sbf.toString();
                     functionSign = Hex.toHexString(digest256.digest(finalFunc.getBytes())).substring(0, 8);
-                    writeContent = "--with param--: " + functionSign + " : " + finalFunc;
+                    writeContent = functionSign + " : " + finalFunc;
                     System.out.println(writeContent);
                 } else {
                     functionSign = Hex.toHexString(digest256.digest((funcName + "()").getBytes())).substring(0, 8);
-                    writeContent = "---no param---: " + functionSign + " : " + funcName + "()";
+                    writeContent = functionSign + " : " + funcName + "()";
                     System.out.println(writeContent);
                 }
                 FileOperate.writeMethodSign(System.getProperty("user.dir") + OUTPUT_FILENAME, writeContent + "\r\n");
             }
         }
     }
+
 }

Original file line number	Diff line number	Diff line change
`@@ -111,6 +111,29 @@ public static void writeMethodSign(String filePath, String content) {`
`111`	`111`	`}`
`112`	`112`	`}`
`113`	`113`
	`114`	`+ /**`
	`115`	`+ * 逐行读取文件，返回集合`
	`116`	`+ *`
	`117`	`+ * @param file`
	`118`	`+ * @return`
	`119`	`+ */`
	`120`	`+ public static List<String> readFileIntoStringArrList(File file) {`
	`121`	`+ List<String> list = new ArrayList<>();`
	`122`	`+ try {`
	`123`	`+ InputStreamReader read = new InputStreamReader(new FileInputStream(file), "UTF-8");`
	`124`	`+ BufferedReader bufferedReader = new BufferedReader(read);`
	`125`	`+ String lineTxt = null;`
	`126`	`+ while ((lineTxt = bufferedReader.readLine()) != null) {`
	`127`	`+ list.add(lineTxt);`
	`128`	`+ }`
	`129`	`+ bufferedReader.close();`
	`130`	`+ read.close();`
	`131`	`+ } catch (Exception e) {`
	`132`	`+ e.printStackTrace();`
	`133`	`+ }`
	`134`	`+ return list;`
	`135`	`+ }`
	`136`	`+`
`114`	`137`	`/**`
`115`	`138`	`* 遍历目录下的所有文件`
`116`	`139`	`*`
`@@ -140,9 +163,6 @@ public static List<File> getFile(String fileDir, int way) {`
`140`	`163`	`}`
`141`	`164`	`}`
`142`	`165`	`}`
`143`		`- /*for (int k = 0; k < fileList.size(); k++) {`
`144`		`- System.out.println(k + " :" + fileList.get(k).getName());`
`145`		`- }*/`
`146`	`166`	`return fileList;`
`147`	`167`	`}`
`148`	`168`