From 41868dec515f150416623308375c5b064f593362 Mon Sep 17 00:00:00 2001
From: Csongor Kiss <ckiss@jumptrading.com>
Date: Thu, 1 Dec 2022 14:40:05 +0000
Subject: [PATCH] ethereum: Add make target to flatten contracts

Running `make flattened` will produce flattened contracts into the
`flattened` directory
---
 ethereum/.dockerignore |  1 +
 ethereum/.gitignore    |  1 +
 ethereum/Makefile      | 27 ++++++++++++++++++++++++++-
 3 files changed, 28 insertions(+), 1 deletion(-)

diff --git a/ethereum/.dockerignore b/ethereum/.dockerignore
index 3c3629e64..7bc6cc20a 100644
--- a/ethereum/.dockerignore
+++ b/ethereum/.dockerignore
@@ -1 +1,2 @@
 node_modules
+flattened
diff --git a/ethereum/.gitignore b/ethereum/.gitignore
index 5674d39c8..35a2b8795 100644
--- a/ethereum/.gitignore
+++ b/ethereum/.gitignore
@@ -1,3 +1,4 @@
 ganache.log
 lib/*
 !lib/README.md
+flattened
diff --git a/ethereum/Makefile b/ethereum/Makefile
index bd624cf28..b4c417c00 100644
--- a/ethereum/Makefile
+++ b/ethereum/Makefile
@@ -1,5 +1,20 @@
 SOURCE_FILES:=$(shell find contracts -name "*.sol")
 
+# List of files to flatten. These are typically the 'leaves' of the
+# dependendency tree, i.e. the contracts that actually get deployed on-chain.
+# Proxies and implementations.
+FLATTEN_FILES:= contracts/Implementation.sol \
+                contracts/Wormhole.sol \
+                contracts/bridge/BridgeImplementation.sol \
+                contracts/bridge/TokenBridge.sol \
+                contracts/bridge/token/TokenImplementation.sol \
+                contracts/bridge/token/Token.sol \
+                contracts/nft/NFTBridgeImplementation.sol \
+                contracts/nft/NFTBridgeEntrypoint.sol \
+                contracts/nft/token/NFT.sol \
+                contracts/nft/token/NFTImplementation.sol \
+                contracts/nft/token/NFT.sol
+
 .PHONY: dependencies test clean all
 
 all: build
@@ -29,6 +44,16 @@ build: node_modules ${SOURCE_FILES}
 	touch -m build
 	npm run build
 
+flattened/%.sol: contracts/%.sol node_modules
+	@mkdir -p $(dir $@)
+    # We remove the license (SPDX) lines and ABIEncoderV2 lines because they
+    # break compilation in the flattened file if there are multiple conflicting
+    # declarations.
+	npx truffle-flattener $< | grep -v SPDX | grep -v ABIEncoderV2 > $@
+
+.PHONY: flattened
+flattened: $(patsubst contracts/%, flattened/%, $(FLATTEN_FILES))
+
 .env: .env.test
 	cp $< $@
 
@@ -54,4 +79,4 @@ test-forge: dependencies
 	forge test
 
 clean:
-	rm -rf ganache.log .env node_modules build
+	rm -rf ganache.log .env node_modules build flattened