From b9b5bafaf3b74b53b628c8318ba9cc83f140aa13 Mon Sep 17 00:00:00 2001
From: Bill <billsonnn@live.com>
Date: Wed, 16 Mar 2022 02:58:36 -0400
Subject: [PATCH] Add RoomEnterError listener

---
 src/api/nitro/session/VisitDesktop.ts         |  9 ++++++
 src/api/nitro/session/index.ts                |  1 +
 .../navigator/NavigatorMessageHandler.tsx     | 32 +++++++++++++++++--
 src/components/toolbar/ToolbarView.tsx        | 12 ++-----
 4 files changed, 42 insertions(+), 12 deletions(-)
 create mode 100644 src/api/nitro/session/VisitDesktop.ts

diff --git a/src/api/nitro/session/VisitDesktop.ts b/src/api/nitro/session/VisitDesktop.ts
new file mode 100644
index 00000000..e7416a35
--- /dev/null
+++ b/src/api/nitro/session/VisitDesktop.ts
@@ -0,0 +1,9 @@
+import { GetRoomSession, GetRoomSessionManager, GoToDesktop } from '.';
+
+export const VisitDesktop = () =>
+{
+    if(!GetRoomSession()) return;
+
+    GoToDesktop();
+    GetRoomSessionManager().removeSession(-1);
+}
diff --git a/src/api/nitro/session/index.ts b/src/api/nitro/session/index.ts
index 5b222369..a860d7a0 100644
--- a/src/api/nitro/session/index.ts
+++ b/src/api/nitro/session/index.ts
@@ -18,3 +18,4 @@ export * from './IsOwnerOfFloorFurniture';
 export * from './IsOwnerOfFurniture';
 export * from './IsRidingHorse';
 export * from './StartRoomSession';
+export * from './VisitDesktop';
diff --git a/src/components/navigator/NavigatorMessageHandler.tsx b/src/components/navigator/NavigatorMessageHandler.tsx
index 38c5f9d5..93c01d29 100644
--- a/src/components/navigator/NavigatorMessageHandler.tsx
+++ b/src/components/navigator/NavigatorMessageHandler.tsx
@@ -1,6 +1,6 @@
-import { GenericErrorEvent, GetGuestRoomResultEvent, LegacyExternalInterface, NavigatorCategoriesComposer, NavigatorCategoriesEvent, NavigatorHomeRoomEvent, NavigatorMetadataEvent, NavigatorSearchEvent, NavigatorSettingsComposer, RoomCreatedEvent, RoomDataParser, RoomDoorbellAcceptedEvent, RoomDoorbellEvent, RoomDoorbellRejectedEvent, RoomEntryInfoMessageEvent, RoomForwardEvent, RoomInfoComposer, RoomSettingsUpdatedEvent, UserInfoEvent } from '@nitrots/nitro-renderer';
+import { CantConnectMessageParser, GenericErrorEvent, GetGuestRoomResultEvent, LegacyExternalInterface, NavigatorCategoriesComposer, NavigatorCategoriesEvent, NavigatorHomeRoomEvent, NavigatorMetadataEvent, NavigatorSearchEvent, NavigatorSettingsComposer, RoomCreatedEvent, RoomDataParser, RoomDoorbellAcceptedEvent, RoomDoorbellEvent, RoomDoorbellRejectedEvent, RoomEnterErrorEvent, RoomEntryInfoMessageEvent, RoomForwardEvent, RoomInfoComposer, RoomSettingsUpdatedEvent, UserInfoEvent } from '@nitrots/nitro-renderer';
 import { FC, useCallback } from 'react';
-import { CreateRoomSession, GetSessionDataManager, SendMessageComposer } from '../../api';
+import { CreateRoomSession, GetSessionDataManager, LocalizeText, NotificationAlertType, NotificationUtilities, SendMessageComposer, VisitDesktop } from '../../api';
 import { UpdateDoorStateEvent } from '../../events';
 import { DispatchUiEvent, UseMessageEventHook } from '../../hooks';
 import { useNavigatorContext } from './NavigatorContext';
@@ -194,6 +194,33 @@ export const NavigatorMessageHandler: FC<{}> = props =>
         SendMessageComposer(new RoomInfoComposer(parser.roomId, false, false));
     }, []);
 
