Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adding IAM resources for Dataplex Lake, Zone and Asset #5348

Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 9 additions & 0 deletions .changelog/7436.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
```release-note:new-resource
`google_dataplex_lake_iam_*`
```
```release-note:new-resource
`google_dataplex_zone_iam_*`
```
```release-note:new-resource
`google_dataplex_asset_iam_*`
```
4 changes: 4 additions & 0 deletions google-beta/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -211,6 +211,7 @@ type Config struct {
DataformBasePath string
DataFusionBasePath string
DataLossPreventionBasePath string
DataplexBasePath string
DataprocBasePath string
DataprocMetastoreBasePath string
DatastoreBasePath string
Expand Down Expand Up @@ -329,6 +330,7 @@ const DataCatalogBasePathKey = "DataCatalog"
const DataformBasePathKey = "Dataform"
const DataFusionBasePathKey = "DataFusion"
const DataLossPreventionBasePathKey = "DataLossPrevention"
const DataplexBasePathKey = "Dataplex"
const DataprocBasePathKey = "Dataproc"
const DataprocMetastoreBasePathKey = "DataprocMetastore"
const DatastoreBasePathKey = "Datastore"
Expand Down Expand Up @@ -441,6 +443,7 @@ var DefaultBasePaths = map[string]string{
DataformBasePathKey: "https://dataform.googleapis.com/v1beta1/",
DataFusionBasePathKey: "https://datafusion.googleapis.com/v1beta1/",
DataLossPreventionBasePathKey: "https://dlp.googleapis.com/v2/",
DataplexBasePathKey: "https://dataplex.googleapis.com/v1/",
DataprocBasePathKey: "https://dataproc.googleapis.com/v1beta2/",
DataprocMetastoreBasePathKey: "https://metastore.googleapis.com/v1beta/",
DatastoreBasePathKey: "https://datastore.googleapis.com/v1/",
Expand Down Expand Up @@ -1343,6 +1346,7 @@ func ConfigureBasePaths(c *Config) {
c.DataformBasePath = DefaultBasePaths[DataformBasePathKey]
c.DataFusionBasePath = DefaultBasePaths[DataFusionBasePathKey]
c.DataLossPreventionBasePath = DefaultBasePaths[DataLossPreventionBasePathKey]
c.DataplexBasePath = DefaultBasePaths[DataplexBasePathKey]
c.DataprocBasePath = DefaultBasePaths[DataprocBasePathKey]
c.DataprocMetastoreBasePath = DefaultBasePaths[DataprocMetastoreBasePathKey]
c.DatastoreBasePath = DefaultBasePaths[DatastoreBasePathKey]
Expand Down
1 change: 1 addition & 0 deletions google-beta/config_test_utils.go
Original file line number Diff line number Diff line change
Expand Up @@ -56,6 +56,7 @@ func configureTestBasePaths(c *Config, url string) {
c.DataformBasePath = url
c.DataFusionBasePath = url
c.DataLossPreventionBasePath = url
c.DataplexBasePath = url
c.DataprocBasePath = url
c.DataprocMetastoreBasePath = url
c.DatastoreBasePath = url
Expand Down
253 changes: 253 additions & 0 deletions google-beta/iam_dataplex_asset.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,253 @@
// ----------------------------------------------------------------------------
//
// *** 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 DataplexAssetIamSchema = map[string]*schema.Schema{
"project": {
Type: schema.TypeString,
Computed: true,
Optional: true,
ForceNew: true,
},
"location": {
Type: schema.TypeString,
Computed: true,
Optional: true,
ForceNew: true,
},
"lake": {
Type: schema.TypeString,
Required: true,
ForceNew: true,
},
"dataplex_zone": {
Type: schema.TypeString,
Required: true,
ForceNew: true,
},
"asset": {
Type: schema.TypeString,
Required: true,
ForceNew: true,
DiffSuppressFunc: compareSelfLinkOrResourceName,
},
}

type DataplexAssetIamUpdater struct {
project string
location string
lake string
dataplexZone string
asset string
d TerraformResourceData
Config *Config
}

func DataplexAssetIamUpdaterProducer(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
location, _ := getLocation(d, config)
if location != "" {
if err := d.Set("location", location); err != nil {
return nil, fmt.Errorf("Error setting location: %s", err)
}
}
values["location"] = location
if v, ok := d.GetOk("lake"); ok {
values["lake"] = v.(string)
}

if v, ok := d.GetOk("dataplex_zone"); ok {
values["dataplex_zone"] = v.(string)
}

if v, ok := d.GetOk("asset"); ok {
values["asset"] = 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>[^/]+)/lakes/(?P<lake>[^/]+)/zones/(?P<dataplex_zone>[^/]+)/assets/(?P<asset>[^/]+)", "(?P<project>[^/]+)/(?P<location>[^/]+)/(?P<lake>[^/]+)/(?P<dataplex_zone>[^/]+)/(?P<asset>[^/]+)", "(?P<location>[^/]+)/(?P<lake>[^/]+)/(?P<dataplex_zone>[^/]+)/(?P<asset>[^/]+)", "(?P<asset>[^/]+)"}, d, config, d.Get("asset").(string))
if err != nil {
return nil, err
}

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

u := &DataplexAssetIamUpdater{
project: values["project"],
location: values["location"],
lake: values["lake"],
dataplexZone: values["dataplex_zone"],
asset: values["asset"],
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("location", u.location); err != nil {
return nil, fmt.Errorf("Error setting location: %s", err)
}
if err := d.Set("lake", u.lake); err != nil {
return nil, fmt.Errorf("Error setting lake: %s", err)
}
if err := d.Set("dataplex_zone", u.dataplexZone); err != nil {
return nil, fmt.Errorf("Error setting dataplex_zone: %s", err)
}
if err := d.Set("asset", u.GetResourceId()); err != nil {
return nil, fmt.Errorf("Error setting asset: %s", err)
}

return u, nil
}

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

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

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

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

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

u := &DataplexAssetIamUpdater{
project: values["project"],
location: values["location"],
lake: values["lake"],
dataplexZone: values["dataplex_zone"],
asset: values["asset"],
d: d,
Config: config,
}
if err := d.Set("asset", u.GetResourceId()); err != nil {
return fmt.Errorf("Error setting asset: %s", err)
}
d.SetId(u.GetResourceId())
return nil
}

func (u *DataplexAssetIamUpdater) GetResourceIamPolicy() (*cloudresourcemanager.Policy, error) {
url, err := u.qualifyAssetUrl("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 *DataplexAssetIamUpdater) 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.qualifyAssetUrl("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 *DataplexAssetIamUpdater) qualifyAssetUrl(methodIdentifier string) (string, error) {
urlTemplate := fmt.Sprintf("{{DataplexBasePath}}%s:%s", fmt.Sprintf("projects/%s/locations/%s/lakes/%s/zones/%s/assets/%s", u.project, u.location, u.lake, u.dataplexZone, u.asset), methodIdentifier)
url, err := replaceVars(u.d, u.Config, urlTemplate)
if err != nil {
return "", err
}
return url, nil
}

func (u *DataplexAssetIamUpdater) GetResourceId() string {
return fmt.Sprintf("projects/%s/locations/%s/lakes/%s/zones/%s/assets/%s", u.project, u.location, u.lake, u.dataplexZone, u.asset)
}

func (u *DataplexAssetIamUpdater) GetMutexKey() string {
return fmt.Sprintf("iam-dataplex-asset-%s", u.GetResourceId())
}

func (u *DataplexAssetIamUpdater) DescribeResource() string {
return fmt.Sprintf("dataplex asset %q", u.GetResourceId())
}
Loading