From e8eaf2781bf4f64d529d4063121c619888a48303 Mon Sep 17 00:00:00 2001
From: Robert Goldmann <deadlocker@gmx.de>
Date: Sun, 8 Jan 2023 17:48:04 +0100
Subject: [PATCH] #724 - redirect back to csv import page after creation of a
 new transaction

---
 .../budgetmaster/transactions/TransactionController.java | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/BudgetMasterServer/src/main/java/de/deadlocker8/budgetmaster/transactions/TransactionController.java b/BudgetMasterServer/src/main/java/de/deadlocker8/budgetmaster/transactions/TransactionController.java
index b26ff55c4..915c81545 100644
--- a/BudgetMasterServer/src/main/java/de/deadlocker8/budgetmaster/transactions/TransactionController.java
+++ b/BudgetMasterServer/src/main/java/de/deadlocker8/budgetmaster/transactions/TransactionController.java
@@ -31,6 +31,7 @@ import org.springframework.ui.Model;
 import org.springframework.util.StringUtils;
 import org.springframework.validation.BindingResult;
 import org.springframework.web.bind.annotation.*;
+import org.springframework.web.context.request.RequestAttributes;
 import org.springframework.web.context.request.WebRequest;
 
 import javax.servlet.http.HttpServletRequest;
@@ -58,6 +59,7 @@ public class TransactionController extends BaseController
 		public static final String NEW_TRANSACTION = "transactions/newTransactionNormal";
 		public static final String CHANGE_TYPE = "transactions/changeTypeModal";
 		public static final String RECURRING_OVERVIEW = "transactions/recurringOverview";
+		public static final String REDIRECT_IMPORT = "redirect:/transactionImport";
 	}
 
 	private static final String CONTINUE = "continue";
@@ -200,7 +202,6 @@ public class TransactionController extends BaseController
 			redirectUrl = ReturnValues.NEW_TRANSACTION;
 		}
 
-
 		final boolean isContinueActivated = action.equals(CONTINUE);
 		return handleRedirect(servletRequest, request, model, transaction.getID() != null, transaction, bindingResult, date, redirectUrl, isContinueActivated);
 	}
@@ -261,6 +262,12 @@ public class TransactionController extends BaseController
 			return ReturnValues.REDIRECT_NEW_TRANSACTION;
 		}
 
+		// redirect back to csv import if import is active
+		if(request.getAttribute(TransactionImportController.RequestAttributeNames.CSV_TRANSACTIONS, RequestAttributes.SCOPE_SESSION) != null)
+		{
+			return ReturnValues.REDIRECT_IMPORT;
+		}
+
 		return ReturnValues.REDIRECT_ALL_ENTITIES;
 	}
 
-- 
GitLab