From c9fed4af43ea3e076c2a4fc876cd34f3e03e5ba1 Mon Sep 17 00:00:00 2001
From: Ubuntu <howardzchung@gmail.com>
Date: Sun, 7 Jan 2024 09:33:24 +0000
Subject: [PATCH] return 500 on failed parse

---
 src/main/java/opendota/Main.java | 33 +++++++++++++++++++-------------
 1 file changed, 20 insertions(+), 13 deletions(-)

diff --git a/src/main/java/opendota/Main.java b/src/main/java/opendota/Main.java
index 41bfe283..fa064646 100644
--- a/src/main/java/opendota/Main.java
+++ b/src/main/java/opendota/Main.java
@@ -1,5 +1,6 @@
 package opendota;
 
+import java.io.ByteArrayOutputStream;
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.OutputStream;
@@ -45,8 +46,6 @@ public void handle(HttpExchange t) throws IOException {
     static class BlobHandler implements HttpHandler {
         @Override
         public void handle(HttpExchange t) throws IOException {
-            t.sendResponseHeaders(200, 0);
-            OutputStream os = t.getResponseBody();
             try {
                 Map<String, String> query = splitQuery(t.getRequestURI());
                 URL replayUrl = new URL(query.get("replay_url"));
@@ -58,18 +57,26 @@ public void handle(HttpExchange t) throws IOException {
                 // Download, unzip, parse, aggregate
                 Process proc = new ProcessBuilder(new String[] {"bash", "-c", cmd})
                 .start();
-
-                // Write error to console (uncomment for debugging but this causes the process not to exit)
-                // copy(proc.getErrorStream(), System.err);
-
-                // Write output to response
-                copy(proc.getInputStream(), os);
+                ByteArrayOutputStream output = new ByteArrayOutputStream();
+                ByteArrayOutputStream error = new ByteArrayOutputStream();
+                copy(proc.getInputStream(), output);
+                // Write error to console
+                copy(proc.getErrorStream(), error);
+                System.err.println(error.toString());
+                int exitCode = proc.waitFor();
+                if (exitCode != 0) {
+                    // We can send 200 status here if expected error (read the error string)
+                    t.sendResponseHeaders(500, 0);
+                    t.getResponseBody().close();
+                } else {
+                    t.sendResponseHeaders(200, output.size());
+                    output.writeTo(t.getResponseBody());
+                    t.getResponseBody().close();
+                }
+            } 
+            catch(InterruptedException e) {
+                e.printStackTrace();
             }
-            catch (Exception e)
-            {
-            	e.printStackTrace();
-            }
-            os.close();
         }
     }