# -*- python -*-
# ex: set syntax=python:

# <pep8 compliant>

# List of the branches being built automatically overnight
NIGHT_SCHEDULE_BRANCHES = [None, "blender2.8"]

# List of the branches available for force build
FORCE_SCHEDULE_BRANCHES = ["master", "blender2.8", "experimental-build"]

# Dictionary that the buildmaster pays attention to.
c = BuildmasterConfig = {}

# BUILD WORKERS
#
# We load the slaves and their passwords from a separator file, so we can have
# this one in SVN.

from buildbot.worker import Worker
import master_private

c['workers'] = []
for slave in master_private.slaves:
    c['workers'].append(Worker(slave['name'], slave['password']))

# TCP port through which slaves connect
c['protocols'] = {'pb': {'port': 9989}}

# CHANGE SOURCES

from buildbot.changes.svnpoller import SVNPoller
from buildbot.changes.gitpoller import GitPoller

c['change_source'] = GitPoller('git://git.blender.org/blender.git',
                               pollinterval=1200)

# CODEBASES
#
# Allow to control separately things like branches for each repo and submodules.

all_repositories = {
    r'git://git.blender.org/blender.git': 'blender',
    r'git://git.blender.org/blender-translations.git': 'blender-translations',
    r'git://git.blender.org/blender-addons.git': 'blender-addons',
    r'git://git.blender.org/blender-addons-contrib.git': 'blender-addons-contrib',
    r'git://git.blender.org/blender-dev-tools.git': 'blender-dev-tools',
    r'https://svn.blender.org/svnroot/bf-blender/': 'lib svn',
}


def codebaseGenerator(chdict):
    return all_repositories[chdict['repository']]

c['codebaseGenerator'] = codebaseGenerator


# SCHEDULERS
#
# Decide how to react to incoming changes.

# from buildbot.scheduler import Scheduler
from buildbot.schedulers import timed, forcesched

c['schedulers'] = []


def schedule_force_build(name):
    c['schedulers'].append(forcesched.ForceScheduler(name='force_' + name,
        builderNames=[name],
        codebases=[forcesched.CodebaseParameter(
                codebase="blender",
                branch=forcesched.ChoiceStringParameter(
                    name="branch", choices=FORCE_SCHEDULE_BRANCHES, default="master"),
                # Do not hide revision, can be handy!
                repository=forcesched.FixedParameter(name="repository", default="", hide=True),
                project=forcesched.FixedParameter(name="project", default="", hide=True)),
            # For now, hide other codebases.
            forcesched.CodebaseParameter(hide=True, codebase="blender-translations"),
            forcesched.CodebaseParameter(
                codebase="blender-addons",
                branch=forcesched.ChoiceStringParameter(
                    name="branch", choices=["master", "blender2.8"], default="master"),
                repository=forcesched.FixedParameter(name="repository", default="", hide=True),
                project=forcesched.FixedParameter(name="project", default="", hide=True),
                revision=forcesched.FixedParameter(name="revision", default="", hide=True),
            ),
            forcesched.CodebaseParameter(hide=True, codebase="blender-addons-contrib"),
            forcesched.CodebaseParameter(hide=True, codebase="blender-dev-tools"),
            forcesched.CodebaseParameter(hide=True, codebase="lib svn")],
        properties=[]))


def schedule_build(name, hour, minute=0):
    for current_branch in NIGHT_SCHEDULE_BRANCHES:
        scheduler_name = "nightly " + name
        if current_branch:
            scheduler_name += ' ' + current_branch
        # Use special addons submodule branch when building blender2.8 branch.
        addons_branch = "master"
        if current_branch == "blender2.8":
            addons_branch = "blender2.8"
        c['schedulers'].append(timed.Nightly(name=scheduler_name,
            codebases={
                "blender": {"repository": ""},
                "blender-translations": {"repository": "", "branch": "master"},
                "blender-addons": {"repository": "", "branch": addons_branch},
                "blender-addons-contrib": {"repository": "", "branch": "master"},
                "blender-dev-tools": {"repository": "", "branch": "master"},
                "lib svn": {"repository": "", "branch": "trunk"}},
            branch=current_branch,
            builderNames=[name],
            hour=hour,
            minute=minute))


# BUILDERS
#
# The 'builders' list defines the Builders, which tell Buildbot how to
# perform a build: what steps, and which slaves can execute them.
# Note that any particular build will only take place on one slave.

from buildbot.config import BuilderConfig
from buildbot.plugins import steps, util
from buildbot.process.factory import BuildFactory
from buildbot.process.properties import Interpolate
from buildbot.steps.shell import ShellCommand
from buildbot.steps.shell import Compile
from buildbot.steps.shell import Test
from buildbot.steps.transfer import FileUpload
from buildbot.steps.master import MasterShellCommand

# add builder utility

