From 51435b93076237c968fe3bb5210296c7f58cb187 Mon Sep 17 00:00:00 2001
From: Admin <dmparrow@gmail.com>
Date: Thu, 17 Sep 2026 08:37:47 +0000
Subject: [PATCH] Make live update badge injection robust

---
 wrapper.js | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/wrapper.js b/wrapper.js
index ba91bd1..929ef00 100644
--- a/wrapper.js
+++ b/wrapper.js
@@ -3,8 +3,7 @@ import http from 'node:http';
 const originalEnd = http.ServerResponse.prototype.end;
 
 http.ServerResponse.prototype.end = function (chunk, encoding, callback) {
-  const type = String(this.getHeader('content-type') || '');
-  if (typeof chunk === 'string' && type.startsWith('text/html') && chunk.includes('</body>')) {
+  if (typeof chunk === 'string' && chunk.includes('</body>')) {
     const badge = '<div style="position:fixed;right:16px;bottom:16px;background:#1f6feb;color:#fff;padding:8px 12px;border-radius:999px;font:600 12px system-ui;box-shadow:0 8px 24px #0006">v2 · live update</div>';
     chunk = chunk.replace('</body>', `${badge}</body>`);
   }