tools/side_chain_tool/app/api/transaction.py

   1 # -*- coding:utf-8 -*-
   2 import json
   3 from flask import request, jsonify, make_response, current_app, render_template
   4 #from sqlalchemy.exc import SQLAlchemyError, IntegrityError
   5
   6 from . import api
   7 from .Connection import Connection
   8 from .. import db
   9 from ..models import KeyPair, PeginAddress
  10
  11
  12 @api.route('/create_key_pair', methods=['POST', 'GET', 'OPTIONS'])
  13 def create_key_pair():
  14     keypairs = KeyPair.query.all()
  15     if len(keypairs) > 5 :
  16         return json_contents(jsonify(code=200, msg="There are already 5 private-public key pairs"))
  17
  18     connection = Connection("http://127.0.0.1:8888")
  19     response = connection.request("/create-key-pair")
  20     resp_json = json.loads(response.text)
  21     if resp_json['status'] == 'success':
  22         xprv = resp_json['data']['xprv']
  23         xpub = resp_json['data']['xpub']
  24         key = KeyPair(xprv=xprv,xpub=xpub)
  25         db.session.add(key)
  26         db.session.commit()
  27     elif resp_json['status'] == 'fail':
  28         return json_contents(jsonify(code=-1, msg=resp_json['msg']))
  29     else:
  30         return json_contents(jsonify(code=-1, msg="create key pair fail"))
  31
  32     return json_contents(jsonify(code=200, msg=""))
  33
  34 @api.route('/get_key_pair',methods=['POST', 'GET', 'OPTIONS'])
  35 def get_key_pair():
  36     keypairs = KeyPair.query.all()
  37     json_data = ""
  38     index = 0
  39     num = len(keypairs)
  40     for keypair in keypairs:
  41         data = {
  42             "xprv":keypair.xprv.encode('utf-8'),
  43             "xpub": keypair.xpub.encode('utf-8')
  44         }
  45         json_data = json_data + json.dumps(data)
  46         index = index + 1
  47         if index < num:
  48             json_data = json_data + ","
  49     return json_contents(jsonify(code=200, msg="sucess", data="[" + json_data + "]"))
  50
  51
  52 @api.route('/create_pegin_address', methods=['GET', 'POST', 'OPTIONS'])
  53 def create_pegin_address():
  54     if not request.json or not 'account_id' in request.json:
  55         return json_contents(jsonify(code=-1, msg="The json format is incorrect"))
  56     accountID = request.json['account_id']
  57     connSide = Connection("http://127.0.0.1:8888")
  58     body_json = {"account_id": accountID}
  59     response = connSide.request("/get-pegin-address",body_json)
  60     resp_json = json.loads(response.text)
  61     if resp_json['status'] == 'success':
  62         mainchain_address = resp_json['data']['mainchain_address']
  63         claim_script = resp_json['data']['claim_script']
  64         pegin = PeginAddress(address=mainchain_address, claim_script=claim_script)
  65         db.session.add(pegin)
  66         db.session.commit()
  67     elif resp_json['status'] == 'fail':
  68         return json_contents(jsonify(code=-1, msg=resp_json['msg']))
  69     else:
  70         return json_contents(jsonify(code=-1, msg="get pegin address fail"))
  71     #return render_template('peginaddress.html', mainchain_address=mainchain_address,claim_script=claim_script)
  72     return json_contents(jsonify(code=200, msg=""))
  73
  74 @api.route('/get_pegin_address',methods=['POST', 'GET', 'OPTIONS'])
  75 def get_pegin_address():
  76     peginAddrs = PeginAddress.query.all()
  77     json_data = ""
  78     index = 0
  79     num = len(peginAddrs)
  80     for peginAddr in peginAddrs:
  81         data = {
  82             "mainchain_address":peginAddr.address.encode('utf-8'),
  83             "claim_script": peginAddr.claim_script.encode('utf-8')
  84         }
  85         json_data = json_data + json.dumps(data)
  86         index = index + 1
  87         if index < num:
  88             json_data = json_data + ","
  89     return json_contents(jsonify(code=200, msg="sucess", data="[" + json_data + "]"))
  90
  91
  92 @api.route('/claim_tx', methods=['GET', 'POST', 'OPTIONS'])
  93 def claim_tx():
  94     if not request.json or not 'claim_script' in request.json or not 'block_height' in request.json or not 'tx_id' in request.json or not 'password' in request.json:
  95         return json_contents(jsonify(code=-1, msg="The json format is incorrect"))
  96     block_height = int(request.json['block_height'])
  97     tx_id = request.json['tx_id'].encode('utf-8')
  98     password = request.json['password'].encode('utf-8')
  99     claim_script = request.json['claim_script'].encode('utf-8')
 100     connSide = Connection("http://127.0.0.1:8888")
 101     connMain = Connection("http://127.0.0.1:9888")
 102     raw_block = ""
 103     raw_transaction = ""
 104     block_hash = ""
 105     proof=""
 106     # 从主链获取raw_block
 107     body_json = {"block_height": block_height}
 108     response = connMain.request("/get-raw-block",body_json)
 109     resp_json = json.loads(response.text)
 110     if resp_json['status'] == 'success':
 111         raw_block = resp_json['data']['raw_block'].encode('utf-8')
 112     elif resp_json['status'] == 'fail':
 113         return json_contents(jsonify(code=-1, msg="get-raw-block: " + resp_json['msg']))
 114     else:
 115         return json_contents(jsonify(code=-1, msg="get raw block fail"))
 116
 117     # 获取raw transaction
 118     body_json = {"tx_id": tx_id,"raw_block": raw_block}
 119     response = connSide.request("/get-raw-transaction",body_json)
 120     resp_json = json.loads(response.text.encode('utf-8'))
 121     if resp_json['status'] == 'success':
 122         raw_transaction = resp_json['data']['raw_transaction'].encode('utf-8')
 123         block_hash = resp_json['data']['block_hash'].encode('utf-8')
 124     elif resp_json['status'] == 'fail':
 125         return json_contents(jsonify(code=-1, msg="get-raw-transaction: " + resp_json['msg']))
 126     else:
 127         return json_contents(jsonify(code=-1, msg="get raw transaction fail"))
 128
 129     # 主链获取proof
 130     body_json = {"tx_id": tx_id,"block_hash": block_hash}
 131     response = connMain.request("/get-merkle-proof",body_json)
 132     resp_json = json.loads(response.text)
 133     if resp_json['status'] == 'success':
 134         proof = json.dumps(resp_json['data']).strip('{}')
 135     elif resp_json['status'] == 'fail':
 136         return json_contents(jsonify(code=-1, msg="get-merkle-proof:" + resp_json['msg']))
 137     else:
 138         return json_contents(jsonify(code=-1, msg="get raw transaction fail"))
 139
 140     # 调用claimtx
 141     body_json = '{"password": "%s","raw_transaction": "%s","claim_script":"%s",%s}' % (password,raw_transaction,claim_script,proof)
 142     print body_json
 143     response = connSide.request("/claim-pegin-transaction",json.loads(body_json))
 144     resp_json = json.loads(response.text)
 145     if resp_json['status'] == 'success':
 146         return json_contents(jsonify(code=200, msg=resp_json['data']))
 147     elif resp_json['status'] == 'fail':
 148         print resp_json
 149         return json_contents(jsonify(code=-1, msg="claim-pegin-transaction: " + resp_json['msg']))
 150     else:
 151         return json_contents(jsonify(code=-1, msg="claim pegin transaction fail"))
 152
 153
 154 @api.route('/send_to_mainchain', methods=['GET', 'POST', 'OPTIONS'])
 155 def send_to_mainchain():
 156     #if not request.json or not 'claim_script' in request.json or not 'xprvs' in request.json or not 'tx_id' in request.json or not 'id' in request.json or not 'side_tx_id' in request.json or not 'block_height' in request.json or not 'side_block_height' in request.json or not 'alias' in request.json or not 'root_xpubs' in request.json or not 'address' in request.json or not 'control_program' in request.json:
 157     if not request.json or not 'claim_script' in request.json or not 'tx_id' in request.json or not 'id' in request.json or not 'side_tx_id' in request.json or not 'block_height' in request.json or not 'side_block_height' in request.json or not 'alias' in request.json or not 'address' in request.json or not 'control_program' in request.json:
 158         return json_contents(jsonify(code=-1, msg="The json format is incorrect"))
 159
 160     connSide = Connection("http://127.0.0.1:8888")
 161     connMain = Connection("http://127.0.0.1:9888")
 162     tx_id = request.json['tx_id'].encode('utf-8')
 163     id = request.json['id'].encode('utf-8')
 164     block_height = int(request.json['block_height'])
 165     claim_script = request.json['claim_script'].encode('utf-8')
 166
 167     utxo = ""
 168     # 从主链获取raw_block
 169     body_json = {"block_height": block_height}
 170     response = connMain.request("/get-raw-block",body_json)
 171     resp_json = json.loads(response.text)
 172     if resp_json['status'] == 'success':
 173         raw_block = resp_json['data']['raw_block'].encode('utf-8')
 174     elif resp_json['status'] == 'fail':
 175         return json_contents(jsonify(code=-1, msg="get-raw-block: " + resp_json['msg']))
 176     else:
 177         return json_contents(jsonify(code=-1, msg="get raw block fail"))
 178
 179     # 获取主链上的utxo
 180     address = request.json['address'].encode('utf-8')
 181     body_json = {"tx_id": tx_id, "id": id, "raw_block": raw_block, "address": address}
 182     response = connSide.request("/get-utxo-from-transaction",body_json)
 183     resp_json = json.loads(response.text)
 184     if resp_json['status'] == 'success':
 185         utxo = json.dumps(resp_json['data']).strip('{}')+"}"
 186     elif resp_json['status'] == 'fail':
 187         return json_contents(jsonify(code=-1, msg="get-utxo-from-transaction: " + resp_json['msg']))
 188     else:
 189         return json_contents(jsonify(code=-1, msg="fail get utxo from transaction"))
 190
 191     block_height = int(request.json['side_block_height'])
 192     tx_id  = request.json['side_tx_id'].encode('utf-8')
 193     alias = request.json['alias'].encode('utf-8')
 194     #root_xpubs = json.dumps(request.json['root_xpubs']).strip('{}')
 195     root_xpubs = "["
 196     control_program = request.json['control_program'].encode('utf-8')
 197     key_pair = {}
 198     f = open("key_pair","r")
 199     lines = f.readlines()
 200     index = 0
 201     num = len(lines)
 202     for line in lines:
 203         tmp = json.loads(line)
 204         xprv = tmp['data']['xprv'].encode('utf-8')
 205         xpub = tmp['data']['xpub'].encode('utf-8')
 206         key_pair[xprv] = xpub
 207         root_xpubs += "\"" + xpub + "\""
 208         index = index + 1
 209         if index < num:
 210             root_xpubs = root_xpubs + ","
 211
 212     root_xpubs = root_xpubs + "]"
 213     # 获取侧链raw transaction
 214     body_json = {"tx_id": tx_id,"block_height": block_height}
 215     response = connSide.request("/get-side-raw-transaction",body_json)
 216     resp_json = json.loads(response.text.encode('utf-8'))
 217     if resp_json['status'] == 'success':
 218         raw_transaction = resp_json['data']['raw_transaction'].encode('utf-8')
 219         block_hash = resp_json['data']['block_hash'].encode('utf-8')
 220     elif resp_json['status'] == 'fail':
 221         return json_contents(jsonify(code=-1, msg="get-side-raw-transaction: " + resp_json['msg']))
 222     else:
 223         return json_contents(jsonify(code=-1, msg="get side raw transaction fail"))
 224
 225     # 构建主链交易
 226     body_json = '{"claim_script":"%s","raw_transaction": "%s","control_program":"%s","root_xpubs":%s,%s}' % (claim_script,raw_transaction,control_program,root_xpubs,utxo)
 227     response = connSide.request("/build-mainchain-tx",json.loads(body_json))
 228     resp_json = json.loads(response.text.encode('utf-8'))
 229     tmpl = ""
 230     if resp_json['status'] == 'success':
 231         tmpl = json.dumps(resp_json['data'])
 232     elif resp_json['status'] == 'fail':
 233         print resp_json
 234         return json_contents(jsonify(code=-1, msg="build-mainchain-tx: " + resp_json['msg']))
 235     else:
 236         return json_contents(jsonify(code=-1, msg="build mainchain transaction fail"))
 237
 238     # 签名
 239     for key,value in key_pair.items():
 240         body_json = '{"xprv": "%s","xpub":"%s","transaction":%s,"claim_script":"%s"}' % (key,value,tmpl,claim_script)
 241         response = connSide.request("/sign-with-key",json.loads(body_json))
 242         resp_json = json.loads(response.text.encode('utf-8'))
 243         if resp_json['status'] == 'success':
 244             tmpl = json.dumps(resp_json['data']['transaction'])
 245             raw_transaction = resp_json['data']['transaction']['raw_transaction'].encode('utf-8')
 246         elif resp_json['status'] == 'fail':
 247             return json_contents(jsonify(code=-1, msg="sign-with-key: " + resp_json['msg']))
 248         else:
 249             return json_contents(jsonify(code=-1, msg="sign-with-key fail"))
 250
 251     # 提交到主链
 252     body_json = '{"raw_transaction": "%s"}' % (raw_transaction)
 253     response = connMain.request("/submit-transaction",json.loads(body_json))
 254     resp_json = json.loads(response.text.encode('utf-8'))
 255     print resp_json
 256     if resp_json['status'] == 'success':
 257         return json_contents(jsonify(code=200, msg=resp_json['data']))
 258     elif resp_json['status'] == 'fail':
 259         return json_contents(jsonify(code=-1, msg="submit-transaction: " + resp_json['msg']))
 260     else:
 261         return json_contents(jsonify(code=-1, msg="submit-transaction fail"))
 262
 263
 264 def json_contents(jsonify):
 265     response = make_response(jsonify)
 266     response.headers['Access-Control-Allow-Origin'] = '*'
 267     response.headers['Access-Control-Allow-Methods'] = '*'
 268     response.headers['Access-Control-Allow-Headers'] = '*'
 269     return response