SourceProblem: init & add to SourceFormat

Author: PragmaTwice

src/main/java/org/hacksource/core/SourceFormat.java

@@ -1,37 +1,42 @@
 package org.hacksource.core;
 
 import com.github.javaparser.Range;
+import com.github.javaparser.TokenRange;
 import com.github.javaparser.ast.CompilationUnit;
 import com.github.javaparser.ast.Node;
 import com.github.javaparser.ast.NodeList;
 import com.github.javaparser.ast.stmt.*;
 
 import java.io.*;
 import java.nio.file.Paths;
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
 
 
 public class SourceFormat {
 
-    public static String format(CompilationUnit cu) {
+    public static String format(CompilationUnit cu, List<SourceProblem> list) {
 
         Arrays.stream(new Class[]{
                 IfStmt.class, ForStmt.class, ForEachStmt.class, WhileStmt.class
-        }).forEach(c -> addBracket(cu, c));
+        }).forEach(c -> addStmtBracket(cu, c, list));
 
         return cu.toString();
     }
 
-    private static <T extends Statement> void addBracket(CompilationUnit cu, Class<T> c) {
+    private static <T extends Statement> void addStmtBracket(CompilationUnit cu, Class<T> c, List<SourceProblem> list) {
         cu.findAll(c).forEach(stmt -> {
             List<Node> children = stmt.getChildNodes();
+
             if(children.size() != 0) {
                 Statement block = (Statement)children.get(children.size() - 1);
+
                 if(!block.isBlockStmt()) {
-                    stmt.replace(block, new BlockStmt(new NodeList<Statement>(block)));
-                    Range range = stmt.getRange().orElseThrow(() -> new RuntimeException("cannot get range"));
-                    System.out.println(range.toString());
+                    stmt.replace(block, new BlockStmt(new NodeList<>(block)));
+
+                    TokenRange tokenRange = stmt.getTokenRange().orElseThrow(() -> new RuntimeException("cannot get range"));
+                    list.add(new SourceProblem("s4.1.1-braces-always-used", tokenRange));
                 }
             }
         });
@@ -42,7 +47,10 @@ public static void main(String[] args) throws IOException, SourceException {
         String path = SourceFormat.class.getResource("/example.java").getPath();
         path = path.substring(1);
 
-        String output = format(SourceParser.parseFile(Paths.get(path)));
+        List<SourceProblem> problems = new ArrayList<>();
+        String output = format(SourceParser.parseFile(Paths.get(path)), problems);
+
+        problems.forEach(p -> System.out.println(p));
 
         FileWriter fileWriter = new FileWriter("generated/example.java");
         fileWriter.write(output);

src/main/java/org/hacksource/core/SourceProblem.java

@@ -0,0 +1,31 @@
+package org.hacksource.core;
+
+import com.github.javaparser.Range;
+import com.github.javaparser.TokenRange;
+
+public class SourceProblem {
+
+    private String codeStyleTag;
+    private TokenRange tokenRange;
+
+    SourceProblem(String tag, TokenRange range) {
+        codeStyleTag = tag;
+        tokenRange = range;
+    }
+
+    public String getCodeStyleTag() {
+        return codeStyleTag;
+    }
+
+    public TokenRange getTokenRange() {
+        return tokenRange;
+    }
+
+    public Range getRange() {
+        return getTokenRange().toRange().orElseThrow(() -> new RuntimeException("cannot get range"));
+    }
+
+    public String toString() {
+        return getRange().toString() + ":" + codeStyleTag;
+    }
+}