lbry-sdk/lbrynet/wallet/transaction.py

import struct
from binascii import hexlify
from typing import List  # pylint: disable=unused-import

from twisted.internet import defer  # pylint: disable=unused-import

from torba.baseaccount import BaseAccount  # pylint: disable=unused-import
from torba.basetransaction import BaseTransaction, BaseInput, BaseOutput
from torba.hash import hash160

from lbryschema.claim import ClaimDict  # pylint: disable=unused-import
from .script import InputScript, OutputScript


def claim_id_hash(txid, n):
    return hash160(txid + struct.pack('>I', n))


class Input(BaseInput):
    script_class = InputScript


class Output(BaseOutput):
    script_class = OutputScript

    @classmethod
    def pay_claim_name_pubkey_hash(cls, amount, claim_name, claim, pubkey_hash):
        script = cls.script_class.pay_claim_name_pubkey_hash(claim_name, claim, pubkey_hash)
        return cls(amount, script)


class Transaction(BaseTransaction):

    input_class = Input
    output_class = Output

    def get_claim_id(self, output_index):
        output = self.outputs[output_index]  # type: Output
        assert output.script.is_claim_name, 'Not a name claim.'
        return claim_id_hash(self.hash, output_index)

    @classmethod
    def claim(cls, name, meta, amount, holding_address, funding_accounts, change_account):
        # type: (bytes, ClaimDict, int, bytes, List[BaseAccount], BaseAccount) -> defer.Deferred
        ledger = cls.ensure_all_have_same_ledger(funding_accounts, change_account)
        claim_output = Output.pay_claim_name_pubkey_hash(
            amount, name, hexlify(meta.serialized), ledger.address_to_hash160(holding_address)
        )
        return cls.pay([claim_output], funding_accounts, change_account)

    @classmethod
    def abandon(cls, utxo, funding_accounts, change_account):
        # type: (Output, List[BaseAccount], BaseAccount) -> defer.Deferred
        return cls.liquidate([utxo], funding_accounts, change_account)