Merge pull request #43 from VertivSRE/project-membership

Project membership resource

Author: roidelapluie

gitlab/data_source_gitlab_project.go

@@ -0,0 +1,51 @@
+package gitlab
+
+import (
+	"fmt"
+	"log"
+
+	"github.com/hashicorp/terraform/helper/schema"
+	"github.com/xanzy/go-gitlab"
+)
+
+func dataSourceGitlabProject() *schema.Resource {
+	return &schema.Resource{
+		Read: dataSourceGitlabProjectRead,
+
+		Schema: map[string]*schema.Schema{
+			"id": {
+				Type:     schema.TypeInt,
+				Required: true,
+			},
+		},
+	}
+}
+
+func dataSourceGitlabProjectRead(d *schema.ResourceData, meta interface{}) error {
+	client := meta.(*gitlab.Client)
+
+	log.Printf("[INFO] Reading Gitlab project")
+
+	v, _ := d.GetOk("id")
+
+	found, _, err := client.Projects.GetProject(v)
+	if err != nil {
+		return err
+	}
+
+	d.SetId(fmt.Sprintf("%d", found.ID))
+	d.Set("name", found.Name)
+	d.Set("path", found.Path)
+	d.Set("description", found.Description)
+	d.Set("default_branch", found.DefaultBranch)
+	d.Set("issues_enabled", found.IssuesEnabled)
+	d.Set("merge_requests_enabled", found.MergeRequestsEnabled)
+	d.Set("wiki_enabled", found.WikiEnabled)
+	d.Set("snippets_enabled", found.SnippetsEnabled)
+	d.Set("visibility_level", string(found.Visibility))
+	d.Set("namespace_id", found.Namespace.ID)
+	d.Set("ssh_url_to_repo", found.SSHURLToRepo)
+	d.Set("http_url_to_repo", found.HTTPURLToRepo)
+	d.Set("web_url", found.WebURL)
+	return nil
+}

gitlab/data_source_gitlab_project_test.go

@@ -0,0 +1,66 @@
+package gitlab
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/hashicorp/terraform/helper/acctest"
+	"github.com/hashicorp/terraform/helper/resource"
+	"github.com/hashicorp/terraform/terraform"
+)
+
+func TestAccDataGitlabProject_basic(t *testing.T) {
+	projectname := fmt.Sprintf("tf-%s", acctest.RandString(5))
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDataGitlabProjectConfig(projectname),
+				Check: resource.ComposeTestCheckFunc(
+					testAccDataSourceGitlabProject("gitlab_project.test", "data.gitlab_project.foo"),
+				),
+			},
+		},
+	})
+}
+
+func testAccDataSourceGitlabProject(src, n string) resource.TestCheckFunc {
+	return func(s *terraform.State) error {
+
+		project := s.RootModule().Resources[src]
+		projectResource := project.Primary.Attributes
+
+		search := s.RootModule().Resources[n]
+		searchResource := search.Primary.Attributes
+
+		if searchResource["id"] == "" {
+			return fmt.Errorf("Expected to get a project ID from Gitlab")
+		}
+
+		testAttributes := []string{"id", "Name", "Path", "Visibility", "Description"}
+
+		for _, attribute := range testAttributes {
+			if searchResource[attribute] != projectResource[attribute] {
+				return fmt.Errorf("Expected the project %s to be: %s, but got: %s", attribute, projectResource[attribute], searchResource[attribute])
+			}
+		}
+		return nil
+	}
+}
+
+func testAccDataGitlabProjectConfig(projectname string) string {
+	return fmt.Sprintf(`
+resource "gitlab_project" "test"{
+	name = "%s"
+	path = "%s"
+	description = "Terraform acceptance tests"
+	visibility_level = "public"
+}
+
+data "gitlab_project" "foo" {
+	id = "${gitlab_project.test.id}"
+}
+	`, projectname, projectname)
+}

gitlab/data_source_gitlab_user.go

