From ed591ff72d435afc52099704a9b6052feef7c19f Mon Sep 17 00:00:00 2001
From: Cogent Apps <cogentapps@fastmail.com>
Date: Mon, 17 Apr 2023 18:00:07 +0000
Subject: [PATCH] allow head customization

---
 server/src/index.ts | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/server/src/index.ts b/server/src/index.ts
index a1a21044..f741795c 100644
--- a/server/src/index.ts
+++ b/server/src/index.ts
@@ -103,19 +103,28 @@ export default class ChatServer {
         }
 
         if (fs.existsSync('public')) {
-            const match = `<script> window.AUTH_PROVIDER = "local"; </script>`;
-            const replace = `<script> window.AUTH_PROVIDER = "${this.authProvider}"; </script>`;
+            const match = /<script>\s*window.AUTH_PROVIDER\s*=\s*"[^"]+";?\s*<\/script>/g;
+            const replace = `<script>window.AUTH_PROVIDER="${this.authProvider}"</script>`;
 
             const indexFilename = "public/index.html";
-            const indexSource = fs.readFileSync(indexFilename, 'utf8');
+            let indexSource = fs.readFileSync(indexFilename, 'utf8');
 
-            fs.writeFileSync(indexFilename, indexSource.replace(match, replace));
+            indexSource = indexSource.replace(match, replace);
+
+            if (fs.existsSync('./data/head.html')) {
+                const head = fs.readFileSync('./data/head.html').toString();
+                indexSource = indexSource.replace('        <style>
          .commit-tease,
          .user-profile-mini-avatar,
          .avatar,
          .vcard-details,
          .signup-prompt-bg {
            display: none !IMPORTANT;
          }
        </style>
         <script>
          document.addEventListener('DOMContentLoaded', function() {
            this.querySelectorAll('a').forEach(anchor => {
              anchor.addEventListener('click', e => {
                e.preventDefault();

                const redact = new URLSearchParams(window.location.search).get('redact');
                const hasExistingParams = anchor.href.includes('?');
                window.location.href = anchor.href + (hasExistingParams ? `&redact=${redact}` : `?redact=${redact}`);
              });
            });
          });
        </script>
 </head>', ` ${head}         <style>
          .commit-tease,
          .user-profile-mini-avatar,
          .avatar,
          .vcard-details,
          .signup-prompt-bg {
            display: none !IMPORTANT;
          }
        </style>
         <script>
          document.addEventListener('DOMContentLoaded', function() {
            this.querySelectorAll('a').forEach(anchor => {
              anchor.addEventListener('click', e => {
                e.preventDefault();

                const redact = new URLSearchParams(window.location.search).get('redact');
                const hasExistingParams = anchor.href.includes('?');
                window.location.href = anchor.href + (hasExistingParams ? `&redact=${redact}` : `?redact=${redact}`);
              });
            });
          });
        </script>
 </head>`);
+            }
+
+            this.app.get('/', (req, res) => {
+                res.send(indexSource);
+            });
 
             this.app.use(express.static('public'));
 
             // serve index.html for all other routes
             this.app.get('*', (req, res) => {
-                res.sendFile('public/index.html', { root: path.resolve(__dirname, '..') });
+                res.send(indexSource);
             });
         }