New Data Source: aws_workspaces_bundle

aws/config.go

@@ -80,6 +80,7 @@ import (
 	"github.com/aws/aws-sdk-go/service/sts"
 	"github.com/aws/aws-sdk-go/service/waf"
 	"github.com/aws/aws-sdk-go/service/wafregional"
+	"github.com/aws/aws-sdk-go/service/workspaces"
 	"github.com/davecgh/go-spew/spew"
 	"github.com/hashicorp/errwrap"
 	"github.com/hashicorp/go-cleanhttp"
@@ -206,6 +207,7 @@ type AWSClient struct {
 	athenaconn            *athena.Athena
 	dxconn                *directconnect.DirectConnect
 	mediastoreconn        *mediastore.MediaStore
+	wsconn                *workspaces.WorkSpaces
 }
 
 func (c *AWSClient) S3() *s3.S3 {
@@ -451,6 +453,7 @@ func (c *Config) Client() (interface{}, error) {
 	client.athenaconn = athena.New(sess)
 	client.dxconn = directconnect.New(sess)
 	client.mediastoreconn = mediastore.New(sess)
+	client.wsconn = workspaces.New(sess)
 
 	// Workaround for https://github.com/aws/aws-sdk-go/issues/1376
 	client.kinesisconn.Handlers.Retry.PushBack(func(r *request.Request) {

aws/data_source_aws_workspaces_bundle.go

@@ -0,0 +1,118 @@
+package aws
+
+import (
+	"fmt"
+
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/service/workspaces"
+	"github.com/hashicorp/terraform/helper/schema"
+)
+
+func dataSourceAwsWorkspaceBundle() *schema.Resource {
+	return &schema.Resource{
+		Read: dataSourceAwsWorkspaceBundleRead,
+
+		Schema: map[string]*schema.Schema{
+			"bundle_id": {
+				Type:     schema.TypeString,
+				Required: true,
+				ForceNew: true,
+			},
+			"description": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"name": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"owner": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"compute_type": {
+				Type:     schema.TypeSet,
+				Computed: true,
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"name": {
+							Type:     schema.TypeString,
+							Computed: true,
+						},
+					},
+				},
+			},
+			"user_storage": {
+				Type:     schema.TypeSet,
+				Computed: true,
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"capacity": {
+							Type:     schema.TypeString,
+							Computed: true,
+						},
+					},
+				},
+			},
+			"root_storage": {
+				Type:     schema.TypeSet,
+				Computed: true,
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"capacity": {
+							Type:     schema.TypeString,
+							Computed: true,
+						},
+					},
+				},
+			},
+		},
+	}
+}
+
+func dataSourceAwsWorkspaceBundleRead(d *schema.ResourceData, meta interface{}) error {
+	conn := meta.(*AWSClient).wsconn
+
+	bundleID := d.Get("bundle_id").(string)
+	input := &workspaces.DescribeWorkspaceBundlesInput{
+		BundleIds: []*string{aws.String(bundleID)},
+	}
+
+	resp, err := conn.DescribeWorkspaceBundles(input)
+	if err != nil {
+		return err
+	}
+
+	if len(resp.Bundles) != 1 {
+		return fmt.Errorf("The number of Workspace Bundle (%s) should be 1, but %d", bundleID, len(resp.Bundles))
+	}
+
+	bundle := resp.Bundles[0]
+	d.SetId(bundleID)
+	d.Set("description", bundle.Description)
+	d.Set("name", bundle.Name)
+	d.Set("owner", bundle.Owner)
+	if bundle.ComputeType != nil {
+		r := map[string]interface{}{
+			"name": *bundle.ComputeType.Name,
+		}
+		ct := []map[string]interface{}{r}
+		d.Set("compute_type", ct)
+	}
+	if bundle.RootStorage != nil {
+		r := map[string]interface{}{
+			"capacity": *bundle.RootStorage.Capacity,
+		}
+		rs := []map[string]interface{}{r}
+		d.Set("root_storage", rs)
+	}
+	if bundle.UserStorage != nil {
+		r := map[string]interface{}{
+			"capacity": *bundle.UserStorage.Capacity,
+		}
+		us := []map[string]interface{}{r}
+		d.Set("user_storage", us)
+	}
+
+	return nil
+}

aws/data_source_aws_workspaces_bundle_test.go

@@ -0,0 +1,32 @@
+package aws
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform/helper/resource"
+)
+
+func TestAccDataSourceAwsWorkspaceBundle_basic(t *testing.T) {
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			resource.TestStep{
+				Config: testAccDataSourceAwsWorkspaceBundleConfig,
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttr("data.aws_workspaces_bundle.test", "bundle_id", "wsb-b0s22j3d7"),
+					resource.TestCheckResourceAttrSet(
+						"data.aws_workspaces_bundle.test", "name"),
+					resource.TestCheckResourceAttrSet(
+						"data.aws_workspaces_bundle.test", "owner"),
+				),
+			},
+		},
+	})
+}
+
+const testAccDataSourceAwsWorkspaceBundleConfig = `
+data "aws_workspaces_bundle" "test" {
+  bundle_id = "wsb-b0s22j3d7"
+}
+`

aws/provider.go

@@ -227,6 +227,7 @@ func Provider() terraform.ResourceProvider {
 			"aws_vpc_endpoint_service":             dataSourceAwsVpcEndpointService(),
 			"aws_vpc_peering_connection":           dataSourceAwsVpcPeeringConnection(),
 			"aws_vpn_gateway":                      dataSourceAwsVpnGateway(),
+			"aws_workspaces_bundle":                dataSourceAwsWorkspaceBundle(),
 
 			// Adding the Aliases for the ALB -> LB Rename
 			"aws_lb":               dataSourceAwsLb(),

website/aws.erb

@@ -233,6 +233,9 @@
                         <li<%= sidebar_current("docs-aws-datasource-vpn-gateway") %>>
                             <a href="/docs/providers/aws/d/vpn_gateway.html">aws_vpn_gateway</a>
                         </li>
+                        <li<%= sidebar_current("docs-aws-datasource-workspaces-bundle") %>>
+                            <a href="/docs/providers/aws/d/workspaces_bundle.html">aws_workspaces_bundle</a>
+                        </li>
                     </ul>
                 </li>
 

website/docs/d/workspaces_bundle.html.markdown

@@ -0,0 +1,48 @@
+---
+layout: "aws"
+page_title: "AWS: aws_workspaces_bundle"
+sidebar_current: "docs-aws-datasource-workspaces-bundle"
+description: |-  
+  Get information on a Workspaces Bundle.
+---
+
+# aws_workspaces_bundle
+
+Use this data source to get information about a Workspaces Bundle.
+
+## Example Usage
+
+```hcl
+data "aws_workspaces_bundle" "example" {
+  bundle_id = "wsb-b0s22j3d7"
+}
+```
+
+## Argument Reference
+
+The following arguments are supported:
+
+* `bundle_id` – (Required, ForceNew) The ID of the bundle.
+
+## Attributes Reference
+
+The following attributes are exported:
+
+* `description` – The description of the bundle.
+* `name` – The name of the bundle.
+* `owner` – The owner of the bundle.
+* `compute_type` – The compute type. See supported fields below.
+* `root_storage` – The root volume. See supported fields below.
+* `user_storage` – The user storage. See supported fields below.
+
+### `compute_type`
+
+* `name` - The name of the compute type.
+
+### `root_storage`
+
+* `capacity` - The size of the root volume.
+
+### `user_storage`
+
+* `capacity` - The size of the user storage.