@@ -0,0 +1,54 @@
+package gitlab
+
+import (
+	"fmt"
+	"log"
+	"strings"
+
+	"github.com/hashicorp/terraform/helper/schema"
+	gitlab "github.com/xanzy/go-gitlab"
+)
+
+func dataSourceGitlabUser() *schema.Resource {
+	return &schema.Resource{
+		Read: dataSourceGitlabUserRead,
+		Schema: map[string]*schema.Schema{
+			"email": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+		},
+	}
+}
+
+func dataSourceGitlabUserRead(d *schema.ResourceData, meta interface{}) error {
+	client := meta.(*gitlab.Client)
+
+	log.Printf("[INFO] Reading Gitlab user")
+
+	searchEmail := strings.ToLower(d.Get("email").(string))
+	query := &gitlab.ListUsersOptions{
+		Search: &searchEmail,
+	}
+	users, _, err := client.Users.ListUsers(query)
+	if err != nil {
+		return err
+	}
+
+	var found *gitlab.User
+
+	for _, user := range users {
+		if strings.ToLower(user.Email) == searchEmail {
+			found = user
+			break
+		}
+	}
+	if found == nil {
+		return fmt.Errorf("The email '%s' does not match any user email", searchEmail)
+	}
+	d.SetId(fmt.Sprintf("%d", found.ID))
+	d.Set("name", found.Name)
+	d.Set("username", found.Username)
+	d.Set("email", found.Email)
+	return nil
+}

gitlab/data_source_gitlab_user_test.go

@@ -0,0 +1,66 @@
+package gitlab
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/hashicorp/terraform/helper/acctest"
+	"github.com/hashicorp/terraform/helper/resource"
+	"github.com/hashicorp/terraform/terraform"
+)
+
+func TestAccDataGitlabUser_basic(t *testing.T) {
+	userEmail := fmt.Sprintf("tf-%s", acctest.RandString(5))
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccDataGitlabUserConfig(userEmail),
+				Check: resource.ComposeTestCheckFunc(
+					testAccDataSourceGitlabUser("gitlab_user.foo", "data.gitlab_user.foo"),
+				),
+			},
+		},
+	})
+}
+
+func testAccDataSourceGitlabUser(src, n string) resource.TestCheckFunc {
+	return func(s *terraform.State) error {
+
+		user := s.RootModule().Resources[src]
+		userResource := user.Primary.Attributes
+
+		search := s.RootModule().Resources[n]
+		searchResource := search.Primary.Attributes
+
+		if searchResource["email"] == "" {
+			return fmt.Errorf("Expected to get user email from Gitlab")
+		}
+
+		testAttributes := []string{"email"}
+
+		for _, attribute := range testAttributes {
+			if searchResource[attribute] != userResource[attribute] {
+				return fmt.Errorf("Expected the user %s to be: %s, but got: %s", attribute, userResource[attribute], searchResource[attribute])
+			}
+		}
+		return nil
+	}
+}
+
+func testAccDataGitlabUserConfig(userEmail string) string {
+	return fmt.Sprintf(`
+resource "gitlab_user" "foo" {
+  name             = "foo %s"
+  username         = "listest%s"
+  password         = "test%stt"
+  email            = "listest%[email protected]"
+}
+
+data "gitlab_user" "foo" {
+	email = "${gitlab_user.foo.email}"
+}
+	`, userEmail, userEmail, userEmail, userEmail)
+}

gitlab/provider.go

@@ -40,13 +40,20 @@ func Provider() terraform.ResourceProvider {
 				Description: descriptions["insecure"],
 			},
 		},
+
+		DataSourcesMap: map[string]*schema.Resource{
+			"gitlab_project": dataSourceGitlabProject(),
+			"gitlab_user":    dataSourceGitlabUser(),
+		},
+
 		ResourcesMap: map[string]*schema.Resource{
-			"gitlab_group":        resourceGitlabGroup(),
-			"gitlab_project":      resourceGitlabProject(),
-			"gitlab_label":        resourceGitlabLabel(),
-			"gitlab_project_hook": resourceGitlabProjectHook(),
-			"gitlab_deploy_key":   resourceGitlabDeployKey(),
-			"gitlab_user":         resourceGitlabUser(),
+			"gitlab_group":              resourceGitlabGroup(),
+			"gitlab_project":            resourceGitlabProject(),
+			"gitlab_label":              resourceGitlabLabel(),
+			"gitlab_project_hook":       resourceGitlabProjectHook(),
+			"gitlab_deploy_key":         resourceGitlabDeployKey(),
+			"gitlab_user":               resourceGitlabUser(),
+			"gitlab_project_membership": resourceGitlabProjectMembership(),
 		},
 
 		ConfigureFunc: providerConfigure,