3 from flask import request, jsonify, make_response, current_app, render_template
4 #from sqlalchemy.exc import SQLAlchemyError, IntegrityError
7 from .Connection import Connection, WSClient
9 from ..models import KeyPair, PeginAddress
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"))
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)
27 elif resp_json['status'] == 'fail':
28 return json_contents(jsonify(code=-1, msg=resp_json['msg']))
30 return json_contents(jsonify(code=-1, msg="create key pair fail"))
32 return json_contents(jsonify(code=200, msg=""))
34 @api.route('/get_key_pair',methods=['POST', 'GET', 'OPTIONS'])
36 keypairs = KeyPair.query.all()
40 for keypair in keypairs:
42 "xprv":keypair.xprv.encode('utf-8'),
43 "xpub": keypair.xpub.encode('utf-8')
45 json_data = json_data + json.dumps(data)
48 json_data = json_data + ","
49 return json_contents(jsonify(code=200, msg="sucess", data="[" + json_data + "]"))
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-contract-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 control_program = resp_json['data']['control_program']
64 claim_script = resp_json['data']['claim_script']
65 pegin = PeginAddress(address=mainchain_address, control_program=control_program, claim_script=claim_script)
68 elif resp_json['status'] == 'fail':
69 return json_contents(jsonify(code=-1, msg=resp_json['msg']))
71 return json_contents(jsonify(code=-1, msg="get pegin address fail"))
72 #return render_template('peginaddress.html', mainchain_address=mainchain_address,claim_script=claim_script)
73 return json_contents(jsonify(code=200, msg=""))
75 @api.route('/get_pegin_address',methods=['POST', 'GET', 'OPTIONS'])
76 def get_pegin_address():
77 peginAddrs = PeginAddress.query.all()
81 for peginAddr in peginAddrs:
83 "mainchain_address":peginAddr.address.encode('utf-8'),
84 "control_program":peginAddr.control_program.encode('utf-8'),
85 "claim_script": peginAddr.claim_script.encode('utf-8')
87 json_data = json_data + json.dumps(data)
90 json_data = json_data + ","
91 return json_contents(jsonify(code=200, msg="sucess", data="[" + json_data + "]"))
94 @api.route('/claim_tx', methods=['GET', 'POST', 'OPTIONS'])
96 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:
97 return json_contents(jsonify(code=-1, msg="The json format is incorrect"))
98 block_height = int(request.json['block_height'])
99 tx_id = request.json['tx_id'].encode('utf-8')
100 password = request.json['password'].encode('utf-8')
101 claim_script = request.json['claim_script'].encode('utf-8')
102 connSide = Connection("http://127.0.0.1:8888")
103 connMain = Connection("http://127.0.0.1:9888")
109 body_json = {"block_height": block_height}
110 response = connMain.request("/get-raw-block",body_json)
111 resp_json = json.loads(response.text)
112 if resp_json['status'] == 'success':
113 raw_block = resp_json['data']['raw_block'].encode('utf-8')
114 elif resp_json['status'] == 'fail':
116 return json_contents(jsonify(code=-1, msg="get-raw-block: " + resp_json['error_detail']))
118 return json_contents(jsonify(code=-1, msg="get raw block fail"))
121 body_json = {"tx_id": tx_id,"raw_block": raw_block}
122 response = connSide.request("/get-raw-transaction",body_json)
123 resp_json = json.loads(response.text.encode('utf-8'))
124 if resp_json['status'] == 'success':
125 raw_transaction = resp_json['data']['raw_transaction'].encode('utf-8')
126 block_hash = resp_json['data']['block_hash'].encode('utf-8')
127 elif resp_json['status'] == 'fail':
129 return json_contents(jsonify(code=-1, msg="get-raw-transaction: " + resp_json['error_detail']))
131 return json_contents(jsonify(code=-1, msg="get raw transaction fail"))
135 body_json = {"tx_id": tx_id,"block_hash": block_hash}
136 response = connMain.request("/get-merkle-proof",body_json)
137 resp_json = json.loads(response.text)
138 if resp_json['status'] == 'success':
139 proof = json.dumps(resp_json['data']).strip('{}')
140 elif resp_json['status'] == 'fail':
142 return json_contents(jsonify(code=-1, msg="get-merkle-proof:" + resp_json['error_detail']))
144 return json_contents(jsonify(code=-1, msg="get raw transaction fail"))
147 body_json = '{"password": "%s","raw_transaction": "%s","claim_script":"%s",%s}' % (password,raw_transaction,claim_script,proof)
149 response = connSide.request("/claim-pegin-transaction",json.loads(body_json))
150 resp_json = json.loads(response.text)
151 if resp_json['status'] == 'success':
152 return json_contents(jsonify(code=200, msg=resp_json['data']))
153 elif resp_json['status'] == 'fail':
155 return json_contents(jsonify(code=-1, msg="claim-pegin-transaction: " + resp_json['error_detail']))
157 return json_contents(jsonify(code=-1, msg="claim pegin transaction fail"))
160 @api.route('/send_to_mainchain', methods=['GET', 'POST', 'OPTIONS'])
161 def send_to_mainchain():
162 #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:
163 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:
164 return json_contents(jsonify(code=-1, msg="The json format is incorrect"))
166 connSide = Connection("http://127.0.0.1:8888")
167 connMain = Connection("http://127.0.0.1:9888")
168 tx_id = request.json['tx_id'].encode('utf-8')
169 id = request.json['id'].encode('utf-8')
170 block_height = int(request.json['block_height'])
171 claim_script = request.json['claim_script'].encode('utf-8')
175 body_json = {"block_height": block_height}
176 response = connMain.request("/get-raw-block",body_json)
177 resp_json = json.loads(response.text)
178 if resp_json['status'] == 'success':
179 raw_block = resp_json['data']['raw_block'].encode('utf-8')
180 elif resp_json['status'] == 'fail':
181 return json_contents(jsonify(code=-1, msg="get-raw-block: " + resp_json['msg']))
183 return json_contents(jsonify(code=-1, msg="get raw block fail"))
186 address = request.json['address'].encode('utf-8')
187 body_json = {"tx_id": tx_id, "id": id, "raw_block": raw_block, "address": address}
188 response = connSide.request("/get-utxo-from-transaction",body_json)
189 resp_json = json.loads(response.text)
190 if resp_json['status'] == 'success':
191 utxo = json.dumps(resp_json['data']).strip('{}')+"}"
192 elif resp_json['status'] == 'fail':
193 return json_contents(jsonify(code=-1, msg="get-utxo-from-transaction: " + resp_json['msg']))
195 return json_contents(jsonify(code=-1, msg="fail get utxo from transaction"))
197 block_height = int(request.json['side_block_height'])
198 tx_id = request.json['side_tx_id'].encode('utf-8')
199 alias = request.json['alias'].encode('utf-8')
200 #root_xpubs = json.dumps(request.json['root_xpubs']).strip('{}')
202 control_program = request.json['control_program'].encode('utf-8')
204 f = open("key_pair","r")
205 lines = f.readlines()
209 tmp = json.loads(line)
210 xprv = tmp['data']['xprv'].encode('utf-8')
211 xpub = tmp['data']['xpub'].encode('utf-8')
212 key_pair[xprv] = xpub
213 root_xpubs += "\"" + xpub + "\""
216 root_xpubs = root_xpubs + ","
218 root_xpubs = root_xpubs + "]"
219 # 获取侧链raw transaction
220 body_json = {"tx_id": tx_id,"block_height": block_height}
221 response = connSide.request("/get-side-raw-transaction",body_json)
222 resp_json = json.loads(response.text.encode('utf-8'))
223 if resp_json['status'] == 'success':
224 raw_transaction = resp_json['data']['raw_transaction'].encode('utf-8')
225 block_hash = resp_json['data']['block_hash'].encode('utf-8')
226 elif resp_json['status'] == 'fail':
227 return json_contents(jsonify(code=-1, msg="get-side-raw-transaction: " + resp_json['msg']))
229 return json_contents(jsonify(code=-1, msg="get side raw transaction fail"))
232 body_json = '{"claim_script":"%s","raw_transaction": "%s","control_program":"%s","root_xpubs":%s,%s}' % (claim_script,raw_transaction,control_program,root_xpubs,utxo)
233 response = connSide.request("/build-mainchain-tx",json.loads(body_json))
234 resp_json = json.loads(response.text.encode('utf-8'))
236 if resp_json['status'] == 'success':
237 tmpl = json.dumps(resp_json['data'])
238 elif resp_json['status'] == 'fail':
240 return json_contents(jsonify(code=-1, msg="build-mainchain-tx: " + resp_json['msg']))
242 return json_contents(jsonify(code=-1, msg="build mainchain transaction fail"))
245 for key,value in key_pair.items():
246 body_json = '{"xprv": "%s","xpub":"%s","transaction":%s,"claim_script":"%s"}' % (key,value,tmpl,claim_script)
247 response = connSide.request("/sign-with-key",json.loads(body_json))
248 resp_json = json.loads(response.text.encode('utf-8'))
249 if resp_json['status'] == 'success':
250 tmpl = json.dumps(resp_json['data']['transaction'])
251 raw_transaction = resp_json['data']['transaction']['raw_transaction'].encode('utf-8')
252 elif resp_json['status'] == 'fail':
253 return json_contents(jsonify(code=-1, msg="sign-with-key: " + resp_json['msg']))
255 return json_contents(jsonify(code=-1, msg="sign-with-key fail"))
258 body_json = '{"raw_transaction": "%s"}' % (raw_transaction)
259 response = connMain.request("/submit-transaction",json.loads(body_json))
260 resp_json = json.loads(response.text.encode('utf-8'))
262 if resp_json['status'] == 'success':
263 return json_contents(jsonify(code=200, msg=resp_json['data']))
264 elif resp_json['status'] == 'fail':
265 return json_contents(jsonify(code=-1, msg="submit-transaction: " + resp_json['msg']))
267 return json_contents(jsonify(code=-1, msg="submit-transaction fail"))
270 def json_contents(jsonify):
271 response = make_response(jsonify)
272 response.headers['Access-Control-Allow-Origin'] = '*'
273 response.headers['Access-Control-Allow-Methods'] = '*'
274 response.headers['Access-Control-Allow-Headers'] = '*'