Skip to content

Commit

Permalink
Adding IAM resources for Dataplex Lake, Zone and Asset (#7436) (#5348)
Browse files Browse the repository at this point in the history
Signed-off-by: Modular Magician <[email protected]>
  • Loading branch information
modular-magician authored Mar 21, 2023
1 parent 5afdac1 commit 1a35ede
Show file tree
Hide file tree
Showing 14 changed files with 2,320 additions and 14 deletions.
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

0 comments on commit 1a35ede

Please sign in to comment.