teamcityparser.go

package goteamcity

import(
    "encoding/json"
    "io"
    "log"
    "strings"
)

const Success string = "Success"
const Fail string = "Fail"
const Investigating string = "Investigating"

type project struct {
	Name string
	WebUrl string
	LastBuildTime string
	LastBuildLabel string
	LastBuildStatus string
}

type teamcityResponse struct {
	Project []project
}

var investigationsPath string = "/httpAuth/app/rest/investigations?locator=buildType:(name:%s)"

func parseResponse(response io.ReadCloser, r investigationReader) string {
    decoder := json.NewDecoder(response)
    teamCityStatus := teamcityResponse{}
    err := decoder.Decode(&teamCityStatus)
    if err != nil {
        log.Fatalf("Error: %s", err)
    }
    projectCount := len(teamCityStatus.Project)
    successCount := 0
    failureCount := 0
    investigateCount := 0

    for i := 0; i < projectCount; i++ {
        buildStatus := teamCityStatus.Project[i].LastBuildStatus
        if buildStatus == "Success" {
        	successCount++
        } else if buildStatus == "Failure" {
        	name := parseName(teamCityStatus.Project[i].Name)
        	if r.IsInvestigating(name) {
        		investigateCount++
        	}
        	failureCount++
        }
    }

    if successCount > 0 && failureCount == 0 {
    	return Success
    } else if failureCount == investigateCount {
    	return Investigating
    }

    return Fail
}

func parseName(name string) string {
	pos := strings.LastIndex(name, "::")
	if pos < 0 {
		return name
	}

	return strings.Trim(name[pos + 2:len(name)], " ")
}