-
Notifications
You must be signed in to change notification settings - Fork 4.6k
/
virtual_desktop_workspace_data_source.go
129 lines (103 loc) · 3.9 KB
/
virtual_desktop_workspace_data_source.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: MPL-2.0
package desktopvirtualization
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/commonschema"
"github.com/hashicorp/go-azure-helpers/resourcemanager/location"
"github.com/hashicorp/go-azure-sdk/resource-manager/desktopvirtualization/2022-02-10-preview/workspace"
"github.com/hashicorp/terraform-provider-azurerm/internal/sdk"
"github.com/hashicorp/terraform-provider-azurerm/internal/services/desktopvirtualization/validate"
"github.com/hashicorp/terraform-provider-azurerm/internal/tf/pluginsdk"
)
type DesktopVirtualizationWorkspaceDataSource struct{}
type DesktopVirtualizationWorkspaceModel struct {
Name string `tfschema:"name"`
ResourceGroup string `tfschema:"resource_group_name"`
Location string `tfschema:"location"`
FriendlyName string `tfschema:"friendly_name"`
Description string `tfschema:"description"`
PublicNetworkAccess bool `tfschema:"public_network_access_enabled"`
Tags map[string]string `tfschema:"tags"`
}
var _ sdk.DataSource = DesktopVirtualizationWorkspaceDataSource{}
func (d DesktopVirtualizationWorkspaceDataSource) ModelObject() interface{} {
return &DesktopVirtualizationWorkspaceModel{}
}
func (d DesktopVirtualizationWorkspaceDataSource) ResourceType() string {
return "azurerm_virtual_desktop_workspace"
}
func (d DesktopVirtualizationWorkspaceDataSource) Arguments() map[string]*pluginsdk.Schema {
return map[string]*pluginsdk.Schema{
"name": {
Type: pluginsdk.TypeString,
Required: true,
ValidateFunc: validate.WorkspaceName,
},
"resource_group_name": commonschema.ResourceGroupNameForDataSource(),
}
}
func (d DesktopVirtualizationWorkspaceDataSource) Attributes() map[string]*pluginsdk.Schema {
return map[string]*pluginsdk.Schema{
"location": commonschema.LocationComputed(),
"friendly_name": {
Type: pluginsdk.TypeString,
Computed: true,
},
"description": {
Type: pluginsdk.TypeString,
Computed: true,
},
"public_network_access_enabled": {
Type: pluginsdk.TypeBool,
Computed: true,
},
"tags": commonschema.TagsDataSource(),
}
}
func (d DesktopVirtualizationWorkspaceDataSource) Read() sdk.ResourceFunc {
return sdk.ResourceFunc{
Timeout: 5 * time.Minute,
Func: func(ctx context.Context, metadata sdk.ResourceMetaData) error {
client := metadata.Client.DesktopVirtualization.WorkspacesClient
subscriptionId := metadata.Client.Account.SubscriptionId
var state DesktopVirtualizationWorkspaceModel
if err := metadata.Decode(&state); err != nil {
return err
}
id := workspace.NewWorkspaceID(subscriptionId, state.ResourceGroup, state.Name)
resp, err := client.Get(ctx, id)
if err != nil {
if response.WasNotFound(resp.HttpResponse) {
return fmt.Errorf("%s was not found", id)
}
return fmt.Errorf("retrieving %s: %+v", id, err)
}
model := resp.Model
if model == nil {
return fmt.Errorf("retrieving %s: model was nil", id)
}
state.Location = location.NormalizeNilable(model.Location)
state.Tags = pointer.From(model.Tags)
if properties := model.Properties; properties != nil {
if properties.FriendlyName != nil {
state.FriendlyName = pointer.From(properties.FriendlyName)
}
if properties.Description != nil {
state.Description = pointer.From(properties.Description)
}
publicNetworkAccess := true
if properties.PublicNetworkAccess != nil && pointer.From(properties.PublicNetworkAccess) != workspace.PublicNetworkAccessEnabled {
publicNetworkAccess = false
}
state.PublicNetworkAccess = publicNetworkAccess
}
metadata.SetID(id)
return metadata.Encode(&state)
},
}
}