From 6d819cee55d9e9c5b36fbef6a3d21db2703237de Mon Sep 17 00:00:00 2001
From: Muzahid <muzahid.h@cisinlabs.com>
Date: Thu, 17 Feb 2022 22:09:45 +0530
Subject: [PATCH] changes for worker

---
 lib/worker_multibuild.py | 14 ++++----------
 multibuild.py            | 22 +++++++---------------
 2 files changed, 11 insertions(+), 25 deletions(-)

diff --git a/lib/worker_multibuild.py b/lib/worker_multibuild.py
index 2242715..969b132 100644
--- a/lib/worker_multibuild.py
+++ b/lib/worker_multibuild.py
@@ -2,7 +2,6 @@ from os import listdir, walk
 from os.path import exists, isfile, join
 import requests
 import re
-from buildbot.plugins import util
 from .renderers import *
 
 
@@ -71,11 +70,11 @@ def list_jobs(directory=".buildbot"):
     return results
 
 
-def _get_dockerfile_contents(props):
+def _get_dockerfile_contents(props, jobname, os_codename):
     """
     Read contents of a Dockerfile and add extra contents for the given os_codename
     """
-    with open(join(path + props.getProperty('jobname', default=None)), "r") as file:
+    with open(join(path + jobname), "r") as file:
         contents = file.read()
         # remove any line containing CMD or ENTRYFILE keywords
         re.sub(r"(?m)^(CMD|ENTRYFILE).*$", "", contents)
@@ -83,12 +82,7 @@ def _get_dockerfile_contents(props):
     return contents + {
         "focal": dockerfile_extra_contents_focal,
         "bionic": dockerfile_extra_contents_bionic,
-    }[util.Interpolate("%(prop:os_codename)s")]
-
-
-@util.renderer
-def get_dockerfile_contents(props):
-    return _get_dockerfile_contents(props)
+    }[os_codename]
 
 
 def trigger_child_hooks(buildbotUrl: str, os_codename: str, repository, branch, jobname, directory=".buildbot"):
@@ -108,7 +102,7 @@ def trigger_child_hooks(buildbotUrl: str, os_codename: str, repository, branch,
 
         # make a post request
         request_data["properties"] = {
-            "dockerfile": get_dockerfile_contents(
+            "dockerfile": _get_dockerfile_contents(
                 join(directory, job, "Dockerfile"), os_codename
             ),
             "build_script_available": is_build_script_available(build_script_exists),
diff --git a/multibuild.py b/multibuild.py
index 04dac43..d27c139 100644
--- a/multibuild.py
+++ b/multibuild.py
@@ -17,22 +17,11 @@ import re
 from buildbot.plugins import steps, util
 
 from .lib.renderers import *
-from .lib.worker_multibuild import trigger_child_hooks
 
-
-def list_jobs(directory=".buildbot"):
-    """
-    list jobs found in a directory
-    """
-    results = []
-    for _ in next(walk(directory))[1]:
-        if exists(join(directory, _, "Dockerfile")) and (
-            exists(join(directory, _, "build.sh"))
-            or exists(join(directory, _, "test.sh"))
-        ):
-            results.append(_)
-
-    return results
+addStep(steps.ShellCommand(
+    name="download worker",
+    command=["sudo", "wget", "-O", "lib/worker_multibuild.py", "worker_multibuild.py"]
+))
 
 
 def find_artifacts(directory="out"):
@@ -45,6 +34,9 @@ def find_artifacts(directory="out"):
         return join(directory, _)
 
 
+@util.renderer
+def get_dockerfile_contents(props):
+    return _get_dockerfile_contents(props, props.getProperty('jobname', default=None), util.Interpolate("%(prop:os_codename)s"))
 
 
 def add_parent_step(build_factory):