# Copyright 2013-2021 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)

import os

from spack import *


class Libyogrt(AutotoolsPackage):
    """Your One Get Remaining Time Library."""

    homepage = "https://github.com/LLNL/libyogrt"
    url      = "https://github.com/LLNL/libyogrt/releases/download/1.21/libyogrt-1.21.tar.gz"

    version('1.24',   sha256='36695030e72b24b1f22bfcfe42bfd1d3c87f9c0eea5e94ce0120782581ea522f')
    version('1.23',   sha256='c95e7a6be29c0d1ac1b673b0ba1d4e5781981722f93d0da99ae62ff3b5f35b5f')
    version('1.22',   sha256='38e7d1ea3fa030f0169197aa96cde9f01caa595a590764ef1cb2ae07379cb711')
    version('1.21',   sha256='5f8f0942d35ee4e418273e478e632210b3fa648dcb6a2e6a92c6ba4213cdc362')
    version('1.20-7', sha256='735e9d6fa572e239ccc73e11c84b4583338b24df0fa91c48e8bc038d882003f7')
    version('1.20-6', sha256='ba5a2e202f995cf7ae3bf87b451943733e760ede02ca172f712cbf2eea693222')
    version('1.20-5', sha256='1e41bc656abffb121145264bc898421c3f355d3be35f1711b7b5e3ffe7effdd9')
    version('1.20-4', sha256='0858a729068b272d4047d79f6a5187cdbd427bdfec64db4e143524b4789a06c5')
    version('1.20-3', sha256='61a8f28f452aef0e09d700dbaaffd91ae3855f7ac221c7ebe478a028df635e31')
    version('1.20-2', sha256='bf22a82ab3bfede780be3fb6c132cc354234f8d57d3cccd58fe594f074ed7f95')

    # libyogrt supports the following schedulers:
    #     lcrm, lsf, moab, slurm, AIX+slurm

    # however, only slurm exists in spack
    # libyogrt's build system is smart enough to detect the system scheduler
    # the slurm option here connects to a spack-installed slurm
    # if/when other schedulers have spack packages, they can be added

    variant('scheduler', default='system',
            description="Select scheduler integration",
            values=['system', 'slurm', 'lsf'], multi=False)
    depends_on('slurm', when='scheduler=slurm')
    depends_on('lsf', when='scheduler=lsf')

    conflicts('scheduler=lsf', when='@:1.22')

    variant('static', default='False',
            description="build static library")

    def url_for_version(self, version):
        if version < Version(1.21):
            return "https://github.com/LLNL/libyogrt/archive/%s.tar.gz" % version
        else:
            return "https://github.com/LLNL/libyogrt/releases/download/{0}/libyogrt-{0}.tar.gz".format(version)

    def configure_args(self):
        args = []

        sched = self.spec.variants['scheduler'].value
        if sched == "lsf":
            # The LSF library depends on a couple of other libraries,
            # and running the build inside of spack does not find
            # them, and the user has to add them when they want
            # to use -lyogrt. If we explicitly tell it what libraries
            # to use, the user does not need to specify them
            args.append('--with-lsf')
            args.append('LIBS=-llsf -lrt -lnsl')
        elif sched != "system":
            args.append('--with-%s=%s' % (sched, self.spec[sched].prefix))

        if '+static' in self.spec:
            args.append('--enable-static=yes')

        return args

    @run_after('install')
    def create_yogrt_conf(self):
        etcpath = os.path.join(prefix, "etc")

        # create subdirectory to hold yogrt.conf file
        if not os.path.isdir(etcpath):
            mode = 0o755
            os.mkdir(etcpath, mode)

        # if no scheduler is specified, create yogrt conf file
        # with backend=none
        sched = self.spec.variants['scheduler'].value
        if sched == "system":
            sched = "none"

        # create conf file to inform libyogrt about job scheduler
        with open(os.path.join(etcpath, "yogrt.conf"), "w+") as f:
            f.write("backend=%s\n" % sched)