feat: base64 option

1 files changed, 15 insertions, 3 deletions

diff --git a/public/generate.php b/public/generate.php
index 6226c8f..bd539c7 100644
--- a/public/generate.php
+++ b/public/generate.php
@@ -112,7 +112,19 @@ $screenshot = generate_message_screenshot(
 imagedestroy($pfp);
 imagedestroy($old_pfp);
 
-header('Content-Type: image/jpeg');
-header("Content-Disposition: inline; filename*=UTF-8''message-{$username}-{$message_id}-{$channel_login}.jpg");
-imagejpeg($screenshot);
+if (isset($_GET['base64'])) {
+    ob_start();
+    imagejpeg($screenshot);
+    $image_data = ob_get_contents();
+    ob_end_clean();
+
+    $image_data = base64_encode($image_data);
+
+    echo "data:image/jpeg;base64,{$image_data}";
+} else {
+    header('Content-Type: image/jpeg');
+    header("Content-Disposition: inline; filename*=UTF-8''message-{$username}-{$message_id}-{$channel_login}.jpg");
+    imagejpeg($screenshot);
+}
+
 imagedestroy($screenshot);
\ No newline at end of file