resource_cabot_check_jenkins.go

package main

import (
	"github.com/hashicorp/terraform/helper/resource"
	"github.com/hashicorp/terraform/helper/schema"
	"github.com/mrsaints/go-cabot/cabot"
	"strconv"
	"time"
)

func resourceCabotCheckJenkins() *schema.Resource {
	s := map[string]*schema.Schema{
		"max_queued_build_time": &schema.Schema{
			Type:     schema.TypeInt,
			Optional: true,
		},
		"jenkins_config": &schema.Schema{
			Type:     schema.TypeInt,
			Required: true,
		},
	}

	return &schema.Resource{
		Create: resourceCabotCheckJenkinsCreate,
		Read:   resourceCabotCheckJenkinsRead,
		Update: resourceCabotCheckJenkinsUpdate,
		Delete: resourceCabotCheckJenkinsDelete,
		Importer: &schema.ResourceImporter{
			State: schema.ImportStatePassthrough,
		},

		Schema: CombineWithBaseCheckSchema(s),
	}
}

func getJenkinsCheckFromResourceData(d *schema.ResourceData) *cabot.JenkinsCheck {
	checkRequest := &cabot.JenkinsCheck{
		MaxQueuedBuildTime: d.Get("max_queued_build_time").(int),
		JenkinsConfig:      d.Get("jenkins_config").(int),
	}
	checkRequest.StatusCheck = getStatusCheckFromResourceData(d)
	return checkRequest
}

func resourceCabotCheckJenkinsCreate(d *schema.ResourceData, meta interface{}) error {
	client := meta.(*cabot.Client)

	check, err := client.JenkinsChecks.Create(getJenkinsCheckFromResourceData(d))
	if err != nil {
		return err
	}

	d.SetId(strconv.Itoa(check.ID))
	return resourceCabotCheckJenkinsRead(d, meta)
}

func resourceCabotCheckJenkinsRead(d *schema.ResourceData, meta interface{}) error {
	client := meta.(*cabot.Client)

	id, _ := strconv.Atoi(d.Id())

	return resource.Retry(DEFAULT_RETRY_TIMEOUT, func() *resource.RetryError {
		check, err := client.JenkinsChecks.Get(id)
		if err != nil {
			time.Sleep(DEFAULT_GRACE_PERIOD)
			return resource.RetryableError(err)
		}

		d.SetId(strconv.Itoa(check.ID))
		setResourceDataForStatusCheck(d, check.StatusCheck)
		d.Set("max_queued_build_time", check.MaxQueuedBuildTime)

		return nil
	})
}

func resourceCabotCheckJenkinsUpdate(d *schema.ResourceData, meta interface{}) error {
	client := meta.(*cabot.Client)

	id, _ := strconv.Atoi(d.Id())
	check, err := client.JenkinsChecks.Update((id), getJenkinsCheckFromResourceData(d))
	if err != nil {
		return err
	}

	d.SetId(strconv.Itoa(check.ID))
	return resourceCabotCheckJenkinsRead(d, meta)
}

func resourceCabotCheckJenkinsDelete(d *schema.ResourceData, meta interface{}) error {
	client := meta.(*cabot.Client)
	id, _ := strconv.Atoi(d.Id())
	return client.JenkinsChecks.Delete(id)
}