class ZKParameters(NamedTuple):
    """
    The parameters required for creating the Zero-Knowledge Cryptosystem
    """
    curve: str          # the name of the standardized elliptic curve to use for cyclic group point generation
    d: int              # the large prime number used to generate an elliptic curve point and as a modulo

class ZKSignature(NamedTuple):
    """
    A cryptographic signature distinct from a hash that can be used to validate the same input in the future
    """
    params: ZKParameters  # the reference ZK parameters
    signature: int        # the calculated signature derived from the ZK curve and user secret

class ZKChallenge(NamedTuple):
    """
    A cryptographic challenge created by a user based on the signature (derived from the password) 
    """
    params: ZKParameters  # the reference ZK parameters
    token: int            # the server-provided random tokens
    c: int                # the hash derived from the signature, a random point, and the random token
    z: int                # a value derived from the random point and password hash

#!/usr/bin/env python3
"""
Extremely simple example of NoKnow ZK Proof implementation
"""

from noknow import ZKProof
from getpass import getpass


def main():
    # the state of `zk` should be known to both Prover (Client) and Verifier (Server)
    zk = ZKProof.new()
    # `signature` is sent to Server along with `zk.params` for persistent storage
    signature = zk.create_signature(getpass("Create Password: "))

    while True:
        # Server generates random token and sends to Client
        token = ZKProof.random_token()

        # Client generates a challenge with the token and submits to Server
        challenge = zk.create_challenge(getpass("Enter Password: "), token)

        # Server proves the challenge with the shared token, user-submitted challenge, and stored signature
        if zk.prove_challenge(challenge, signature, token):
            print("Authentication Successful!\n")
            break
        else:
            print("Authentication Failed!\n")

if __name__ == "__main__":
    main()