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

New Resource: azurerm_restore_point_collection #26518

Merged
merged 5 commits into from
Jul 3, 2024
Merged
Show file tree
Hide file tree
Changes from 2 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 internal/services/compute/client/client.go
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@ import (
"github.com/hashicorp/go-azure-sdk/resource-manager/compute/2024-03-01/availabilitysets"
"github.com/hashicorp/go-azure-sdk/resource-manager/compute/2024-03-01/dedicatedhostgroups"
"github.com/hashicorp/go-azure-sdk/resource-manager/compute/2024-03-01/dedicatedhosts"
"github.com/hashicorp/go-azure-sdk/resource-manager/compute/2024-03-01/restorepointcollections"
"github.com/hashicorp/go-azure-sdk/resource-manager/compute/2024-03-01/sshpublickeys"
"github.com/hashicorp/go-azure-sdk/resource-manager/compute/2024-03-01/virtualmachineextensions"
"github.com/hashicorp/go-azure-sdk/resource-manager/compute/2024-03-01/virtualmachineimages"
Expand Down Expand Up @@ -61,6 +62,7 @@ type Client struct {
ImagesClient *images.ImagesClient
MarketplaceAgreementsClient *agreements.AgreementsClient
ProximityPlacementGroupsClient *proximityplacementgroups.ProximityPlacementGroupsClient
RestorePointCollectionsClient *restorepointcollections.RestorePointCollectionsClient
SkusClient *skus.SkusClient
SSHPublicKeysClient *sshpublickeys.SshPublicKeysClient
SnapshotsClient *snapshots.SnapshotsClient
Expand Down Expand Up @@ -177,6 +179,12 @@ func NewClient(o *common.ClientOptions) (*Client, error) {
}
o.Configure(proximityPlacementGroupsClient.Client, o.Authorizers.ResourceManager)

restorePointCollectionsClient, err := restorepointcollections.NewRestorePointCollectionsClientWithBaseURI(o.Environment.ResourceManager)
if err != nil {
return nil, fmt.Errorf("building RestorePointCollections client: %+v", err)
}
o.Configure(restorePointCollectionsClient.Client, o.Authorizers.ResourceManager)

skusClient, err := skus.NewSkusClientWithBaseURI(o.Environment.ResourceManager)
if err != nil {
return nil, fmt.Errorf("building Skus client: %+v", err)
Expand Down Expand Up @@ -261,6 +269,7 @@ func NewClient(o *common.ClientOptions) (*Client, error) {
ImagesClient: imagesClient,
MarketplaceAgreementsClient: marketplaceAgreementsClient,
ProximityPlacementGroupsClient: proximityPlacementGroupsClient,
RestorePointCollectionsClient: restorePointCollectionsClient,
SkusClient: skusClient,
SSHPublicKeysClient: sshPublicKeysClient,
SnapshotsClient: snapshotsClient,
Expand Down
1 change: 1 addition & 0 deletions internal/services/compute/registration.go
Original file line number Diff line number Diff line change
Expand Up @@ -92,6 +92,7 @@ func (r Registration) Resources() []sdk.Resource {
VirtualMachineRunCommandResource{},
GalleryApplicationResource{},
GalleryApplicationVersionResource{},
RestorePointCollectionResource{},
VirtualMachineGalleryApplicationAssignmentResource{},
}
}
211 changes: 211 additions & 0 deletions internal/services/compute/restore_point_collection_resource.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,211 @@
package compute

import (
"context"
"fmt"
"time"

"github.com/hashicorp/go-azure-helpers/lang/pointer"
"github.com/hashicorp/go-azure-helpers/lang/response"
"github.com/hashicorp/go-azure-helpers/resourcemanager/commonids"
"github.com/hashicorp/go-azure-helpers/resourcemanager/commonschema"
"github.com/hashicorp/go-azure-helpers/resourcemanager/tags"
"github.com/hashicorp/go-azure-sdk/resource-manager/compute/2024-03-01/restorepointcollections"
"github.com/hashicorp/terraform-provider-azurerm/internal/sdk"
"github.com/hashicorp/terraform-provider-azurerm/internal/tf/pluginsdk"
)

type RestorePointCollectionResource struct{}

var _ sdk.ResourceWithUpdate = RestorePointCollectionResource{}

func (r RestorePointCollectionResource) ModelObject() interface{} {
return &RestorePointCollectionResourceModel{}
}

type RestorePointCollectionResourceModel struct {
Name string `tfschema:"name"`
ResourceGroup string `tfschema:"resource_group_name"`
Location string `tfschema:"location"`
SourceVirtualMachineId string `tfschema:"source_virtual_machine_id"`
Tags map[string]interface{} `tfschema:"tags"`
}

func (r RestorePointCollectionResource) IDValidationFunc() pluginsdk.SchemaValidateFunc {
return restorepointcollections.ValidateRestorePointCollectionID
}

func (r RestorePointCollectionResource) ResourceType() string {
return "azurerm_restore_point_collection"
}

func (r RestorePointCollectionResource) Arguments() map[string]*pluginsdk.Schema {
return map[string]*pluginsdk.Schema{
"name": {
ForceNew: true,
Required: true,
Type: pluginsdk.TypeString,
},

"resource_group_name": commonschema.ResourceGroupName(),

"location": commonschema.Location(),

"source_virtual_machine_id": {
Type: pluginsdk.TypeString,
Required: true,
ForceNew: true,
ValidateFunc: commonids.ValidateVirtualMachineID,
},

"tags": commonschema.Tags(),
}
}

func (r RestorePointCollectionResource) Attributes() map[string]*pluginsdk.Schema {
return map[string]*pluginsdk.Schema{}
}

func (r RestorePointCollectionResource) Create() sdk.ResourceFunc {
return sdk.ResourceFunc{
Timeout: 30 * time.Minute,
Func: func(ctx context.Context, metadata sdk.ResourceMetaData) error {
client := metadata.Client.Compute.RestorePointCollectionsClient
subscriptionId := metadata.Client.Account.SubscriptionId

var config RestorePointCollectionResourceModel
if err := metadata.Decode(&config); err != nil {
return fmt.Errorf("decoding: %+v", err)
}

id := restorepointcollections.NewRestorePointCollectionID(subscriptionId, config.ResourceGroup, config.Name)

existing, err := client.Get(ctx, id, restorepointcollections.DefaultGetOperationOptions())
if err != nil {
if !response.WasNotFound(existing.HttpResponse) {
return fmt.Errorf("checking for the presence of an existing %s: %+v", id, err)
}
}
if !response.WasNotFound(existing.HttpResponse) {
return metadata.ResourceRequiresImport(r.ResourceType(), id)
}

parameters := restorepointcollections.RestorePointCollection{
Location: config.Location,
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
Location: config.Location,
Location: location.Normalize(config.Location),

should we normalize the location here?

Properties: &restorepointcollections.RestorePointCollectionProperties{
Source: &restorepointcollections.RestorePointCollectionSourceProperties{
Id: pointer.To(config.SourceVirtualMachineId),
},
},
Tags: tags.Expand(config.Tags),
}

if _, err = client.CreateOrUpdate(ctx, id, parameters); err != nil {
return fmt.Errorf("creating %s: %+v", id, err)
}

metadata.SetID(id)
return nil
},
}
}

func (r RestorePointCollectionResource) Read() sdk.ResourceFunc {
return sdk.ResourceFunc{
Timeout: 5 * time.Minute,
Func: func(ctx context.Context, metadata sdk.ResourceMetaData) error {
client := metadata.Client.Compute.RestorePointCollectionsClient

schema := RestorePointCollectionResourceModel{}

id, err := restorepointcollections.ParseRestorePointCollectionID(metadata.ResourceData.Id())
if err != nil {
return err
}

resp, err := client.Get(ctx, *id, restorepointcollections.DefaultGetOperationOptions())
if err != nil {
if response.WasNotFound(resp.HttpResponse) {
return metadata.MarkAsGone(*id)
}
return fmt.Errorf("retrieving %s: %+v", *id, err)
}

if model := resp.Model; model != nil {
schema.Name = id.RestorePointCollectionName
schema.ResourceGroup = id.ResourceGroupName

if props := model.Properties; props != nil {
if source := props.Source; source != nil {
schema.SourceVirtualMachineId = pointer.From(source.Id)
schema.Location = pointer.From(source.Location)
}
}

schema.Tags = tags.Flatten(model.Tags)
}

return metadata.Encode(&schema)
},
}
}

func (r RestorePointCollectionResource) Update() sdk.ResourceFunc {
return sdk.ResourceFunc{
Timeout: 30 * time.Minute,
Func: func(ctx context.Context, metadata sdk.ResourceMetaData) error {
client := metadata.Client.Compute.RestorePointCollectionsClient

id, err := restorepointcollections.ParseRestorePointCollectionID(metadata.ResourceData.Id())
if err != nil {
return err
}

existing, err := client.Get(ctx, *id, restorepointcollections.DefaultGetOperationOptions())
if err != nil {
return fmt.Errorf("checking for presence of existing %s: %+v", id, err)
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
return fmt.Errorf("checking for presence of existing %s: %+v", id, err)
return fmt.Errorf("retrieving %s: %+v", id, err)

}

if existing.Model == nil {
return fmt.Errorf("retrieving %s: `model` was nil", id)
}

payload := *existing.Model

var config RestorePointCollectionResourceModel
if err := metadata.Decode(&config); err != nil {
return fmt.Errorf("decoding: %+v", err)
}

if metadata.ResourceData.HasChange("tags") {
payload.Tags = tags.Expand(config.Tags)
}

if _, err = client.CreateOrUpdate(ctx, *id, payload); err != nil {
return fmt.Errorf("updating %s: %+v", *id, err)
}

return nil
},
}
}

func (r RestorePointCollectionResource) Delete() sdk.ResourceFunc {
return sdk.ResourceFunc{
Timeout: 30 * time.Minute,
Func: func(ctx context.Context, metadata sdk.ResourceMetaData) error {
client := metadata.Client.Compute.RestorePointCollectionsClient

id, err := restorepointcollections.ParseRestorePointCollectionID(metadata.ResourceData.Id())
if err != nil {
return err
}

if err := client.DeleteThenPoll(ctx, *id); err != nil {
return fmt.Errorf("deleting %s: %+v", *id, err)
}

return nil
},
}
}
Loading
Loading