From 71e90c7b64e0315028e0f657cea85b5ebe9869f7 Mon Sep 17 00:00:00 2001
From: Alexander Zinchuk <alx.zinchuk@gmail.com>
Date: Sun, 20 Jun 2021 16:50:42 +0300
Subject: [PATCH] Better support for non-production envs

---
 src/components/left/main/LeftMainHeader.tsx | 33 +++++++++++++--------
 src/config.ts                               |  2 +-
 src/serviceWorker/pushNotification.ts       |  2 +-
 webpack.config.js                           |  2 +-
 4 files changed, 23 insertions(+), 16 deletions(-)

diff --git a/src/components/left/main/LeftMainHeader.tsx b/src/components/left/main/LeftMainHeader.tsx
index 56afcd3b4..b131067db 100644
--- a/src/components/left/main/LeftMainHeader.tsx
+++ b/src/components/left/main/LeftMainHeader.tsx
@@ -52,6 +52,7 @@ type DispatchProps = Pick<GlobalActions, (
 
 const ANIMATION_LEVEL_OPTIONS = [0, 1, 2];
 
+const PRODUCTION_HOSTNAME = 'web.telegram.org';
 const LEGACY_VERSION_URL = 'https://web.telegram.org/?legacy=1';
 const WEBK_VERSION_URL = 'https://web.telegram.org/k/';
 const PERMANENT_VERSION_KEY = 'kz_version';
@@ -101,6 +102,8 @@ const LeftMainHeader: FC<OwnProps & StateProps & DispatchProps> = ({
     }, 0);
   }, [hasMenu, chatsById]);
 
+  const withOtherVersions = window.location.hostname === PRODUCTION_HOSTNAME;
+
   const MainButton: FC<{ onTrigger: () => void; isOpen?: boolean }> = useMemo(() => {
     return ({ onTrigger, isOpen }) => (
       <Button
@@ -229,19 +232,23 @@ const LeftMainHeader: FC<OwnProps & StateProps & DispatchProps> = ({
           >
             Report Bug
           </MenuItem>
-          <MenuItem
-            icon="char-K"
-            href={WEBK_VERSION_URL}
-            onClick={handleSwitchToWebK}
-          >
-            Switch to K Version
-          </MenuItem>
-          <MenuItem
-            icon="char-W"
-            href={LEGACY_VERSION_URL}
-          >
-            Switch to Old Version
-          </MenuItem>
+          {withOtherVersions && (
+            <>
+              <MenuItem
+                icon="char-K"
+                href={WEBK_VERSION_URL}
+                onClick={handleSwitchToWebK}
+              >
+                Switch to K Version
+              </MenuItem>
+              <MenuItem
+                icon="char-W"
+                href={LEGACY_VERSION_URL}
+              >
+                Switch to Old Version
+              </MenuItem>
+            </>
+          )}
         </DropdownMenu>
         <SearchInput
           inputId="telegram-search-input"
diff --git a/src/config.ts b/src/config.ts
index 570ca4308..d697d8b22 100644
--- a/src/config.ts
+++ b/src/config.ts
@@ -1,7 +1,7 @@
 export const PAGE_TITLE = 'Telegram';
 export const INACTIVE_MARKER = ' [Inactive]';
 
-export const APP_NAME = 'Telegram WebZ';
+export const APP_NAME = process.env.APP_NAME || 'Telegram WebZ';
 export const APP_VERSION = process.env.APP_VERSION || 'dev';
 
 export const DEBUG = (
diff --git a/src/serviceWorker/pushNotification.ts b/src/serviceWorker/pushNotification.ts
index 002d5bab3..5780038dc 100644
--- a/src/serviceWorker/pushNotification.ts
+++ b/src/serviceWorker/pushNotification.ts
@@ -141,7 +141,7 @@ async function focusChatMessage(client: WindowClient, data: { chatId?: number; m
 }
 
 export function handleNotificationClick(e: NotificationEvent) {
-  const appUrl = process.env.APP_URL!;
+  const appUrl = self.location.href.replace('serviceWorker.js', '');
   e.notification.close(); // Android needs explicit close.
   const { data } = e.notification;
   const notifyClients = async () => {
diff --git a/webpack.config.js b/webpack.config.js
index 1b5acc8e0..93fca3c60 100644
--- a/webpack.config.js
+++ b/webpack.config.js
@@ -103,9 +103,9 @@ module.exports = (env = {}, argv = {}) => {
         ignoreOrder: true,
       }),
       new EnvironmentPlugin({
+        APP_NAME: 'Telegram WebZ',
         APP_VERSION: 'dev',
         APP_ENV: 'production',
-        APP_URL: 'https://web.telegram.org/z/',
         TELEGRAM_T_API_ID: '',
         TELEGRAM_T_API_HASH: '',
       }),