From afd0ae6428ab6627cd14d751df00d5f0c6e63518 Mon Sep 17 00:00:00 2001
From: Nathan Neulinger <nneul@neulinger.org>
Date: Fri, 27 Dec 2024 14:21:18 -0600
Subject: [PATCH] Initial support for setting a server identifier

---
 server/client.js                       |  1 +
 server/notification-providers/slack.js |  4 ++++
 server/notification-providers/smtp.js  |  5 +++++
 src/components/settings/General.vue    | 21 +++++++++++++++++++++
 src/layouts/Layout.vue                 | 18 +++++++++++++++---
 src/mixins/public.js                   | 12 ++++++++++++
 src/pages/Setup.vue                    |  2 +-
 src/pages/SetupDatabase.vue            |  2 +-
 8 files changed, 60 insertions(+), 5 deletions(-)

diff --git a/server/client.js b/server/client.js
index 72f0a4e8e..421f89441 100644
--- a/server/client.js
+++ b/server/client.js
@@ -160,6 +160,7 @@ async function sendInfo(socket, hideVersion = false) {
         isContainer,
         dbType,
         primaryBaseURL: await setting("primaryBaseURL"),
+        serverIdentifier: await setting("serverIdentifier"),
         serverTimezone: await server.getTimezone(),
         serverTimezoneOffset: server.getTimezoneOffset(),
     });
diff --git a/server/notification-providers/slack.js b/server/notification-providers/slack.js
index 5e25a1fbc..202aef12b 100644
--- a/server/notification-providers/slack.js
+++ b/server/notification-providers/slack.js
@@ -142,8 +142,12 @@ class Slack extends NotificationProvider {
             }
 
             const baseURL = await setting("primaryBaseURL");
+            const serverIdentifier = await setting("serverIdentifier");
 
             const title = "Uptime Kuma Alert";
