import argparse
import mysql.connector
import configparser
import subprocess

def main():
    """
    Entry point of the script.
    """
    # Set up the argument parser
    parser = argparse.ArgumentParser()
    parser.add_argument('-c', '--config', help='path to the configuration file', default="config.ini")
    args = parser.parse_args()

    # Read the MariaDB server details from the configuration file
    config = configparser.ConfigParser()
    config.read(args.config)

    # Connect to the database
    conn = mysql.connector.connect(
        host=config['database']['host'],
        user=config['database']['user'],
        password=config['database']['password'],
        database=config['database']['database']
    )
    cursor = conn.cursor()

    # Create the table if it doesn't already exist
    cursor.execute('''
    CREATE TABLE IF NOT EXISTS onion_keys (hostname text, public_key text, secret_key text)
    ''')

    # Start mkp224o
    process = subprocess.Popen(['mkp224o', '-y', '-f', 'filter'], stdout=subprocess.PIPE)

    # Run mkp224o until the user interrupts the script
    while True:
        try:
            # Read the output from mkp224o in blocks of four lines
            hostname = None
            public_key = None
            secret_key = None
            while True:
                line = process.stdout.readline().decode().strip()
                if line == '---':
                    break

                # Split the line into fields
                fields = line.split(':', 1)
                if len(fields) != 2:
                    continue
                key, value = fields
                key = key.strip()
                value = value.strip()

                # Extract the hostname, public key, and secret key from the output
                if key == 'hostname':
                    hostname = value
                elif key == 'hs_ed25519_public_key':
                    public_key = value
                elif key == 'hs_ed25519_secret_key':
                    secret_key = value
    
            if hostname:
                # Insert the data into the database
                cursor.execute('''
                INSERT INTO onion_keys (hostname, public_key, secret_key) VALUES (%s, %s, %s)
                ''', (hostname, public_key, secret_key))

                # Save the changes to the database
                conn.commit()
        except KeyboardInterrupt:
            # If the user interrupts the script, break out of the loop
            break

main()