o @Bi5d@sddlZddlZddlmZddlZddlZddlZddlZddl Zddl Zddl Z ddl m Z ddlmZddlmZmZddlZddlmZmZmZGdd d eZGd d d eZdS) N)pformat)ApplicationError)ApplicationSession)unpack_uint256 pack_uint256)hlsign_eip712_datarecover_eip712_signerc@s$eZdZddZddZddZdS) Transactionc Cs4||_||_||_||_||_||_||_||_dSNchanneldelegatepubkeykey_id channel_seqamountbalance signature) selfrrrrrrrrrS/var/www/html/Trade-python/venv/lib/python3.10/site-packages/autobahn/xbr/_buyer.py__init__4s zTransaction.__init__c Cs*|j|j|j|j|j|j|j|jd}|S)Nr r )rresrrrmarshal>s zTransaction.marshalcCs t|Sr )rrrrrr__str__Ks zTransaction.__str__N)__name__ __module__ __qualname__rrrrrrrr 2s r c@seZdZdZdZddZddZddZd d Zdd d Z dd dZ ddZ ddZ ddZ d ddZddZddZddZdS)! SimpleBuyerz Simple XBR buyer component. This component can be used by a XBR buyer delegate to handle the automated buying of data encryption keys from the XBR market maker. NcCst|tkr t|dksJd|t|tkrt|dks&Jd|t|tkr0|dks2Jt|_||_||_ t j ||_ d|_|j j|_d|_tjj |_||_d|_d|_d|_i|_d|_d|_d|_i|_g|_ dS) a :param market_maker_adr: :type market_maker_adr: :param buyer_key: Consumer delegate (buyer) private Ethereum key. :type buyer_key: bytes :param max_price: Maximum price we are willing to buy per key. :type max_price: int z0market_maker_adr must be bytes[20], but got "{}" z.buyer delegate must be bytes[32], but got "{}"rNFT)!typebyteslenformatinttxaio make_loggerlog_market_maker_adr _pkey_raweth_keyskeys PrivateKey_pkey_acct public_keyto_canonical_address_addr_caddrnaclpublicgenerate _receive_key _max_price_channel_balance_seq_keys_session_running_auto_close_channel_transaction_idx _transactions)rmarket_maker_adr buyer_key max_pricerrrrVs*& &  zSimpleBuyer.__init__cs,t|tsJt|tksJ|jrJ||_d|_|jjdt|j t |j j dddz)t|jtkrAt|jdksCJ|d|jIdH|_|d|jd IdH}Wn||d |_t|jtkrvt|j|_|d |_|jjd td t |jd |j|jd|jS)a Start buying keys to decrypt XBR data by calling ``unwrap()``. :param session: WAMP session over which to communicate with the XBR market maker. :type session: :class:`autobahn.wamp.protocol.ApplicationSession` :param consumer_id: XBR consumer ID. :type consumer_id: str :return: Current remaining balance in payment channel. :rtype: int TzSStart buying from consumer delegate address {address} (public key 0x{public_key}..)N )addressr3r"z*xbr.marketmaker.get_active_payment_channel+xbr.marketmaker.get_payment_channel_balancer remainingseqztDelegate has current payment channel address {payment_channel_adr} (remaining balance {remaining} at sequence {seq})0x)payment_channel_adrrKrL) isinstancerr$strrAr@r+inforr6binasciib2a_hexr1r3decoder5r%r&callr<leaver=rr>)rsession consumer_idpayment_balancerrrstarts4      zSimpleBuyer.startcs|jsJd|_dS)z# Stop buying keys. FN)rArrrrstops  zSimpleBuyer.stopcs4|jr |js J|jd|jdIdH}|S)a Return current balance of payment channel: * ``amount``: The initial amount with which the payment channel was opened. * ``remaining``: The remaining amount of XBR in the payment channel that can be spent. * ``inflight``: The amount of XBR allocated to buy transactions that are currently processed. :return: Current payment balance. :rtype: dict rJrN)r@ is_attachedrUr<)rrYrrrrs zSimpleBuyer.balancecsX|jr |js Jtd}|jd||j||IdH}|d|d|dd}|S)z :param amount: :type amount: :param details: :type details: :return: :rtype: @z$xbr.marketmaker.open_payment_channelNrrKinflight)rrKr^)r@r\osurandomrUr5)r buyer_addrrdetailsrpayment_channelrrrr open_channels  zSimpleBuyer.open_channelcsdS)z[ Requests to close the currently active payment channel. :return: Nr)rrbrrr close_channelszSimpleBuyer.close_channelc! sjt|tkr t|dksJt|tkr|dvsJt|tks#Jt|jd}||jvr?|jjdtt j |ddnj|jjdtt j |ddd|j|<|j d |Id H}t |d }|jjd tt j |dtt|d d||jkrtdd|jjt j |dt|d t|jd |j|}|dkr|jrd }|} | r| d}|r|} |j} |j} d} t|j| | | | }|jjdt| | t| d | d|j d| | t| | |Id Htdd|jjt j |dt| t| d tdd|jjt j |dt|jd t|d |jjj t!j"j#d}|jd}d}t|j||||d}|$||j%||||||z|j d|j%||||t|t|| Id H}Wn7tyw}z |j&dkrq|'|d }~wt(y}z|j&d|j)||j|<|d }~ww|d}|d }t |d!}t |d"}t |d#}t*|||d|}||j+kr|jj,d$|jjtt|tt|j+d%td&d'|jj|jd|krtd(d)|jj|j|d |j||krtd(d*|jj|j||d"||_||_|-||j+|||||||d+}t!j./|j}z|0|}Wnt!j1j2yW}z||j|<td,d-|jj|d }~wwt!j34||j|<|5}|jjd.|jjtd/d0d1tt j |dttt|d d2d0d1tt|d3tt|d tt|d tt||d4 d}|j|dur|d5r|jjd6|jjtt j |dd7|d7}t67d8Id H|j|dust8|j|t(r|j|}|z |j|0|}Wnt!j1j2y}z td,d9|jj|d }~wwzt9:|} W| St9j;j<y4}z td:d;|jj|d }~ww)Key {key_id} already in key store (or currently being bought).)r%)rz1Key {key_id} not yet in key store - buying key ..Fzxbr.marketmaker.get_quoteNpricez-Key {key_id} has current price quote {amount}lNZo)rrzxbr.error.max_price_exceededzR{}.unwrap() - key {} needed cannot be bought: price {} exceeds maximum price of {}rTzauto-closing payment channel {close_adr} [close_seq={close_seq}, close_balance={close_balance}, close_is_final={close_is_final}]) close_adr close_seq close_balanceclose_is_finalzxbr.marketmaker.close_channelzxbr.error.channel_closedzw{}.unwrap() - key {} cannot be bought: payment channel 0x{} ran empty and we initiated close at remaining balance of {}zxbr.error.insufficient_balancez_{}.unwrap() - key {} cannot be bought: insufficient balance {} in payment channel for amount {})encoderr)is_finalzxbr.marketmaker.buy_keyz8Encountered error while calling market maker to buy key!rr amount_paidrKr^zv{klass}.unwrap()::XBRSIG[8/8] - EIP712 signature invalid: signer_address={signer_address}, delegate_adr={delegate_adr})klasssigner_address delegate_adrzxbr.error.invalid_signaturezQ{}.unwrap()::XBRSIG[8/8] - EIP712 signature invalid or not signed by market makerzxbr.error.invalid_transactionz^{}.buy_key(): invalid transaction (channel sequence number mismatch - expected {}, but got {})z_{}.buy_key(): invalid transaction (channel remaining amount mismatch - expected {}, but got {}) sealed_keyzxbr.error.decryption_failedz6{}.unwrap() - could not unseal data encryption key: {}z{klass}.unwrap() - {tx_type} key {key_id} bought for {amount_paid} [payment_channel={payment_channel}, remaining={remaining}, inflight={inflight}, buyer_pubkey={buyer_pubkey}, transactions={transactions}]z XBR BUY magenta)colorz XBRrc) rptx_typerrorcrKr^ buyer_pubkey transactionsdzG{klass}.unwrap() - waiting for key "{key_id}" currently being bought ..)rprg?z1{}.unwrap() - failed to unwrap encrypted data: {}z xbr.error.deserialization_failedz;{}.unwrap() - failed to deserialize application payload: {})=r$r%r&rPr<r?r+rQruuidUUIDr@rUrr(r;rr' __class__rr=rBpast_transactionsr>r r-rRrSrTrr:r3encoder7encoding RawEncoder_save_transaction_phase1r5errorr[ Exceptionfailurer r,warn_save_transaction_phase2r8 SealedBoxdecrypt exceptions CryptoErrorsecret SecretBoxcount_transactionsr)sleeprOcbor2loadsdecoderCBORDecodeError)!rr serializer ciphertext channel_adrquoterrlast_txtxnsrirjrkrlrrwrrnreceiptemarketmaker_signaturemarketmaker_channel_seqmarketmaker_amount_paidmarketmaker_remainingmarketmaker_inflightrqrs unseal_boxkeytransactions_count log_countermessagepayloadrrrunwrap s<     ,                             zSimpleBuyer.unwrapc Cs\||jvrtdt|t||||||||} t|j} |j| dg| |j|<dS) :param channel_adr: :param delegate_adr: :param buyer_pubkey: :param key_id: :param channel_seq: :param amount: :param balance: :param signature: :return: z;save_transaction_phase1: duplicate transaction for key 0x{}N) rC RuntimeErrorr'rRrSr r&rDappend) rrrrrwrrrrrtx1key_idxrrrrs  z$SimpleBuyer._save_transaction_phase1c Cs||jvrtdt||j|} |j| dr%tdt||j| d} t||||||||} | j| jks?J| j| jksGJ| j | j ksOJ| j | j ksWJ| j | j ks_J| j | j ksgJ| j | j ksoJ| |j| d<dS)rz;save_transaction_phase2: transaction for key 0x{} not foundrz;save_transaction_phase2: duplicate transaction for key 0x{}rN)rCrr'rRrSrDr rrrrrrr) rrrrrwrrrrrrrtx2rrrrs"  z$SimpleBuyer._save_transaction_phase2TrcCst|tksJt|tkr|dksJd}g}||krDt|j|kr=|j| }|r3|dr<|dr<|||d7}n |S||ks|S)zQ :param filter_complete: :param limit: :return: rr)r$boolr(r&rDr)rfilter_completelimitnrtxrrrr}s  zSimpleBuyer.past_transactionscCsHddd}|jD]\}}|r|r|dd7<q|dd7<q|S)z :return: r)completependingrrr)rD)rrrrrrrr2szSimpleBuyer.count_transactionscCs |j|d}|r|j|SdS)2 :param key_id: :return: NrCgetrD)rridxrrrget_transactionBs zSimpleBuyer.get_transactioncCs,|j|d}|r|j|\}}|o|SdS)rNFr)rrrrrrrr is_completeLs zSimpleBuyer.is_completer )Tr)rrr __doc__r+rrZr[rrdrerrrr}rrrrrrrr!Os"D0  X & r!)rzrRpprintrr_r nacl.secretr7 nacl.utilsnacl.exceptions nacl.publicr)autobahn.wamp.exceptionrautobahn.wamp.protocolr_utilrrr.rr r objectr r!rrrrs"