+    const onRoomEnterErrorEvent = useCallback((event: RoomEnterErrorEvent) =>
+    {
+        const parser = event.getParser();
+
+        switch(parser.reason)
+        {
+            case CantConnectMessageParser.REASON_FULL:
+                NotificationUtilities.simpleAlert(LocalizeText('navigator.guestroomfull.text'), NotificationAlertType.DEFAULT, null, null, LocalizeText('navigator.guestroomfull.title'));
+
+                break;
+            case CantConnectMessageParser.REASON_QUEUE_ERROR:
+                NotificationUtilities.simpleAlert(LocalizeText(`room.queue.error.${ parser.parameter }`), NotificationAlertType.DEFAULT, null, null, LocalizeText('room.queue.error.title'));
+
+                break;
+            case CantConnectMessageParser.REASON_BANNED:
+                NotificationUtilities.simpleAlert(LocalizeText('navigator.banned.text'), NotificationAlertType.DEFAULT, null, null, LocalizeText('navigator.banned.title'));
+
+                break;
+            default:
+                NotificationUtilities.simpleAlert(LocalizeText('room.queue.error.title'), NotificationAlertType.DEFAULT, null, null, LocalizeText('room.queue.error.title'));
+
+                break;
+        }
+
+        VisitDesktop();
+    }, []);
+
     UseMessageEventHook(UserInfoEvent, onUserInfoEvent);
     UseMessageEventHook(RoomForwardEvent, onRoomForwardEvent);
     UseMessageEventHook(RoomEntryInfoMessageEvent, onRoomEntryInfoMessageEvent);
@@ -208,6 +235,7 @@ export const NavigatorMessageHandler: FC<{}> = props =>
     UseMessageEventHook(RoomCreatedEvent, onRoomCreatedEvent);
     UseMessageEventHook(NavigatorHomeRoomEvent, onNavigatorHomeRoomEvent);
     UseMessageEventHook(RoomSettingsUpdatedEvent, onRoomSettingsUpdatedEvent);
+    UseMessageEventHook(RoomEnterErrorEvent, onRoomEnterErrorEvent);
 
     return null;
 }
diff --git a/src/components/toolbar/ToolbarView.tsx b/src/components/toolbar/ToolbarView.tsx
index 06a69731..70decda4 100644
--- a/src/components/toolbar/ToolbarView.tsx
+++ b/src/components/toolbar/ToolbarView.tsx
@@ -1,6 +1,6 @@
 import { Dispose, DropBounce, EaseOut, FigureUpdateEvent, JumpBy, Motions, NitroToolbarAnimateIconEvent, PerkAllowancesMessageEvent, PerkEnum, Queue, UserInfoDataParser, UserInfoEvent, Wait } from '@nitrots/nitro-renderer';
 import { FC, useCallback, useState } from 'react';
-import { CreateLinkEvent, GetRoomSession, GetRoomSessionManager, GetSessionDataManager, GetUserProfile, GoToDesktop, OpenMessengerChat } from '../../api';
+import { CreateLinkEvent, GetSessionDataManager, GetUserProfile, OpenMessengerChat, VisitDesktop } from '../../api';
 import { Base, Flex, LayoutAvatarImageView, LayoutItemCountView, TransitionAnimation, TransitionAnimationTypes } from '../../common';
 import { AchievementsUIEvent, AchievementsUIUnseenCountEvent, FriendsEvent, FriendsMessengerIconEvent, FriendsRequestCountEvent, GuideToolEvent, InventoryEvent, ModToolsEvent, NavigatorEvent, UnseenItemTrackerUpdateEvent, UserSettingsUIEvent } from '../../events';
 import { BatchUpdates, DispatchUiEvent, UseMessageEventHook, UseRoomEngineEvent, UseUiEvent } from '../../hooks';
@@ -178,14 +178,6 @@ export const ToolbarView: FC<ToolbarViewProps> = props =>
         }
     }, []);
 
-    const visitDesktop = useCallback(() =>
-    {
-        if(!GetRoomSession()) return;
-
-        GoToDesktop();
-        GetRoomSessionManager().removeSession(-1);
-    }, []);
-
     return (
         <>
             <TransitionAnimation type={ TransitionAnimationTypes.FADE_IN } inProp={ isMeExpanded } timeout={ 300 }>
@@ -200,7 +192,7 @@ export const ToolbarView: FC<ToolbarViewProps> = props =>
                                 <LayoutItemCountView count={ unseenAchievementCount } /> }
                         </Flex>
                         { isInRoom &&
-                            <Base pointer className="navigation-item icon icon-habbo" onClick={ visitDesktop } /> }
+                            <Base pointer className="navigation-item icon icon-habbo" onClick={ event => VisitDesktop() } /> }
                         { !isInRoom &&
                             <Base pointer className="navigation-item icon icon-house" onClick={ event => CreateLinkEvent('navigator/goto/home') } /> }
                         <Base pointer className="navigation-item icon icon-rooms" onClick={ event => handleToolbarItemClick(ToolbarViewItems.NAVIGATOR_ITEM) } />