from hashlib import sha512
import fire
from urllib.parse import urlparse
from pathlib import Path
import re

home = Path.home()
credential_path = Path(home / ".ssh/secret_key")

with credential_path.open("r") as f:
    chunks = list(map(lambda x:x.strip(), f.readlines()))
    chunks = list(filter(lambda x: not re.search("^-----[BEGIN|END]", x) and x != "", chunks))
    secret_key = "".join(chunks)

chrs = list("0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ!@#$%^&*()")
def to_chr(i: int) -> str:
    return chrs[i%len(chrs)]

def main(url="https://google.com"):
    """
    Parameters:
        url(str): fullpath of url
    Returns:
        None
    """
    o = urlparse(url)
    salt = o.netloc

    public_key = sha512(bytes(secret_key + salt, "utf8")).digest()
    public_key = "".join([to_chr(c) for c in public_key])
    print(f"salt = {salt}")
    print(f"public_key = {public_key}")
    print(f"public_key of top 10 = {public_key[:10]}")

if __name__ == "__main__":
    fire.Fire(main)