From 77e283d9344b5242b54d58fec2a70df50d918ec2 Mon Sep 17 00:00:00 2001
From: Tim <tymmesyde@gmail.com>
Date: Thu, 30 Jan 2025 14:57:37 +0100
Subject: [PATCH] refactor: use useShell hook for UpdaterBanner

---
 src/App/UpdaterBanner/UpdaterBanner.tsx | 9 +++++----
 src/common/Platform/Platform.tsx        | 2 +-
 src/common/index.js                     | 2 ++
 src/common/{Platform => }/useShell.ts   | 0
 4 files changed, 8 insertions(+), 5 deletions(-)
 rename src/common/{Platform => }/useShell.ts (100%)

diff --git a/src/App/UpdaterBanner/UpdaterBanner.tsx b/src/App/UpdaterBanner/UpdaterBanner.tsx
index a6b48066d..cf80c198a 100644
--- a/src/App/UpdaterBanner/UpdaterBanner.tsx
+++ b/src/App/UpdaterBanner/UpdaterBanner.tsx
@@ -1,7 +1,7 @@
 import React, { useEffect } from 'react';
 import Icon from '@stremio/stremio-icons/react';
 import { useServices } from 'stremio/services';
-import { useBinaryState } from 'stremio/common';
+import { useBinaryState, useShell } from 'stremio/common';
 import { Button, Transition } from 'stremio/components';
 import styles from './UpdaterBanner.less';
 
@@ -11,17 +11,18 @@ type Props = {
 
 const UpdaterBanner = ({ className }: Props) => {
     const { shell } = useServices();
+    const shellTransport = useShell();
     const [visible, show, hide] = useBinaryState(false);
 
     const onInstallClick = () => {
-        shell.transport && shell.transport.send('autoupdater-notif-clicked');
+        shellTransport.send('autoupdater-notif-clicked');
     };
 
     useEffect(() => {
-        shell.on('autoupdater-show-notif', show);
+        shell.transport && shell.transport.on('autoupdater-show-notif', show);
 
         return () => {
-            shell.off('autoupdater-show-notif', show);
+            shell.transport && shell.transport.off('autoupdater-show-notif', show);
         };
     });
 
diff --git a/src/common/Platform/Platform.tsx b/src/common/Platform/Platform.tsx
index 41e74b24f..375fe5e18 100644
--- a/src/common/Platform/Platform.tsx
+++ b/src/common/Platform/Platform.tsx
@@ -1,6 +1,6 @@
 import React, { createContext, useContext } from 'react';
 import { WHITELISTED_HOSTS } from 'stremio/common/CONSTANTS';
-import useShell from './useShell';
+import { useShell } from 'stremio/common';
 import { name, isMobile } from './device';
 
 interface PlatformContext {
diff --git a/src/common/index.js b/src/common/index.js
index 4c514dfbe..61d2a3933 100644
--- a/src/common/index.js
+++ b/src/common/index.js
@@ -19,6 +19,7 @@ const useModelState = require('./useModelState');
 const useNotifications = require('./useNotifications');
 const useOnScrollToBottom = require('./useOnScrollToBottom');
 const useProfile = require('./useProfile');
+const { default: useShell } = require('./useShell');
 const useStreamingServer = require('./useStreamingServer');
 const useTorrent = require('./useTorrent');
 const useTranslate = require('./useTranslate');
@@ -47,6 +48,7 @@ module.exports = {
     useNotifications,
     useOnScrollToBottom,
     useProfile,
+    useShell,
     useStreamingServer,
     useTorrent,
     useTranslate,
diff --git a/src/common/Platform/useShell.ts b/src/common/useShell.ts
similarity index 100%
rename from src/common/Platform/useShell.ts
rename to src/common/useShell.ts