104 lines
3.6 KiB
Python
104 lines
3.6 KiB
Python
import zXcat
|
|
import bXcat
|
|
from xcat import *
|
|
|
|
htlcTrade = Trade()
|
|
print("Starting test of xcat...")
|
|
|
|
def get_initiator_addresses():
|
|
return {'bitcoin': 'myfFr5twPYNwgeXyjCmGcrzXtCmfmWXKYp', 'zcash': 'tmFRXyju7ANM7A9mg75ZjyhFW1UJEhUPwfQ'}
|
|
|
|
def get_fulfiller_addresses():
|
|
return {'bitcoin': 'mrQzUGU1dwsWRx5gsKKSDPNtrsP65vCA3Z', 'zcash': 'tmTjZSg4pX2Us6V5HttiwFZwj464fD2ZgpY'}
|
|
|
|
def initiate(trade):
|
|
# Get amounts
|
|
amounts = {"sell": {"currency": "bitcoin", "amount": "0.5"}, "buy": {"currency": "zcash", "amount": "1.12"}}
|
|
sell = amounts['sell']
|
|
buy = amounts['buy']
|
|
sell_currency = sell['currency']
|
|
buy_currency = buy['currency']
|
|
# Get addresses
|
|
init_addrs = get_initiator_addresses()
|
|
sell['initiator'] = init_addrs[sell_currency]
|
|
buy['initiator'] = init_addrs[buy_currency]
|
|
fulfill_addrs = get_fulfiller_addresses()
|
|
sell['fulfiller'] = fulfill_addrs[sell_currency]
|
|
buy['fulfiller'] = fulfill_addrs[buy_currency]
|
|
# initializing contract classes with addresses, currencies, and amounts
|
|
trade.sellContract = Contract(sell)
|
|
trade.buyContract = Contract(buy)
|
|
print(trade.sellContract.__dict__)
|
|
print(trade.buyContract.__dict__)
|
|
|
|
secret = generate_password()
|
|
print("Generating secret to lock funds:", secret)
|
|
save_secret(secret)
|
|
# TODO: Implement locktimes and mock block passage of time
|
|
sell_locktime = 2
|
|
buy_locktime = 4 # Must be more than first tx
|
|
|
|
create_sell_p2sh(trade, secret, sell_locktime)
|
|
txid = fund_sell_contract(trade)
|
|
print("Sent")
|
|
create_buy_p2sh(trade, secret, buy_locktime)
|
|
|
|
def fulfill(trade):
|
|
buy = trade.buyContract
|
|
sell = trade.sellContract
|
|
buy_p2sh_balance = check_p2sh(buy.currency, buy.p2sh)
|
|
sell_p2sh_balance = check_p2sh(sell.currency, sell.p2sh)
|
|
|
|
if buy_p2sh_balance == 0:
|
|
print("Buy amt:", buy.amount)
|
|
txid = fund_buy_contract(trade)
|
|
print("Fund tx txid:", txid)
|
|
else:
|
|
raise ValueError("Sell p2sh not funded, buyer cannot redeem")
|
|
|
|
def redeem_one(trade):
|
|
buy = trade.buyContract
|
|
if trade.sellContract.get_status() == 'redeemed':
|
|
raise RuntimeError("Sell contract status was already redeemed before seller could redeem buyer's tx")
|
|
else:
|
|
secret = get_secret()
|
|
print("GETTING SECRET IN TEST:", secret)
|
|
tx_type, txid = redeem_p2sh(trade.buyContract, secret)
|
|
print("\nTX Type", tx_type)
|
|
setattr(trade.buyContract, tx_type, txid)
|
|
save(trade)
|
|
print("You have redeemed {0} {1}!".format(buy.amount, buy.currency))
|
|
|
|
def redeem_two(trade):
|
|
if trade.sellContract.get_status() == 'redeemed':
|
|
raise RuntimeError("Sell contract was redeemed before buyer could retrieve funds")
|
|
elif trade.buyContract.get_status() == 'refunded':
|
|
print("buyContract was refunded to buyer")
|
|
else:
|
|
# Buy contract is where seller disclosed secret in redeeming
|
|
if trade.buyContract.currency == 'bitcoin':
|
|
secret = bXcat.parse_secret(trade.buyContract.redeem_tx)
|
|
else:
|
|
secret = zXcat.parse_secret(trade.buyContract.redeem_tx)
|
|
print("Found secret in seller's redeem tx", secret)
|
|
redeem_tx = redeem_p2sh(trade.sellContract, secret)
|
|
setattr(trade.sellContract, 'redeem_tx', redeem_tx)
|
|
save(trade)
|
|
|
|
def generate_blocks(num):
|
|
bXcat.generate(num)
|
|
zXcat.generate(num)
|
|
|
|
initiate(htlcTrade)
|
|
fulfill(htlcTrade)
|
|
|
|
generate_blocks(6)
|
|
|
|
redeem_one(htlcTrade)
|
|
redeem_two(htlcTrade)
|
|
|
|
# addr = CBitcoinAddress('tmFRXyju7ANM7A9mg75ZjyhFW1UJEhUPwfQ')
|
|
# print(addr)
|
|
# # print(b2x('tmFRXyju7ANM7A9mg75ZjyhFW1UJEhUPwfQ'))
|
|
# print(b2x(addr))
|