import logging
import sys

from cool_domain.word_list import WordList

from . import PKG_NAME, PKG_VERSION
from .cache import Cache
from .constants import LANGS
from .factory import Factory
from .logs import setup_logs
from .params import parse_parameters
from .tld_list import TLDList


def main() -> int:
    params = parse_parameters(sys.argv[1:])
    setup_logs(params)
    logging.getLogger().info("%s %s", PKG_NAME, PKG_VERSION)
    langs = LANGS
    langs["custom"] = WordList("custom", params.custom_words)
    with (
        Cache(filepath=params.cache_file, no_load=params.cache) as cache,
        TLDList(cache=cache, remote_uri=params.tld_list_uri) as tld_list,
        langs[params.lang] as word_list,
    ):
        factory = Factory(tld_list=tld_list, word_list=word_list, words=params.words)
        logging.getLogger().info(
            "Searching %s domain%s", params.count, "s" if params.count > 1 else ""
        )
        for _ in range(params.count):
            if params.dig:
                logging.getLogger().info(factory.random_available_domain())
            else:
                logging.getLogger().info(factory.random_domain())
    return 0