import logging
import os
from glob import glob

from setuptools import Command, setup
from setuptools.command.build import build as _build
from setuptools.command.install import install as _install

## Snippet of code found on
## http://developer.berlios.de/snippet/detail.php?type=snippet&id=100019
## Useful to handle *.po files under distutils


class messages(Command):
    description = "Extract messages from source files"

    user_options = [
        ("merge", "m", "Merge message catalogs after " "extracting template")
    ]

    boolean_options = ["merge"]

    def initialize_options(self):
        self.merge = False

    def finalize_options(self):
        pass

    def run(self):
        wd = os.getcwd()
        name = self.distribution.get_name()
        logging.info("Extracting messages from %s" % name)
        os.chdir("po")
        os.system("pygettext -k i18n -o %s.pot ../%s" % (name, name))
        os.chdir(wd)
        if self.merge:
            self.run_command("merge")


class merge(Command):
    description = "Merge message catalogs with template"

    user_options = [("lang=", "l", "Only merge specified language")]

    def initialize_options(self):
        self.lang = None

    def finalize_options(self):
        pass

    def run(self):
        wd = os.getcwd()
        os.chdir("po")
        if self.lang is not None:
            filename = "%s.po" % self.lang
            if os.path.exists(filename):
                self.merge(filename)
            else:
                logging.error("No catalog found for language %s" % self.lang)
        else:
            files = glob("*.po")
            for f in files:
                self.merge(f)
        os.chdir(wd)

    def merge(self, f):
        name = self.distribution.get_name()
        logging.info("Merging catalog %s" % f)
        os.system("msgmerge %s %s.pot -o %s" % (f, name, f))


class build_messages(Command):
    description = "Compile message catalogs"
    user_options = [("build-dir=", "d", "directory to build message catalogs in")]

    def initialize_options(self):
        self.build_dir = None

    def finalize_options(self):
        self.set_undefined_options("build", ("build_messages", "build_dir"))

    def run(self):
        self.mkpath(self.build_dir)
        po_files = os.path.join("po", "*.po")
        po_files = glob(po_files)
        for f in po_files:
            base = os.path.basename(f)
            base = os.path.splitext(base)[0]
            out_file = os.path.join(self.build_dir, "%s.gmo" % base)
            logging.info("Building catalog %s > %s" % (f, out_file))
            cmd = 'msgfmt -o "%s" %s' % (out_file, f)
            os.system(cmd)


class build(_build):
    # integrate build_message
    def has_messages(self):
        return True

    def initialize_options(self):
        _build.initialize_options(self)
        self.build_messages = None

    def finalize_options(self):
        _build.finalize_options(self)
        if self.build_messages is None:
            self.build_messages = os.path.join(self.build_base, "po")

    _build.user_options.append(
        ("build-messages=", None, "build directory for messages")
    )
    _build.sub_commands.append(("build_messages", has_messages))


class install_messages(Command):
    description = "Installs message catalogs"

    user_options = [
        ("install-dir=", "d", "directory to install scripts to"),
        ("build-dir=", "b", "build directory (where to install from)"),
        ("skip-build", None, "skip the build steps"),
    ]

    boolean_options = ["skip-build"]

    def initialize_options(self):
        self.install_dir = None
        self.build_dir = None
        self.skip_build = None

    def finalize_options(self):
        self.set_undefined_options("build", ("build_messages", "build_dir"))
        self.set_undefined_options(
            "install",
            ("install_data", "install_dir"),
            ("force", "force"),
            ("skip_build", "skip_build"),
        )
        self.install_dir = os.path.join(self.install_dir, "share", "locale")

    def run(self):
        if not self.skip_build:
            self.run_command("build_messages")
        catalogs = glob(os.path.join(self.build_dir, "*.gmo"))
        for c in catalogs:
            base = os.path.basename(c)
            base = os.path.splitext(base)[0]
            out_dir = os.path.join(self.install_dir, base, "LC_MESSAGES")
            self.mkpath(out_dir)
            out_file = os.path.join(out_dir, "%s.mo" % self.distribution.get_name())
            self.copy_file(c, out_file)


## Class modified to add manpages command
class install(_install):
    def has_messages(self):
        return True

    def has_manpages(self):
        return True

    _install.sub_commands.append(("install_messages", has_messages))
    _install.sub_commands.append(("install_manpages", has_manpages))


### End of snippet ###


class install_manpages(Command):
    description = "Install man pages"

    user_options = []

    def initialize_options(self):
        self.man_pages = man_pages
        self.install_dir = None
        self.root = None

    def finalize_options(self):
        self.set_undefined_options(
            "install",
            ("install_data", "install_dir"),
            ("root", "root"),
        )

    def run(self):
        for f in self.man_pages:
            attrs = f.split(".")
            dest = os.path.join(self.install_dir, "share", "man")
            try:
                dest = os.path.join(dest, attrs[2])
            except IndexError:
                pass
            dest = os.path.join(dest, "man" + attrs[1])
            self.mkpath(dest)
            self.copy_file(f, dest)


man_pages = glob("man/*")

setup(
    packages=[],
    scripts=["elogv"],
    cmdclass={
        "extract_messages": messages,
        "merge": merge,
        "build_messages": build_messages,
        "build": build,
        "install_messages": install_messages,
        "install_manpages": install_manpages,
        "install": install,
    },
)