statuscake/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 castInterfaceToSliceStrings(in interface{}) []string {
  14         input := in.([]interface{})
  15         res := make([]string, len(input))
  16
  17         for i, element := range input {
  18                 res[i] = element.(string)
  19         }
  20         return res
  21 }
  22
  23 func resourceStatusCakeTest() *schema.Resource {
  24         return &schema.Resource{
  25                 Create: CreateTest,
  26                 Update: UpdateTest,
  27                 Delete: DeleteTest,
  28                 Read:   ReadTest,
  29
  30                 Schema: map[string]*schema.Schema{
  31                         "test_id": {
  32                                 Type:     schema.TypeString,
  33                                 Computed: true,
  34                         },
  35
  36                         "website_name": {
  37                                 Type:     schema.TypeString,
  38                                 Required: true,
  39                         },
  40
  41                         "website_url": {
  42                                 Type:     schema.TypeString,
  43                                 Required: true,
  44                         },
  45
  46                         "contact_id": {
  47                                 Type:     schema.TypeInt,
  48                                 Optional: true,
  49                         },
  50
  51                         "check_rate": {
  52                                 Type:     schema.TypeInt,
  53                                 Optional: true,
  54                                 Default:  300,
  55                         },
  56
  57                         "test_type": {
  58                                 Type:     schema.TypeString,
  59                                 Required: true,
  60                         },
  61
  62                         "paused": {
  63                                 Type:     schema.TypeBool,
  64                                 Optional: true,
  65                                 Default:  false,
  66                         },
  67
  68                         "timeout": {
  69                                 Type:     schema.TypeInt,
  70                                 Optional: true,
  71                                 Default:  40,
  72                         },
  73
  74                         "confirmations": {
  75                                 Type:     schema.TypeInt,
  76                                 Optional: true,
  77                         },
  78
  79                         "port": {
  80                                 Type:     schema.TypeInt,
  81                                 Optional: true,
  82                         },
  83
  84                         "trigger_rate": {
  85                                 Type:     schema.TypeInt,
  86                                 Optional: true,
  87                                 Default:  5,
  88                         },
  89                         "custom_header": {
  90                                 Type:     schema.TypeString,
  91                                 Optional: true,
  92                         },
  93
  94                         "user_agent": {
  95                                 Type:     schema.TypeString,
  96                                 Optional: true,
  97                         },
  98
  99                         "status": {
 100                                 Type:     schema.TypeString,
 101                                 Computed: true,
 102                         },
 103
 104                         "uptime": {
 105                                 Type:     schema.TypeFloat,
 106                                 Computed: true,
 107                         },
 108
 109                         "node_locations": {
 110                                 Type:     schema.TypeList,
 111                                 Elem:     &schema.Schema{Type: schema.TypeString},
 112                                 Optional: true,
 113                         },
 114
 115                         "ping_url": {
 116                                 Type:     schema.TypeString,
 117                                 Optional: true,
 118                         },
 119
 120                         "basic_user": {
 121                                 Type:     schema.TypeString,
 122                                 Optional: true,
 123                         },
 124
 125                         "basic_pass": {
 126                                 Type:      schema.TypeString,
 127                                 Optional:  true,
 128                                 Sensitive: true,
 129                         },
 130
 131                         "public": {
 132                                 Type:     schema.TypeInt,
 133                                 Optional: true,
 134                         },
 135
 136                         "logo_image": {
 137                                 Type:     schema.TypeString,
 138                                 Optional: true,
 139                         },
 140
 141                         "branding": {
 142                                 Type:     schema.TypeInt,
 143                                 Optional: true,
 144                         },
 145
 146                         "website_host": {
 147                                 Type:     schema.TypeString,
 148                                 Optional: true,
 149                         },
 150
 151                         "virus": {
 152                                 Type:     schema.TypeInt,
 153                                 Optional: true,
 154                         },
 155
 156                         "find_string": {
 157                                 Type:     schema.TypeString,
 158                                 Optional: true,
 159                         },
 160
 161                         "do_not_find": {
 162                                 Type:     schema.TypeBool,
 163                                 Optional: true,
 164                         },
 165
 166                         "real_browser": {
 167                                 Type:     schema.TypeInt,
 168                                 Optional: true,
 169                         },
 170
 171                         "test_tags": {
 172                                 Type:     schema.TypeString,
 173                                 Optional: true,
 174                         },
 175
 176                         "status_codes": {
 177                                 Type:     schema.TypeString,
 178                                 Optional: true,
 179                                 Default: "204, 205, 206, 303, 400, 401, 403, 404, 405, 406, " +
 180                                         "408, 410, 413, 444, 429, 494, 495, 496, 499, 500, 501, 502, 503, " +
 181                                         "504, 505, 506, 507, 508, 509, 510, 511, 521, 522, 523, 524, 520, " +
 182                                         "598, 599",
 183                         },
 184
 185                         "use_jar": {
 186                                 Type:     schema.TypeInt,
 187                                 Optional: true,
 188                         },
 189
 190                         "post_raw": {
 191                                 Type:     schema.TypeString,
 192                                 Optional: true,
 193                         },
 194
 195                         "final_endpoint": {
 196                                 Type:     schema.TypeString,
 197                                 Optional: true,
 198                         },
 199
 200                         "follow_redirect": {
 201                                 Type:     schema.TypeBool,
 202                                 Optional: true,
 203                         },
 204                 },
 205         }
 206 }
 207
 208 func CreateTest(d *schema.ResourceData, meta interface{}) error {
 209         client := meta.(*statuscake.Client)
 210
 211         newTest := &statuscake.Test{
 212                 WebsiteName:    d.Get("website_name").(string),
 213                 WebsiteURL:     d.Get("website_url").(string),
 214                 CheckRate:      d.Get("check_rate").(int),
 215                 TestType:       d.Get("test_type").(string),
 216                 Paused:         d.Get("paused").(bool),
 217                 Timeout:        d.Get("timeout").(int),
 218                 ContactID:      d.Get("contact_id").(int),
 219                 Confirmation:   d.Get("confirmations").(int),
 220                 Port:           d.Get("port").(int),
 221                 TriggerRate:    d.Get("trigger_rate").(int),
 222                 CustomHeader:   d.Get("custom_header").(string),
 223                 UserAgent:      d.Get("user_agent").(string),
 224                 Status:         d.Get("status").(string),
 225                 Uptime:         d.Get("uptime").(float64),
 226                 NodeLocations:  castInterfaceToSliceStrings(d.Get("node_locations")),
 227                 PingURL:        d.Get("ping_url").(string),
 228                 BasicUser:      d.Get("basic_user").(string),
 229                 BasicPass:      d.Get("basic_pass").(string),
 230                 Public:         d.Get("public").(int),
 231                 LogoImage:      d.Get("logo_image").(string),
 232                 Branding:       d.Get("branding").(int),
 233                 WebsiteHost:    d.Get("website_host").(string),
 234                 Virus:          d.Get("virus").(int),
 235                 FindString:     d.Get("find_string").(string),
 236                 DoNotFind:      d.Get("do_not_find").(bool),
 237                 RealBrowser:    d.Get("real_browser").(int),
 238                 TestTags:       d.Get("test_tags").(string),
 239                 StatusCodes:    d.Get("status_codes").(string),
 240                 UseJar:         d.Get("use_jar").(int),
 241                 PostRaw:        d.Get("post_raw").(string),
 242                 FinalEndpoint:  d.Get("final_endpoint").(string),
 243                 FollowRedirect: d.Get("follow_redirect").(bool),
 244         }
 245
 246         log.Printf("[DEBUG] Creating new StatusCake Test: %s", d.Get("website_name").(string))
 247
 248         response, err := client.Tests().Update(newTest)
 249         if err != nil {
 250                 return fmt.Errorf("Error creating StatusCake Test: %s", err.Error())
 251         }
 252
 253         d.Set("test_id", fmt.Sprintf("%d", response.TestID))
 254         d.Set("status", response.Status)
 255         d.Set("uptime", fmt.Sprintf("%.3f", response.Uptime))
 256         d.SetId(fmt.Sprintf("%d", response.TestID))
 257
 258         return ReadTest(d, meta)
 259 }
 260
 261 func UpdateTest(d *schema.ResourceData, meta interface{}) error {
 262         client := meta.(*statuscake.Client)
 263
 264         params := getStatusCakeTestInput(d)
 265
 266         log.Printf("[DEBUG] StatusCake Test Update for %s", d.Id())
 267         _, err := client.Tests().Update(params)
 268         if err != nil {
 269                 return fmt.Errorf("Error Updating StatusCake Test: %s", err.Error())
 270         }
 271         return nil
 272 }
 273
 274 func DeleteTest(d *schema.ResourceData, meta interface{}) error {
 275         client := meta.(*statuscake.Client)
 276
 277         testId, parseErr := strconv.Atoi(d.Id())
 278         if parseErr != nil {
 279                 return parseErr
 280         }
 281         log.Printf("[DEBUG] Deleting StatusCake Test: %s", d.Id())
 282         err := client.Tests().Delete(testId)
 283         if err != nil {
 284                 return err
 285         }
 286
 287         return nil
 288 }
 289
 290 func ReadTest(d *schema.ResourceData, meta interface{}) error {
 291         client := meta.(*statuscake.Client)
 292
 293         testId, parseErr := strconv.Atoi(d.Id())
 294         if parseErr != nil {
 295                 return parseErr
 296         }
 297         testResp, err := client.Tests().Detail(testId)
 298         if err != nil {
 299                 return fmt.Errorf("Error Getting StatusCake Test Details for %s: Error: %s", d.Id(), err)
 300         }
 301         d.Set("website_name", testResp.WebsiteName)
 302         d.Set("website_url", testResp.WebsiteURL)
 303         d.Set("check_rate", testResp.CheckRate)
 304         d.Set("test_type", testResp.TestType)
 305         d.Set("paused", testResp.Paused)
 306         d.Set("timeout", testResp.Timeout)
 307         d.Set("contact_id", testResp.ContactID)
 308         d.Set("confirmations", testResp.Confirmation)
 309         d.Set("port", testResp.Port)
 310         d.Set("trigger_rate", testResp.TriggerRate)
 311         d.Set("custom_header", testResp.CustomHeader)
 312         d.Set("user_agent", testResp.UserAgent)
 313         d.Set("status", testResp.Status)
 314         d.Set("uptime", testResp.Uptime)
 315         if err := d.Set("node_locations", testResp.NodeLocations); err != nil {
 316                 return fmt.Errorf("[WARN] Error setting node locations: %s", err)
 317         }
 318         d.Set("ping_url", testResp.PingURL)
 319         d.Set("basic_user", testResp.BasicUser)
 320         d.Set("basic_pass", testResp.BasicPass)
 321         d.Set("public", testResp.Public)
 322         d.Set("logo_image", testResp.LogoImage)
 323         d.Set("branding", testResp.Branding)
 324         d.Set("website_host", testResp.WebsiteHost)
 325         d.Set("virus", testResp.Virus)
 326         d.Set("find_string", testResp.FindString)
 327         d.Set("do_not_find", testResp.DoNotFind)
 328         d.Set("real_browser", testResp.RealBrowser)
 329         d.Set("test_tags", testResp.TestTags)
 330         d.Set("status_codes", testResp.StatusCodes)
 331         d.Set("use_jar", testResp.UseJar)
 332         d.Set("post_raw", testResp.PostRaw)
 333         d.Set("final_endpoint", testResp.FinalEndpoint)
 334         d.Set("follow_redirect", testResp.FollowRedirect)
 335
 336         return nil
 337 }
 338
 339 func getStatusCakeTestInput(d *schema.ResourceData) *statuscake.Test {
 340         testId, parseErr := strconv.Atoi(d.Id())
 341         if parseErr != nil {
 342                 log.Printf("[DEBUG] Error Parsing StatusCake TestID: %s", d.Id())
 343         }
 344         test := &statuscake.Test{
 345                 TestID: testId,
 346         }
 347         if v, ok := d.GetOk("website_name"); ok {
 348                 test.WebsiteName = v.(string)
 349         }
 350         if v, ok := d.GetOk("website_url"); ok {
 351                 test.WebsiteURL = v.(string)
 352         }
 353         if v, ok := d.GetOk("check_rate"); ok {
 354                 test.CheckRate = v.(int)
 355         }
 356         if v, ok := d.GetOk("contact_id"); ok {
 357                 test.ContactID = v.(int)
 358         }
 359         if v, ok := d.GetOk("test_type"); ok {
 360                 test.TestType = v.(string)
 361         }
 362         if v, ok := d.GetOk("paused"); ok {
 363                 test.Paused = v.(bool)
 364         }
 365         if v, ok := d.GetOk("timeout"); ok {
 366                 test.Timeout = v.(int)
 367         }
 368         if v, ok := d.GetOk("confirmations"); ok {
 369                 test.Confirmation = v.(int)
 370         }
 371         if v, ok := d.GetOk("port"); ok {
 372                 test.Port = v.(int)
 373         }
 374         if v, ok := d.GetOk("trigger_rate"); ok {
 375                 test.TriggerRate = v.(int)
 376         }
 377         if v, ok := d.GetOk("custom_header"); ok {
 378                 test.CustomHeader = v.(string)
 379         }
 380         if v, ok := d.GetOk("user_agent"); ok {
 381                 test.UserAgent = v.(string)
 382         }
 383         if v, ok := d.GetOk("node_locations"); ok {
 384                 test.NodeLocations = castInterfaceToSliceStrings(v)
 385         }
 386         if v, ok := d.GetOk("ping_url"); ok {
 387                 test.PingURL = v.(string)
 388         }
 389         if v, ok := d.GetOk("basic_user"); ok {
 390                 test.BasicUser = v.(string)
 391         }
 392         if v, ok := d.GetOk("basic_pass"); ok {
 393                 test.BasicPass = v.(string)
 394         }
 395         if v, ok := d.GetOk("public"); ok {
 396                 test.Public = v.(int)
 397         }
 398         if v, ok := d.GetOk("logo_image"); ok {
 399                 test.LogoImage = v.(string)
 400         }
 401         if v, ok := d.GetOk("branding"); ok {
 402                 test.Branding = v.(int)
 403         }
 404         if v, ok := d.GetOk("website_host"); ok {
 405                 test.WebsiteHost = v.(string)
 406         }
 407         if v, ok := d.GetOk("virus"); ok {
 408                 test.Virus = v.(int)
 409         }
 410         if v, ok := d.GetOk("find_string"); ok {
 411                 test.FindString = v.(string)
 412         }
 413         if v, ok := d.GetOk("do_not_find"); ok {
 414                 test.DoNotFind = v.(bool)
 415         }
 416         if v, ok := d.GetOk("real_browser"); ok {
 417                 test.RealBrowser = v.(int)
 418         }
 419         if v, ok := d.GetOk("test_tags"); ok {
 420                 test.TestTags = v.(string)
 421         }
 422         if v, ok := d.GetOk("status_codes"); ok {
 423                 test.StatusCodes = v.(string)
 424         }
 425         if v, ok := d.GetOk("use_jar"); ok {
 426                 test.UseJar = v.(int)
 427         }
 428         if v, ok := d.GetOk("post_raw"); ok {
 429                 test.PostRaw = v.(string)
 430         }
 431         if v, ok := d.GetOk("final_endpoint"); ok {
 432                 test.FinalEndpoint = v.(string)
 433         }
 434         if v, ok := d.GetOk("follow_redirect"); ok {
 435                 test.FollowRedirect = v.(bool)
 436         }
 437
 438         return test
 439 }