# Copyright (C) 2012-2015, 2019 Codethink Limited
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; version 2 of the License.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.


"""Setup.py for lorry."""


import compileall
import distutils.dir_util
import glob
import os
import shutil
import subprocess
import sys
from distutils.cmd import Command
from distutils.command.build import build
from distutils.command.clean import clean
from distutils.command.install import install
from distutils.core import setup


class Build(build):
    def run(self):
        build.run(self)

        compileall.compile_dir("hg-fast-export/plugins")

        print("building manpages")
        for x in ["lorry"]:
            with open("%s.1" % x, "w") as f:
                subprocess.check_call(
                    [
                        "python3",
                        x,
                        "--generate-manpage=%s.1.in" % x,
                        "--output=%s.1" % x,
                    ],
                    stdout=f,
                )


class Clean(clean):
    clean_files = [".coverage", "build", "lorry.1"]
    clean_globs = [
        "*/*.py[co]",
        "hg-fast-export/plugins/*/__pycache__",
    ]

    def run(self):
        clean.run(self)
        itemses = [self.clean_files] + [glob.glob(x) for x in self.clean_globs]
        for items in itemses:
            for filename in items:
                if os.path.isdir(filename):
                    shutil.rmtree(filename)
                elif os.path.exists(filename):
                    os.remove(filename)


class Check(Command):
    user_options = []

    def initialize_options(self):
        pass

    def finalize_options(self):
        pass

    def run(self):
        subprocess.check_call(["./check"])


class Install(install):
    def run(self):
        install.run(self)

        # Install hg-fast-export plugins in a private directory
        distutils.dir_util.copy_tree(
            "hg-fast-export/plugins",
            os.path.join(self.install_data, "share/lorry/hg-fast-export/plugins"),
        )


if sys.version_info < (3, 6):
    print("lorry requires Python 3.6+", file=sys.stderr)
    sys.exit(1)

setup(
    name="lorry",
    description="FIXME",
    long_description="""\
FIXME
""",
    author="Baserock",
    author_email="baserock-dev@baserock.org",
    url="http://wiki.baserock.org/",
    scripts=[
        "lorry",
        "lorry.tar-importer",
        "lorry.gzip-importer",
        "lorry.zip-importer",
        "lorry-ssh-wrapper",
        "lorry.raw-file-importer",
    ],
    data_files=[("share/man/man1", glob.glob("*.[1-8]"))],
    cmdclass={
        "build": Build,
        "check": Check,
        "clean": Clean,
        "install": Install,
    },
)