"""Prints environment information.
"""
import os
from pathlib import Path

from milc import cli
from qmk_cli.helpers import is_qmk_firmware, is_qmk_userspace


@cli.argument('var', arg_only=True, default=None, nargs='?', help='Optional variable to query')
@cli.subcommand('Prints environment information.')
def env(cli):
    home = os.environ.get('QMK_HOME', '')
    userspace = os.environ.get('QMK_USERSPACE', '')
    data = {
        'QMK_HOME': home,
        'QMK_FIRMWARE': home if is_qmk_firmware(Path(home)) else '',
        'QMK_USERSPACE': userspace if is_qmk_userspace(Path(userspace)) else '',
        'QMK_DISTRIB_DIR': os.environ.get('QMK_DISTRIB_DIR', ''),
        'QMK_PATH_PREFIX': os.environ.get('QMK_PATH_PREFIX', '')
    }

    # Now munge the current cli config
    for key, val in cli.config.general.items():
        converted_key = 'QMK_' + key.upper()
        data[converted_key] = val

    if cli.args.var:
        # dump out requested arg
        print(data[cli.args.var])
    else:
        # dump out everything
        for key, val in data.items():
            print(f'{key}="{val}"')