#!/usr/bin/env python3

# sudo apt install python3-pip
# pip3 install hdwallet tqdm web3 bip32utils

# 132x43 terminal size is recommended

from hdwallet import HDWallet
from hdwallet.cryptocurrencies import Bitcoin as BTC
from hdwallet.hds import BIP32HD
from hdwallet.mnemonics import BIP39Mnemonic
from hdwallet.seeds import BIP39Seed
from os import system, name
from subprocess import check_output
from tqdm import tqdm
from web3 import Web3
import base58
import binascii
import bip32utils
import ecdsa
import ecdsa.der
import ecdsa.util
import hashlib
import math
import mnemonic
import os
import pprint
import random
import re
import requests
import struct
import unittest

alchemy_url = "https://eth-mainnet.g.alchemy.com/v2/"

def clear():
	if name == 'nt':
		_ = system('cls')
	else:
		_ = system('clear')
	return

def bw2wif1(s):
	sha=hashlib.sha256(s.encode()).digest()
	tmp=b'\x80'+sha
	return base58.b58encode_check(tmp).decode()

def bw2wifmany(i,o):
	f1=open(i,'r')
	cnt=sum(1 for line in open(i))
	f2=open(o,'w')
	for line in tqdm(f1, total=cnt, unit=" lines"):
		x=line.rstrip('\n').encode()
		sha=hashlib.sha256(x).digest()
		tmp=b'\x80'+sha
		h=base58.b58encode_check(tmp)
		i=h+b" 0 # "+x+b'\n'
		f2.write(i.decode())
	f1.close()
	f2.flush()
	f2.close()
	return

def checkBTCbal(a):
	r=requests.get('https://blockchain.info/q/addressbalance/'+a)
	if not r.status_code==200:
		print('Error',r.status_code)
		exit(1)
	b=int(r.text)
	return b

def checkETHbal(a,k):
	w3 = Web3(Web3.HTTPProvider(alchemy_url+k))
	cksum=Web3.to_checksum_address(a)
	bal=w3.eth.get_balance(cksum)
	return bal

def bip39(mnemonic_words,n1,n2):
	mobj = mnemonic.Mnemonic("english")
	seed = mobj.to_seed(mnemonic_words)
	bip32_root_key_obj = bip32utils.BIP32Key.fromEntropy(seed)
	bip32_child_key_obj = bip32_root_key_obj.ChildKey(
		n1 + bip32utils.BIP32_HARDEN
	).ChildKey(
		n2 + bip32utils.BIP32_HARDEN
	).ChildKey(
		0 + bip32utils.BIP32_HARDEN
	).ChildKey(0).ChildKey(0)
	return bip32_child_key_obj.WalletImportFormat()

def b58dec(s):
	return base58.b58decode_check(s).hex()

def b58enc(h):
	return base58.b58encode_check(bytes.fromhex(h)).decode()

def hex_to_bytes(hex):
	return bytes.fromhex(hex)

def bytes_to_hex(b):
	return b.hex()

def bytes_to_hex2(b):
	return hex(b)

def count_lines(file):
	return sum(1 for line in open(file, 'r'))

def hex_to_int(hex):
	return int(hex, 16)

def int_to_hex(i):
	return hex(int(i))

def int_to_bytes3(value, length = None):
	if not length and value == 0:
		result = [0]
	else:
		result = []
		for i in range(0, length or 1+int(math.log(value, 2**8))):
			result.append(value >> (i * 8) & 0xff)
		result.reverse()
	return bytearray(result)

def pubkey_to_addrs(pk):
	hdwallet = HDWallet(cryptocurrency=BTC, hd=BIP32HD)
	hdwallet.from_public_key(public_key=pk)
	print()
	print(f'P2PKH:          {hdwallet.address("P2PKH")}')
	print(f'P2SH:           {hdwallet.address("P2SH")}')
	print(f'P2TR:           {hdwallet.address("P2TR")}')
	print(f'P2WPKH:         {hdwallet.address("P2WPKH")}')
	print(f'P2WPKH-In-P2SH: {hdwallet.address("P2WPKH-In-P2SH")}')
	print(f'P2WSH:          {hdwallet.address("P2WSH")}')
	print(f'P2WSH-In-P2SH:  {hdwallet.address("P2WSH-In-P2SH")}')
	print()
	return 

def pvk_to_wif2(key_hex):
	return base58.b58encode_check(b'\x80' + bytes.fromhex(key_hex))

