Host-customized fork of https://github.com/tecnovert/basicswap/
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

351 lines
11 KiB

# -*- coding: utf-8 -*-
# Copyright (c) 2019-2020 tecnovert
# Distributed under the MIT software license, see the accompanying
# file LICENSE or http://www.opensource.org/licenses/mit-license.php.
import struct
import time
import sqlalchemy as sa
from sqlalchemy.ext.declarative import declarative_base
from enum import IntEnum, auto
CURRENT_DB_VERSION = 5
Base = declarative_base()
class DBKVInt(Base):
__tablename__ = 'kv_int'
5 years ago
key = sa.Column(sa.String, primary_key=True)
value = sa.Column(sa.Integer)
class DBKVString(Base):
__tablename__ = 'kv_string'
5 years ago
key = sa.Column(sa.String, primary_key=True)
value = sa.Column(sa.String)
class Offer(Base):
__tablename__ = 'offers'
offer_id = sa.Column(sa.LargeBinary, primary_key=True)
active_ind = sa.Column(sa.Integer)
coin_from = sa.Column(sa.Integer)
coin_to = sa.Column(sa.Integer)
amount_from = sa.Column(sa.BigInteger)
rate = sa.Column(sa.BigInteger)
min_bid_amount = sa.Column(sa.BigInteger)
time_valid = sa.Column(sa.BigInteger)
lock_type = sa.Column(sa.Integer)
lock_value = sa.Column(sa.Integer)
swap_type = sa.Column(sa.Integer)
proof_address = sa.Column(sa.String)
proof_signature = sa.Column(sa.LargeBinary)
pkhash_seller = sa.Column(sa.LargeBinary)
secret_hash = sa.Column(sa.LargeBinary)
addr_from = sa.Column(sa.String)
created_at = sa.Column(sa.BigInteger)
expire_at = sa.Column(sa.BigInteger)
was_sent = sa.Column(sa.Boolean)
4 years ago
from_feerate = sa.Column(sa.BigInteger)
to_feerate = sa.Column(sa.BigInteger)
# Local fields
auto_accept_bids = sa.Column(sa.Boolean)
withdraw_to_addr = sa.Column(sa.String) # Address to spend lock tx to - address from wallet if empty TODO
state = sa.Column(sa.Integer)
states = sa.Column(sa.LargeBinary) # Packed states and times
def setState(self, new_state):
now = int(time.time())
self.state = new_state
if self.states is None:
self.states = struct.pack('<iq', new_state, now)
else:
self.states += struct.pack('<iq', new_state, now)
class Bid(Base):
__tablename__ = 'bids'
bid_id = sa.Column(sa.LargeBinary, primary_key=True)
offer_id = sa.Column(sa.LargeBinary, sa.ForeignKey('offers.offer_id'))
active_ind = sa.Column(sa.Integer)
was_sent = sa.Column(sa.Boolean)
was_received = sa.Column(sa.Boolean)
contract_count = sa.Column(sa.Integer)
created_at = sa.Column(sa.BigInteger)
expire_at = sa.Column(sa.BigInteger)
bid_addr = sa.Column(sa.String)
proof_address = sa.Column(sa.String)
withdraw_to_addr = sa.Column(sa.String) # Address to spend lock tx to - address from wallet if empty TODO
recovered_secret = sa.Column(sa.LargeBinary)
amount_to = sa.Column(sa.BigInteger) # amount * offer.rate
pkhash_buyer = sa.Column(sa.LargeBinary)
amount = sa.Column(sa.BigInteger)
accept_msg_id = sa.Column(sa.LargeBinary)
pkhash_seller = sa.Column(sa.LargeBinary)
initiate_txn_redeem = sa.Column(sa.LargeBinary)
initiate_txn_refund = sa.Column(sa.LargeBinary)
participate_txn_redeem = sa.Column(sa.LargeBinary)
participate_txn_refund = sa.Column(sa.LargeBinary)
in_progress = sa.Column(sa.Integer)
state = sa.Column(sa.Integer)
state_time = sa.Column(sa.BigInteger) # Timestamp of last state change
states = sa.Column(sa.LargeBinary) # Packed states and times
state_note = sa.Column(sa.String)
debug_ind = sa.Column(sa.Integer)
initiate_tx = None
participate_tx = None
xmr_a_lock_tx = None
xmr_a_lock_spend_tx = None
xmr_b_lock_tx = None # TODO: Can't move to txns due to error: Exception UPDATE statement on table expected to update 1 row(s); 0 were matched
txns = {}
def getITxState(self):
if self.initiate_tx is None:
return None
return self.initiate_tx.state
def setITxState(self, new_state):
if self.initiate_tx is not None:
self.initiate_tx.setState(new_state)
def getPTxState(self):
if self.participate_tx is None:
return None
return self.participate_tx.state
def setPTxState(self, new_state):
if self.participate_tx is not None:
self.participate_tx.setState(new_state)
4 years ago
def setState(self, new_state, state_note=None):
now = int(time.time())
self.state = new_state
self.state_time = now
4 years ago
if state_note is not None:
self.state_note = state_note
if self.states is None:
self.states = struct.pack('<iq', new_state, now)
else:
self.states += struct.pack('<iq', new_state, now)
class SwapTx(Base):
__tablename__ = 'transactions'
bid_id = sa.Column(sa.LargeBinary, sa.ForeignKey('bids.bid_id'))
tx_type = sa.Column(sa.Integer) # TxTypes
__table_args__ = (
sa.PrimaryKeyConstraint('bid_id', 'tx_type'),
{},
)
txid = sa.Column(sa.LargeBinary)
vout = sa.Column(sa.Integer)
script = sa.Column(sa.LargeBinary)
tx_fee = sa.Column(sa.BigInteger)
chain_height = sa.Column(sa.Integer)
conf = sa.Column(sa.Integer)
spend_txid = sa.Column(sa.LargeBinary)
spend_n = sa.Column(sa.Integer)
state = sa.Column(sa.Integer)
states = sa.Column(sa.LargeBinary) # Packed states and times
def setState(self, new_state):
self.state = new_state
self.states = (self.states if self.states is not None else bytes()) + struct.pack('<iq', new_state, int(time.time()))
class PooledAddress(Base):
__tablename__ = 'addresspool'
addr_id = sa.Column(sa.Integer, primary_key=True, autoincrement=True)
coin_type = sa.Column(sa.Integer)
addr = sa.Column(sa.String)
bid_id = sa.Column(sa.LargeBinary)
tx_type = sa.Column(sa.Integer)
class SentOffer(Base):
__tablename__ = 'sentoffers'
offer_id = sa.Column(sa.LargeBinary, primary_key=True)
class SmsgAddress(Base):
__tablename__ = 'smsgaddresses'
5 years ago
addr_id = sa.Column(sa.Integer, primary_key=True, autoincrement=True)
addr = sa.Column(sa.String)
use_type = sa.Column(sa.Integer)
class EventQueue(Base):
__tablename__ = 'eventqueue'
5 years ago
event_id = sa.Column(sa.Integer, primary_key=True, autoincrement=True)
active_ind = sa.Column(sa.Integer)
created_at = sa.Column(sa.BigInteger)
trigger_at = sa.Column(sa.BigInteger)
linked_id = sa.Column(sa.LargeBinary)
event_type = sa.Column(sa.Integer)
5 years ago
event_data = sa.Column(sa.LargeBinary)
4 years ago
class EventLog(Base):
__tablename__ = 'eventlog'
event_id = sa.Column(sa.Integer, primary_key=True, autoincrement=True)
active_ind = sa.Column(sa.Integer)
created_at = sa.Column(sa.BigInteger)
linked_type = sa.Column(sa.Integer)
linked_id = sa.Column(sa.LargeBinary)
event_type = sa.Column(sa.Integer)
event_msg = sa.Column(sa.String)
__table_args__ = (sa.Index('main_index', 'linked_type', 'linked_id'), )
4 years ago
class XmrOffer(Base):
__tablename__ = 'xmr_offers'
swap_id = sa.Column(sa.Integer, primary_key=True, autoincrement=True)
offer_id = sa.Column(sa.LargeBinary, sa.ForeignKey('offers.offer_id'))
a_fee_rate = sa.Column(sa.BigInteger)
b_fee_rate = sa.Column(sa.BigInteger)
lock_time_1 = sa.Column(sa.Integer) # Delay before the chain a lock refund tx can be mined
lock_time_2 = sa.Column(sa.Integer) # Delay before the follower can spend from the chain a lock refund tx
class XmrSwap(Base):
__tablename__ = 'xmr_swaps'
swap_id = sa.Column(sa.Integer, primary_key=True, autoincrement=True)
bid_id = sa.Column(sa.LargeBinary, sa.ForeignKey('bids.bid_id'))
bid_msg_id2 = sa.Column(sa.LargeBinary)
bid_msg_id3 = sa.Column(sa.LargeBinary)
bid_accept_msg_id = sa.Column(sa.LargeBinary)
bid_accept_msg_id2 = sa.Column(sa.LargeBinary)
bid_accept_msg_id3 = sa.Column(sa.LargeBinary)
coin_a_lock_tx_sigs_l_msg_id = sa.Column(sa.LargeBinary) # MSG3L F -> L
coin_a_lock_refund_spend_tx_msg_id = sa.Column(sa.LargeBinary) # MSG4F L -> F
4 years ago
contract_count = sa.Column(sa.Integer)
sv = sa.Column(sa.LargeBinary) # Secret value
4 years ago
sh = sa.Column(sa.LargeBinary) # Secret hash
dest_af = sa.Column(sa.LargeBinary) # Destination for coin A amount to follower when swap completes successfully
4 years ago
pkal = sa.Column(sa.LargeBinary)
pkarl = sa.Column(sa.LargeBinary)
pkasl = sa.Column(sa.LargeBinary)
4 years ago
pkaf = sa.Column(sa.LargeBinary)
pkarf = sa.Column(sa.LargeBinary)
pkasf = sa.Column(sa.LargeBinary)
4 years ago
vkbvl = sa.Column(sa.LargeBinary)
vkbsl = sa.Column(sa.LargeBinary)
pkbvl = sa.Column(sa.LargeBinary)
pkbsl = sa.Column(sa.LargeBinary)
vkbvf = sa.Column(sa.LargeBinary)
vkbsf = sa.Column(sa.LargeBinary)
pkbvf = sa.Column(sa.LargeBinary)
pkbsf = sa.Column(sa.LargeBinary)
kbsl_dleag = sa.Column(sa.LargeBinary)
kbsf_dleag = sa.Column(sa.LargeBinary)
vkbv = sa.Column(sa.LargeBinary)
pkbv = sa.Column(sa.LargeBinary)
pkbs = sa.Column(sa.LargeBinary)
4 years ago
a_lock_tx = sa.Column(sa.LargeBinary)
a_lock_tx_script = sa.Column(sa.LargeBinary)
a_lock_tx_id = sa.Column(sa.LargeBinary)
a_lock_tx_vout = sa.Column(sa.Integer)
4 years ago
a_lock_refund_tx = sa.Column(sa.LargeBinary)
a_lock_refund_tx_script = sa.Column(sa.LargeBinary)
a_lock_refund_tx_id = sa.Column(sa.LargeBinary)
4 years ago
a_swap_refund_value = sa.Column(sa.BigInteger)
al_lock_refund_tx_sig = sa.Column(sa.LargeBinary)
af_lock_refund_tx_sig = sa.Column(sa.LargeBinary)
4 years ago
a_lock_refund_spend_tx = sa.Column(sa.LargeBinary)
a_lock_refund_spend_tx_id = sa.Column(sa.LargeBinary)
4 years ago
af_lock_refund_spend_tx_esig = sa.Column(sa.LargeBinary)
af_lock_refund_spend_tx_sig = sa.Column(sa.LargeBinary)
a_lock_spend_tx = sa.Column(sa.LargeBinary)
a_lock_spend_tx_id = sa.Column(sa.LargeBinary)
al_lock_spend_tx_esig = sa.Column(sa.LargeBinary)
a_lock_refund_swipe_tx = sa.Column(sa.LargeBinary) # Follower spends script coin lock refund tx
b_lock_tx_id = sa.Column(sa.LargeBinary)
start_chain_a_height = sa.Column(sa.Integer) # Height of script chain before the swap
b_restore_height = sa.Column(sa.Integer) # Height of scriptless chain before the swap
4 years ago
class XmrSplitData(Base):
__tablename__ = 'xmr_split_data'
record_id = sa.Column(sa.Integer, primary_key=True, autoincrement=True)
bid_id = sa.Column(sa.LargeBinary)
msg_type = sa.Column(sa.Integer)
msg_sequence = sa.Column(sa.Integer)
dleag = sa.Column(sa.LargeBinary)
created_at = sa.Column(sa.BigInteger)
class RevokedMessage(Base):
__tablename__ = 'revoked_messages'
record_id = sa.Column(sa.Integer, primary_key=True, autoincrement=True)
active_ind = sa.Column(sa.Integer)
msg_id = sa.Column(sa.LargeBinary)
created_at = sa.Column(sa.BigInteger)
expires_at = sa.Column(sa.BigInteger)
class TableTypes(IntEnum):
OFFER = auto()
BID = auto()