From a3c4b44c2d12d93eaa92b3a7825596a6f523422c Mon Sep 17 00:00:00 2001
From: Ben Hauser <ben@hauser.id>
Date: Fri, 18 Dec 2020 00:22:16 +0100
Subject: [PATCH] feat: flatten contracts

---
 brownie/network/contract.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/brownie/network/contract.py b/brownie/network/contract.py
index 3af406170..b754b51ce 100644
--- a/brownie/network/contract.py
+++ b/brownie/network/contract.py
@@ -127,6 +127,32 @@ def decode_input(self, calldata: Union[str, bytes]) -> Tuple[str, Any]:
 
         return function_sig, input_args
 
+    def flatten(self) -> str:
+        """
+        Return a single, deployable source code for this contract.
+        """
+        language = self._build["language"]
+        if language == "Vyper":
+            return self._build["source"]
+        elif language == "Solidity":
+            flattened_source = ""
+
+            for name in self._build["dependencies"]:
+                build_json = self._project._build.get(name)
+                offset = slice(*build_json["offset"])
+                source = build_json["source"][offset]
+                flattened_source = f"{flattened_source}\n\n{source}"
+
+            build_json = self._build
+            version = build_json["compiler"]["version"]
+            offset = slice(*build_json["offset"])
+            source = build_json["source"][offset]
+            flattened_source = f"pragma solidity {version};{flattened_source}\n\n{source}\n"
+
+            return flattened_source
+        else:
+            raise TypeError(f"Unsupported language for flattening: {language}")
+
 
 class ContractContainer(_ContractBase):