def pvk_to_pubkey(h):
	sk = ecdsa.SigningKey.from_string(h, curve=ecdsa.SECP256k1)
	vk = sk.verifying_key
	return (b'\04' + sk.verifying_key.to_string()).hex()

def pvk_bin_to_addr(pvk):
	signing_key = ecdsa.SigningKey.from_string(pvk, curve = ecdsa.SECP256k1)
	verifying_key = signing_key.get_verifying_key()
	public_key = bytes.fromhex("04") + verifying_key.to_string()
	sha256_1 = hashlib.sha256(public_key)
	ripemd160 = hashlib.new("ripemd160")
	ripemd160.update(sha256_1.digest())
	hashed_public_key = bytes.fromhex("00") + ripemd160.digest()
	checksum_full = hashlib.sha256(hashlib.sha256(hashed_public_key).digest()).digest()
	checksum = checksum_full[:4]
	bin_addr = hashed_public_key + checksum
	return base58.b58encode(bin_addr).decode()

def str_to_hex(text):
	return binascii.hexlify(text.encode()).decode()

def wif_to_pvk(s):
	return base58.b58decode_check(s)[0:].hex()[2:]

def sha256binary(a,b):
	i=open(a,'rb')
	d=i.read()
	h=hashlib.sha256(d).digest()
	o=open(b,'wb')
	o.write(h)
	i.close()
	o.flush()
	o.close()
	return

def ripemd160binary(a,b):
	i=open(a,'rb')
	d=i.read()
	h=hashlib.new('ripemd160',d).digest()
	o=open(b,'wb')
	o.write(h)
	i.close()
	o.flush()
	o.close()
	return

def sha256hex(h):
	return hashlib.sha256(hex_to_bytes(h)).hexdigest()

def ripemd160hex(h):
	t=hashlib.new('ripemd160',hex_to_bytes(h)).hexdigest()
	return t

def hex_to_string(h):
	return bytes.fromhex(h).decode('utf-8')

