From d40e1b0722810a33bd97c7e06647807eac782c0f Mon Sep 17 00:00:00 2001
From: Stephan Kulow <coolo@suse.de>
Date: Thu, 7 Mar 2019 14:20:19 +0100
Subject: [PATCH] Add all checks from staging-required-checks-adi to main repo
 of adi projects

For Factory I want to switch away from repo checker reviews, but a required
check on project level. So first step: make it required for adi projects.
---
 osclib/conf.py       |  1 +
 osclib/stagingapi.py | 32 ++++++++++++++++++++++++++------
 2 files changed, 27 insertions(+), 6 deletions(-)

diff --git a/osclib/conf.py b/osclib/conf.py
index d2460016..ff1fb55a 100644
--- a/osclib/conf.py
+++ b/osclib/conf.py
@@ -161,6 +161,7 @@ DEFAULT = {
         'staging-group': None,
         'staging-archs': '',
         'staging-dvd-archs': '',
+        'staging-required-checks-adi': '',
         'onlyadi': False,
         'rings': None,
         'nonfree': None,
diff --git a/osclib/stagingapi.py b/osclib/stagingapi.py
index 48681def..ba799f3d 100644
--- a/osclib/stagingapi.py
+++ b/osclib/stagingapi.py
@@ -1234,11 +1234,13 @@ class StagingAPI(object):
 
         return tar_pkg
 
-    def ensure_staging_archs(self, project):
-        url = self.makeurl(['source', project, '_meta'])
-        meta = ET.parse(http_GET(url))
+    def project_meta_url(self, project):
+        return self.makeurl(['source', project, '_meta'])
 
+    def ensure_staging_archs(self, project):
+        meta = ET.parse(http_GET(self.project_meta_url(project)))
         repository = meta.find('repository[@name="{}"]'.format(self.cmain_repo))
+
         changed = False
         for arch in self.cstaging_archs:
             if not repository.xpath('./arch[text()="{}"]'.format(arch)):
@@ -1246,9 +1248,11 @@ class StagingAPI(object):
                 elm.text = arch
                 changed = True
 
-        if changed:
-            meta = ET.tostring(meta)
-            http_PUT(url, data=meta)
+        if not changed:
+            return
+
+        meta = ET.tostring(meta)
+        http_PUT(self.project_meta_url(project), data=meta)
 
     def prj_from_letter(self, letter):
         if ':' in letter:  # not a letter
@@ -1703,8 +1707,24 @@ class StagingAPI(object):
         if use_frozenlinks:
             self.update_adi_frozenlinks(name, src_prj)
 
+        for required_check in self.cstaging_required_checks_adi.split():
+            self.add_required_check(name, required_check)
         return name
 
+    def add_required_check(self, project, check):
+        root = ET.Element('required_checks')
+        name = ET.SubElement(root, 'name')
+        name.text = check
+
+        meta = ET.parse(http_GET(self.project_meta_url(project)))
+        repository = meta.find('repository[@name="{}"]'.format(self.cmain_repo))
+
+        for arch_element in repository.findall('arch'):
+            architecture = arch_element.text
+            url = self.makeurl(['status_reports', 'built_repositories', project,
+                                self.cmain_repo, architecture, 'required_checks'])
+            http_POST(url, data=ET.tostring(root))
+
     def is_user_member_of(self, user, group):
         root = ET.fromstring(get_group(self.apiurl, group))