Separate compute method from result

Author: SMadani

SNIPPETS.md

@@ -20,6 +20,7 @@ This file was generated by running [AggregateSnippets.java](src/main/java/Aggreg
 - [**Verify**](#verify)
 - [**Verify v2**](#verify-v2)
 - [**Voice**](#voice)
+
 ## Initialize
 ### Application Auth With Key Contents
 

src/main/java/AggregateSnippets.java

@@ -7,25 +7,47 @@
 import java.util.*;
 
 public final class AggregateSnippets {
-    record CodeSnippetFile(
+    public record CodeSnippetFile(
             Path file,
             int mainStartIndex, int mainEndIndex,
             int clientStartIndex, int clientEndIndex
     ) {
 
     }
 
-    private final StringBuilder sb = new StringBuilder(1 << 17);
+    private StringBuilder sb;
     private final Path snippetsSrcRoot;
-    private final Collection<CodeSnippetFile> snippetFiles = new LinkedHashSet<>(256);
+    private Collection<CodeSnippetFile> snippetFiles;
 
     public AggregateSnippets(Path snippetsSrcRoot) {
         this.snippetsSrcRoot = Objects.requireNonNull(snippetsSrcRoot);
     }
 
-    public String computeContents() throws IOException {
-        final String classFileName = "AggregateSnippets.java";
-        sb.append("# Vonage Java SDK Code Snippets\n")
+    private void checkComputed() {
+        if (sb == null || sb.isEmpty()) {
+            throw new IllegalStateException("Contents not computed yet.");
+        }
+    }
+
+    public Collection<CodeSnippetFile> getLineNumbers() {
+        checkComputed();
+        return snippetFiles;
+    }
+
+    public String getContents() {
+        checkComputed();
+        return sb.toString();
+    }
+
+    public void saveToFile(Path destPath) throws IOException {
+        Files.writeString(destPath, getContents(), StandardOpenOption.CREATE);
+    }
+
+    public void computeContents() throws IOException {
+        snippetFiles = new LinkedHashSet<>(256);
+        final String classFileName = getClass().getSimpleName() + ".java";
+        sb = new StringBuilder(1 << 17)
+                .append("# Vonage Java SDK Code Snippets\n")
                 .append("Here are all the snippets in this repository.\n")
                 .append("This file was generated by running [").append(classFileName)
                 .append("](src/main/java/").append(classFileName)
@@ -39,13 +61,11 @@ public String computeContents() throws IOException {
             sb.append("\n- [**").append(title).append("**](#")
                     .append(title.toLowerCase().replace(' ', '-')).append(")");
         }
-        sb.append("\n");
+        sb.append("\n\n");
 
         for (var file : allDirs) {
             appendSnippetContent(file, 2);
         }
-
-        return sb.toString();
     }
 
     private List<File> getAllDirsSorted() {
@@ -95,6 +115,10 @@ else if (level > 2 && path.getName().endsWith(".java")) {
                     .stripTrailing().stripIndent().replace("\t", "    ");
 
             sb.append("\n```java\n").append(nugget).append("\n```\n");
+
+            snippetFiles.add(new CodeSnippetFile(
+                    path.toPath(), startIndex, endIndex, clientInitStartIndex, clientInitEndIndex
+            ));
         }
     }
 
@@ -130,9 +154,9 @@ private static String toHeadingTitle(String title) {
     public static void main(String[] args) throws Throwable {
         final var repoRoot = Paths.get("").toAbsolutePath();
         final var snippetsSrcRoot = repoRoot.resolve("src/main/java/com/vonage/quickstart");
-        final String contents = new AggregateSnippets(snippetsSrcRoot).computeContents();
+        final var aggregator = new AggregateSnippets(snippetsSrcRoot);
+        aggregator.computeContents();
         var destPath = repoRoot.resolve("SNIPPETS.md");
-        Files.deleteIfExists(destPath);
-        Files.writeString(destPath, contents, StandardOpenOption.CREATE_NEW);
+        aggregator.saveToFile(destPath);
     }
 }