Merge pull request #396 from autonomic-ai/data-github-membership

Add a github_membership data source.

github/data_source_github_membership.go

@@ -0,0 +1,53 @@
+package github
+
+import (
+	"context"
+	"log"
+
+	"github.com/hashicorp/terraform-plugin-sdk/helper/schema"
+)
+
+func dataSourceGithubMembership() *schema.Resource {
+	return &schema.Resource{
+		Read: dataSourceGithubMembershipRead,
+
+		Schema: map[string]*schema.Schema{
+			"username": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"role": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"etag": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+		},
+	}
+}
+
+func dataSourceGithubMembershipRead(d *schema.ResourceData, meta interface{}) error {
+	username := d.Get("username").(string)
+	log.Printf("[INFO] Refreshing GitHub membership: %s", username)
+
+	client := meta.(*Organization).client
+	orgName := meta.(*Organization).name
+
+	ctx := context.Background()
+
+	membership, resp, err := client.Organizations.GetOrgMembership(ctx,
+		username, orgName)
+
+	if err != nil {
+		return err
+	}
+
+	d.SetId(buildTwoPartID(membership.Organization.Login, membership.User.Login))
+
+	d.Set("username", membership.User.Login)
+	d.Set("role", membership.Role)
+	d.Set("etag", resp.Header.Get("ETag"))
+	return nil
+}

github/data_source_github_membership_test.go

@@ -0,0 +1,55 @@
+package github
+
+import (
+	"fmt"
+	"regexp"
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-sdk/helper/resource"
+)
+
+func TestAccGithubMembershipDataSource_noMatchReturnsError(t *testing.T) {
+	username := "admin"
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck: func() {
+			testAccPreCheck(t)
+		},
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config:      testAccCheckGithubMembershipDatasourceConfig(username),
+				ExpectError: regexp.MustCompile(`Not Found`),
+			},
+		},
+	})
+}
+
+func TestAccGithubMembershipDataSource_existing(t *testing.T) {
+	if testUser == "" {
+		t.Skip("This test requires you to set the test user (set it by exporting GITHUB_TEST_USER)")
+	}
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck: func() {
+			testAccPreCheck(t)
+		},
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccCheckGithubMembershipDatasourceConfig(testUser),
+				Check: resource.ComposeAggregateTestCheckFunc(
+					resource.TestCheckResourceAttr("data.github_membership.test", "username", testUser),
+					resource.TestCheckResourceAttrSet("data.github_membership.test", "role"),
+					resource.TestCheckResourceAttrSet("data.github_membership.test", "etag"),
+				),
+			},
+		},
+	})
+}
+
+func testAccCheckGithubMembershipDatasourceConfig(username string) string {
+	return fmt.Sprintf(`
+data "github_membership" "test" {
+  username = "%s"
+}
+`, username)
+}

github/provider.go

@@ -71,6 +71,7 @@ func Provider() terraform.ResourceProvider {
 		DataSourcesMap: map[string]*schema.Resource{
 			"github_collaborators":      dataSourceGithubCollaborators(),
 			"github_ip_ranges":          dataSourceGithubIpRanges(),
+			"github_membership":         dataSourceGithubMembership(),
 			"github_release":            dataSourceGithubRelease(),
 			"github_repositories":       dataSourceGithubRepositories(),
 			"github_repository":         dataSourceGithubRepository(),

website/docs/d/membership.html.markdown

@@ -0,0 +1,31 @@
+---
+layout: "github"
+page_title: "GitHub: github_membership"
+description: |-
+  Get information on user membership in an organization.
+-------------------------------------------
+
+# github_membership
+
+Use this data source to find out if a user is a member of your organization, as well
+as what role they have within it.
+If the user's membership in the organization is pending their acceptance of an invite,
+the role they would have once they accept will be returned.
+
+## Example Usage
+
+```hcl
+data "github_membership" "membership_for_some_user" {
+    username = "SomeUser"
+}
+```
+
+## Argument Reference
+
+ * `username` - (Required) The username to lookup in the organization.
+
+## Attributes Reference
+
+ * `username` - The username.
+ * `role` - `admin` or `member` -- the role the user has within the organization.
+ * `etag` - An etag representing the membership object.

website/github.erb

@@ -22,6 +22,9 @@
             <li>
               <a href="/docs/providers/github/d/ip_ranges.html">github_ip_ranges</a>
             </li>
+            <li>
+            <a href="/docs/providers/github/d/membership.html">github_membership</a>
+            </li>
             <li>
               <a href="/docs/providers/github/d/release.html">github_release</a>
             </li>