]> git.immae.eu Git - perso/Immae/Config/Nix.git/blobdiff - modules/private/buildbot/projects/test/__init__.py
Rework buildbot configuration
[perso/Immae/Config/Nix.git] / modules / private / buildbot / projects / test / __init__.py
index e2f6f82e4e6ddcee48c7f1b705d0dbdd010efafa..2898bbc1d5faa49416d78ffd2ff294110a6bf20a 100644 (file)
@@ -18,7 +18,6 @@ class E():
     GIT_URL       = "https://git.immae.eu/perso/Immae/TestProject.git"
     SSH_KEY_PATH  = "/var/lib/buildbot/buildbot_key"
     LIBVIRT_URL   = os.environ["BUILDBOT_VIRT_URL"] + "?keyfile=" + SSH_KEY_PATH
-    PUPPET_HOST   = "root@backup-1.v.immae.eu"
     LDAP_HOST     = "ldap.immae.eu"
     LDAP_DN       = "cn=buildbot,ou=services,dc=immae,dc=eu"
     LDAP_ROLES_BASE = "ou=roles,ou=hosts,dc=immae,dc=eu"
@@ -61,53 +60,48 @@ class CustomBase(webhooks.base):
 
         return ([args], None)
 
-def deploy_hook_scheduler(project, timer=1):
-    return schedulers.AnyBranchScheduler(
-            change_filter=util.ChangeFilter(category="deploy_webhook", project=project),
-            name="{}_deploy".format(project), treeStableTimer=timer, builderNames=["{}_deploy".format(project)])
-
 def configure(c):
     c["buildbotURL"] = E.BUILDBOT_URL
     c["www"]["port"] = E.SOCKET
 
-    c["www"]["change_hook_dialects"]["base"] = {
-            "custom_class": CustomBase
-            }
-    c['workers'].append(ilibvirt.LibVirtWorker("test-build",
-                                             open(E.SECRETS_FILE + "/worker_password", "r").read().rstrip(),
-                                             ilibvirt.Connection(E.LIBVIRT_URL),
-                                             E.WORKER_HOST))
-    c['workers'].append(ilibvirt.LibVirtWorker("test-deploy",
-                                             open(E.SECRETS_FILE + "/worker_password", "r").read().rstrip(),
-                                             ilibvirt.Connection(E.LIBVIRT_URL),
-                                             E.WORKER_HOST))
-
-    c['schedulers'].append(hook_scheduler("TestProject", timer=1))
-    c['schedulers'].append(force_scheduler("force_test", ["TestProject_build"]))
-    c['schedulers'].append(deploy_scheduler("deploy_test", ["TestProject_deploy"]))
-    c['schedulers'].append(deploy_hook_scheduler("TestProject", timer=1))
-
-    c['builders'].append(factory())
-    c['builders'].append(deploy_factory())
-
-    c['services'].append(SlackStatusPush(
-        name="slack_status_test_project",
-        builders=["TestProject_build", "TestProject_deploy"],
-        serverUrl=open(E.SECRETS_FILE + "/slack_webhook", "r").read().rstrip()))
-    c['services'].append(XMPPStatusPush(
-        name="xmpp_status_test_project",
-        builders=["TestProject_build", "TestProject_deploy"],
-        recipients=E.XMPP_RECIPIENTS,
-        password=open(E.SECRETS_FILE + "/notify_xmpp_password", "r").read().rstrip()))
-
-def factory():
+    c["www"]["change_hook_dialects"]["base"] = { "custom_class": CustomBase }
+
+    configure_build(c)
+    configure_deploy(c)
+
+    configure_slack_push(c, E.SECRETS_FILE, all_builder_names(c))
+    configure_xmpp_push(c, E.SECRETS_FILE, all_builder_names(c), E.XMPP_RECIPIENTS)
+
+def configure_build(c):
+    builder_name = "TestProject_build"
+    worker_name = "test-build"
+    c['schedulers'].append(force_scheduler("force_test", [builder_name]))
+    c['schedulers'].append(git_hook_scheduler("TestProject", [builder_name]))
+    c['workers'].append(libvirt_worker(worker_name))
+    c['builders'].append(util.BuilderConfig(name=builder_name, workernames=[worker_name], factory=build_factory()))
+
+def configure_deploy(c):
+    builder_name = "TestProject_deploy"
+    worker_name = "test-deploy"
+    c['workers'].append(libvirt_worker(worker_name))
+    c['schedulers'].append(deploy_hook_scheduler("TestProject", [builder_name]))
+    c['schedulers'].append(deploy_scheduler("deploy_test", [builder_name]))
+    c['builders'].append(util.BuilderConfig(name=builder_name, workernames=[worker_name], factory=deploy_factory()))
+
+def libvirt_worker(name):
+    return ilibvirt.LibVirtWorker(name,
+            open(E.SECRETS_FILE + "/worker_password", "r").read().rstrip(),
+            ilibvirt.Connection(E.LIBVIRT_URL),
+            E.WORKER_HOST)
+
+def build_factory():
     package = util.Interpolate("test_%(kw:clean_branch)s.tar.gz", clean_branch=clean_branch)
     package_dest = util.Interpolate("{}/test_%(kw:clean_branch)s.tar.gz".format(E.RELEASE_PATH), clean_branch=clean_branch)
     package_url = util.Interpolate("{}/test_%(kw:clean_branch)s.tar.gz".format(E.RELEASE_URL), clean_branch=clean_branch)
 
     factory = util.BuildFactory()
     factory.addStep(steps.Git(logEnviron=False,
-        repourl=E.GIT_URL, mode="full", method="copy"))
+        repourl=E.GIT_URL, mode="full", method="fresh"))
     factory.addStep(steps.ShellCommand(name="env",
         logEnviron=False, command=["env"]))
     factory.addStep(steps.ShellCommand(name="pwd",
@@ -118,80 +112,19 @@ def factory():
         logEnviron=False, command=["echo", package]))
     factory.addSteps(package_and_upload(package, package_dest, package_url))
 
