KaikOut/kaikout/__main__.py

#!/usr/bin/env python3

# Kaikout: Kaiko Out anti-spam bot for XMPP
# Copyright (C) 2024 Schimon Zackary
# This file is part of Kaikout.
# See the file LICENSE for copying permission.

# from kaikout.about import Documentation
from kaikout.utilities import Config, Toml
# from kaikout.xmpp.chat import XmppChat
from kaikout.xmpp.client import XmppClient
from getpass import getpass
from argparse import ArgumentParser
import logging
import os
import shutil
# import slixmpp
# import sys

def main():

    directory = os.path.dirname(__file__)

    # Copy data files
    directory_data = Config.get_default_config_directory()
    # TODO Utilize the actual data directory
    #directory_data = Config.get_default_data_directory()
    if not os.path.exists(directory_data):
        directory_assets = os.path.join(directory, 'assets')
        directory_assets_new = shutil.copytree(directory_assets, directory_data)
        print(f'Data directory {directory_assets_new} has been created and populated.')

    # Copy settings files
    directory_settings = Config.get_default_config_directory()
    if not os.path.exists(directory_settings):
        directory_configs = os.path.join(directory, 'configs')
        directory_settings_new = shutil.copytree(directory_configs, directory_settings)
        print(f'Configuration directory {directory_settings_new} has been created and populated.')

    # Setup the command line arguments.
    parser = ArgumentParser(description=XmppClient.__doc__)

    # Output verbosity options.
    parser.add_argument("-q", "--quiet", help="set logging to ERROR",
                        action="store_const", dest="loglevel",
                        const=logging.ERROR, default=logging.INFO)
    parser.add_argument("-d", "--debug", help="set logging to DEBUG",
                        action="store_const", dest="loglevel",
                        const=logging.DEBUG, default=logging.INFO)

    args = parser.parse_args()

    # Setup logging.
    logging.basicConfig(level=args.loglevel,
                        format='%(levelname)-8s %(message)s')

    # Configure settings file
    file_settings = os.path.join(directory_settings, 'accounts.toml')
    if not os.path.exists(file_settings):
        directory_configs = os.path.join(directory, 'configs')
        file_settings_empty = os.path.join(directory_configs, 'accounts.toml')
        shutil.copyfile(file_settings_empty, file_settings)

    data_settings = Toml.open_file(file_settings)

    # Configure account
    data_settings_account = data_settings['xmpp']['client']

    settings_account = {
        'alias': 'Set an Alias',
        'jid': 'Set a Jabber ID',
        'password': 'Input Password'
    }

    for key in settings_account:
        data_settings_account_value = data_settings_account[key]
        if not data_settings_account_value:
            settings_account_message = settings_account[key]
            while not data_settings_account_value:
                if key == 'password':
                    data_settings_account_value = getpass(f'{settings_account_message}: ')
                else:
                    data_settings_account_value = input(f'{settings_account_message}: ')
            data_settings_account[key] = data_settings_account_value

    Toml.save_file(file_settings, data_settings)

    # Try configuration file
    account_xmpp = Config.get_values('accounts.toml', 'xmpp')
    if 'client' in account_xmpp:
        account_xmpp_client = account_xmpp['client']
        jid = account_xmpp_client['jid']
        password = account_xmpp_client['password']
        alias = account_xmpp_client['alias'] if 'alias' in account_xmpp_client else None
        hostname = account_xmpp_client['hostname'] if 'hostname' in account_xmpp_client else None
        port = account_xmpp_client['port'] if 'port' in account_xmpp_client else None
        XmppClient(jid, password, hostname, port, alias)

if __name__ == '__main__':
    main()