From cfe5ea3f9bf27cd6b2b69f4348f9615608e5ad18 Mon Sep 17 00:00:00 2001
From: Greyson Parrelli <greyson@signal.org>
Date: Tue, 7 Nov 2023 09:07:59 -0500
Subject: [PATCH] Add the ability to download the current perfetto trace in
 Spinner.

---
 spinner/lib/src/main/assets/partials/prefix.hbs       |  6 ++++++
 .../src/main/java/org/signal/spinner/SpinnerServer.kt | 11 +++++++++++
 2 files changed, 17 insertions(+)

diff --git a/spinner/lib/src/main/assets/partials/prefix.hbs b/spinner/lib/src/main/assets/partials/prefix.hbs
index 69c93c49c1..1242fe2a99 100644
--- a/spinner/lib/src/main/assets/partials/prefix.hbs
+++ b/spinner/lib/src/main/assets/partials/prefix.hbs
@@ -9,6 +9,12 @@
   {{/each}}
 </table>
 
+<div>
+  Download Trace: <a href="/trace">Link</a>
+</div>
+
+<br />
+
 <div>
   Database:
   <select id="database-selector">
diff --git a/spinner/lib/src/main/java/org/signal/spinner/SpinnerServer.kt b/spinner/lib/src/main/java/org/signal/spinner/SpinnerServer.kt
index 420d6b41c1..759284f33c 100644
--- a/spinner/lib/src/main/java/org/signal/spinner/SpinnerServer.kt
+++ b/spinner/lib/src/main/java/org/signal/spinner/SpinnerServer.kt
@@ -12,7 +12,9 @@ import org.signal.core.util.ExceptionUtil
 import org.signal.core.util.ForeignKeyConstraint
 import org.signal.core.util.getForeignKeys
 import org.signal.core.util.logging.Log
+import org.signal.core.util.tracing.Tracer
 import org.signal.spinner.Spinner.DatabaseConfig
+import java.io.ByteArrayInputStream
 import java.lang.IllegalArgumentException
 import java.security.NoSuchAlgorithmException
 import java.text.SimpleDateFormat
@@ -71,6 +73,7 @@ internal class SpinnerServer(
         session.method == Method.GET && session.uri == "/query" -> getQuery(dbParam)
         session.method == Method.POST && session.uri == "/query" -> postQuery(dbParam, dbConfig, session)
         session.method == Method.GET && session.uri == "/recent" -> getRecent(dbParam)
+        session.method == Method.GET && session.uri == "/trace" -> getTrace()
         session.method == Method.GET && session.uri == "/logs" -> getLogs(dbParam)
         isWebsocketRequested(session) && session.uri == "/logs/websocket" -> getLogWebSocket(session)
         else -> {
@@ -217,6 +220,14 @@ internal class SpinnerServer(
     )
   }
 
+  private fun getTrace(): Response {
+    return newChunkedResponse(
+      Response.Status.OK,
+      "application/octet-stream",
+      ByteArrayInputStream(Tracer.getInstance().serialize())
+    )
+  }
+
   private fun getLogs(dbName: String): Response {
     return renderTemplate(
       "logs",