From dfcc863191e5209a0b1645f16792d7f4290e7079 Mon Sep 17 00:00:00 2001
From: karim-en <karim.alabtakh@nearone.org>
Date: Tue, 26 Nov 2024 15:21:28 +0000
Subject: [PATCH] fix: make `get_token_id` public

---
 near/omni-bridge/src/lib.rs | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/near/omni-bridge/src/lib.rs b/near/omni-bridge/src/lib.rs
index 709e56b6..17cde52c 100644
--- a/near/omni-bridge/src/lib.rs
+++ b/near/omni-bridge/src/lib.rs
@@ -796,6 +796,16 @@ impl Contract {
         self.token_id_to_address.get(&(chain_kind, token))
     }
 
+    pub fn get_token_id(&self, address: &OmniAddress) -> AccountId {
+        if let OmniAddress::Near(token_account_id) = address {
+            token_account_id.clone()
+        } else {
+            self.token_address_to_id
+                .get(address)
+                .sdk_expect("ERR_TOKEN_NOT_REGISTERED")
+        }
+    }
+
     pub fn get_transfer_message(&self, transfer_id: TransferId) -> TransferMessage {
         self.pending_transfers
             .get(&transfer_id)
@@ -1015,16 +1025,6 @@ impl Contract {
         env::log_str(&OmniBridgeEvent::FinTransferEvent { transfer_message }.to_log_string());
     }
 
-    fn get_token_id(&self, address: &OmniAddress) -> AccountId {
-        if let OmniAddress::Near(token_account_id) = address {
-            token_account_id.clone()
-        } else {
-            self.token_address_to_id
-                .get(address)
-                .sdk_expect("ERR_TOKEN_NOT_REGISTERED")
-        }
-    }
-
     fn get_native_token_id(&self, origin_chain: ChainKind) -> AccountId {
         let native_token_address =
             OmniAddress::new_zero(origin_chain).sdk_expect("ERR_FAILED_TO_GET_ZERO_ADDRESS");