updated watchonly

2020-12-01 19:54:16 +00:00
parent 52956a62a2
commit a838706090
12 changed files with 1419 additions and 0 deletions
@@ -0,0 +1,194 @@
+import hashlib
+from quart import g, jsonify, request, url_for
+from http import HTTPStatus
+
+from lnbits.core.crud import get_user
+from lnbits.decorators import api_check_wallet_key, api_validate_post_request
+
+from lnbits.extensions.watchonly import watchonly_ext
+from .crud import (
+    create_watch_wallet,
+    get_watch_wallet,
+    get_watch_wallets,
+    update_watch_wallet,
+    delete_watch_wallet,
+    create_payment,
+    get_payment,
+    get_payments,
+    delete_payment,
+    create_mempool,
+    update_mempool,
+    get_mempool,
+    get_addresses,
+    get_fresh_address,
+    get_address
+)
+
+###################WALLETS#############################
+
+@watchonly_ext.route("/api/v1/wallet", methods=["GET"])
+@api_check_wallet_key("invoice")
+async def api_wallets_retrieve():
+
+    try:
+        return (
+            jsonify([wallet._asdict() for wallet in get_watch_wallets(g.wallet.user)]), HTTPStatus.OK
+        )
+    except:
+        return (
+            jsonify({"message": "Cant fetch."}),
+            HTTPStatus.UPGRADE_REQUIRED,
+        )
+
+@watchonly_ext.route("/api/v1/wallet/<wallet_id>", methods=["GET"])
+@api_check_wallet_key("invoice")
+async def api_wallet_retrieve(wallet_id):
+    wallet = get_watch_wallet(wallet_id) 
+        
+    if not wallet:
+        return jsonify({"message": "wallet does not exist"}), HTTPStatus.NOT_FOUND
+
+    return jsonify({wallet}), HTTPStatus.OK
+
+
+@watchonly_ext.route("/api/v1/wallet", methods=["POST"])
+@watchonly_ext.route("/api/v1/wallet/<wallet_id>", methods=["PUT"])
+@api_check_wallet_key("invoice")
+@api_validate_post_request(
+    schema={
+        "masterpub": {"type": "string", "empty": False, "required": True},
+        "title": {"type": "string", "empty": False, "required": True},
+    }
+)
+async def api_wallet_create_or_update(wallet_id=None):
+    print("g.data")
+    if not wallet_id:
+        wallet = create_watch_wallet(user=g.wallet.user, masterpub=g.data["masterpub"], title=g.data["title"])
+        mempool = get_mempool(g.wallet.user) 
+        if not mempool:
+            create_mempool(user=g.wallet.user)
+        return jsonify(wallet._asdict()), HTTPStatus.CREATED
+
+    else:
+        wallet = update_watch_wallet(wallet_id=wallet_id, **g.data) 
+        return jsonify(wallet._asdict()), HTTPStatus.OK 
+
+
+@watchonly_ext.route("/api/v1/wallet/<wallet_id>", methods=["DELETE"])
+@api_check_wallet_key("invoice")
+async def api_wallet_delete(wallet_id):
+    wallet = get_watch_wallet(wallet_id)
+
+    if not wallet:
+        return jsonify({"message": "Wallet link does not exist."}), HTTPStatus.NOT_FOUND
+
+    delete_watch_wallet(wallet_id)
+
+    return jsonify({"deleted": "true"}), HTTPStatus.NO_CONTENT
+
+
+#############################ADDRESSES##########################
+
+@watchonly_ext.route("/api/v1/address/<wallet_id>", methods=["GET"])
+@api_check_wallet_key("invoice")
+async def api_fresh_address(wallet_id):
+    address = get_fresh_address(wallet_id) 
+        
+    if not address:
+        return jsonify({"message": "something went wrong"}), HTTPStatus.NOT_FOUND
+
+    return jsonify({address}), HTTPStatus.OK
+
+
+@watchonly_ext.route("/api/v1/addresses/<wallet_id>", methods=["GET"])
+@api_check_wallet_key("invoice")
+async def api_get_addresses(wallet_id):
+    addresses = get_addresses(wallet_id) 
+    print(addresses)
+    if not addresses:
+        return jsonify({"message": "wallet does not exist"}), HTTPStatus.NOT_FOUND
+
+    return jsonify([address._asdict() for address in addresses]), HTTPStatus.OK
+
+#############################PAYEMENTS##########################
+
+@watchonly_ext.route("/api/v1/payment", methods=["GET"])
+@api_check_wallet_key("invoice")
+async def api_payments_retrieve():
+
+    try:
+        return (
+            jsonify(get_payments(g.wallet.user)),
+            HTTPStatus.OK,
+        )
+    except:
+        return (
+            jsonify({"message": "Cant fetch."}),
+            HTTPStatus.UPGRADE_REQUIRED,
+        )
+
+@watchonly_ext.route("/api/v1/payment/<payment_id>", methods=["GET"])
+@api_check_wallet_key("invoice")
+async def api_payment_retrieve(payment_id):
+    payment = get_payment(payment_id) 
+        
+    if not payment:
+        return jsonify({"message": "payment does not exist"}), HTTPStatus.NOT_FOUND
+
+    return jsonify({payment}), HTTPStatus.OK
+
+
+@watchonly_ext.route("/api/v1/payment", methods=["POST"])
+@watchonly_ext.route("/api/v1/payment/<payment_id>", methods=["PUT"])
+@api_check_wallet_key("invoice")
+@api_validate_post_request(
+    schema={
+        "ex_key": {"type": "string", "empty": False, "required": True},
+        "pub_key": {"type": "string", "empty": False, "required": True},
+        "time_to_pay": {"type": "integer", "min": 1, "required": True},
+        "amount": {"type": "integer", "min": 1, "required": True},
+    }
+)
+async def api_payment_create_or_update(payment_id=None):
+
+    if not payment_id:
+        payment = create_payment(g.wallet.user, g.data.ex_key, g.data.pub_key, g.data.amount)
+        return jsonify(get_payment(payment)), HTTPStatus.CREATED
+
+    else:
+        payment = update_payment(payment_id, g.data) 
+        return jsonify({payment}), HTTPStatus.OK 
+
+
+@watchonly_ext.route("/api/v1/payment/<payment_id>", methods=["DELETE"])
+@api_check_wallet_key("invoice")
+async def api_payment_delete(payment_id):
+    payment = get_watch_wallet(payment_id)
+
+    if not payment:
+        return jsonify({"message": "Wallet link does not exist."}), HTTPStatus.NOT_FOUND
+
+    delete_watch_wallet(payment_id)
+
+    return "", HTTPStatus.NO_CONTENT
+
+#############################MEMPOOL##########################
+
+@watchonly_ext.route("/api/v1/mempool", methods=["PUT"])
+@api_check_wallet_key("invoice")
+@api_validate_post_request(
+    schema={
+        "endpoint": {"type": "string", "empty": False, "required": True},
+    }
+)
+async def api_update_mempool():
+    mempool = update_mempool(user=g.wallet.user, **g.data)
+    return jsonify(mempool._asdict()), HTTPStatus.OK 
+
+@watchonly_ext.route("/api/v1/mempool", methods=["GET"])
+@api_check_wallet_key("invoice")
+async def api_get_mempool():
+    mempool = get_mempool(g.wallet.user) 
+    if not mempool:
+        mempool = create_mempool(user=g.wallet.user)
+    return jsonify(mempool._asdict()), HTTPStatus.OK