Skip to content

Commit

Permalink
datafusion - add args + iam support (hashicorp#6270)
Browse files Browse the repository at this point in the history
Closes hashicorp/terraform-provider-google#10536
Closes hashicorp/terraform-provider-google#9065
Closes hashicorp/terraform-provider-google#9068

Signed-off-by: Modular Magician <magic-modules@google.com>
  • Loading branch information
modular-magician committed Nov 29, 2022
1 parent 04ec1a6 commit 6dba89a
Show file tree
Hide file tree
Showing 8 changed files with 1,007 additions and 31 deletions.
15 changes: 15 additions & 0 deletions .changelog/6270.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
```release-note:enhancement
datafusion: add `zone`, `display_name`, `crypto_key_config`, `event_publish_config`, and `enable_rbac` args to `google_data_fusion_instance`
```
```release-note:enhancement
datafusion: add `api_endpoint` and `p4_service_account ` attributes to `google_data_fusion_instance`
```
```release-note:new-resource
google_data_fusion_instance_iam_member
```
```release-note:new-resource
google_data_fusion_instance_iam_binding
```
```release-note:new-resource
google_data_fusion_instance_iam_policy
```
223 changes: 223 additions & 0 deletions google-beta/iam_data_fusion_instance.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,223 @@
// ----------------------------------------------------------------------------
//
// *** AUTO GENERATED CODE *** Type: MMv1 ***
//
// ----------------------------------------------------------------------------
//
// This file is automatically generated by Magic Modules and manual
// changes will be clobbered when the file is regenerated.
//
// Please read more about how to change this file in
// .github/CONTRIBUTING.md.
//
// ----------------------------------------------------------------------------

package google

import (
"fmt"

"github.com/hashicorp/errwrap"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
"google.golang.org/api/cloudresourcemanager/v1"
)

var DataFusionInstanceIamSchema = map[string]*schema.Schema{
"project": {
Type: schema.TypeString,
Computed: true,
Optional: true,
ForceNew: true,
},
"region": {
Type: schema.TypeString,
Computed: true,
Optional: true,
ForceNew: true,
},
"name": {
Type: schema.TypeString,
Required: true,
ForceNew: true,
DiffSuppressFunc: compareSelfLinkOrResourceName,
},
}

type DataFusionInstanceIamUpdater struct {
project string
region string
name string
d TerraformResourceData
Config *Config
}

func DataFusionInstanceIamUpdaterProducer(d TerraformResourceData, config *Config) (ResourceIamUpdater, error) {
values := make(map[string]string)

project, _ := getProject(d, config)
if project != "" {
if err := d.Set("project", project); err != nil {
return nil, fmt.Errorf("Error setting project: %s", err)
}
}
values["project"] = project
region, _ := getRegion(d, config)
if region != "" {
if err := d.Set("region", region); err != nil {
return nil, fmt.Errorf("Error setting region: %s", err)
}
}
values["region"] = region
if v, ok := d.GetOk("name"); ok {
values["name"] = v.(string)
}

// We may have gotten either a long or short name, so attempt to parse long name if possible
m, err := getImportIdQualifiers([]string{"projects/(?P<project>[^/]+)/locations/(?P<location>[^/]+)/instances/(?P<name>[^/]+)", "(?P<project>[^/]+)/(?P<location>[^/]+)/(?P<name>[^/]+)", "(?P<location>[^/]+)/(?P<name>[^/]+)", "(?P<name>[^/]+)"}, d, config, d.Get("name").(string))
if err != nil {
return nil, err
}

for k, v := range m {
values[k] = v
}

u := &DataFusionInstanceIamUpdater{
project: values["project"],
region: values["region"],
name: values["name"],
d: d,
Config: config,
}

if err := d.Set("project", u.project); err != nil {
return nil, fmt.Errorf("Error setting project: %s", err)
}
if err := d.Set("region", u.region); err != nil {
return nil, fmt.Errorf("Error setting region: %s", err)
}
if err := d.Set("name", u.GetResourceId()); err != nil {
return nil, fmt.Errorf("Error setting name: %s", err)
}

return u, nil
}

func DataFusionInstanceIdParseFunc(d *schema.ResourceData, config *Config) error {
values := make(map[string]string)

project, _ := getProject(d, config)
if project != "" {
values["project"] = project
}

region, _ := getRegion(d, config)
if region != "" {
values["region"] = region
}

m, err := getImportIdQualifiers([]string{"projects/(?P<project>[^/]+)/locations/(?P<location>[^/]+)/instances/(?P<name>[^/]+)", "(?P<project>[^/]+)/(?P<location>[^/]+)/(?P<name>[^/]+)", "(?P<location>[^/]+)/(?P<name>[^/]+)", "(?P<name>[^/]+)"}, d, config, d.Id())
if err != nil {
return err
}

for k, v := range m {
values[k] = v
}

u := &DataFusionInstanceIamUpdater{
project: values["project"],
region: values["region"],
name: values["name"],
d: d,
Config: config,
}
if err := d.Set("name", u.GetResourceId()); err != nil {
return fmt.Errorf("Error setting name: %s", err)
}
d.SetId(u.GetResourceId())
return nil
}

func (u *DataFusionInstanceIamUpdater) GetResourceIamPolicy() (*cloudresourcemanager.Policy, error) {
url, err := u.qualifyInstanceUrl("getIamPolicy")
if err != nil {
return nil, err
}

project, err := getProject(u.d, u.Config)
if err != nil {
return nil, err
}
var obj map[string]interface{}

userAgent, err := generateUserAgentString(u.d, u.Config.userAgent)
if err != nil {
return nil, err
}

policy, err := sendRequest(u.Config, "GET", project, url, userAgent, obj)
if err != nil {
return nil, errwrap.Wrapf(fmt.Sprintf("Error retrieving IAM policy for %s: {{err}}", u.DescribeResource()), err)
}

out := &cloudresourcemanager.Policy{}
err = Convert(policy, out)
if err != nil {
return nil, errwrap.Wrapf("Cannot convert a policy to a resource manager policy: {{err}}", err)
}

return out, nil
}

func (u *DataFusionInstanceIamUpdater) SetResourceIamPolicy(policy *cloudresourcemanager.Policy) error {
json, err := ConvertToMap(policy)
if err != nil {
return err
}

obj := make(map[string]interface{})
obj["policy"] = json

url, err := u.qualifyInstanceUrl("setIamPolicy")
if err != nil {
return err
}
project, err := getProject(u.d, u.Config)
if err != nil {
return err
}

userAgent, err := generateUserAgentString(u.d, u.Config.userAgent)
if err != nil {
return err
}

_, err = sendRequestWithTimeout(u.Config, "POST", project, url, userAgent, obj, u.d.Timeout(schema.TimeoutCreate))
if err != nil {
return errwrap.Wrapf(fmt.Sprintf("Error setting IAM policy for %s: {{err}}", u.DescribeResource()), err)
}

return nil
}

func (u *DataFusionInstanceIamUpdater) qualifyInstanceUrl(methodIdentifier string) (string, error) {
urlTemplate := fmt.Sprintf("{{DataFusionBasePath}}%s:%s", fmt.Sprintf("projects/%s/locations/%s/instances/%s", u.project, u.region, u.name), methodIdentifier)
url, err := replaceVars(u.d, u.Config, urlTemplate)
if err != nil {
return "", err
}
return url, nil
}

func (u *DataFusionInstanceIamUpdater) GetResourceId() string {
return fmt.Sprintf("projects/%s/locations/%s/instances/%s", u.project, u.region, u.name)
}

func (u *DataFusionInstanceIamUpdater) GetMutexKey() string {
return fmt.Sprintf("iam-datafusion-instance-%s", u.GetResourceId())
}

func (u *DataFusionInstanceIamUpdater) DescribeResource() string {
return fmt.Sprintf("datafusion instance %q", u.GetResourceId())
}

0 comments on commit 6dba89a

Please sign in to comment.