#!/usr/bin/env python3
# Copyright (c) 2015-2021 The Bitcoin Core developers
# Copyright (c) 2023 The Floofy Core developers
# Distributed under the MIT software license, see the accompanying 
# file COPYING or http://www.opensource.org/licenses/mit-license.php.

import sys
from secrets import token_hex, token_urlsafe
import hmac

def generate_salt(size):
    """Create size byte hex salt"""
    return token_hex(size)

def generate_password():
    """Create 32 byte b64 password"""
    return token_urlsafe(32)

def password_to_hmac(salt, password):
    m = hmac.new(salt.encode('utf-8'), password.encode('utf-8'), 'SHA256')
    return m.hexdigest()

if len(sys.argv) < 2:
    sys.stderr.write('Please include username as an argument.\n')
    sys.exit(0)

username = sys.argv[1]
password = generate_password()

# Create 16 byte hex salt
salt = generate_salt(16)
password_hmac = password_to_hmac(salt, password)

print('String to be appended to floofy.conf:')
print(f'rpcauth={username}:{salt}${password_hmac}')
print(f'Your password:\n{password}')
