From 00862c2b157a617c238bd0669055ae79fe991030 Mon Sep 17 00:00:00 2001
From: Goran Vladika <goran.vladika@gmail.com>
Date: Thu, 28 Mar 2024 18:20:20 +0100
Subject: [PATCH] Add option to customize token bridge branch

---
 docker-compose.yaml    | 5 ++++-
 tokenbridge/Dockerfile | 3 ++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/docker-compose.yaml b/docker-compose.yaml
index 2eb00ace..27b61677 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -329,7 +329,10 @@ services:
       - geth
       - sequencer
     pid: host
-    build: tokenbridge/
+    build:
+      context: tokenbridge/
+      args:
+        TOKEN_BRIDGE_BRANCH: ${TOKEN_BRIDGE_BRANCH:-main}
     environment:
       - ARB_URL=http://sequencer:8547
       - ETH_URL=http://geth:8545
diff --git a/tokenbridge/Dockerfile b/tokenbridge/Dockerfile
index 344e537d..ed53eae2 100644
--- a/tokenbridge/Dockerfile
+++ b/tokenbridge/Dockerfile
@@ -1,8 +1,9 @@
 FROM node:16-bullseye-slim
+ARG TOKEN_BRIDGE_BRANCH=main
 RUN apt-get update && \
     apt-get install -y git docker.io python3 build-essential
 WORKDIR /workspace  
-RUN git clone https://github.com/OffchainLabs/token-bridge-contracts.git ./
+RUN git clone -b ${TOKEN_BRIDGE_BRANCH} https://github.com/OffchainLabs/token-bridge-contracts.git ./
 RUN yarn install
 RUN yarn build
 ENTRYPOINT ["yarn"]