c['builders'] = []
buildernames = []


def add_builder(c, name, libdir, factory, branch='',
                rsync=False, hour=3, minute=0):
    workernames = []

    for slave in master_private.slaves:
        if name in slave['builders']:
            workernames.append(slave['name'])

    if workernames:
        f = factory(name, libdir, branch, rsync)
        c['builders'].append(BuilderConfig(name=name,
                                           workernames=workernames,
                                           factory=f,
                                           tags=['blender']))
        buildernames.append(name)

        schedule_build(name, hour, minute)
        schedule_force_build(name)

# common steps


def git_submodule_step(submodule):
    return steps.Git(name=submodule + '.git',
                     repourl='git://git.blender.org/' + submodule + '.git',
                     mode='incremental',
                     codebase=submodule,
                     workdir=submodule + '.git')


def git_step(branch=''):
    if branch:
        return steps.Git(name='blender.git',
                         repourl='git://git.blender.org/blender.git',
                         mode='incremental',
                         branch=branch,
                         codebase='blender',
                         workdir='blender.git',
                         submodules=True)
    else:
        return steps.Git(name='blender.git',
                         repourl='git://git.blender.org/blender.git',
                         mode='incremental',
                         codebase='blender',
                         workdir='blender.git',
                         submodules=True)


def git_submodules_update():
    command = ['git', 'submodule', 'update', '--remote']
    return ShellCommand(name='Submodules Update',
                        command=command,
                        description='updating',
                        descriptionDone='up to date',
                        workdir='blender.git')


def lib_svn_step(dir):
    return steps.SVN(name='lib svn',
                     repourl=util.Interpolate('https://svn.blender.org/svnroot/bf-blender/%(src::branch)s/lib/' + dir),
                     codebase='lib svn',
                     mode='incremental',
                     workdir='lib/' + dir)


def rsync_step(id, branch, rsync_script):
    return ShellCommand(name='rsync',
                        command=['python', rsync_script, id, branch],
                        description='uploading',
                        descriptionDone='uploaded',
                        workdir='install')

# generic builder


def generic_builder(id, libdir='', branch='', rsync=False):
    filename = 'uploaded/buildbot_upload_' + id + '.zip'
    compile_script = '../blender.git/build_files/buildbot/slave_compile.py'
    test_script = '../blender.git/build_files/buildbot/slave_test.py'
    pack_script = '../blender.git/build_files/buildbot/slave_pack.py'
    rsync_script = '../blender.git/build_files/buildbot/slave_rsync.py'
    unpack_script = 'master_unpack.py'

    f = BuildFactory()
    if libdir != '':
        f.addStep(lib_svn_step(libdir))

    for submodule in ('blender-translations',
                      'blender-addons',
                      'blender-addons-contrib',
                      'blender-dev-tools'):
        f.addStep(git_submodule_step(submodule))

    f.addStep(git_step(branch))
    f.addStep(git_submodules_update())

    f.addStep(Compile(command=['python', compile_script, id], timeout=3600))
    f.addStep(Test(command=['python', test_script, id]))
    f.addStep(ShellCommand(name='package',
                           command=['python', pack_script, id, branch or Interpolate('%(src:blender:branch)s')],
                           description='packaging',
                           descriptionDone='packaged'))
    if rsync:
        f.addStep(rsync_step(id, branch, rsync_script))
    else:
        f.addStep(FileUpload(name='upload',
                             workersrc='buildbot_upload.zip',
                             masterdest=filename,
                             maxsize=180 * 1024 * 1024,
                             workdir='install'))
    f.addStep(MasterShellCommand(name='unpack',
                                 command=['python2.7', unpack_script, filename],
                                 description='unpacking',
                                 descriptionDone='unpacked'))
    return f

# Builders

add_builder(c, 'mac_x86_64_10_9_cmake', 'darwin', generic_builder, hour=1)
add_builder(c, 'linux_glibc219_i686_cmake', '', generic_builder, hour=2)
add_builder(c, 'linux_glibc219_x86_64_cmake', '', generic_builder, hour=1)
add_builder(c, 'win32_cmake_vc2013', 'windows_vc12', generic_builder, hour=1)
add_builder(c, 'win64_cmake_vc2013', 'win64_vc12', generic_builder, hour=2)
add_builder(c, 'win32_cmake_vc2015', 'windows_vc14', generic_builder, hour=3)
add_builder(c, 'win64_cmake_vc2015', 'win64_vc14', generic_builder, hour=4)

# WWW
c['www'] = dict(port=8010)

# PROJECT IDENTITY
c['projectName'] = "Blender"
c['projectURL'] = "http://www.blender.org"

# Buildbot information
c['buildbotURL'] = "http://builder.blender.org/"
c['buildbotNetUsageData'] = 'basic'

# Various
c['db_url'] = "sqlite:///state.sqlite"