import base58
import hashlib

def sha256_to_wif(hex_str, compressed=True):
    if len(hex_str) != 64:
        raise ValueError("Invalid SHA256 hex string length")
   
    # Step 1: Add 0x80 prefix
    prefix = b'\x80'
    priv_key_bytes = bytes.fromhex(hex_str)
   
    if compressed:
        priv_key_bytes += b'\x01'  # Add compression byte

    extended_key = prefix + priv_key_bytes

    # Step 2: Double SHA-256 and get checksum
    checksum = hashlib.sha256(hashlib.sha256(extended_key).digest()).digest()[:4]

    # Step 3: Append checksum
    wif_bytes = extended_key + checksum

    # Step 4: Base58Check encode
    return base58.b58encode(wif_bytes).decode()

def convert_file_to_wif(input_file, output_file, compressed=True):
    with open(input_file, 'r') as infile, open(output_file, 'w') as outfile:
        for line_num, line in enumerate(infile, start=1):
            hex_str = line.strip()
            try:
                wif = sha256_to_wif(hex_str, compressed=compressed)
                outfile.write(wif + '\n')
            except Exception as e:
                print(f"")

# --- Usage ---
if __name__ == '__main__':
    input_file_path = 'sha256_hashes.txt'   # Input: one hex hash per line
    output_file_path = 'wif_keys.txt'       # Output: WIF keys
    convert_file_to_wif(input_file_path, output_file_path)
    print("Done.")