import json
from typing import List
from pathlib import Path
from importlib import import_module

import click

from ... import config, __version__
from ..utils import pretty_info
from ...binaries.platform import binary_platform


@click.command('version', short_help='Display Prisma Client Python version information.')
@click.option(
    '--json',
    'output_json',
    is_flag=True,
    help='Output version information in JSON format.',
)
def cli(output_json: bool) -> None:
    """Display Prisma Client Python version information."""
    extras = {
        'dev': 'nox',
        'docs': 'mkdocs',
    }
    installed: List[str] = []
    for extra, module in extras.items():
        try:
            import_module(module)
        except ImportError:
            continue
        else:
            installed.append(extra)

    info = {
        'prisma': config.prisma_version,
        'prisma client python': __version__,
        'platform': binary_platform(),
        'expected engine version': config.expected_engine_version,
        'installed extras': installed,
        'install path': str(Path(__file__).resolve().parent.parent.parent),
        'binary cache dir': str(config.binary_cache_dir),
    }

    if output_json:
        click.echo(json.dumps({k.replace(' ', '-'): v for k, v in info.items()}, indent=2))
    else:
        click.echo(pretty_info(info))