cpg1111/maestro

/*
Copyright 2016 Christian Grabowski All rights reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
    http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package pipeline

import (
    "log"

    git "gopkg.in/libgit2/git2go.v24"

    "github.com/cpg1111/maestro/statecom"
)

func build(srv *DepService, index string, stateCom *statecom.StateCom, done chan string, errChan chan error, shouldDeploy *bool) {
    buildName := srv.build.conf.Name
    stateCom.Services[buildName].SetState("building", true)
    err := srv.build.execBuild()
    if err != nil {
        stateCom.Services[buildName].SetState("built", false)
        errChan <- err
        return
    }
    stateCom.Services[buildName].SetState("built", true)
    log.Println("Run tests")
    stateCom.Services[buildName].SetState("testing", true)
    err = RunTests(srv.build)
    if err != nil {
        stateCom.Services[buildName].SetState("tested", false)
        errChan <- err
        return
    }
    log.Println("Tests done")
    stateCom.Services[buildName].SetState("tested", true)
    if !*shouldDeploy {
        stateCom.Services[buildName].SetState("finished", true)
        log.Println("No Deploy")
        done <- index
        return
    }
    log.Println("Checking deployment...")
    stateCom.Services[buildName].SetState("deploying", true)
    err = check(srv.build)
    if err != nil {
        stateCom.Services[buildName].SetState("finished", false)
        errChan <- err
        return
    }
    stateCom.Services[buildName].SetState("finished", true)
    srv.build.shouldBuild = false
    done <- index
    return
}

func runServiceBuild(srvs map[string]*DepService, stateCom *statecom.StateCom, testAll, shouldDeploy *bool) error {
    doneChan := make(chan string)
    errChan := make(chan error)
    buildTotal := 0
    for i := range srvs {
        if srvs[i].build.shouldBuild || *testAll {
            buildTotal++
            log.Println("Building", srvs[i].build.conf.Name)
            go build(srvs[i], i, stateCom, doneChan, errChan, shouldDeploy)
        } else {
            err := runServiceBuild(srvs[i].Children, stateCom, testAll, shouldDeploy)
            if err != nil {
                return err
            }
        }
    }
    total := 0
    var siblingFailed bool
    if buildTotal > 0 {
        for {
            select {
            case index := <-doneChan:
                total++
                if len(srvs[index].Children) > 0 && !siblingFailed {
                    runErr := runServiceBuild(srvs[index].Children, stateCom, testAll, shouldDeploy)
                    if runErr != nil {
                        return runErr
                    }
                }
                if total == buildTotal {
                    return nil
                }
            case errMsg := <-errChan:
                if errMsg != nil {
                    total++
                    log.Println(errMsg)
                    if total < buildTotal {
                        siblingFailed = true
                        continue
                    }
                    return errMsg
                }
            }
        }
    }
    return nil
}

// Run runs the build for all changed services
func Run(depTree *DepTree, repo *git.Repository, stateCom *statecom.StateCom, lastBuildCommit, currBuildCommit *string, testAll, shouldDeploy *bool) error {
    currNode := depTree.CurrNode
    travErr := TraverseTree(currNode, repo, lastBuildCommit, currBuildCommit)
    if travErr != nil {
        return travErr
    }
    rootMap := make(map[string]*DepService)
    rootMap["root"] = currNode
    err := runServiceBuild(rootMap, stateCom, testAll, shouldDeploy)
    if err != nil {
        return err
    }
    return nil
}