from buildbot.plugins import *
from buildbot_common.build_helpers import *
import os
from buildbot.util import bytes2unicode
import json

__all__ = [ "configure", "E" ]

class E():
    PROJECT           = "denise"
    BUILDBOT_URL      = "https://git.immae.eu/buildbot/{}/".format(PROJECT)
    SOCKET            = "unix:/run/buildbot/{}.sock".format(PROJECT)
    PB_SOCKET         = "unix:address=/run/buildbot/{}_pb.sock".format(PROJECT)
    SSH_KEY_PATH      = "/var/lib/buildbot/buildbot_key"
    SSH_HOST_KEY      = "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIIFbhFTl2A2RJn5L51yxJM4XfCS2ZaiSX/jo9jFSdghF"

    OMS_RELEASE_PATH  = "/var/lib/buildbot/outputs/denise/oms"
    OMS_GIT_URL       = "gitolite@git.immae.eu:perso/Denise/oms"

    # master.cfg
    SECRETS_FILE       = os.getcwd() + "/secrets"
    LDAP_URL           = "ldaps://ldap.immae.eu:636"
    LDAP_ADMIN_USER    = "cn=buildbot,ou=services,dc=immae,dc=eu"
    LDAP_BASE          = "dc=immae,dc=eu"
    LDAP_PATTERN       = "(uid=%(username)s)"
    LDAP_GROUP_PATTERN = "(&(memberOf=cn=groups,ou=denise,cn=buildbot,ou=services,dc=immae,dc=eu)(member=%(dn)s))"
    TITLE_URL          = "https://oms.syanni.eu"
    TITLE              = "Syanni website"

class CustomBase(webhooks.base):
    def getChanges(self, request):
        try:
            content = request.content.read()
            args = json.loads(bytes2unicode(content))
        except Exception as e:
            raise ValueError("Error loading JSON: " + str(e))

        args.setdefault("comments", "")
        args.setdefault("repository", "")
        args.setdefault("author", args.get("who", "unknown"))

        return ([args], None)

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(worker.LocalWorker("generic-worker-denise"))

    c['schedulers'].append(hook_scheduler("DeniseOMS", timer=1))
    c['schedulers'].append(force_scheduler("force_denise", [
        "DeniseOMS_build"
        ]))

    c['builders'].append(denise_oms_factory())

def denise_oms_factory():
    factory = util.BuildFactory()
    factory.addStep(steps.Git(logEnviron=False, repourl=E.OMS_GIT_URL,
        submodules=True, sshPrivateKey=open(E.SSH_KEY_PATH).read().rstrip(),
        sshHostKey=E.SSH_HOST_KEY, mode="full", method="copy"))
    factory.addStep(steps.MasterShellCommand(command="rm -rf {}".format(E.OMS_RELEASE_PATH)))
    factory.addStep(steps.DirectoryUpload(workersrc="../source",
        masterdest=E.OMS_RELEASE_PATH,
        url="https://oms.syanni.eu"))
    factory.addStep(steps.MasterShellCommand(command="chmod -R a+rX {}".format(E.OMS_RELEASE_PATH)))
    factory.addStep(steps.MasterShellCommand(command="kill $(cat /run/denise_oms/gunicorn.pid)"))

    return util.BuilderConfig(name="DeniseOMS_build", workernames=["generic-worker-denise"], factory=factory)