Source code for pylorax.api.gitrpm

# Copyright (C) 2019 Red Hat, Inc.
#
# 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; either version 2 of the License, or
# (at your option) any later version.
#
# 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, see <http://www.gnu.org/licenses/>.
#
""" Clone a git repository and package it as an rpm

This module contains functions for cloning a git repo, creating a tar archive of
the selected commit, branch, or tag, and packaging the files into an rpm that will
be installed by anaconda when creating the image.
"""
import logging
log = logging.getLogger("lorax-composer")

import os
from rpmfluff import SimpleRpmBuild
import shutil
import subprocess
import tempfile
import time

from pylorax.sysutils import joinpaths

[docs]def get_repo_description(gitRepo): """ Return a description including the git repo and reference :param gitRepo: A dict with the repository details :type gitRepo: dict :returns: A string with the git repo url and reference :rtype: str """ return "Created from %s, reference '%s', on %s" % (gitRepo["repo"], gitRepo["ref"], time.ctime())
[docs]class GitArchiveTarball: """Create a git archive of the selected git repo and reference""" def __init__(self, gitRepo): self._gitRepo = gitRepo self.sourceName = self._gitRepo["rpmname"]+".tar.xz"
[docs] def write_file(self, sourcesDir): """ Create the tar archive :param sourcesDir: Path to use for creating the archive :type sourcesDir: str This clones the git repository and creates a git archive from the specified reference. The result is in RPMNAME.tar.xz under the sourcesDir """ # Clone the repository into a temporary location cmd = ["git", "clone", self._gitRepo["repo"], joinpaths(sourcesDir, "gitrepo")] log.debug(cmd) try: subprocess.check_output(cmd, stderr=subprocess.STDOUT) except subprocess.CalledProcessError as e: log.error("Failed to clone %s: %s", self._gitRepo["repo"], e.output) raise RuntimeError("Failed to clone %s" % self._gitRepo["repo"]) oldcwd = os.getcwd() try: os.chdir(joinpaths(sourcesDir, "gitrepo")) # Configure archive to create a .tar.xz cmd = ["git", "config", "tar.tar.xz.command", "xz -c"] log.debug(cmd) subprocess.check_call(cmd) cmd = ["git", "archive", "--prefix", self._gitRepo["rpmname"] + "/", "-o", joinpaths(sourcesDir, self.sourceName), self._gitRepo["ref"]] log.debug(cmd) try: subprocess.check_output(cmd, stderr=subprocess.STDOUT) except subprocess.CalledProcessError as e: log.error("Failed to archive %s: %s", self._gitRepo["repo"], e.output) raise RuntimeError('Failed to archive %s from ref "%s"' % (self._gitRepo["repo"], self._gitRepo["ref"])) finally: # Cleanup even if there was an error os.chdir(oldcwd) shutil.rmtree(joinpaths(sourcesDir, "gitrepo"))
[docs]class GitRpmBuild(SimpleRpmBuild): """Build an rpm containing files from a git repository""" def __init__(self, *args, **kwargs): self._base_dir = None super().__init__(*args, **kwargs)
[docs] def check(self): raise NotImplementedError
[docs] def get_base_dir(self): """Place all the files under a temporary directory + rpmbuild/ """ if not self._base_dir: self._base_dir = tempfile.mkdtemp(prefix="lorax-git-rpm.") return joinpaths(self._base_dir, "rpmbuild")
[docs] def cleanup_tmpdir(self): """Remove the temporary directory and all of its contents """ if len(self._base_dir) < 5: raise RuntimeError("Invalid base_dir: %s" % self.get_base_dir()) shutil.rmtree(self._base_dir)
[docs] def clean(self): """Remove the base directory from inside the tmpdir""" if len(self.get_base_dir()) < 5: raise RuntimeError("Invalid base_dir: %s" % self.get_base_dir()) shutil.rmtree(self.get_base_dir(), ignore_errors=True)
[docs] def add_git_tarball(self, gitRepo): """Add a tar archive of a git repository to the rpm :param gitRepo: A dict with the repository details :type gitRepo: dict This populates the rpm with the URL of the git repository, the summary describing the repo, the description of the repository and reference used, and sets up the rpm to install the archive contents into the destination path. """ self.addUrl(gitRepo["repo"]) self.add_summary(gitRepo["summary"]) self.add_description(get_repo_description(gitRepo)) self.addLicense("Unknown") sourceIndex = self.add_source(GitArchiveTarball(gitRepo)) self.section_build += "tar -xvf %s\n" % self.sources[sourceIndex].sourceName dest = os.path.normpath(gitRepo["destination"]) # Prevent double slash root if dest == "/": dest = "" self.create_parent_dirs(dest) self.section_install += "cp -r %s/. $RPM_BUILD_ROOT/%s\n" % (gitRepo["rpmname"], dest) sub = self.get_subpackage(None) if not dest: # / is special, we don't want to include / itself, just what's under it sub.section_files += "/*\n" else: sub.section_files += "%s/\n" % dest
[docs]def make_git_rpm(gitRepo, dest): """ Create an rpm from the specified git repo :param gitRepo: A dict with the repository details :type gitRepo: dict This will clone the git repository, create an archive of the selected reference, and build an rpm that will install the files from the repository under the destination directory. The gitRepo dict should have the following fields:: rpmname: "server-config" rpmversion: "1.0" rpmrelease: "1" summary: "Setup files for server deployment" repo: "PATH OF GIT REPO TO CLONE" ref: "v1.0" destination: "/opt/server/" * rpmname: Name of the rpm to create, also used as the prefix name in the tar archive * rpmversion: Version of the rpm, eg. "1.0.0" * rpmrelease: Release of the rpm, eg. "1" * summary: Summary string for the rpm * repo: URL of the get repo to clone and create the archive from * ref: Git reference to check out. eg. origin/branch-name, git tag, or git commit hash * destination: Path to install the / of the git repo at when installing the rpm """ gitRpm = GitRpmBuild(gitRepo["rpmname"], gitRepo["rpmversion"], gitRepo["rpmrelease"], ["noarch"]) try: gitRpm.add_git_tarball(gitRepo) gitRpm.do_make() rpmfile = gitRpm.get_built_rpm("noarch") shutil.move(rpmfile, dest) except Exception as e: log.error("Creating git repo rpm: %s", e) raise RuntimeError("Creating git repo rpm: %s" % e) finally: gitRpm.cleanup_tmpdir() return os.path.basename(rpmfile)
# Create the git rpms, if any, and return the path to the repo under results_dir
[docs]def create_gitrpm_repo(results_dir, recipe): """Create a dnf repository with the rpms from the recipe :param results_dir: Path to create the repository under :type results_dir: str :param recipe: The recipe to get the repos.git entries from :type recipe: Recipe :returns: Path to the dnf repository or "" :rtype: str This function creates a dnf repository directory at results_dir+"repo/", creates rpms for all of the repos.git entries in the recipe, runs createrepo_c on the dnf repository so that Anaconda can use it, and returns the path to the repository to the caller. """ if "repos" not in recipe or "git" not in recipe["repos"]: return "" gitrepo = joinpaths(results_dir, "repo/") if not os.path.exists(gitrepo): os.makedirs(gitrepo) for r in recipe["repos"]["git"]: make_git_rpm(r, gitrepo) cmd = ["createrepo_c", gitrepo] log.debug(cmd) try: subprocess.check_output(cmd, stderr=subprocess.STDOUT) except subprocess.CalledProcessError as e: log.error("Failed to create repo at %s: %s", gitrepo, e.output) raise RuntimeError("Failed to create repo at %s" % gitrepo) return gitrepo