-    return util.BuilderConfig(name="TestProject_build", workernames=["test-build"], factory=factory)
-
-
-def compute_build_infos():
-    @util.renderer
-    def compute(props):
-        import re, hashlib
-        build_file = props.getProperty("build")
-        package_dest = "{}/{}".format(E.RELEASE_PATH, build_file)
-        version = re.match(r"{0}_(.*).tar.gz".format("test"), build_file).group(1)
-        with open(package_dest, "rb") as f:
-            sha = hashlib.sha256(f.read()).hexdigest()
-        return {
-                "build_version": version,
-                "build_hash": sha,
-                }
-    return compute
-
-@util.renderer
-def puppet_host(props):
-    return E.PUPPET_HOST
+    return factory
 
 def deploy_factory():
     package_dest = util.Interpolate("{}/%(prop:build)s".format(E.RELEASE_PATH))
 
     factory = util.BuildFactory()
     factory.addStep(steps.MasterShellCommand(command=["test", "-f", package_dest]))
-    factory.addStep(steps.SetProperties(properties=compute_build_infos()))
-    factory.addStep(LdapPush(environment=util.Property("environment"),
+    factory.addStep(steps.SetProperties(properties=compute_build_infos("test", E.RELEASE_PATH)))
+    factory.addStep(LdapEdit(environment=util.Property("environment"),
         build_version=util.Property("build_version"),
         build_hash=util.Property("build_hash"),
-        ldap_password=util.Secret("ldap")))
-    factory.addStep(steps.MasterShellCommand(command=[
-        "ssh", "-o", "UserKnownHostsFile=/dev/null", "-o", "StrictHostKeyChecking=no", "-o", "CheckHostIP=no", "-i", E.SSH_KEY_PATH, puppet_host]))
-    return util.BuilderConfig(name="TestProject_deploy", workernames=["test-deploy"], factory=factory)
-
-from twisted.internet import defer
-from buildbot.process.buildstep import FAILURE
-from buildbot.process.buildstep import SUCCESS
-from buildbot.process.buildstep import BuildStep
-
-class LdapPush(BuildStep):
-    name = "LdapPush"
-    renderables = ["environment", "build_version", "build_hash", "ldap_password"]
-
-    def __init__(self, **kwargs):
-        self.environment = kwargs.pop("environment")
-        self.build_version = kwargs.pop("build_version")
-        self.build_hash = kwargs.pop("build_hash")
-        self.ldap_password = kwargs.pop("ldap_password")
-        self.ldap_host = kwargs.pop("ldap_host", E.LDAP_HOST)
-        super().__init__(**kwargs)
-
-    def run(self):
-        import json
-        from ldap3 import Reader, Writer, Server, Connection, ObjectDef
-        server = Server(self.ldap_host)
-        conn = Connection(server,
-                user=E.LDAP_DN,
-                password=self.ldap_password)
-        conn.bind()
-        obj = ObjectDef("immaePuppetClass", conn)
-        r = Reader(conn, obj,
-                "cn=test.{},{}".format(self.environment, E.LDAP_ROLES_BASE))
-        r.search()
-        if len(r) > 0:
-            w = Writer.from_cursor(r)
-            for value in w[0].immaePuppetJson.values:
-                config = json.loads(value)
-                if "test_version" in config:
-                    config["test_version"] = self.build_version
-                    config["test_sha256"] = self.build_hash
-                    w[0].immaePuppetJson -= value
-                    w[0].immaePuppetJson += json.dumps(config, indent="  ")
-                    w.commit()
-                    return defer.succeed(SUCCESS)
-        return defer.succeed(FAILURE)
+        config_key="test",
+        ldap_host=E.LDAP_HOST, ldap_roles_base=E.LDAP_ROLES_BASE, ldap_dn=E.LDAP_DN,
+        ldap_cn_template="test.{}", ldap_password=util.Secret("ldap")))
+    factory.addStep(steps.MasterShellCommand(command=deploy_ssh_command(E.SSH_KEY_PATH, {})))
+    return factory