Fix welcomeMessage

Author: stevenjoezhang

build/message.d.ts

@@ -3,6 +3,6 @@ type Time = {
     text: string;
 }[];
 declare function showMessage(text: string | string[], timeout: number, priority: number, override?: boolean): void;
-declare function welcomeMessage(time: Time, welcomeTemplate: string, referrerTemplate: string): string;
+declare function welcomeMessage(time: Time, welcomeTemplate?: string, referrerTemplate?: string): string;
 declare function i18n(template: string, ...args: string[]): string;
 export { showMessage, welcomeMessage, i18n, Time };

build/message.js

@@ -33,14 +33,15 @@ function welcomeMessage(time, welcomeTemplate, referrerTemplate) {
             }
         }
     }
+    if (!welcomeTemplate)
+        return '';
     const text = i18n(welcomeTemplate, document.title);
-    if (document.referrer !== '') {
-        const referrer = new URL(document.referrer);
-        if (location.hostname === referrer.hostname)
-            return text;
-        return `${i18n(referrerTemplate, referrer.hostname)}<br>${text}`;
-    }
-    return text;
+    if (document.referrer === '' || !referrerTemplate)
+        return text;
+    const referrer = new URL(document.referrer);
+    if (location.hostname === referrer.hostname)
+        return text;
+    return `${i18n(referrerTemplate, referrer.hostname)}<br>${text}`;
 }
 function i18n(template, ...args) {
     return template.replace(/\$(\d+)/g, (_, idx) => {

dist/waifu-tips.js

@@ -61,9 +61,11 @@ function showMessage(
 /**
  * Show welcome message based on time.
  * @param {Time} time - Time message configuration.
+ * @param {string} [welcomeTemplate] - Welcome message template.
+ * @param {string} [referrerTemplate] - Referrer message template.
  * @returns {string} Welcome message.
  */
-function welcomeMessage(time: Time, welcomeTemplate: string, referrerTemplate: string): string {
+function welcomeMessage(time: Time, welcomeTemplate?: string, referrerTemplate?: string): string {
   if (location.pathname === '/') {
     // If on the homepage
     for (const { hour, text } of time) {
@@ -78,13 +80,13 @@ function welcomeMessage(time: Time, welcomeTemplate: string, referrerTemplate: s
       }
     }
   }
+  if (!welcomeTemplate) return '';
   const text = i18n(welcomeTemplate, document.title);
-  if (document.referrer !== '') {
-    const referrer = new URL(document.referrer);
-    if (location.hostname === referrer.hostname) return text;
-    return `${i18n(referrerTemplate, referrer.hostname)}<br>${text}`;
-  }
-  return text;
+  if (document.referrer === '' || !referrerTemplate) return text;
+
+  const referrer = new URL(document.referrer);
+  if (location.hostname === referrer.hostname) return text;
+  return `${i18n(referrerTemplate, referrer.hostname)}<br>${text}`;
 }
 
 function i18n(template: string, ...args: string[]) {