Use VM install repo URL on the installed system

Signed-off-by: Jiri Jaburek <jjaburek@redhat.com>

tests/install_vm.py

@@ -3,6 +3,7 @@
 import argparse
 import os
 import sys
+import subprocess
 
 def parse_args():
     parser = argparse.ArgumentParser()
@@ -82,6 +83,23 @@ def parse_args():
 
     return parser.parse_args()
 
+def detect_url_os(url):
+    """Given a repository URL, guess an OS type and version."""
+    cmd = ["osinfo-detect", "-f", "env", "-t", "tree", url]
+    try:
+        proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    except OSError as e:
+        if e.errno == os.errno.ENOENT:
+            return None
+        else:
+            raise
+
+    contents = proc.stdout.read().rstrip()
+    if not contents:
+        return None
+    pairs = map(lambda x: x.split('='), contents.splitlines())
+    osinfo = dict(pairs)
+    return osinfo['OSINFO_TREE']
 
 def main():
     data = parse_args()
@@ -125,9 +143,12 @@ def main():
 
     tmp_kickstart = "/tmp/" + data.ks_basename
     with open(data.kickstart) as infile, open(tmp_kickstart, "w") as outfile:
-        old_content = infile.read()
-        new_content = old_content.replace("&&HOST_PUBLIC_KEY&&", pub_key)
-        outfile.write(new_content)
+        content = infile.read()
+        content = content.replace("&&HOST_PUBLIC_KEY&&", pub_key)
+        ostype = detect_url_os(data.url)
+        if any(filter(lambda x: x in ostype, ['centos.org', 'redhat.com'])):
+            content = content.replace("&&YUM_REPO_URL&&", data.url)
+        outfile.write(content)
     data.kickstart = tmp_kickstart
     print("Using kickstart file: {0}".format(data.kickstart))
 

tests/kickstarts/test_suite.cfg

@@ -116,6 +116,17 @@ printf "%s\n" "&&HOST_PUBLIC_KEY&&" >> /root/.ssh/authorized_keys
 chmod og-rw /root/.ssh /root/.ssh/authorized_keys
 systemctl enable sshd
 
+# create yum/dnf repository from URL if replaced by install_vm.py
+if ! [[ '&&YUM_REPO_URL&&' =~ YUM_REPO_URL ]]; then
+	cat > /etc/yum.repos.d/inst-ks.repo <<EOF
+[inst-ks]
+name=Installation repo from kickstart
+baseurl=&&YUM_REPO_URL&&
+enabled=1
+gpgcheck=0
+EOF
+fi
+
 %end
 
 # Reboot after the installation is complete (optional)