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