chore: optimize reboot button

OneKeyHQ · Jul 22, 2024 · 7a5ca9c · 7a5ca9c
1 parent e0a1730
commit 7a5ca9c
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/src/components/Firmware/Firmware.tsx b/src/components/Firmware/Firmware.tsx
@@ -93,17 +93,26 @@ const RebootToBoard: FC = () => {
   const intl = useIntl();
   const device = useSelector((state: RootState) => state.runtime.device);
   const [loading, setLoading] = useState(false);
+  const isBootLoader = device?.features?.bootloader_mode;
 
   const onRebootToBoard = useCallback(async () => {
     setLoading(true);
     await serviceHardware.rebootToBoard();
     setLoading(false);
   }, []);
 
-  if (['touch', 'pro'].includes(getDeviceType(device?.features))) {
+  if (
+    !isBootLoader &&
+    ['touch', 'pro'].includes(getDeviceType(device?.features))
+  ) {
     return (
-      <div className="flex items-center justify-between text-sm text-gray-800 py-1">
-        <Button loading={loading} className="w-full" onClick={onRebootToBoard}>
+      <div className="flex items-center justify-end py-1">
+        <Button
+          className="sm:w-[50%] w-full"
+          size="sm"
+          loading={loading}
+          onClick={onRebootToBoard}
+        >
           {intl.formatMessage({ id: 'TR_CLICK_TO_BOARD' })}
         </Button>
       </div>