From 00b05e06e0252a60b29e02beb650665a7683503d Mon Sep 17 00:00:00 2001
From: Neeraj Kashyap <nkashy1@gmail.com>
Date: Mon, 5 Sep 2022 17:17:29 -0700
Subject: [PATCH] Added script to create pool on TErminus contracts

Handles ERC20 allowances as well as setting of pool URI
---
 scripts/create_pool.py | 70 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 70 insertions(+)
 create mode 100644 scripts/create_pool.py

diff --git a/scripts/create_pool.py b/scripts/create_pool.py
new file mode 100644
index 0000000..9413a6f
--- /dev/null
+++ b/scripts/create_pool.py
@@ -0,0 +1,70 @@
+"""
+Create a Terminus pool and set its metadata
+"""
+
+import argparse
+
+from brownie import network
+
+from dao import TerminusFacet, ERC20Facet
+
+parser = argparse.ArgumentParser("Create a Terminus pool and set its metadata")
+TerminusFacet.add_default_arguments(parser, transact=True)
+parser.add_argument(
+    "--capacity",
+    type=int,
+    default=2**256 - 1,
+    help="Maximum number of tokens in pool (default: 2^256 - 1)",
+)
+parser.add_argument(
+    "--transferable",
+    required=True,
+    type=TerminusFacet.boolean_argument_type,
+    help="Should pool be transferable? ('y' or 'yes' or 1 for yes, 'n' or 'no' or 0 for no)",
+)
+parser.add_argument(
+    "--burnable",
+    required=True,
+    type=TerminusFacet.boolean_argument_type,
+    help="Should pool be burnable? ('y' or 'yes' or 1 for yes, 'n' or 'no' or 0 for no)",
+)
+parser.add_argument(
+    "--uri", default=None, help="Set this if you would like to set a pool metadata URI"
+)
+
+
+args = parser.parse_args()
+
+network.connect(args.network)
+transaction_config = TerminusFacet.get_transaction_config(args)
+sender = transaction_config["from"]
+
+terminus_contract = TerminusFacet.TerminusFacet(args.address)
+payment_token_address = terminus_contract.payment_token()
+pool_price = terminus_contract.pool_base_price()
+
+payment_token = ERC20Facet.ERC20Facet(payment_token_address)
+
+sender_balance = payment_token.balance_of(sender.address)
+assert (
+    sender_balance >= pool_price
+), f"Insufficient balance on payment token ({payment_token_address}). Balance: {sender_balance}, required: {pool_price}"
+
+sender_allowance = payment_token.allowance(sender.address, args.address)
+if sender_allowance < pool_price:
+    print(f"Approve purchase with payment token ({payment_token_address}):")
+    payment_token.approve(args.address, pool_price, transaction_config)
+
+print("Creating pool")
+receipt = terminus_contract.create_pool_v1(
+    args.capacity, args.transferable, args.burnable, transaction_config
+)
+
+pool_id = terminus_contract.total_pools(block_number=receipt.block_number)
+print(f"Pool created on Terminus contract ({args.address}). Pool ID: {pool_id}")
+
+if args.uri:
+    print(
+        f"Setting pool URI. Terminus: {args.address}, Pool ID: {pool_id}, URI: {args.uri}"
+    )
+    terminus_contract.set_uri(pool_id, args.uri, transaction_config)