Fix padding before colon when single digit

ballerina-platform · Feb 27, 2024 · bc654cc · bc654cc
1 parent 92e0db6
commit bc654cc
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/cli/ballerina-cli/src/main/java/io/ballerina/cli/utils/DiagnosticAnnotation.java b/cli/ballerina-cli/src/main/java/io/ballerina/cli/utils/DiagnosticAnnotation.java
@@ -74,9 +74,15 @@ public String toString() {
 
         int max_length_line = Math.max(lines.get(0).length(), lines.get(lines.size() - 1).length());
         int n_digits_end = (int) Math.log10(startLineNumber + lines.size() - 1) + 1;
-        String padding = " ".repeat(n_digits_end + 1);
-        String paddingWithColon = " :" + " ".repeat(n_digits_end - 1);
-
+        String padding;
+        String paddingWithColon;
+        if (n_digits_end == 1) {
+            padding = " ".repeat(n_digits_end + 1);
+            paddingWithColon = ":" + " ".repeat(n_digits_end);
+        } else {
+            padding = " ".repeat(n_digits_end + 1);
+            paddingWithColon = " :" + " ".repeat(n_digits_end - 1);
+        }
         int tabsInLastLine = countTabChars(lines.get(lines.size() - 1), this.endOffset);
         lines.set(lines.size() - 1, replaceTabs(lines.get(lines.size() - 1), this.endOffset));
         int maxLength = terminalWidth - n_digits_end - 3;