o i m@sddlZddlZddlmZddlZddlZddlZddlZddl Zddl Zddl Z ddl m Z ddlmZddlmZmZddlZddlmZmZdd lmZmZGd d d eZGd d d eZdS)N)pformat)ApplicationError)ApplicationSession)unpack_uint256 pack_uint256)hlhlval)sign_eip712_channel_closerecover_eip712_channel_closec@s$eZdZddZddZddZdS) Transactionc Cs4||_||_||_||_||_||_||_||_dSNchanneldelegatepubkeykey_id channel_seqamountbalance signature) selfrrrrrrrrrYC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\autobahn/xbr/_buyer.py__init__3s zTransaction.__init__c Cs*|j|j|j|j|j|j|j|jd}|S)Nrr)rresrrrmarshal=s zTransaction.marshalcCs t|Sr)rrrrrr__str__Js zTransaction.__str__N)__name__ __module__ __qualname__rrrrrrrr 1s 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|_||_d|_ ||_ t j ||_d|_|jj|_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)"typebyteslenformatinttxaioZ make_loggerlog_market_maker_adr _xbrmm_config _pkey_raweth_keyskeysZ PrivateKey_pkeyZ_acct public_keyZto_canonical_address_addr_caddrnaclpublicgenerate _receive_key _max_price_channel_balance_seq_keys_session_running_auto_close_channel_transaction_idx _transactions)rZmarket_maker_adrZ buyer_keyZ max_pricerrrrUs,& &  zSimpleBuyer.__init__cst|tsJt|tksJ|jrJ||_d|_|jjdt|j t |j j dddzR|dIdH|_t|jtkrJt|jdksLJ|d|jIdH|_|js^td |jd }t|tkrot|d ksqJtj|d |_|d |jjIdH}Wn|t|dtkrt|d|d<|ddkstd|d|_|d|_|jjdt|jt|jt|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 )addressr3zxbr.marketmaker.get_configr$z*xbr.marketmaker.get_active_payment_channelzno active payment channel found channel_oidr'+xbr.marketmaker.get_payment_channel_balance remainingrz1no off-chain balance remaining on payment channelseqzvOk, buyer delegate started [active payment channel {channel_oid} with remaining balance {remaining} at sequence {seq}])rFrJrK) isinstancerr&strr@r?r,debugr r5binasciib2a_hexr2r3decodecallr.r4r'r(r; ExceptionuuidUUIDZ _channel_oidZleaverr<r=infor )rsessionZ consumer_idrFpayment_balancerrrstartsB       zSimpleBuyer.startcs"|jsJd|_|jddS)z# Stop buying keys. FzOk, buyer delegate stopped.N)r@r,rVrrrrstops 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 rIrFN)r? is_attachedrRr;)rrXrrrrs 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_channelNrrJinflight)rrJr])r?r[osurandomrRr4)rZ buyer_addrrdetailsrpayment_channelrrrr open_channels  zSimpleBuyer.open_channelcsdS)z[ Requests to close the currently active payment channel. :return: Nr)rr`rrr close_channelszSimpleBuyer.close_channelc$ st|tkr t|dksJt|tkr|dvsJt|tks#J|jd}|jd}d}|jd}t|jddd }||jvrU|j j d t t j |d d np|j j d t t j |d d d|j|<|jd|Id H} t| d} |j j dt t j |d t t| dd| |jkrtdd|jjt j |d t| dt|jd|j| } | dkr3|jrd } |} | r| d} | r|j}|j}d}t|j|||||||| }|j j dt j |d |t|d|d|jd||||t|||| Id Htdd|jjt j |d |t|dtdd|jjt j |d t|jdt| d|jjj t!j"j#d}|jd}d}t|j||||||| | }|$||j%|||| | |z|jd |j%||||t| t| | Id H}Wn7ty}z |j&dkr|'|d }~wt(y}z|j &d!|j )||j|<|d }~ww|d"}|d#}t|d$}t|d%}t|d&}t*|||||||d| }||j+kr|j j,d'|jjt t-|.t t-|j+.d(td)d*|jj|jd|krtd+d,|jj|j|d#|j| |kr/td+d-|jj|j| |d%||_||_|/||j+|||||||d.}t!j01|j}z|2|}Wnt!j3j4ys}z||j|<td/d0|jj|d }~wwt!j56||j|<|7} |j j8d1|jjt d2d3d4t t j |d t tt|dd5d3d4t t-|d6.t t|dt t|dt t-|.| d7 d}!|j|dur|!d8r|j j d9|jjt t j |d d:|!d7}!t9:d;Id H|j|dust;|j|t(r |j|}|z |j|2|}"Wnt!j3j4y-}z td/d<|jj|d }~wwzt<=|"}#W|#Stj?yP}z td=d>|jj|d }~ww)?aO Decrypt XBR data. This functions will potentially make the buyer call the XBR market maker to buy data encryption keys from the XBR provider. :param key_id: ID of the data encryption used for decryption of application payload. :type key_id: bytes :param serializer: Application payload serializer. :type serializer: str :param ciphertext: Ciphertext of encrypted application payload to decrypt. :type ciphertext: bytes :return: Decrypted application payload. :rtype: object rG)Zcbor market_oidrFrverifying_chain_idverifying_contract_adrrNz>Key {key_id} already in key store (or currently being bought).rH)rz1Key {key_id} not yet in key store - buying key ..Fzxbr.marketmaker.get_quoteZpricez-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 {channel_oid} [close_seq={close_seq}, close_balance={close_balance}, close_is_final={close_is_final}])rF close_seq close_balanceclose_is_finalzxbr.marketmaker.close_channelzxbr.error.channel_closedzu{}.unwrap() - key {} cannot be bought: payment channel {} 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 {})encoderzxbr.marketmaker.buy_keyz8Encountered error while calling market maker to buy key!rr amount_paidrJr]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 Zmagenta)colorz XBRra) rlZtx_typerrkrarJr] buyer_pubkeyZ transactionsdzG{klass}.unwrap() - waiting for key "{key_id}" currently being bought ..)rlrg?z1{}.unwrap() - failed to unwrap encrypted data: {}z xbr.error.deserialization_failedz;{}.unwrap() - failed to deserialize application payload: {})@r&r'r(rMr;r.rOa2b_hexr>r,rNr rTrUr?rRrr*r:rr) __class__r r<rApast_transactionsr=r r/rr9r3encoder6encodingZ RawEncoder_save_transaction_phase1r4errorrZrSZfailurer r-warnrPrQ_save_transaction_phase2r7Z SealedBoxZdecrypt exceptionsZ CryptoErrorsecretZ SecretBoxcount_transactionsrVr+sleeprLcbor2loadsdecoderZCBORDecodeError)$rr serializerZ ciphertextrdrFZcurrent_block_numberrerfquoterrZlast_txZtxnsrgrhrirrqris_finalZreceipteZmarketmaker_signatureZmarketmaker_channel_seqZmarketmaker_amount_paidZmarketmaker_remainingZmarketmaker_inflightrmroZ unseal_boxkeyZtransactions_countZ log_countermessagepayloadrrrunwrapst        ,                             zSimpleBuyer.unwrapc Cs\||jvrtdt|t||||||||} t|j} |j| dg| |j|<dS) :param channel_oid: :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) rB RuntimeErrorr)rOrPr r(rCappend) rrFrnrqrrrrrtx1key_idxrrrrxs  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)rBrr)rOrPrCr rrrrrrr) rrFrnrqrrrrrrrtx2rrrr{s"  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: rrN)r&boolr*r(rCr)rZfilter_completelimitnrZtxrrrruBs  zSimpleBuyer.past_transactionscCsHddd}|jD]\}}|r|r|dd7<q|dd7<q|S)z :return: r)completependingrrrN)rC)rrrrrrrr~XszSimpleBuyer.count_transactionscCs |j|d}|r|j|SdS)2 :param key_id: :return: NrBgetrC)rridxrrrget_transactionhs zSimpleBuyer.get_transactioncCs,|j|d}|r|j|\}}|o|SdS)rNFr)rrrrrrrr is_completers zSimpleBuyer.is_completer)Tr)r r!r"__doc__r,rrYrZrrbrcrrxr{rur~rrrrrrr#Ns"E:  r & r#)rTrOpprintrr^rZ nacl.secretr6Z nacl.utilsZnacl.exceptionsZ nacl.publicr+Zautobahn.wamp.exceptionrZautobahn.wamp.protocolrZ_utilrrr0utilr r Z_eip712_channel_closer r objectr r#rrrrs$