from setuptools import setup, Command
from setuptools.command.easy_install import ScriptWriter
from setuptools.command.install import install as orig_install
from distutils.command.build import build as orig_build

from textwrap import dedent
from subprocess import call
import logging
from os import path
from glob import glob


comp_files = glob('completions/zsh/_*')
languages  = [path.splitext(path.split(po_file)[1])[0]
              for po_file in glob('lang/*.po')]


class build(orig_build):
    """Subclass build command to add a subcommand for building .mo files."""
    sub_commands = orig_build.sub_commands + [('build_mo', None)]


class build_mo(Command):

    """Create machine specific translation files (for i18n via gettext)."""

    description = 'Compile .po files into .mo files'

    def initialize_options(self):
        pass

    def finalize_options(self):
        pass

    def run(self):
        for lang in languages:
            po_file = 'lang/{}.po'.format(lang)
            mo_file = 'build/locale/{}/LC_MESSAGES/udiskie.mo'.format(lang)
            self.mkpath(path.dirname(mo_file))
            self.make_file(
                po_file, mo_file, self.make_mo,
                [po_file, mo_file])

    def make_mo(self, po_filename, mo_filename):
        """Create a machine object (.mo) from a portable object (.po) file."""
        try:
            call(['msgfmt', po_filename, '-o', mo_filename])
        except OSError as e:
            # ignore failures since i18n support is optional:
            logging.warning(e)


# NOTE: Subclassing the setuptools install command alters its behaviour to use
# the distutils code. This is due to some really odd call-context checks in
# the setuptools command.
#
# In fact this is desirable because distutils (correctly) installs data files
# to `sys.prefix` whereas setuptools by default installs to the egg folder
# (which is pretty much useless) and doesn't invoke build commands before
# install. The only real drawback with the distutils behaviour is that it does
# not automatically install dependencies, but we can easily live with that.
#
# Note further that we need to subclass the *setuptools* install command
# rather than the *distutils* one to prevent errors when installing with pip
# from the source distribution.
class install(orig_install):
    pass


def fast_entrypoint_script_template():
    """
    Replacement for ``easy_install.ScriptWriter.template`` to generate faster
    entry points that don't depend on and import pkg_resources.

    NOTE: `pip install` already does the right thing (at least for pip 19.0)
    without our help, but this is still needed for setuptools install, i.e.
    ``python setup.py install`` or develop.
    """
    SCRIPT_TEMPLATE = dedent(r'''
        # encoding: utf-8
        import sys
        from {ep.module_name} import {ep.attrs[0]}

        if __name__ == '__main__':
            sys.exit({func}())
    ''').lstrip()

    class ScriptTemplate(str):
        def __mod__(self, context):
            func = '.'.join(context['ep'].attrs)
            return self.format(func=func, **context)

    return ScriptTemplate(SCRIPT_TEMPLATE)


ScriptWriter.template = fast_entrypoint_script_template()
setup(
    cmdclass={
        'install': install,
        'build': build,
        'build_mo': build_mo,
    },
    data_files=[
        ('share/zsh/site-functions', comp_files),
        *[('share/locale/{}/LC_MESSAGES'.format(lang),
           ['build/locale/{}/LC_MESSAGES/udiskie.mo'.format(lang)])
          for lang in languages],
    ],
)

@@ -327,22 +327,73 @@
 b%0A%0A%0Acomp
-_files
+letions_zsh = glob('completions/zsh/_*')%0Acompletions_bash
  = glob(
@@ -401,29 +401,29 @@
 completions/
-z
+ba
 sh/
-_
 *')%0Alanguage
@@ -3401,16 +3401,82 @@
 files=%5B%0A
+        ('share/bash-completions/completions', completions_bash),%0A
         
@@ -3508,22 +3508,27 @@
 s', comp
-_files
+letions_zsh
 ),%0A     

Install bash completions $PREFIX/share/bash-completions