matejcik
6 years ago
parent
a5e3d50fb0
commit
f87d931ce7
@ -0,0 +1,140 @@
|
||||
import binascii
|
||||
import json
|
||||
from . import messages as proto
|
||||
|
||||
TYPE_MOSAIC_TRANSFER = 0x0101
|
||||
TYPE_IMPORTANCE_TRANSFER = 0x0801
|
||||
TYPE_MULTISIG_CHANGE = 0x1001
|
||||
TYPE_MULTISIG_SIGN = 0x1002
|
||||
TYPE_MULTISIG_TX = 0x1004
|
||||
TYPE_PROVISION_NAMESPACE = 0x2001
|
||||
TYPE_MOSAIC_DEFINITION_CREATION = 0x4001
|
||||
TYPE_MOSAIC_SUPPLY_CHANGE = 0x4002
|
||||
|
||||
|
||||
def create_transaction_common(transaction):
|
||||
msg = proto.NEMTransactionCommon()
|
||||
msg.network = (transaction["version"] >> 24) & 0xFF
|
||||
msg.timestamp = transaction["timeStamp"]
|
||||
msg.fee = transaction["fee"]
|
||||
msg.deadline = transaction["deadline"]
|
||||
|
||||
if "signed" in transaction:
|
||||
msg.signer = binascii.unhexlify(transaction["signer"])
|
||||
|
||||
return msg
|
||||
|
||||
|
||||
def create_transfer(transaction):
|
||||
msg = proto.NEMTransfer()
|
||||
msg.recipient = transaction["recipient"]
|
||||
msg.amount = transaction["amount"]
|
||||
|
||||
if "payload" in transaction["message"]:
|
||||
msg.payload = binascii.unhexlify(transaction["message"]["payload"])
|
||||
|
||||
if transaction["message"]["type"] == 0x02:
|
||||
msg.public_key = binascii.unhexlify(transaction["message"]["publicKey"])
|
||||
|
||||
if "mosaics" in transaction:
|
||||
msg.mosaics = [proto.NEMMosaic(
|
||||
namespace=mosaic["mosaicId"]["namespaceId"],
|
||||
mosaic=mosaic["mosaicId"]["name"],
|
||||
quantity=mosaic["quantity"],
|
||||
) for mosaic in transaction["mosaics"]]
|
||||
|
||||
return msg
|
||||
|
||||
|
||||
def create_aggregate_modification(transactions):
|
||||
msg = proto.NEMAggregateModification()
|
||||
msg.modifications = [proto.NEMCosignatoryModification(
|
||||
type=modification["modificationType"],
|
||||
public_key=binascii.unhexlify(modification["cosignatoryAccount"]),
|
||||
) for modification in transactions["modifications"]]
|
||||
|
||||
if "minCosignatories" in transactions:
|
||||
msg.relative_change = transactions["minCosignatories"]["relativeChange"]
|
||||
|
||||
return msg
|
||||
|
||||
|
||||
def create_provision_namespace(transaction):
|
||||
msg = proto.NEMProvisionNamespace()
|
||||
msg.namespace = transaction["newPart"]
|
||||
|
||||
if transaction["parent"]:
|
||||
msg.parent = transaction["parent"]
|
||||
|
||||
msg.sink = transaction["rentalFeeSink"]
|
||||
msg.fee = transaction["rentalFee"]
|
||||
|
||||
|
||||
def create_mosaic_creation(transaction):
|
||||
definition = transaction["mosaicDefinition"]
|
||||
msg = proto.NEMMosaicCreation()
|
||||
msg.definition = proto.NEMMosaicDefinition()
|
||||
msg.definition.namespace = definition["id"]["namespaceId"]
|
||||
msg.definition.mosaic = definition["id"]["name"]
|
||||
|
||||
if definition["levy"]:
|
||||
msg.definition.levy = definition["levy"]["type"]
|
||||
msg.definition.fee = definition["levy"]["fee"]
|
||||
msg.definition.levy_address = definition["levy"]["recipient"]
|
||||
msg.definition.levy_namespace = definition["levy"]["mosaicId"]["namespaceId"]
|
||||
msg.definition.levy_mosaic = definition["levy"]["mosaicId"]["name"]
|
||||
|
||||
msg.definition.description = definition["description"]
|
||||
|
||||
for property in definition["properties"]:
|
||||
name = property["name"]
|
||||
value = json.loads(property["value"])
|
||||
|
||||
if name == "divisibility":
|
||||
msg.definition.divisibility = value
|
||||
elif name == "initialSupply":
|
||||
msg.definition.supply = value
|
||||
elif name == "supplyMutable":
|
||||
msg.definition.mutable_supply = value
|
||||
elif name == "transferable":
|
||||
msg.definition.transferable = value
|
||||
|
||||
msg.sink = transaction["creationFeeSink"]
|
||||
msg.fee = transaction["creationFee"]
|
||||
return msg
|
||||
|
||||
|
||||
def create_supply_change(transaction):
|
||||
msg = proto.NEMMosaicSupplyChange()
|
||||
msg.namespace = transaction["mosaicId"]["namespaceId"]
|
||||
msg.mosaic = transaction["mosaicId"]["name"]
|
||||
msg.type = transaction["supplyType"]
|
||||
msg.delta = transaction["delta"]
|
||||
return msg
|
||||
|
||||
|
||||
def create_sign_tx(transaction):
|
||||
msg = proto.NEMSignTx()
|
||||
msg.transaction = create_transaction_common(transaction)
|
||||
msg.cosigning = (transaction["type"] == TYPE_MULTISIG_SIGN)
|
||||
|
||||
if transaction["type"] in (TYPE_MULTISIG_SIGN, TYPE_MULTISIG_TX):
|
||||
transaction = transaction["otherTrans"]
|
||||
msg.multisig = create_transaction_common(transaction)
|
||||
elif "otherTrans" in transaction:
|
||||
raise ValueError("Transaction does not support inner transaction")
|
||||
|
||||
if transaction["type"] == TYPE_MOSAIC_TRANSFER:
|
||||
msg.transfer = create_transfer(transaction)
|
||||
elif transaction["type"] == TYPE_MULTISIG_CHANGE:
|
||||
msg.aggregate_modification = create_aggregate_modification(transaction)
|
||||
elif transaction["type"] == TYPE_PROVISION_NAMESPACE:
|
||||
msg.provision_namespace = create_provision_namespace(transaction)
|
||||
elif transaction["type"] == TYPE_MOSAIC_DEFINITION_CREATION:
|
||||
msg.mosaic_creation = create_mosaic_creation(transaction)
|
||||
elif transaction["type"] == TYPE_MOSAIC_SUPPLY_CHANGE:
|
||||
msg.mosaic_supply_change = create_supply_change(transaction)
|
||||
else:
|
||||
raise ValueError("Unknown transaction type")
|
||||
|
||||
return msg
|
||||
@ -0,0 +1,38 @@
|
||||
import binascii
|
||||
from trezorlib import nem
|
||||
|
||||
|
||||
def test_nem_basic():
|
||||
transaction = {
|
||||
"timeStamp": 76809215,
|
||||
"amount": 1000000,
|
||||
"fee": 1000000,
|
||||
"recipient": "TALICE2GMA34CXHD7XLJQ536NM5UNKQHTORNNT2J",
|
||||
"type": nem.TYPE_MOSAIC_TRANSFER,
|
||||
"deadline": 76895615,
|
||||
"version": (0x98 << 24),
|
||||
"message": {
|
||||
"payload": binascii.hexlify(b'hello world'),
|
||||
"type": 1,
|
||||
},
|
||||
"mosaics": [
|
||||
{
|
||||
"mosaicId": {
|
||||
"namespaceId": "nem",
|
||||
"name": "xem",
|
||||
},
|
||||
"quantity": 1000000,
|
||||
},
|
||||
],
|
||||
}
|
||||
|
||||
msg = nem.create_sign_tx(transaction)
|
||||
|
||||
# this is basically just a random sampling of expected properties
|
||||
assert msg.transaction is not None
|
||||
assert msg.transfer is not None
|
||||
assert len(msg.transfer.mosaics) == 1
|
||||
assert msg.transfer.mosaics[0].namespace == "nem"
|
||||
|
||||
assert msg.aggregate_modification is None
|
||||
assert msg.provision_namespace is None
|
||||
Loading…
Reference in new issue