update

[bytom/bytom-kit.git] / app / api / resources.py

diff --git a/app/api/resources.py b/app/api/resources.py
index d507e0b..b97ceb3 100644 (file)
--- a/app/api/resources.py
+++ b/app/api/resources.py
@@ -1,3 +1,5 @@
+# from flask_restplus import inputs
+from flask_restful import inputs
 from flask_restful import Resource
 from flask_restful import reqparse
 from app.model.hello import get_hello_result
@@ -8,6 +10,14 @@ from app.model.key import entropy_to_mnemonic
 from app.model.key import mnemonic_to_seed
 from app.model.key import seed_to_root_xprv
 from app.model.key import xprv_to_expanded_private_key
+from app.model.key import xpub_to_public_key
+from app.model.key import xpub_verify
+from app.model.key import xprv_sign
+from app.model.key import xprv_to_xpub
+from app.model.key import xprv_sign
+from app.model.key import xprv_to_child_xprv
+from app.model.key import xpub_to_child_xpub
+from app.model.receiver import create_P2WPKH_program
 
 parser = reqparse.RequestParser()
 parser.add_argument('private_key_str', type=str)
@@ -18,6 +28,11 @@ parser.add_argument('entropy_str', type=str)
 parser.add_argument('mnemonic_str', type=str)
 parser.add_argument('seed_str', type=str)
 parser.add_argument('xprv_str', type=str)
+parser.add_argument('xpub_str', type=str)
+parser.add_argument('path_list', type=str, action='append')
+parser.add_argument('account_index_int', type=int)
+parser.add_argument('address_index_int', type=int)
+parser.add_argument('change_bool', type=inputs.boolean)
 
 class Hello(Resource):
 
@@ -79,4 +94,68 @@ class Xprv_To_Expanded_Private_Key(Resource):
         args = parser.parse_args()
         xprv_str = args.get('xprv_str')
         expanded_private_key_str = xprv_to_expanded_private_key(xprv_str)
-        return expanded_private_key_str
\ No newline at end of file
+        return expanded_private_key_str
+
+class Xpub_To_Public_Key(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        xpub_str = args.get('xpub_str')
+        public_key_str = xpub_to_public_key(xpub_str)
+        return public_key_str
+
+class Xpub_Verify(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        xpub = args.get('xpub_str')
+        message = args.get('message_str')
+        signature = args.get('signature_str')
+        result = xpub_verify(xpub, message, signature)
+        return result
+
+class Xprv_Sign(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        xprv = args.get('xprv_str')
+        message = args.get('message_str')
+        signature = xprv_sign(xprv, message)
+        return signature
+
+class Xprv_To_Xpub(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        xprv = args.get('xprv_str')
+        xpub = xprv_to_xpub(xprv)
+        return xpub
+
+class Xprv_To_Child_Xprv(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        xprv = args.get('xprv_str')
+        path = args.get('path_list')
+        child_xprv = xprv_to_child_xprv(xprv, path)
+        return child_xprv
+
+class Xpub_To_Child_Xpub(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        xpub = args.get('xpub_str')
+        path = args.get('path_list')
+        child_xpub = xpub_to_child_xpub(xpub, path)
+        return child_xpub
+
+class Create_P2WPKH_Program(Resource):
+
+    def post(self):
+        args = parser.parse_args()
+        account_index = args.get('account_index_int')
+        address_index = args.get('address_index_int')
+        change = args.get('change_bool')
+        xpub = args.get('xpub_str')
+        control_program = create_P2WPKH_program(account_index, address_index, change, xpub)
+        return control_program
\ No newline at end of file