From 7d98448cdab9c225227f244c6e49efbc42b07651 Mon Sep 17 00:00:00 2001 From: Jake Champlin Date: Fri, 9 Jun 2017 17:53:46 +0000 Subject: Transfer statuscake provider --- statuscake/resource_statuscaketest.go | 216 ++++++++++++++++++++++++++++++++++ 1 file changed, 216 insertions(+) create mode 100644 statuscake/resource_statuscaketest.go (limited to 'statuscake/resource_statuscaketest.go') diff --git a/statuscake/resource_statuscaketest.go b/statuscake/resource_statuscaketest.go new file mode 100644 index 0000000..101ee83 --- /dev/null +++ b/statuscake/resource_statuscaketest.go @@ -0,0 +1,216 @@ +package statuscake + +import ( + "fmt" + "strconv" + + "log" + + "github.com/DreamItGetIT/statuscake" + "github.com/hashicorp/terraform/helper/schema" +) + +func resourceStatusCakeTest() *schema.Resource { + return &schema.Resource{ + Create: CreateTest, + Update: UpdateTest, + Delete: DeleteTest, + Read: ReadTest, + + Schema: map[string]*schema.Schema{ + "test_id": { + Type: schema.TypeString, + Computed: true, + }, + + "website_name": { + Type: schema.TypeString, + Required: true, + }, + + "website_url": { + Type: schema.TypeString, + Required: true, + }, + + "contact_id": { + Type: schema.TypeInt, + Optional: true, + }, + + "check_rate": { + Type: schema.TypeInt, + Optional: true, + Default: 300, + }, + + "test_type": { + Type: schema.TypeString, + Required: true, + }, + + "paused": { + Type: schema.TypeBool, + Optional: true, + Default: false, + }, + + "timeout": { + Type: schema.TypeInt, + Optional: true, + Default: 40, + }, + + "confirmations": { + Type: schema.TypeInt, + Optional: true, + }, + + "port": { + Type: schema.TypeInt, + Optional: true, + }, + + "trigger_rate": { + Type: schema.TypeInt, + Optional: true, + Default: 5, + }, + }, + } +} + +func CreateTest(d *schema.ResourceData, meta interface{}) error { + client := meta.(*statuscake.Client) + + newTest := &statuscake.Test{ + WebsiteName: d.Get("website_name").(string), + WebsiteURL: d.Get("website_url").(string), + CheckRate: d.Get("check_rate").(int), + TestType: d.Get("test_type").(string), + Paused: d.Get("paused").(bool), + Timeout: d.Get("timeout").(int), + ContactID: d.Get("contact_id").(int), + Confirmation: d.Get("confirmations").(int), + Port: d.Get("port").(int), + TriggerRate: d.Get("trigger_rate").(int), + } + + log.Printf("[DEBUG] Creating new StatusCake Test: %s", d.Get("website_name").(string)) + + response, err := client.Tests().Update(newTest) + if err != nil { + return fmt.Errorf("Error creating StatusCake Test: %s", err.Error()) + } + + d.Set("test_id", fmt.Sprintf("%d", response.TestID)) + d.SetId(fmt.Sprintf("%d", response.TestID)) + + return ReadTest(d, meta) +} + +func UpdateTest(d *schema.ResourceData, meta interface{}) error { + client := meta.(*statuscake.Client) + + params := getStatusCakeTestInput(d) + + log.Printf("[DEBUG] StatusCake Test Update for %s", d.Id()) + _, err := client.Tests().Update(params) + if err != nil { + return fmt.Errorf("Error Updating StatusCake Test: %s", err.Error()) + } + return nil +} + +func DeleteTest(d *schema.ResourceData, meta interface{}) error { + client := meta.(*statuscake.Client) + + testId, parseErr := strconv.Atoi(d.Id()) + if parseErr != nil { + return parseErr + } + log.Printf("[DEBUG] Deleting StatusCake Test: %s", d.Id()) + err := client.Tests().Delete(testId) + if err != nil { + return err + } + + return nil +} + +func ReadTest(d *schema.ResourceData, meta interface{}) error { + client := meta.(*statuscake.Client) + + testId, parseErr := strconv.Atoi(d.Id()) + if parseErr != nil { + return parseErr + } + testResp, err := client.Tests().Detail(testId) + if err != nil { + return fmt.Errorf("Error Getting StatusCake Test Details for %s: Error: %s", d.Id(), err) + } + d.Set("website_name", testResp.WebsiteName) + d.Set("website_url", testResp.WebsiteURL) + d.Set("check_rate", testResp.CheckRate) + d.Set("test_type", testResp.TestType) + d.Set("paused", testResp.Paused) + d.Set("timeout", testResp.Timeout) + d.Set("contact_id", testResp.ContactID) + d.Set("confirmations", testResp.Confirmation) + d.Set("port", testResp.Port) + d.Set("trigger_rate", testResp.TriggerRate) + + return nil +} + +func getStatusCakeTestInput(d *schema.ResourceData) *statuscake.Test { + testId, parseErr := strconv.Atoi(d.Id()) + if parseErr != nil { + log.Printf("[DEBUG] Error Parsing StatusCake TestID: %s", d.Id()) + } + test := &statuscake.Test{ + TestID: testId, + } + if v, ok := d.GetOk("website_name"); ok { + test.WebsiteName = v.(string) + } + if v, ok := d.GetOk("website_url"); ok { + test.WebsiteURL = v.(string) + } + if v, ok := d.GetOk("check_rate"); ok { + test.CheckRate = v.(int) + } + if v, ok := d.GetOk("contact_id"); ok { + test.ContactID = v.(int) + } + if v, ok := d.GetOk("test_type"); ok { + test.TestType = v.(string) + } + if v, ok := d.GetOk("paused"); ok { + test.Paused = v.(bool) + } + if v, ok := d.GetOk("timeout"); ok { + test.Timeout = v.(int) + } + if v, ok := d.GetOk("contact_id"); ok { + test.ContactID = v.(int) + } + if v, ok := d.GetOk("confirmations"); ok { + test.Confirmation = v.(int) + } + if v, ok := d.GetOk("port"); ok { + test.Port = v.(int) + } + if v, ok := d.GetOk("trigger_rate"); ok { + test.TriggerRate = v.(int) + } + + defaultStatusCodes := "204, 205, 206, 303, 400, 401, 403, 404, 405, 406, " + + "408, 410, 413, 444, 429, 494, 495, 496, 499, 500, 501, 502, 503, " + + "504, 505, 506, 507, 508, 509, 510, 511, 521, 522, 523, 524, 520, " + + "598, 599" + + test.StatusCodes = defaultStatusCodes + + return test +} -- cgit v1.2.3