-
Notifications
You must be signed in to change notification settings - Fork 13
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
68b363a
commit 57fbdf3
Showing
3 changed files
with
375 additions
and
83 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,219 @@ | ||
| import json | ||
| import subprocess | ||
| import requests | ||
| import click # pip install click | ||
|
|
||
|
|
||
| @click.group() | ||
| def cli(): | ||
| pass | ||
|
|
||
|
|
||
| # Shared function used internally but not standalone from CLI | ||
|
|
||
| def request_router(target_url, body): | ||
| headers = {'Content-type': 'application/json'} | ||
| r = requests.post(target_url, data=json.dumps(body), headers=headers) | ||
| return r.json() | ||
|
|
||
|
|
||
| def preprocess_transaction(rosetta_host_url, root_originator, operations, metadata=None): | ||
| endpoint = "/construction/preprocess" | ||
| target_url = rosetta_host_url + endpoint | ||
| data = { | ||
| "network_identifier": { | ||
| "blockchain": "flow", | ||
| "network": "localnet" | ||
| }, | ||
| "operations": operations, | ||
| "metadata": { | ||
| "payer": root_originator | ||
| } | ||
| } | ||
| if metadata: | ||
| for key in metadata: | ||
| data["metadata"][key] = metadata[key] | ||
| return request_router(target_url, data) | ||
|
|
||
|
|
||
| def metadata_transaction(rosetta_host_url, options): | ||
| endpoint = "/construction/metadata" | ||
| target_url = rosetta_host_url + endpoint | ||
| data = { | ||
| "network_identifier": { | ||
| "blockchain": "flow", | ||
| "network": "localnet" | ||
| }, | ||
| "options": options | ||
| } | ||
| return request_router(target_url, data) | ||
|
|
||
|
|
||
| def payloads_transaction(rosetta_host_url, operations, protobuf): | ||
| endpoint = "/construction/payloads" | ||
| target_url = rosetta_host_url + endpoint | ||
| data = { | ||
| "network_identifier": { | ||
| "blockchain": "flow", | ||
| "network": "localnet" | ||
| }, | ||
| "operations": operations, | ||
| "metadata": { | ||
| "protobuf": protobuf | ||
| } | ||
| } | ||
| return request_router(target_url, data) | ||
|
|
||
|
|
||
| def combine_transaction(rosetta_host_url, unsigned_tx, root_originator, hex_bytes, rosetta_key, signed_tx): | ||
| endpoint = "/construction/combine" | ||
| target_url = rosetta_host_url + endpoint | ||
| data = { | ||
| "network_identifier": { | ||
| "blockchain": "flow", | ||
| "network": "localnet" | ||
| }, | ||
| "unsigned_transaction": unsigned_tx, | ||
| "signatures": [ | ||
| { | ||
| "signing_payload": { | ||
| "account_identifier": { | ||
| "address": root_originator | ||
| }, | ||
| "address": root_originator, | ||
| "hex_bytes": hex_bytes, | ||
| "signature_type": "ecdsa" | ||
| }, | ||
| "public_key": { | ||
| "hex_bytes": rosetta_key, | ||
| "curve_type": "secp256k1" | ||
| }, | ||
| "signature_type": "ecdsa", | ||
| "hex_bytes": signed_tx | ||
| } | ||
| ] | ||
| } | ||
| return request_router(target_url, data) | ||
|
|
||
|
|
||
| def submit_transaction(rosetta_host_url, signed_tx): | ||
| endpoint = "/construction/submit" | ||
| target_url = rosetta_host_url + endpoint | ||
| data = { | ||
| "network_identifier": { | ||
| "blockchain": "flow", | ||
| "network": "localnet" | ||
| }, | ||
| "signed_transaction": signed_tx | ||
| } | ||
| return request_router(target_url, data) | ||
|
|
||
|
|
||
| ###################################################################################### | ||
| # Rosetta Construction helper functions callable from CLI | ||
| ###################################################################################### | ||
|
|
||
| @click.command() | ||
| @click.argument('rosetta_host_url', envvar='ROSETTA_HOST_URL', required=True) | ||
| @click.argument('root_originator_address', envvar='ROOT_ORIGINATOR_ADDRESS', required=True) | ||
| @click.argument('root_originator_public_rosetta_key', envvar='ROOT_ORIGINATOR_PUBLIC_ROSETTA_KEY', required=True) | ||
| @click.argument('root_originator_private_key', envvar='ROOT_ORIGINATOR_PRIVATE_KEY', required=True) | ||
| @click.argument('new_account_public_rosetta_key', envvar='NEW_ACCOUNT_PUBLIC_ROSETTA_KEY', required=True) | ||
| def rosetta_create_derived_account(rosetta_host_url, root_originator_address, root_originator_public_rosetta_key, | ||
| root_originator_private_key, new_account_public_rosetta_key): | ||
| print("Enter create: " + root_originator_address + ", new: " + new_account_public_rosetta_key) | ||
| transaction = "create_account" | ||
| metadata = {"public_key": new_account_public_rosetta_key} | ||
| operations = [ | ||
| { | ||
| "type": transaction, | ||
| "operation_identifier": { | ||
| "index": 0 | ||
| }, | ||
| "metadata": metadata | ||
| } | ||
| ] | ||
| preprocess_response = preprocess_transaction(rosetta_host_url, root_originator_address, operations) | ||
| print(preprocess_response) | ||
| metadata_response = metadata_transaction(rosetta_host_url, preprocess_response["options"]) | ||
| payloads_response = payloads_transaction(rosetta_host_url, operations, metadata_response["metadata"]["protobuf"]) | ||
| hex_bytes = payloads_response["payloads"][0]["hex_bytes"] | ||
| unsigned_tx = payloads_response["unsigned_transaction"] | ||
| sign_tx_cmd = "go run cmd/sign/sign.go " + root_originator_private_key + " " + hex_bytes | ||
| result = subprocess.run(sign_tx_cmd.split(" "), stdout=subprocess.PIPE) | ||
| signed_tx = result.stdout.decode('utf-8')[:-1] | ||
| combine_response = combine_transaction(rosetta_host_url, unsigned_tx, root_originator_address, hex_bytes, | ||
| root_originator_public_rosetta_key, signed_tx) | ||
| submit_transaction_response = submit_transaction(rosetta_host_url, combine_response["signed_transaction"]) | ||
| return submit_transaction_response["transaction_identifier"]["hash"] | ||
|
|
||
|
|
||
| @click.command() | ||
| @click.argument('rosetta_host_url', envvar='ROSETTA_HOST_URL', required=True) | ||
| @click.argument('root_originator_address', envvar='ROOT_ORIGINATOR_ADDRESS', required=True) | ||
| @click.argument('root_originator_public_rosetta_key', envvar='ROOT_ORIGINATOR_PUBLIC_ROSETTA_KEY', required=True) | ||
| @click.argument('root_originator_private_key', envvar='ROOT_ORIGINATOR_PRIVATE_KEY', required=True) | ||
| @click.argument('recipient_address', envvar='RECIPIENT', required=True) | ||
| @click.argument('amount', envvar='AMOUNT', required=True) | ||
| def rosetta_transfer_funds(rosetta_host_url, root_originator_address, root_originator_public_rosetta_key, | ||
| root_originator_private_key, recipient_address, amount, i=0): | ||
| transaction = "transfer" | ||
| operations = [ | ||
| { | ||
| "type": transaction, | ||
| "operation_identifier": { | ||
| "index": i | ||
| }, | ||
| "account": { | ||
| "address": root_originator_address | ||
| }, | ||
| "amount": { | ||
| "currency": { | ||
| "decimals": 8, | ||
| "symbol": "FLOW" | ||
| }, | ||
| "value": str(-1 * amount * 10 ** 7) | ||
| } | ||
| }, | ||
| { | ||
| "type": transaction, | ||
| "operation_identifier": { | ||
| "index": i + 1 | ||
| }, | ||
| "related_operations": [ | ||
| { | ||
| "index": i | ||
| } | ||
| ], | ||
| "account": { | ||
| "address": recipient_address | ||
| }, | ||
| "amount": { | ||
| "currency": { | ||
| "decimals": 8, | ||
| "symbol": "FLOW" | ||
| }, | ||
| "value": str(amount * 10 ** 7) | ||
| } | ||
| } | ||
| ] | ||
| preprocess_response = preprocess_transaction(rosetta_host_url, root_originator_address, operations) | ||
| metadata_response = metadata_transaction(rosetta_host_url, preprocess_response["options"]) | ||
| payloads_response = payloads_transaction(rosetta_host_url, operations, metadata_response["metadata"]["protobuf"]) | ||
| hex_bytes = payloads_response["payloads"][0]["hex_bytes"] | ||
| unsigned_tx = payloads_response["unsigned_transaction"] | ||
| sign_tx_cmd = "go run cmd/sign/sign.go " + root_originator_private_key + " " + hex_bytes | ||
| result = subprocess.run(sign_tx_cmd.split(" "), stdout=subprocess.PIPE) | ||
| signed_tx = result.stdout.decode('utf-8')[:-1] | ||
| combine_response = combine_transaction(rosetta_host_url, unsigned_tx, root_originator_address, hex_bytes, | ||
| root_originator_public_rosetta_key, signed_tx) | ||
| submit_transaction_response = submit_transaction(rosetta_host_url, combine_response["signed_transaction"]) | ||
| return submit_transaction_response["transaction_identifier"]["hash"] | ||
|
|
||
|
|
||
| # CLI bindings | ||
| cli.add_command(rosetta_create_derived_account) | ||
| cli.add_command(rosetta_transfer_funds) | ||
|
|
||
| if __name__ == '__main__': | ||
| cli() |
Oops, something went wrong.