| // Copyright (c) HashiCorp, Inc. |
| // SPDX-License-Identifier: MPL-2.0 |
| package tags |
| |
| import ( |
| "encoding/json" |
| "fmt" |
| "regexp" |
| "strings" |
| "time" |
| |
| "github.com/hashicorp/terraform-provider-google-beta/google-beta/tpgresource" |
| transport_tpg "github.com/hashicorp/terraform-provider-google-beta/google-beta/transport" |
| ) |
| |
| type TagsLocationOperationWaiter struct { |
| Config *transport_tpg.Config |
| UserAgent string |
| tpgresource.CommonOperationWaiter |
| } |
| |
| func (w *TagsLocationOperationWaiter) QueryOp() (interface{}, error) { |
| if w == nil { |
| return nil, fmt.Errorf("Cannot query operation, it's unset or nil.") |
| } |
| location := GetLocationFromOpName(w.CommonOperationWaiter.Op.Name) |
| if location != w.CommonOperationWaiter.Op.Name { |
| // Found location in Op.Name, fill it in TagsLocationBasePath and rewrite URL |
| url := fmt.Sprintf("%s%s", strings.Replace(w.Config.TagsLocationBasePath, "{{location}}", location, 1), w.CommonOperationWaiter.Op.Name) |
| return transport_tpg.SendRequest(transport_tpg.SendRequestOptions{ |
| Config: w.Config, |
| Method: "GET", |
| RawURL: url, |
| UserAgent: w.UserAgent, |
| }) |
| } else { |
| url := fmt.Sprintf("%s%s", w.Config.TagsBasePath, w.CommonOperationWaiter.Op.Name) |
| return transport_tpg.SendRequest(transport_tpg.SendRequestOptions{ |
| Config: w.Config, |
| Method: "GET", |
| RawURL: url, |
| UserAgent: w.UserAgent, |
| }) |
| } |
| } |
| |
| func createTagsLocationWaiter(config *transport_tpg.Config, op map[string]interface{}, activity, userAgent string) (*TagsLocationOperationWaiter, error) { |
| w := &TagsLocationOperationWaiter{ |
| Config: config, |
| UserAgent: userAgent, |
| } |
| if err := w.CommonOperationWaiter.SetOp(op); err != nil { |
| return nil, err |
| } |
| return w, nil |
| } |
| |
| func TagsLocationOperationWaitTimeWithResponse(config *transport_tpg.Config, op map[string]interface{}, response *map[string]interface{}, activity, userAgent string, timeout time.Duration) error { |
| w, err := createTagsLocationWaiter(config, op, activity, userAgent) |
| if err != nil { |
| return err |
| } |
| if err := tpgresource.OperationWait(w, activity, timeout, config.PollInterval); err != nil { |
| return err |
| } |
| return json.Unmarshal([]byte(w.CommonOperationWaiter.Op.Response), response) |
| } |
| |
| func TagsLocationOperationWaitTime(config *transport_tpg.Config, op map[string]interface{}, activity, userAgent string, timeout time.Duration) error { |
| if val, ok := op["name"]; !ok || val == "" { |
| // This was a synchronous call - there is no operation to wait for. |
| return nil |
| } |
| w, err := createTagsLocationWaiter(config, op, activity, userAgent) |
| if err != nil { |
| // If w is nil, the op was synchronous. |
| return err |
| } |
| return tpgresource.OperationWait(w, activity, timeout, config.PollInterval) |
| } |
| |
| func GetLocationFromOpName(opName string) string { |
| re := regexp.MustCompile("operations/(?:rctb|rdtb)\\.([a-zA-Z0-9-]*)\\.([0-9]*)") |
| switch { |
| case re.MatchString(opName): |
| if res := re.FindStringSubmatch(opName); len(res) == 3 && res[1] != "" { |
| return res[1] |
| } |
| } |
| return opName |
| } |