Fix missing end quote of value with more cases

Author: coderphonui

src/main/java/com/cdpn/jsonautorepair/internal/EscapeProcessor.java

@@ -75,22 +75,6 @@ private boolean hasNextQuoteRightAfterCurrentQuoteWithoutComma(int position) {
     }
 
     private void handleNonQuoteCharacter(char currentChar, int position) {
-        if (currentChar == COMMA) {
-            char nextNonSpaceChar = findNextNonSpaceChar(position + 1);
-            if (inQuotes) {
-                if (nextNonSpaceChar == DOUBLE_QUOTE_CHAR ) {
-                    escapedJson.append(DOUBLE_QUOTE_CHAR);
-                    inQuotes = false;
-                }
-                if (nextNonSpaceChar == CLOSED_BRACKET ) {
-                    escapedJson.append(DOUBLE_QUOTE_CHAR);
-                    inQuotes = false;
-                    return;
-                }
-            }
-            escapedJson.append(currentChar);
-            return;
-        }
         if (!inQuotes) {
             escapedJson.append(currentChar);
             return;
@@ -99,6 +83,38 @@ private void handleNonQuoteCharacter(char currentChar, int position) {
             escapedJson.append(getEscapeStringFromChar(currentChar));
             return;
         }
+        if (currentChar == COMMA) {
+            handleCommaToFixMissingClosedQuote(currentChar, position);
+            return;
+        }
+        if (currentChar == CLOSED_BRACKET) {
+            handleClosedBracket(currentChar, position);
+            return;
+        }
+        escapedJson.append(currentChar);
+    }
+
+    private void handleClosedBracket(char currentChar, int position) {
+        int previousNonSpaceCharPosition = getPreviousNonSpaceCharPosition(position - 1);
+        if (previousNonSpaceCharPosition != -1) {
+           escapedJson = new StringBuilder(escapedJson.substring(0, previousNonSpaceCharPosition + 1 ));
+        }
+        escapedJson.append(DOUBLE_QUOTE_CHAR);
+        escapedJson.append(currentChar);
+        inQuotes = false;
+    }
+
+    private void handleCommaToFixMissingClosedQuote(char currentChar, int position) {
+        char nextNonSpaceChar = findNextNonSpaceChar(position + 1);
+        if (nextNonSpaceChar == DOUBLE_QUOTE_CHAR ) {
+            escapedJson.append(DOUBLE_QUOTE_CHAR);
+            inQuotes = false;
+        }
+        if (nextNonSpaceChar == CLOSED_BRACKET ) {
+            escapedJson.append(DOUBLE_QUOTE_CHAR);
+            inQuotes = false;
+            return;
+        }
         escapedJson.append(currentChar);
     }
 
@@ -128,6 +144,7 @@ private char findNextNonSpaceChar(int position) {
         return EOF;
     }
 
+
     private int getNextNonSpaceCharPosition(int position) {
         for (int i = position; i < inputString.length(); i++) {
             char currentChar = inputString.charAt(i);
@@ -138,4 +155,14 @@ private int getNextNonSpaceCharPosition(int position) {
         return -1;
     }
 
+    private int getPreviousNonSpaceCharPosition(int position) {
+        for (int i = position; i >= 0; i--) {
+            char currentChar = inputString.charAt(i);
+            if (currentChar != SPACE_CHAR && currentChar != BREAK_LINE_CHAR && currentChar != TAB_CHAR) {
+                return i;
+            }
+        }
+        return -1;
+    }
+
 }

src/test/java/com/cdpn/jsonautorepair/JSONAutoRepairerTest.java

@@ -39,29 +39,39 @@ public void repair_should_cleanup_markdown_code_block_when_string_contain_json_c
 
     @Test
     public void repair_should_fix_missing_end_quote_of_value() {
-        String originalJSON = """
+
+        assertEquals(JsonParser.parseString("""
                 {
-                    "name": "Alice,
+                    "name": "Alice",
                     "age": 30
                 }
-                """;
-        assertEquals(JsonParser.parseString("""
+                """).toString(), jsonAutoRepairer.repair("""
                 {
-                    "name": "Alice",
+                    "name": "Alice,
                     "age": 30
                 }
-                """).toString(), jsonAutoRepairer.repair(originalJSON));
+                """));
 
-        originalJSON = """
+        assertEquals(JsonParser.parseString("""
+                {
+                    "name": "Alice"
+                }
+                """).toString(), jsonAutoRepairer.repair("""
                 {
                     "name": "Alice,
                 }
-                """;
+                """));
+
+
         assertEquals(JsonParser.parseString("""
                 {
                     "name": "Alice"
                 }
-                """).toString(), jsonAutoRepairer.repair(originalJSON));
+                """).toString(), jsonAutoRepairer.repair("""
+                {
+                    "name": "Alice
+                }
+                """));
     }