#!/usr/bin/env python

##############################################################################
# Copyright (c) 2016 Daniel Farrell and Others.  All rights reserved.
#
# This program and the accompanying materials are made available under the
# terms of the Eclipse Public License v1.0 which accompanies this distribution,
# and is available at http://www.eclipse.org/legal/epl-v10.html
##############################################################################

"""Build OpenDaylight's RPMs using build description and Jinja2 templates."""

import imp
import os
import shutil
from string import Template
import subprocess

import specs.build_specs as build_specs

# This file is assumed to be in the root of the RPM build logic's dir structure
rpm_root = os.path.dirname(os.path.abspath(__file__))

# FIXME: Surely there is a better way to do this
pkg_lib = imp.load_source("", os.path.join(rpm_root, os.pardir, "lib.py"))

# Common paths used in this script
rpmbuild_dir = os.path.join(os.path.expanduser("~"), "rpmbuild")
src_in_dir = os.path.join(rpmbuild_dir, "SOURCES")
spec_in_dir = os.path.join(rpmbuild_dir, "SPECS")

# Templates that can be specialized per-build
# NB: Templates can't be concatenated with other Templates or strings, or
# cast to strings for concatenation. If they could, we would do elegant
# refactoring like concatenating paths to templates here and only calling
# Template.substitute in the build_rpm function.
rpm_template = Template("opendaylight-$version_major.$version_minor."
                        "$version_patch-$pkg_version.el7.noarch.rpm")
srpm_template = Template("opendaylight-$version_major.$version_minor."
                         "$version_patch-$pkg_version.el7.src.rpm")
spec_template = Template("opendaylight-$version_major.$version_minor."
                         "$version_patch-$pkg_version.spec")


def build_rpm(build):
    """Build the RPMs described by the given build description

    :param build: Description of an RPM build
    :type build: dict

    """
    # Specialize templates for the given build
    distro_tar_path = os.path.join(
        pkg_lib.cache_dir,
        pkg_lib.distro_template.substitute(build)) + ".tar.gz"
    rpm_out_path = os.path.join(rpmbuild_dir, "RPMS", "noarch",
                                rpm_template.substitute(build))
    spec_in_path = os.path.join(rpmbuild_dir, "SPECS",
                                spec_template.substitute(build))
    spec_path = os.path.join(rpm_root, "specs",
                             spec_template.substitute(build))
    srpm_out_path = os.path.join(rpmbuild_dir, "SRPMS",
                                 srpm_template.substitute(build))
    unitfile_tar_path = os.path.join(
        pkg_lib.cache_dir,
        pkg_lib.unitfile_template.substitute(build)) + ".tar.gz"

    # Call helper script to build the required RPM .spec files
    build_specs.build_spec(build)

    # Clean up old rpmbuild dir structure if it exists
    if os.path.isdir(rpmbuild_dir):
        shutil.rmtree(rpmbuild_dir)

    # Create rpmbuild dir structure
    subprocess.call("rpmdev-setuptree")

    # Cache ODL distro and systemd unit file to package
    distro_tar_path = pkg_lib.cache_distro(build)
    unitfile_tar_path = pkg_lib.cache_sysd(build)["unitfile_tar_path"]

    # Move unit file, tarball and specfile to correct rpmbuild dirs
    shutil.copy(distro_tar_path, src_in_dir)
    shutil.copy(unitfile_tar_path, src_in_dir)
    shutil.copy(spec_path, spec_in_dir)

    # Call rpmbuild, build both SRPMs/RPMs
    subprocess.call(["rpmbuild", "-ba", spec_in_path])

    # Copy the RPMs/SRPMs from their output dir to the cache dir
    shutil.copy(rpm_out_path, pkg_lib.cache_dir)
    shutil.copy(srpm_out_path, pkg_lib.cache_dir)