From 3c0fcb40966f28d6e5afba799ba00921286e1a99 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ond=C5=99ej=20Jodas?= <ondra.jodas@gmail.com>
Date: Mon, 25 Nov 2024 21:11:26 +0100
Subject: [PATCH] Fix use warehouse

---
 src/MigrateStructure.php | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/MigrateStructure.php b/src/MigrateStructure.php
index 5c0e90e..80110e1 100644
--- a/src/MigrateStructure.php
+++ b/src/MigrateStructure.php
@@ -747,7 +747,14 @@ private function copyFunctions(string $database, ProjectRoles $projectRoles): vo
                 ->getRole($ownershipRole->getGrantedBy())
                 ->getAssignedGrants()
                 ->getWarehouseGrants();
-            $this->destinationConnection->useWarehouse(current($warehouseGrants)->getName());
+
+            $warehouseName = current($warehouseGrants)->getName();
+            try {
+                $this->destinationConnection->useWarehouse($warehouseName);
+            } catch (Throwable $e) {
+                $warehouseName = str_ireplace(['_SMALL', '_MEDIUM', '_LARGE'], '', $warehouseName);
+                $this->destinationConnection->useWarehouse($warehouseName);
+            }
 
             $this->destinationConnection->useRole($ownershipRole->getGrantedBy());