From fb727b5921e2ca5551597ee2cfa5e7a7630aa1fb Mon Sep 17 00:00:00 2001
From: Joel Jeremy Marquez <joeljeremy.marquez@gmail.com>
Date: Thu, 11 Apr 2024 14:12:46 -0700
Subject: [PATCH] Focus amount on mount

---
 .../components/modals/ReportCategoryBudgetMenuModal.tsx    | 4 ++++
 .../components/modals/RolloverCategoryBudgetMenuModal.tsx  | 7 ++++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/packages/desktop-client/src/components/modals/ReportCategoryBudgetMenuModal.tsx b/packages/desktop-client/src/components/modals/ReportCategoryBudgetMenuModal.tsx
index 1751d17133d..656c0bcdf6a 100644
--- a/packages/desktop-client/src/components/modals/ReportCategoryBudgetMenuModal.tsx
+++ b/packages/desktop-client/src/components/modals/ReportCategoryBudgetMenuModal.tsx
@@ -43,6 +43,10 @@ export function ReportCategoryBudgetMenuModal({
     onUpdateBudget?.(amountToInteger(amount));
   };
 
+  useEffect(() => {
+    setAmountFocused(true);
+  }, []);
+
   return (
     <Modal
       title={`Budget: ${category?.name}`}
diff --git a/packages/desktop-client/src/components/modals/RolloverCategoryBudgetMenuModal.tsx b/packages/desktop-client/src/components/modals/RolloverCategoryBudgetMenuModal.tsx
index 8be98ea74d2..c04c421acb3 100644
--- a/packages/desktop-client/src/components/modals/RolloverCategoryBudgetMenuModal.tsx
+++ b/packages/desktop-client/src/components/modals/RolloverCategoryBudgetMenuModal.tsx
@@ -1,4 +1,4 @@
-import React, { useState, type ComponentPropsWithoutRef } from 'react';
+import React, { useState, type ComponentPropsWithoutRef, useEffect } from 'react';
 
 import { rolloverBudget } from 'loot-core/client/queries';
 import { amountToInteger, integerToAmount } from 'loot-core/shared/util';
@@ -11,6 +11,7 @@ import { View } from '../common/View';
 import { FocusableAmountInput } from '../mobile/transactions/FocusableAmountInput';
 import { type CommonModalProps } from '../Modals';
 import { useSheetValue } from '../spreadsheet/useSheetValue';
+import { useInitialMount } from '../../hooks/useInitialMount';
 
 type RolloverCategoryBudgetMenuModalProps = ComponentPropsWithoutRef<
   typeof CategoryBudgetMenu
@@ -43,6 +44,10 @@ export function RolloverCategoryBudgetMenuModal({
     onUpdateBudget?.(amountToInteger(amount));
   };
 
+  useEffect(() => {
+    setAmountFocused(true);
+  }, []);
+
   return (
     <Modal
       title={`Budget: ${category?.name}`}