providence/bin/providence.py

import json
import sys
import argparse
from pathlib import Path
import os
import re
import orjson

import requests

parser = argparse.ArgumentParser(
    description="providence is an open source to find company domain names based on a given company name",
    epilog="More info: https://github.com/adulau/providence",
)

parser.add_argument("-v", help="increase output verbosity", action="store_true")
parser.add_argument("-n", "--name", type=str, help="Company name to find")
parser.add_argument(
    "-t",
    "--tld",
    type=str,
    help="Limit to a specific TLD (if not, all known TLDs are tested)",
    action="append"
)
#parser.add_argument("","")

args = parser.parse_args()

homedir = str(Path.home())

if not args.name:
    parser.print_usage()
    parser.exit()


def cache_suffixes(url="https://publicsuffix.org/list/public_suffix_list.dat"):
    tlds = set()
    cachedfile = os.path.join(homedir, ".providence-suffix")
    if not os.path.exists(cachedfile):
        r = requests.get(url)
        with open(cachedfile, 'wb') as fd:
            fd.write(r.content)

    with open(cachedfile, 'r') as fd:
        for line in fd:
            tld = line.rstrip()
            if not tld.startswith("//"):
                tlds.add(tld)
    return tlds


def guess_name(name=None, tlds=None):
    guessed_names = set()
    if name is None:
        return False
    for t in tlds:
        guessed_names.add(f"{name}.{t}")
    return guessed_names


if not args.tld:
    tlds = cache_suffixes()
else:
    tlds = args.tld

results = set()

if re.search(r' {1,}', args.name):
    # replace with -
    name = args.name
    reversed_name = " ".join(args.name.split(" ")[::-1])
    p = name.replace(" ", "-")
    results.update(guess_name(name=p, tlds=tlds))
    p = reversed_name.replace(" ", "-")
    results.update(guess_name(name=p, tlds=tlds))
    p = name.replace(" ", "")
    results.update(guess_name(name=p, tlds=tlds))
    p = reversed_name.replace(" ", "")
    results.update(guess_name(name=p, tlds=tlds))
    # try with removal of word
    for word in args.name.split(" "):
        results.update(guess_name(name=word, tlds=tlds))
else:
    results.update(guess_name(name=args.name, tlds=tlds))

output = {}
output['results'] = list(results)
j = orjson.dumps(output).decode()

print(j)