resource_statuscaketest.go

   1 package statuscake
   2
   3 import (
   4         "fmt"
   5         "strconv"
   6
   7         "log"
   8
   9         "github.com/DreamItGetIT/statuscake"
  10         "github.com/hashicorp/terraform/helper/schema"
  11 )
  12
  13 func resourceStatusCakeTest() *schema.Resource {
  14         return &schema.Resource{
  15                 Create: CreateTest,
  16                 Update: UpdateTest,
  17                 Delete: DeleteTest,
  18                 Read:   ReadTest,
  19
  20                 Schema: map[string]*schema.Schema{
  21                         "test_id": &schema.Schema{
  22                                 Type:     schema.TypeString,
  23                                 Computed: true,
  24                         },
  25
  26                         "website_name": &schema.Schema{
  27                                 Type:     schema.TypeString,
  28                                 Required: true,
  29                         },
  30
  31                         "website_url": &schema.Schema{
  32                                 Type:     schema.TypeString,
  33                                 Required: true,
  34                         },
  35
  36                         "contact_id": &schema.Schema{
  37                                 Type:     schema.TypeInt,
  38                                 Optional: true,
  39                         },
  40
  41                         "check_rate": &schema.Schema{
  42                                 Type:     schema.TypeInt,
  43                                 Optional: true,
  44                                 Default:  300,
  45                         },
  46
  47                         "test_type": &schema.Schema{
  48                                 Type:     schema.TypeString,
  49                                 Required: true,
  50                         },
  51
  52                         "paused": &schema.Schema{
  53                                 Type:     schema.TypeBool,
  54                                 Optional: true,
  55                                 Default:  false,
  56                         },
  57                         "timeout": &schema.Schema{
  58                                 Type:     schema.TypeInt,
  59                                 Computed: true,
  60                         },
  61                 },
  62         }
  63 }
  64
  65 func CreateTest(d *schema.ResourceData, meta interface{}) error {
  66         client := meta.(*statuscake.Client)
  67
  68         newTest := &statuscake.Test{
  69                 WebsiteName: d.Get("website_name").(string),
  70                 WebsiteURL:  d.Get("website_url").(string),
  71                 TestType:    d.Get("test_type").(string),
  72                 CheckRate:   d.Get("check_rate").(int),
  73                 ContactID:   d.Get("contact_id").(int),
  74         }
  75
  76         log.Printf("[DEBUG] Creating new StatusCake Test: %s", d.Get("website_name").(string))
  77
  78         response, err := client.Tests().Update(newTest)
  79         if err != nil {
  80                 return fmt.Errorf("Error creating StatusCake Test: %s", err.Error())
  81         }
  82
  83         d.Set("test_id", fmt.Sprintf("%d", response.TestID))
  84         d.SetId(fmt.Sprintf("%d", response.TestID))
  85
  86         return ReadTest(d, meta)
  87 }
  88
  89 func UpdateTest(d *schema.ResourceData, meta interface{}) error {
  90         client := meta.(*statuscake.Client)
  91
  92         params := getStatusCakeTestInput(d)
  93
  94         log.Printf("[DEBUG] StatusCake Test Update for %s", d.Id())
  95         _, err := client.Tests().Update(params)
  96         if err != nil {
  97                 return fmt.Errorf("Error Updating StatusCake Test: %s", err.Error())
  98         }
  99         return nil
 100 }
 101
 102 func DeleteTest(d *schema.ResourceData, meta interface{}) error {
 103         client := meta.(*statuscake.Client)
 104
 105         testId, parseErr := strconv.Atoi(d.Id())
 106         if parseErr != nil {
 107                 return parseErr
 108         }
 109         log.Printf("[DEBUG] Deleting StatusCake Test: %s", d.Id())
 110         err := client.Tests().Delete(testId)
 111         if err != nil {
 112                 return err
 113         }
 114
 115         return nil
 116 }
 117
 118 func ReadTest(d *schema.ResourceData, meta interface{}) error {
 119         client := meta.(*statuscake.Client)
 120
 121         testId, parseErr := strconv.Atoi(d.Id())
 122         if parseErr != nil {
 123                 return parseErr
 124         }
 125         testResp, err := client.Tests().Detail(testId)
 126         if err != nil {
 127                 return fmt.Errorf("Error Getting StatusCake Test Details for %s: Error: %s", d.Id(), err)
 128         }
 129         d.Set("check_rate", testResp.CheckRate)
 130
 131         return nil
 132 }
 133
 134 func getStatusCakeTestInput(d *schema.ResourceData) *statuscake.Test {
 135         testId, parseErr := strconv.Atoi(d.Id())
 136         if parseErr != nil {
 137                 log.Printf("[DEBUG] Error Parsing StatusCake TestID: %s", d.Id())
 138         }
 139         test := &statuscake.Test{
 140                 TestID: testId,
 141         }
 142         if v, ok := d.GetOk("website_name"); ok {
 143                 test.WebsiteName = v.(string)
 144         }
 145         if v, ok := d.GetOk("website_url"); ok {
 146                 test.WebsiteURL = v.(string)
 147         }
 148         if v, ok := d.GetOk("check_rate"); ok {
 149                 test.CheckRate = v.(int)
 150         }
 151         if v, ok := d.GetOk("contact_id"); ok {
 152                 test.ContactID = v.(int)
 153         }
 154         if v, ok := d.GetOk("test_type"); ok {
 155                 test.TestType = v.(string)
 156         }
 157         if v, ok := d.GetOk("paused"); ok {
 158                 test.Paused = v.(bool)
 159         }
 160         if v, ok := d.GetOk("timeout"); ok {
 161                 test.Timeout = v.(int)
 162         }
 163         return test
 164 }