From 96f32c2de0316a11000765959b82fe1b4ae59327 Mon Sep 17 00:00:00 2001
From: TheGamerdk <5618080+TheGamerdk@users.noreply.github.com>
Date: Sun, 15 Oct 2023 21:40:21 +0200
Subject: [PATCH] Update DropshipFlightControl.tsx

---
 .../tgui/interfaces/DropshipFlightControl.tsx | 33 ++++++++-----------
 1 file changed, 13 insertions(+), 20 deletions(-)

diff --git a/tgui/packages/tgui/interfaces/DropshipFlightControl.tsx b/tgui/packages/tgui/interfaces/DropshipFlightControl.tsx
index 6943c96a79fa..06bbcdb773d7 100644
--- a/tgui/packages/tgui/interfaces/DropshipFlightControl.tsx
+++ b/tgui/packages/tgui/interfaces/DropshipFlightControl.tsx
@@ -129,6 +129,9 @@ export const DropshipDestinationSelection = (_, context) => {
           selected={siteselection}
           onClick={(value) => {
             setSiteSelection(value);
+            if (data.flight_configuration === 'flyby') {
+              act('set-ferry');
+            }
             act('button-push');
           }}
         />
@@ -190,27 +193,20 @@ const FlybyControl = (props, context) => {
       className="flybyControl"
       buttons={
         <>
-          {data.flight_configuration === 'flyby' && (
-            <Button icon="road" onClick={() => act('set-ferry')}>
-              Set ferry
-            </Button>
-          )}
-          {data.has_flyby_skill === 1 && data.flight_configuration === 'ferry' && (
-            <Button icon="jet-fighter" onClick={() => act('set-flyby')}>
-              Set flyby
+          {data.has_flyby_skill === 1 && data.shuttle_mode === 'idle' && (
+            <Button
+              color="red"
+              icon="jet-fighter"
+              onClick={() => {
+                act('set-flyby');
+                act('move', { target: null });
+              }}>
+              Launch Flyby
             </Button>
           )}
           {data.has_flyby_skill === 1 && data.shuttle_mode === 'called' && (
             <Button onClick={() => act('cancel-flyby')}>cancel flyby</Button>
           )}
-          {data.has_flyby_skill === 1 && data.shuttle_mode === 'idle' && (
-            <Button
-              icon="rocket"
-              disabled={data.flight_configuration === 'ferry'}
-              onClick={() => act('move')}>
-              Launch
-            </Button>
-          )}
         </>
       }
     />
@@ -321,10 +317,7 @@ const RenderScreen = (props, context) => {
         (data.shuttle_mode === 'idle' || data.shuttle_mode === 'called') && (
           <FlybyControl />
         )}
-      {data.shuttle_mode === 'idle' &&
-        data.flight_configuration !== 'flyby' && (
-          <DropshipDestinationSelection />
-        )}
+      {data.shuttle_mode === 'idle' && <DropshipDestinationSelection />}
       {data.shuttle_mode === 'igniting' && <LaunchCountdown />}
       {data.shuttle_mode === 'pre-arrival' && <TouchdownCooldown />}
       {data.shuttle_mode === 'recharging' && <ShuttleRecharge />}