import re
from grapheneapi.grapheneapi import GrapheneAPI
from pprint import pprint
from math import log10

rpc_host = "localhost"
rpc_port = 19988
rpc_user = "xeroc"
rpc_pass = "kKhVzWPkcE4KFEKi"

account = "xeroc"
wallet_name = "default"

client = GrapheneAPI(rpc_host, rpc_port, rpc_user, rpc_pass)

info = client.info()
if not info["wallet_open"]:
    client.wallet_open(wallet_name)

entries =  client.wallet_account_transaction_history(account)

"""
print("=" * 80)
print("Transfers")
print("=" * 80)
for entry in entries:
    if entry["is_market"]:
        continue

    for ledger in entry["ledger_entries"]:
        asset = client.get_asset(ledger["amount"]["asset_id"])
        amount = int(ledger["amount"]["amount"]) / int(asset["precision"])
        print(
            '{blocknum} ; {:.{prec}f} ; {asset} ; {src} ; {dest}'.format(
                amount,
                prec=int(log10(int(asset["precision"]))),
                blocknum=entry["timestamp"],
                asset=asset["symbol"],
                src=ledger["from_account"],
                dest=ledger["to_account"],
            )
        )
"""

print("=" * 80)
print("Market stuff")
print("=" * 80)
for entry in entries:
    if not entry["is_market"]:
        continue
    if entry["is_market_cancel"]:
        continue
    if not entry["is_virtual"]:  # filled, shorts and calls
        continue

    for ledger in entry["ledger_entries"]:
        asset = client.get_asset(ledger["amount"]["asset_id"])
        amount = int(ledger["amount"]["amount"]) / int(asset["precision"])
        if re.search("^sell collateral", ledger["memo"]):
            pass
        elif re.search("^cover proceeds", ledger["memo"]):
            pass
        elif re.search("^payoff debt", ledger["memo"]):
            pass
        elif re.search("^short proceeds", ledger["memo"]):
            pass
        elif re.search("^add collateral", ledger["memo"]):
            pass
        elif re.search("^bid proceeds", ledger["memo"]):
            pass
        elif re.search("^bid pay bid", ledger["memo"]):
            pass
        elif re.search("^pay bid", ledger["memo"]):
            pass
        else:
            print("Unhandled ledger entry:")
            pprint(ledger)

        print(
            '{blocknum} ; {:.{prec}f} ; {asset} ; {memo} ; {ref}'.format(
                amount,
                prec=int(log10(int(asset["precision"]))),
                blocknum=entry["timestamp"],
                asset=asset["symbol"],
                memo=ledger["memo"],
                ref=ledger["from_account"]
            )
        )