clear()
while True:
	print('Tool for cc v0.56 (C) 2023-2025 Aftermath @Tzeeck')
	print('Mostly all options are for BTC if not mentioned differently')
	print('1. seed phrase = mnemonic to HDWallet')
	print('5. seed hex to HDWallet')
	print('2. mnemonic to WIF - BCH Bitcoin Cash')
	print('3. mnemonic to WIF - BTC Bitcoin')
	print('4. mnemonic to WIF - LTC Litecoin')
	print('6. private key integer to WIF')
	print('7. private key integer to HDWallet')
	print('8. private key hex to WIF')
	print('9. private key hex to public key')
	print('a. private key hex to HDWallet')
	print('b. brainwallet to WIF - single')
	print('c. brainwallet to WIF - many (a file)')
	print('d. public key to address')
	print('h. public key hex to hash160')
	print('g. address to string')
	print('i. address to public key')
	print('s. hex to string')
	print('r. string to hex')
	print('f. string to address')
	print('t. hex to int')
	print('u. int to hex')
	print('n. decode Base58Check to hex')
	print('o. encode hex to Base58Check')
	print('j. generate set')
	print('k. generate HD wallet')
	print('l. check BTC balance - single')
	print('m. check ETH balance - single')
	print('e. WIF to private key hex')
	print('p. bytes (file) to hex')
	print('q. convert hex to bytes (to file)')
	print('v. count lines in file')
	print('w. binary SHA256 (files)')
	print('x. hex SHA256 (a file)')
	print('y. binary RIPEMD160 (files)')
	print('z. hex RIPEMD160 (a file)')
	print('A. get SHA256 of hex (hex converted to binary)')
	print('B. get RIPEMD160 of hex (hex converted to binary)')
	print('C. ETH mnemonic (seed phrase) to address and private key')
	print()
	m=input('Select option or enter empty to quit: ')

	match m:
		case 'b':
			a=input('Enter brainwallet: ')
			print('\nWIF: '+bw2wif1(a)+'\n')
		case 'c':
			a=input('Enter input filename [input.txt]: ')
			b=input('Enter output filename [output.txt]: ')
			if a=='':
				a='input.txt'
			if b=='':
				b='output.txt'
			bw2wifmany(a,b)
			print('Done!\n')
		case 'l':
			a=input('Enter BTC address: ')
			sat=checkBTCbal(a)
			print('\n',a,'\t',sat,'sat\t',sat/100000,'mBTC\t',sat/100000000,'BTC\n')
		case 'm':
			a=input('Enter ETH address: ')
			k=input('Enter Alchemy API key: ')
			sat=checkETHbal(a,k)
			print('\n',a,' = ',sat/1e18,' ETH\n')
		case '2':
			a=input('Enter BCH mnemonic (seed phrase, usually 12 words): ')
			print('\nWIF: '+bip39(a,44,145)+'\n')
		case '3':
			a=input('Enter BTC mnemonic (seed phrase, usually 12 words): ')
			print('\nWIF: '+bip39(a,84,0)+'\n')
		case '4':
			a=input('Enter LTC mnemonic (seed phrase, usually 12 words): ')
			print('\nWIF: '+bip39(a,84,2)+'\n')
		case 'n':
			a=input('Enter Base58Check encoded string: ')
			print('\n'+b58dec(a)+'\n')
		case 'o':
			a=input('Enter hex string: ')
			print('\n'+b58enc(a)+'\n')
		case 'q':
			a=input('Enter hex string: ')
			open('output.bin','wb').write(hex_to_bytes(a))
			print('\nWritten to output.bin file\n')
		case 'v':
			a=input('Enter filename [input.txt]: ')
			if a=='':
				a='input.txt'
			print('\nLines count: '+str(count_lines(a))+'\n')
		case 't':
			a=input('Enter hex: ')
			print('\nInteger: '+str(hex_to_int(a))+'\n')
		case 'u':
			a=input('Enter int: ')
			print('\nHex: '+int_to_hex(a)+'\n')
		case 'd':
			a=input('Enter public key: ')
			pubkey_to_addrs(a)
		case '8':
			a=input('Enter private key in hex: ')
			a=a.zfill(64)
			print('\nWIF uncomp: '+pvk_to_wif2(a).decode('ascii'))
			hdwallet = HDWallet(cryptocurrency=BTC, hd=BIP32HD)
			hdwallet.from_private_key(private_key=a)
			print(f'WIF comp: {hdwallet.wif()}\n')
		case '9':
			a=input('Enter private key in hex: ')
			a=a.zfill(64)
			print('\nPublic key: '+pvk_to_pubkey(hex_to_bytes(a))+'\n')
		case 'r':
			a=input('Enter string: ')
			print('\nHex: '+str_to_hex(a)+'\n')
		case 'e':
			w=input('Enter WIF: ')
			p=wif_to_pvk(w)
			l=len(p)
			if l==66:
				p=p[:-2]
			elif l==68:
				p=p[:-4]
			print('\nPrivate key: '+p+'\n')
		case 'w':
			a=input('Enter input filename [input.bin]: ')
			b=input('Enter output filename [output.bin]: ')
			if a=='':
				a='input.bin'
			if b=='':
				b='output.bin'
			sha256binary(a,b)
		case 'x':
			a=input('Enter input filename [input.bin]: ')
			if a=='':
				a='input.bin'
			i=open(a,'rb')
			d=i.read()
			print('\nSHA256: '+hashlib.sha256(d).hexdigest()+'\n')
			i.close()
		case 'y':
			a=input('Enter input filename [input.bin]: ')
			b=input('Enter output filename [output.bin]: ')
			if a=='':
				a='input.bin'
			if b=='':
				b='output.bin'
			ripemd160binary(a,b)
		case 'z':
			a=input('Enter input filename [input.bin]: ')
			if a=='':
				a='input.bin'
			i=open(a,'rb')
			d=i.read()
			h=hashlib.new('ripemd160',d).hexdigest()
			print('\nRIPEMD160: '+h+'\n')
			i.close()
		case 'j':
			pvk=os.urandom(32)
			pvkhex=pvk.hex()
			pvkhex=pvkhex.zfill(64)
			print('\nPrivate key: '+pvkhex)
			wif1=pvk_to_wif2(pvkhex).decode('ascii')
			hdwallet = HDWallet(cryptocurrency=BTC, hd=BIP32HD)
			hdwallet.from_private_key(private_key=pvkhex)
			wif2=hdwallet.wif()
			print('WIF uncomp : '+wif1)
			print('WIF comp   : '+wif2)
			print('Public key : '+pvk_to_pubkey(pvk))
			pubkey_to_addrs(pvk_to_pubkey(pvk))
		case 'A':
			h=input('Enter hex: ')
			print('\nSHA256: '+sha256hex(h))
			print()
		case 'B':
			h=input('Enter hex: ')
			print('\nRIPEMD160: '+ripemd160hex(h))
			print()
		case 'f':
			h=input('Enter string (up to 20 chars): ')
			if len(h) > 20:
				print('Too long!')
			else:
				h=b'\x00'+str.encode(h)
				while len(h)<21:
					h=h+b'\x20'
				print('\nAddress: '+base58.b58encode_check(h).decode('utf-8'))
				print()
		case 'g':
			s=input('Enter address: ')
			print('\nString: '+base58.b58decode_check(s).decode('utf-8'))
			print()
		case 'h':
			h=input('Enter public key: ')
			print('\nHASH160: '+ripemd160hex(sha256hex(h)))
			print()
		case 's':
			a=input('Enter hex: ')
			try:
				print('\nString: '+hex_to_string(a)+'\n')
			except:
				print('\nNot UTF-8 printable bytes, can\'t convert!\n')
		case 'p':
			a=input('Enter input filename [input.bin]: ')
			if a=='':
				a='input.bin'
			i=open(a,'rb')
			d=i.read()
			print('\nHex: '+d.hex())
			print()
			i.close()
		case 'i':
			a=input('Enter address: ')
			req = ('https://blockchain.info/q/pubkeyaddr/' + a)
			resp = requests.get(req)
			if resp.status_code == 404: print('\nNot found\n')
			if resp.status_code == 200: print('\nPublic key:\n'+resp.text+'\n')
		case '6':
			i=input('Enter integer: ')
			h=hex(int(i))[2:]
			h=h.zfill(64)
			if len(h)<66:
				h='80'+h
			p=bytes.fromhex(h)
			b=base58.b58encode_check(p)
			print('\nWIF uncomp: '+pvk_to_wif2(h).decode('ascii'))
			print((b'WIF comp: '+b+b'\n').decode('utf-8'))
		case '7':
			a=input('Enter integer: ')
			b=int_to_hex(a)[2:]
			c=b.zfill(64)
			hdwallet = HDWallet(cryptocurrency=BTC, hd=BIP32HD)
			hdwallet.from_private_key(private_key=c)
			d=hdwallet.dump()
			pp = pprint.PrettyPrinter(indent=4)
			print('\n'+pp.pformat(d))
			print('\nWIF uncomp: '+pvk_to_wif2(c).decode('ascii')+'\n')
		case '5':
			j=input('Enter seed hex: ')
			j=j.zfill(128)
			hdwallet = HDWallet(cryptocurrency=BTC, hd=BIP32HD)
			hdwallet.from_seed(seed=BIP39Seed(j))
			d=hdwallet.dump()
			print()
			pprint.pprint(d)
			print()
		case 'a':
			j=input('Enter private key hex: ')
			j=j.zfill(64)
			hdwallet = HDWallet(cryptocurrency=BTC, hd=BIP32HD)
			hdwallet.from_private_key(private_key=j)
			d=hdwallet.dump()
			print()
			pprint.pprint(d)
			print('\nWIF uncomp: '+pvk_to_wif2(j).decode('ascii')+'\n')
		case 'k':
			hdwallet = HDWallet(cryptocurrency=BTC, hd=BIP32HD)
			hdwallet.from_seed(seed=BIP39Seed(seed=os.urandom(64).hex()))
			pp = pprint.PrettyPrinter(indent=4)
			d = hdwallet.dump()
			print('\n'+pp.pformat(d)+'\n')
		case '1':
			j=input('Enter seed phrase = mnemonic: ')
			hdwallet = HDWallet(cryptocurrency=BTC, hd=BIP32HD)
			hdwallet.from_mnemonic(mnemonic=BIP39Mnemonic(mnemonic=j))
			d=hdwallet.dump()
			print()
			pprint.pprint(d)
			print()
		case 'C':
			k=input('Enter Alchemy API key: ')
			j=input('Enter seed phrase = mnemonic: ')
			alchemy_url = 'https://eth-mainnet.g.alchemy.com/v2/'+k
			w3 = Web3(Web3.HTTPProvider(alchemy_url))
			w3.eth.account.enable_unaudited_hdwallet_features()
			try:
				acc = w3.eth.account.from_mnemonic(j)
			except:
				print('Error: bad mnemonic')
				pass
			address = w3.to_checksum_address(acc.address)
			h=acc._private_key.hex()
			print()
			print('Checksum address: '+address)
			print('Private key: '+h)
			print()
		case '':
			exit(0)
	input('Press Enter...')