add shebang and make executable

[vv01f]

basically let the script run directly

[vv01f]

as I had difficulties (on NixOS) with the dependecies I did a little rewrite:

0. create the two files, make makenpub.py executable (chmod u+x makenpub.py)
1. run the shell.nix to get the dependencies right: nix-shell
2. use either generation from bip39 or vanity with the shown commands

with import <nixpkgs> {};

mkShell {
  buildInputs = [
    python3
    pkg-config    # required by secp256k1 build
    secp256k1     # secp256k1 C library
    gcc           # compiler
    gnumake       # correct name for make
    parallel-full # to use all cores if needed
  ];

  shellHook = ''
    printf "Setting up virtual environment…\n"
    if [ ! -d ".venv" ]; then
      python3 -m venv .venv
    fi
    source .venv/bin/activate
    pip install --upgrade pip
    pip install nostr-tools bech32 mnemonic
    printf "Virtual environment ready.\nEither use './makenpub.py vanity <target>'\nOr './makenpub.py bip39 <seedphase>'\n"
  '';
}

#!/usr/bin/env python3
import sys
import os
from mnemonic import Mnemonic
from secp256k1 import PrivateKey as SecpPrivateKey
from nostr_tools import generate_keypair, to_bech32

def generate_vanity(prefix: str):
    attempts = 0
    while True:
        priv_bytes = os.urandom(32)
        priv = SecpPrivateKey(priv_bytes)
        pub_bytes = priv.pubkey.serialize(compressed=False)[1:]
        npub = to_bech32("npub", pub_bytes.hex())
        npriv = to_bech32("nsec", priv_bytes.hex())
        attempts += 1
        if npub.lower().startswith(prefix.lower()):
            print(f"Attempts: {attempts}")
            print("Private key (hex):", priv_bytes.hex())
            print("Private key (Bech32):", npriv)
            print("Public key (hex):", pub_bytes.hex())
            print("Public key (Bech32):", npub)
            break

def generate_from_bip39(seed_phrase: str):
    mnemo = Mnemonic("english")
    if not mnemo.check(seed_phrase):
        print("Error: invalid BIP39 seed phrase (checksum failed).")
        sys.exit(1)
    seed = mnemo.to_seed(seed_phrase)
    priv_bytes = seed[:32]
    priv = SecpPrivateKey(priv_bytes)
    pub_bytes = priv.pubkey.serialize(compressed=False)[1:]
    npub = to_bech32("npub", pub_bytes.hex())
    npriv = to_bech32("nsec", priv_bytes.hex())
    print("Private key (hex):", priv_bytes.hex())
    print("Private key (Bech32):", npriv)
    print("Public key (hex):", pub_bytes.hex())
    print("Public key (Bech32):", npub)

def main():
    if len(sys.argv) < 3:
        print("Usage:")
        print("  ./makenpub.py vanity <prefix>")
        print("  ./makenpub.py bip39 <12‑or‑24‑word‑seed>")
        sys.exit(1)

    mode = sys.argv[1].lower()
    arg = " ".join(sys.argv[2:])

    if mode == "vanity":
        generate_vanity(arg)
    elif mode == "bip39":
        generate_from_bip39(arg)
    else:
        print("Unknown mode:", mode)
        sys.exit(1)

if __name__ == "__main__":
    main()