"""Definition of connect function."""

import logging
import os

from minio import Minio


def connect_minio() -> Minio:
    """Connect to MinIO server."""
    # ensure necessary env vars available
    env_var_list = [
        'MINIO_ENDPOINT',
        'MINIO_ACCESS_KEY',
        'MINIO_SECRET_KEY',
        'MINIO_BUCKET_NAME',
    ]
    for env_var in env_var_list:
        # ensure env var set
        assert (
            env_var in os.environ
        ), f"environment variable not set: {
            env_var
        }"
    # prepare arguments
    minio_endpoint = os.getenv('MINIO_ENDPOINT')
    minio_access_key = os.getenv('MINIO_ACCESS_KEY')
    minio_secret_key = os.getenv('MINIO_SECRET_KEY')
    minio_bucket_name = os.getenv('MINIO_BUCKET_NAME')
    # connect client
    client = Minio(
        endpoint=minio_endpoint,
        access_key=minio_access_key,
        secret_key=minio_secret_key,
        secure=False,
    )
    # ensure bucket exists
    if not client.bucket_exists(bucket_name=minio_bucket_name):
        logging.info('creating bucket: %s', minio_bucket_name)
        client.make_bucket(bucket_name=minio_bucket_name)
    logging.debug('finished')
    return client