]>
git.immae.eu Git - perso/Immae/Config/Nix.git/blob - nixops/modules/buildbot/projects/test/__init__.py
1 from buildbot
.plugins
import *
2 from buildbot_common
.build_helpers
import *
4 from buildbot
.util
import bytes2unicode
7 __all__
= [ "configure", "E" ]
11 BUILDBOT_URL
= "https://git.immae.eu/buildbot/{}/".format(PROJECT
)
12 SOCKET
= "unix:/run/buildbot/{}.sock".format(PROJECT
)
13 PB_SOCKET
= "unix:address=/run/buildbot/{}_pb.sock".format(PROJECT
)
14 RELEASE_PATH
= "/var/lib/ftp/release.immae.eu/{}".format(PROJECT
)
15 RELEASE_URL
= "https://release.immae.eu/{}".format(PROJECT
)
16 GIT_URL
= "https://git.immae.eu/perso/Immae/TestProject.git"
17 SSH_KEY_PATH
= "/var/lib/buildbot/buildbot_key"
18 PUPPET_HOST
= "root@backup-1.v.immae.eu"
19 LDAP_HOST
= "ldap.immae.eu"
20 LDAP_DN
= "cn=buildbot,ou=services,dc=immae,dc=eu"
21 LDAP_ROLES_BASE
= "ou=roles,ou=hosts,dc=immae,dc=eu"
24 SECRETS_FILE
= os
.getcwd() + "/secrets"
25 LDAP_URL
= "ldaps://ldap.immae.eu:636"
26 LDAP_ADMIN_USER
= "cn=buildbot,ou=services,dc=immae,dc=eu"
27 LDAP_BASE
= "dc=immae,dc=eu"
28 LDAP_PATTERN
= "(uid=%(username)s)"
29 LDAP_GROUP_PATTERN
= "(&(memberOf=cn=groups,ou=test,cn=buildbot,ou=services,dc=immae,dc=eu)(member=%(dn)s))"
30 TITLE_URL
= "https://git.immae.eu/?p=perso/Immae/TestProject.git;a=summary"
31 TITLE
= "Test project"
33 class CustomBase(webhooks
.base
):
34 def getChanges(self
, request
):
36 content
= request
.content
.read()
37 args
= json
.loads(bytes2unicode(content
))
38 except Exception as e
:
39 raise ValueError("Error loading JSON: " + str(e
))
41 args
.setdefault("comments", "")
42 args
.setdefault("repository", "")
43 args
.setdefault("author", args
.get("who", "unknown"))
45 if args
["category"] == "deploy_webhook":
47 "category": "deploy_webhook",
51 "project": "TestProject",
53 "environment": args
.get("environment", "integration"),
54 "build": "test_{}.tar.gz".format(args
.get("branch", "master"))
60 def deploy_hook_scheduler(project
, timer
=1):
61 return schedulers
.AnyBranchScheduler(
62 change_filter
=util
.ChangeFilter(category
="deploy_webhook", project
=project
),
63 name
="{}_deploy".format(project), treeStableTimer=timer, builderNames=["{}_deploy".format(project
)])
66 c
["buildbotURL"] = E
.BUILDBOT_URL
67 c
["www"]["port"] = E
.SOCKET
69 c
["www"]["change_hook_dialects"]["base"] = {
70 "custom_class": CustomBase
72 c
['workers'].append(worker
.LocalWorker("generic-worker-test"))
73 c
['workers'].append(worker
.LocalWorker("deploy-worker-test"))
75 c
['schedulers'].append(hook_scheduler("TestProject", timer
=1))
76 c
['schedulers'].append(force_scheduler("force_test", ["TestProject_build"]))
77 c
['schedulers'].append(deploy_scheduler("deploy_test", ["TestProject_deploy"]))
78 c
['schedulers'].append(deploy_hook_scheduler("TestProject", timer
=1))
80 c
['builders'].append(factory())
81 c
['builders'].append(deploy_factory())
83 c
['services'].append(SlackStatusPush(
84 name
="slack_status_test_project",
85 builders
=["TestProject_build", "TestProject_deploy"],
86 serverUrl
=open(E
.SECRETS_FILE
+ "/slack_webhook", "r").read().rstrip()))
89 package
= util
.Interpolate("test_%(kw:clean_branch)s.tar.gz", clean_branch
=clean_branch
)
90 package_dest
= util
.Interpolate("{}/test_%(kw:clean_branch)s.tar.gz".format(E
.RELEASE_PATH
), clean_branch
=clean_branch
)
91 package_url
= util
.Interpolate("{}/test_%(kw:clean_branch)s.tar.gz".format(E
.RELEASE_URL
), clean_branch
=clean_branch
)
93 factory
= util
.BuildFactory()
94 factory
.addStep(steps
.Git(logEnviron
=False,
95 repourl
=E
.GIT_URL
, mode
="full", method
="copy"))
96 factory
.addStep(steps
.ShellCommand(name
="env",
97 logEnviron
=False, command
=["env"]))
98 factory
.addStep(steps
.ShellCommand(name
="pwd",
99 logEnviron
=False, command
=["pwd"]))
100 factory
.addStep(steps
.ShellCommand(name
="true",
101 logEnviron
=False, command
=["true"]))
102 factory
.addStep(steps
.ShellCommand(name
="echo",
103 logEnviron
=False, command
=["echo", package
]))
104 factory
.addSteps(package_and_upload(package
, package_dest
, package_url
))
106 return util
.BuilderConfig(name
="TestProject_build", workernames
=["generic-worker-test"], factory
=factory
)
109 def compute_build_infos():
113 build_file
= props
.getProperty("build")
114 package_dest
= "{}/{}".format(E
.RELEASE_PATH
, build_file
)
115 version
= re
.match(r
"{0}_(.*).tar.gz".format("test"), build_file
).group(1)
116 with open(package_dest
, "rb") as f
:
117 sha
= hashlib
.sha256(f
.read()).hexdigest()
119 "build_version": version
,
125 def puppet_host(props
):
128 def deploy_factory():
129 package_dest
= util
.Interpolate("{}/%(prop:build)s".format(E
.RELEASE_PATH
))
131 factory
= util
.BuildFactory()
132 factory
.addStep(steps
.MasterShellCommand(command
=["test", "-f", package_dest
]))
133 factory
.addStep(steps
.SetProperties(properties
=compute_build_infos()))
134 factory
.addStep(LdapPush(environment
=util
.Property("environment"),
135 build_version
=util
.Property("build_version"),
136 build_hash
=util
.Property("build_hash"),
137 ldap_password
=util
.Secret("ldap")))
138 factory
.addStep(steps
.MasterShellCommand(command
=[
139 "ssh", "-o", "UserKnownHostsFile=/dev/null", "-o", "StrictHostKeyChecking=no", "-o", "CheckHostIP=no", "-i", E
.SSH_KEY_PATH
, puppet_host
]))
140 return util
.BuilderConfig(name
="TestProject_deploy", workernames
=["deploy-worker-test"], factory
=factory
)
142 from twisted
.internet
import defer
143 from buildbot
.process
.buildstep
import FAILURE
144 from buildbot
.process
.buildstep
import SUCCESS
145 from buildbot
.process
.buildstep
import BuildStep
147 class LdapPush(BuildStep
):
149 renderables
= ["environment", "build_version", "build_hash", "ldap_password"]
151 def __init__(self
, **kwargs
):
152 self
.environment
= kwargs
.pop("environment")
153 self
.build_version
= kwargs
.pop("build_version")
154 self
.build_hash
= kwargs
.pop("build_hash")
155 self
.ldap_password
= kwargs
.pop("ldap_password")
156 self
.ldap_host
= kwargs
.pop("ldap_host", E
.LDAP_HOST
)
157 super().__init
__(**kwargs
)
161 from ldap3
import Reader
, Writer
, Server
, Connection
, ObjectDef
162 server
= Server(self
.ldap_host
)
163 conn
= Connection(server
,
165 password
=self
.ldap_password
)
167 obj
= ObjectDef("immaePuppetClass", conn
)
168 r
= Reader(conn
, obj
,
169 "cn=test.{},{}".format(self
.environment
, E
.LDAP_ROLES_BASE
))
172 w
= Writer
.from_cursor(r
)
173 for value
in w
[0].immaePuppetJson
.values
:
174 config
= json
.loads(value
)
175 if "test_version" in config
:
176 config
["test_version"] = self
.build_version
177 config
["test_sha256"] = self
.build_hash
178 w
[0].immaePuppetJson
-= value
179 w
[0].immaePuppetJson
+= json
.dumps(config
, indent
=" ")
181 return defer
.succeed(SUCCESS
)
182 return defer
.succeed(FAILURE
)