from flask import Flask, request
from tinydb import TinyDB, Query

app = Flask(__name__)
db = TinyDB("./wgenroll.json")

@app.route("/createPin", methods=["Post"])
def create_pin():
    r = request.get_json()

    issuer = r["issuer_code"]
    pin = r["pin"]

    if db.contains(Query().issuer == issuer):
        db.insert({"pin": pin, "active": True})
        return {"status": "pin created"}
    else:
        return {"status": "invalid issuer"}


@app.route("/enrollPeer", methods=["Post"])
def enroll_peer():
    r = request.get_json()

    pin = r["pin"]
    public_key = r["public_key"]


def setIssuerCode(code: str):
    db.insert({"issuer": code})

def deleteIssuerCode(code: str):
    db.remove({"issuer": code})