From bfec74b0cc6afb4280013fe5b779c35086d9e276 Mon Sep 17 00:00:00 2001
From: Shebin John <admin@remedcu.com>
Date: Wed, 25 Sep 2024 13:47:39 +0200
Subject: [PATCH] Add migration deployment retrieval (#807)

Safe deployments for the migration contracts were added in #777 but
retrieving those deployments was not added. This PR adds the same.
---
 src/deployments.ts | 15 ++++++++++++
 src/libs.ts        | 57 ++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 72 insertions(+)

diff --git a/src/deployments.ts b/src/deployments.ts
index 02f8cd8f8..6c70231a2 100644
--- a/src/deployments.ts
+++ b/src/deployments.ts
@@ -76,6 +76,18 @@ import SignMessageLib141 from './assets/v1.4.1/sign_message_lib.json';
 
 const _SIGN_MESSAGE_LIB_DEPLOYMENTS = [SignMessageLib141, SignMessageLib130] as SingletonDeploymentJSON[];
 
+import SafeMigration141 from './assets/v1.4.1/safe_migration.json';
+
+const _SAFE_MIGRATION_DEPLOYMENTS = [SafeMigration141] as SingletonDeploymentJSON[];
+
+import SafeToL2Migration141 from './assets/v1.4.1/safe_to_l2_migration.json';
+
+const _SAFE_TO_L2_MIGRATION_DEPLOYMENTS = [SafeToL2Migration141] as SingletonDeploymentJSON[];
+
+import SafeToL2Setup141 from './assets/v1.4.1/safe_to_l2_setup.json';
+
+const _SAFE_TO_L2_SETUP_DEPLOYMENTS = [SafeToL2Setup141] as SingletonDeploymentJSON[];
+
 export {
   _ACCESSOR_DEPLOYMENTS,
   _FACTORY_DEPLOYMENTS,
@@ -87,4 +99,7 @@ export {
   _MULTI_SEND_CALL_ONLY_DEPLOYMENTS,
   _CREATE_CALL_DEPLOYMENTS,
   _SIGN_MESSAGE_LIB_DEPLOYMENTS,
+  _SAFE_MIGRATION_DEPLOYMENTS,
+  _SAFE_TO_L2_MIGRATION_DEPLOYMENTS,
+  _SAFE_TO_L2_SETUP_DEPLOYMENTS,
 };
diff --git a/src/libs.ts b/src/libs.ts
index a066fbbfc..e79634c74 100644
--- a/src/libs.ts
+++ b/src/libs.ts
@@ -3,6 +3,9 @@ import {
   _MULTI_SEND_CALL_ONLY_DEPLOYMENTS,
   _MULTI_SEND_DEPLOYMENTS,
   _SIGN_MESSAGE_LIB_DEPLOYMENTS,
+  _SAFE_MIGRATION_DEPLOYMENTS,
+  _SAFE_TO_L2_MIGRATION_DEPLOYMENTS,
+  _SAFE_TO_L2_SETUP_DEPLOYMENTS,
 } from './deployments';
 import { DeploymentFilter, DeploymentFormats, SingletonDeployment, SingletonDeploymentV2 } from './types';
 import { findDeployment } from './utils';
@@ -78,3 +81,57 @@ export const getSignMessageLibDeployment = (filter?: DeploymentFilter): Singleto
 export const getSignMessageLibDeployments = (filter?: DeploymentFilter): SingletonDeploymentV2 | undefined => {
   return findDeployment(filter, _SIGN_MESSAGE_LIB_DEPLOYMENTS, DeploymentFormats.MULTIPLE);
 };
+
+/**
+ * Get the SafeMigration deployment based on the provided filter.
+ * @param {DeploymentFilter} [filter] - The filter criteria for the deployment.
+ * @returns {SingletonDeployment | undefined} - The matched deployment or undefined if not found.
+ */
+export const getSafeMigrationDeployment = (filter?: DeploymentFilter): SingletonDeployment | undefined => {
+  return findDeployment(filter, _SAFE_MIGRATION_DEPLOYMENTS);
+};
+
+/**
+ * Get all SafeMigration deployments based on the provided filter.
+ * @param {DeploymentFilter} [filter] - The filter criteria for the deployments.
+ * @returns {SingletonDeploymentV2 | undefined} - The matched deployments or undefined if not found.
+ */
+export const getSafeMigrationDeployments = (filter?: DeploymentFilter): SingletonDeploymentV2 | undefined => {
+  return findDeployment(filter, _SAFE_MIGRATION_DEPLOYMENTS, DeploymentFormats.MULTIPLE);
+};
+
+/**
+ * Get the SafeToL2Migration deployment based on the provided filter.
+ * @param {DeploymentFilter} [filter] - The filter criteria for the deployment.
+ * @returns {SingletonDeployment | undefined} - The matched deployment or undefined if not found.
+ */
+export const getSafeToL2MigrationDeployment = (filter?: DeploymentFilter): SingletonDeployment | undefined => {
+  return findDeployment(filter, _SAFE_TO_L2_MIGRATION_DEPLOYMENTS);
+};
+
+/**
+ * Get all SafeToL2Migration deployments based on the provided filter.
+ * @param {DeploymentFilter} [filter] - The filter criteria for the deployments.
+ * @returns {SingletonDeploymentV2 | undefined} - The matched deployments or undefined if not found.
+ */
+export const getSafeToL2MigrationDeployments = (filter?: DeploymentFilter): SingletonDeploymentV2 | undefined => {
+  return findDeployment(filter, _SAFE_TO_L2_MIGRATION_DEPLOYMENTS, DeploymentFormats.MULTIPLE);
+};
+
+/**
+ * Get the SafeToL2Setup deployment based on the provided filter.
+ * @param {DeploymentFilter} [filter] - The filter criteria for the deployment.
+ * @returns {SingletonDeployment | undefined} - The matched deployment or undefined if not found.
+ */
+export const getSafeToL2SetupDeployment = (filter?: DeploymentFilter): SingletonDeployment | undefined => {
+  return findDeployment(filter, _SAFE_TO_L2_SETUP_DEPLOYMENTS);
+};
+
+/**
+ * Get all SafeToL2Setup deployments based on the provided filter.
+ * @param {DeploymentFilter} [filter] - The filter criteria for the deployments.
+ * @returns {SingletonDeploymentV2 | undefined} - The matched deployments or undefined if not found.
+ */
+export const getSafeToL2SetupDeployments = (filter?: DeploymentFilter): SingletonDeploymentV2 | undefined => {
+  return findDeployment(filter, _SAFE_TO_L2_SETUP_DEPLOYMENTS, DeploymentFormats.MULTIPLE);
+};