'''utility functions for printing version information


see pandas/pandas/util/_print_versions.py'''

import codecs
import importlib
import locale
import os
import platform
import struct
import subprocess
import sys


def get_sys_info():
    'Returns system information as a dict'

    blob = []

    # get full commit hash
    commit = None
    if os.path.isdir('.git') and os.path.isdir('xarray'):
        try:
            pipe = subprocess.Popen(
                'git log --format="%H" -n 1'.split(' '),
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
            )
            so, _ = pipe.communicate()
        except Exception:
            pass
        else:
            if pipe.returncode == 0:
                commit = so
                try:
                    commit = so.decode('utf-8')
                except ValueError:
                    pass
                commit = commit.strip().strip('"')

    blob.append(('commit', commit))

    try:
        (sysname, _, release, _, machine, processor) = platform.uname()
        blob.extend(
            [
                ('python', sys.version),
                ('python-bits', struct.calcsize('P') * 8),
                ('OS', '%s' % (sysname)),
                ('OS-release', '%s' % (release)),
                # ("Version", "%s" % (version)),
                ('machine', '%s' % (machine)),
                ('processor', '%s' % (processor)),
                ('byteorder', '%s' % sys.byteorder),
                ('LC_ALL', '%s' % os.environ.get('LC_ALL', 'None')),
                ('LANG', '%s' % os.environ.get('LANG', 'None')),
                ('LOCALE', '%s.%s' % locale.getlocale()),
            ]
        )
    except Exception:
        pass

    return blob


def show_versions(as_json=False):
    sys_info = get_sys_info()

    deps = [
        # (MODULE_NAME, f(mod) -> mod version)
        ('esmlab', lambda mod: mod.__version__),
        ('xarray', lambda mod: mod.__version__),
        ('pandas', lambda mod: mod.__version__),
        ('numpy', lambda mod: mod.__version__),
        ('scipy', lambda mod: mod.__version__),
        ('xesmf', lambda mod: mod.__version__),
        ('cftime', lambda mod: mod.__version__),
        ('dask', lambda mod: mod.__version__),
        ('distributed', lambda mod: mod.__version__),
        # setup/test
        ('setuptools', lambda mod: mod.__version__),
        ('pip', lambda mod: mod.__version__),
        ('conda', lambda mod: mod.__version__),
        ('pytest', lambda mod: mod.__version__),
        # Misc.
        ('IPython', lambda mod: mod.__version__),
        ('sphinx', lambda mod: mod.__version__),
    ]

    deps_blob = list()
    for (modname, ver_f) in deps:
        try:
            if modname in sys.modules:
                mod = sys.modules[modname]
            else:
                mod = importlib.import_module(modname)
        except Exception:
            deps_blob.append((modname, None))
        else:
            try:
                ver = ver_f(mod)
                deps_blob.append((modname, ver))
            except Exception:
                deps_blob.append((modname, 'installed'))

    if as_json:
        try:
            import json
        except Exception:
            import simplejson as json

        j = dict(system=dict(sys_info), dependencies=dict(deps_blob))

        if as_json is True:
            print(j)
        else:
            with codecs.open(as_json, 'wb', encoding='utf8') as f:
                json.dump(j, f, indent=2)

    else:

        print('\nINSTALLED VERSIONS')
        print('------------------')

        for k, stat in sys_info:
            print('%s: %s' % (k, stat))

        print('')
        for k, stat in deps_blob:
            print('%s: %s' % (k, stat))


def main():
    from optparse import OptionParser

    parser = OptionParser()
    parser.add_option(
        '-j',
        '--json',
        metavar='FILE',
        nargs=1,
        help='Save output as JSON into file, pass in ' "'-' to output to stdout",
    )

    (options, args) = parser.parse_args()

    if options.json == '-':
        options.json = True

    show_versions(as_json=options.json)

    return 0


if __name__ == '__main__':
    sys.exit(main())