From 8ba0a4dc0074844b7f1023e8c7fcefab5e2d3325 Mon Sep 17 00:00:00 2001
From: rfresh2 <89827146+rfresh2@users.noreply.github.com>
Date: Sat, 21 Sep 2024 16:21:50 -0700
Subject: [PATCH] modify swagger UI page title

---
 .../config/SwaggerCodeBlockTransformer.java   | 34 +++++++++----------
 1 file changed, 17 insertions(+), 17 deletions(-)

diff --git a/src/main/java/vc/config/SwaggerCodeBlockTransformer.java b/src/main/java/vc/config/SwaggerCodeBlockTransformer.java
index 417e39c..18a4795 100644
--- a/src/main/java/vc/config/SwaggerCodeBlockTransformer.java
+++ b/src/main/java/vc/config/SwaggerCodeBlockTransformer.java
@@ -13,8 +13,8 @@
 
 import java.io.BufferedReader;
 import java.io.IOException;
-import java.io.InputStream;
 import java.io.InputStreamReader;
+import java.util.ArrayList;
 import java.util.List;
 import java.util.stream.Collectors;
 
@@ -35,24 +35,24 @@ public SwaggerCodeBlockTransformer(final SwaggerUiConfigProperties swaggerUiConf
     public Resource transform(HttpServletRequest request,
                               Resource resource,
                               ResourceTransformerChain transformer) throws IOException {
-        if (resource.toString().contains("index.html")) {
-            final InputStream is = resource.getInputStream();
-            final InputStreamReader isr = new InputStreamReader(is);
-            try (BufferedReader br = new BufferedReader(isr)) {
-                final List<String> lines = br.lines().collect(Collectors.toList());
-                int endHeadIndex = -1;
-                for (int i = 0; i < lines.size(); i++) {
-                    if (lines.get(i).contains("</head>")) {
-                        endHeadIndex = i;
-                        break;
-                    }
+        if (!resource.toString().contains("index.html")) return super.transform(request, resource, transformer);
+        try (BufferedReader br = new BufferedReader(new InputStreamReader(resource.getInputStream()))) {
+            final List<String> lines = br.lines().collect(Collectors.toCollection(ArrayList::new));
+            for (int i = 0; i < lines.size(); i++) {
+                String line = lines.get(i);
+                if (line.contains("</head>")) {
+                    lines.add(i, "    <link rel=\"stylesheet\" type=\"text/css\" href=\"../swagger-theme/theme-material.css\" />");
+                    break;
                 }
-                if (endHeadIndex != -1) {
-                    lines.add(endHeadIndex, "    <link rel=\"stylesheet\" type=\"text/css\" href=\"../swagger-theme/theme-material.css\" />");
+            }
+            for (int i = 0; i < lines.size(); i++) {
+                String line = lines.get(i);
+                if (line.contains("<title>")) {
+                    lines.set(i, line.replace("Swagger UI", "2b2t.vc API Explorer"));
+                    break;
                 }
-                return new TransformedResource(resource, String.join("\n", lines).getBytes());
-            } // AutoCloseable br > isr > is
+            }
+            return new TransformedResource(resource, String.join("\n", lines).getBytes());
         }
-        return super.transform(request, resource, transformer);
     }
 }