+            if (serverIdentifier) {
+                title = title + " (" + serverIdentifier + ")"
+            }
             let data = {
                 "channel": notification.slackchannel,
                 "username": notification.slackusername,
diff --git a/server/notification-providers/smtp.js b/server/notification-providers/smtp.js
index 9f3defa5e..4e614fa06 100644
--- a/server/notification-providers/smtp.js
+++ b/server/notification-providers/smtp.js
@@ -46,6 +46,11 @@ class SMTP extends NotificationProvider {
         let body = msg;
         if (heartbeatJSON) {
             body = `${msg}\nTime (${heartbeatJSON["timezone"]}): ${heartbeatJSON["localDateTime"]}`;
+
+            const serverIdentifier = await setting("serverIdentifier");
+            if (serverIdentifier) {
+                body = body + `\nServer Identifier: ${serverIdentifier}`
+            }
         }
         // subject and body are templated
         if ((monitorJSON && heartbeatJSON) || msg.endsWith("Testing")) {
diff --git a/src/components/settings/General.vue b/src/components/settings/General.vue
index 487c3ba3a..82df5e958 100644
--- a/src/components/settings/General.vue
+++ b/src/components/settings/General.vue
@@ -132,6 +132,27 @@
                 <div class="form-text"></div>
             </div>
 
+            <!-- Server Identifier -->
+            <div class="mb-4">
+                <label class="form-label" for="serverIdentifier">
+                    {{ $t("Server Identifier") }}
+                </label>
+
+                <div class="input-group mb-3">
+                    <input
+                        id="serverIdentifier"
+                        v-model="settings.serverIdentifier"
+                        class="form-control"
+                        name="serverIdentifier"
+                        placeholder=""
+                        pattern=".+"
+                        autocomplete="new-password"
+                    />
+                </div>
+
+                <div class="form-text"></div>
+            </div>
+
             <!-- Steam API Key -->
             <div class="mb-4">
                 <label class="form-label" for="steamAPIKey">
diff --git a/src/layouts/Layout.vue b/src/layouts/Layout.vue
index 9faedf589..2fe7d4020 100644
--- a/src/layouts/Layout.vue
+++ b/src/layouts/Layout.vue
@@ -13,7 +13,7 @@
         <header v-if="! $root.isMobile" class="d-flex flex-wrap justify-content-center py-3 mb-3 border-bottom">
             <router-link to="/dashboard" class="d-flex align-items-center mb-3 mb-md-0 me-md-auto text-dark text-decoration-none">
                 <object class="bi me-2 ms-4" width="40" height="40" data="/icon.svg" />
-                <span class="fs-4 title">{{ $t("Uptime Kuma") }}</span>
+                <span class="fs-4 title">{{ $t("Uptime Kuma") }}{{ $root.serverIdentifierSuffix }}</span>
             </router-link>
 
             <a v-if="hasNewVersion" target="_blank" href="https://github.com/louislam/uptime-kuma/releases" class="btn btn-info me-3">
@@ -85,7 +85,7 @@
         <header v-else class="d-flex flex-wrap justify-content-center pt-2 pb-2 mb-3">
             <router-link to="/dashboard" class="d-flex align-items-center text-dark text-decoration-none">
                 <object class="bi" width="40" height="40" data="/icon.svg" />
-                <span class="fs-4 title ms-2">Uptime Kuma</span>
+                <span class="fs-4 title ms-2">{{ $t("Uptime Kuma") }}{{ $root.serverIdentifierSuffix }}</span>
             </router-link>
         </header>
 
@@ -170,7 +170,11 @@ export default {
     },
 
     watch: {
-
+        $route(to) {
+            if (this.serverIdentifierSuffix) {
+                this.updateTitle("Uptime Kuma" + this.$root.serverIdentifierSuffix);
+            }
+        }
     },
 
     mounted() {
@@ -188,6 +192,10 @@ export default {
         if (this.toastContainer != null) {
             this.toastContainerObserver.observe(this.toastContainer, { childList: true });
         }
+
+        if (this.$root.serverIdentifierSuffix) {
+            this.updateTitle("Uptime Kuma" + this.$root.serverIdentifierSuffix);
+        }
     },
 
     beforeUnmount() {
@@ -201,6 +209,10 @@ export default {
          */
         clearToasts() {
             toast.clear();
+        },
+
+        updateTitle(title) {
+            document.title = title;
         }
     },
 
diff --git a/src/mixins/public.js b/src/mixins/public.js
index c87bfb358..f809a09c1 100644
--- a/src/mixins/public.js
+++ b/src/mixins/public.js
@@ -51,5 +51,17 @@ export default {
                 return location.protocol + "//" + location.host;
             }
         },
+
+        serverIdentifierSuffix() {
+            if (this.$root.info.serverIdentifier) {
+                return " (" + this.$root.info.serverIdentifier + ")";
+            }
+
+            if (env === "development" || localStorage.dev === "dev") {
+                return " (dev)";
+            } else {
+                return "";
+            }
+        },
     }
 };
diff --git a/src/pages/Setup.vue b/src/pages/Setup.vue
index f681b7a82..f88758ce7 100644
--- a/src/pages/Setup.vue
+++ b/src/pages/Setup.vue
@@ -5,7 +5,7 @@
                 <div>
                     <object width="64" height="64" data="/icon.svg" />
                     <div style="font-size: 28px; font-weight: bold; margin-top: 5px;">
-                        Uptime Kuma
+                        {{ $t("Uptime Kuma") }}
                     </div>
                 </div>
 
diff --git a/src/pages/SetupDatabase.vue b/src/pages/SetupDatabase.vue
index 81738a98b..f4e6963b7 100644
--- a/src/pages/SetupDatabase.vue
+++ b/src/pages/SetupDatabase.vue
@@ -4,7 +4,7 @@
             <div>
                 <object width="64" height="64" data="/icon.svg" />
                 <div style="font-size: 28px; font-weight: bold; margin-top: 5px;">
-                    Uptime Kuma
+                    {{ $t("Uptime Kuma") }}
                 </div>
             </div>