Add resource_team.go

Author: palson-cf

codefresh/resource_team.go

@@ -0,0 +1,190 @@
+package codefresh
+
+import (
+	"fmt"
+	cfClient "github.com/codefresh-io/terraform-provider-codefresh/client"
+	"github.com/hashicorp/terraform-plugin-sdk/helper/schema"
+)
+
+func resourceTeam() *schema.Resource {
+	return &schema.Resource{
+		Create: resourceTeamCreate,
+		Read:   resourceTeamRead,
+		Update: resourceTeamUpdate,
+		Delete: resourceTeamDelete,
+		Importer: &schema.ResourceImporter{
+			State: schema.ImportStatePassthrough,
+		},
+		Schema: map[string]*schema.Schema{
+			"name": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"type": {
+				Type:     schema.TypeString,
+				ForceNew: true,
+				Optional: true,
+			},
+			"account": {
+				Type:     schema.TypeString,
+				Optional: true,
+				Computed: true,
+			},
+			"tags": {
+				Type:     schema.TypeSet,
+				Optional: true,
+				ForceNew: true,
+				Elem: &schema.Schema{
+					Type: schema.TypeString,
+				},
+			},
+			"users": {
+				Type:     schema.TypeList,
+				Optional: true,
+				ForceNew: true,
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"user_id": {
+							Type:     schema.TypeString,
+							Required: true,
+						},
+						"name": {
+							Type:     schema.TypeString,
+							Optional: true,
+						},
+					},
+				},
+			},
+		},
+	}
+}
+
+func resourceTeamCreate(d *schema.ResourceData, meta interface{}) error {
+	client := meta.(*cfClient.Client)
+
+	team := *mapResourceToTeam(d)
+
+	resp, err := client.CreateTeam(&team)
+	if err != nil {
+		return err
+	}
+
+	d.SetId(resp.ID)
+
+	return nil
+}
+
+func resourceTeamRead(d *schema.ResourceData, meta interface{}) error {
+
+	client := meta.(*cfClient.Client)
+
+	teamID := d.Id()
+	if teamID == "" {
+		d.SetId("")
+		return nil
+	}
+
+	team, err := client.GetTeamByID(teamID)
+	if err != nil {
+		return err
+	}
+
+	err = mapTeamToResource(team, d)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func resourceTeamUpdate(d *schema.ResourceData, meta interface{}) error {
+	client := meta.(*cfClient.Client)
+
+	team := *mapResourceToTeam(d)
+
+	_, err := client.CreateTeam(&team)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func resourceTeamDelete(d *schema.ResourceData, meta interface{}) error {
+	client := meta.(*cfClient.Client)
+
+	err := client.DeleteTeam(d.Id())
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func mapTeamToResource(team *cfClient.Team, d *schema.ResourceData) error {
+
+	err := d.Set("name", team.Name)
+	if err != nil {
+		return err
+	}
+
+	err = d.Set("type", team.Type)
+	if err != nil {
+		return err
+	}
+
+	err = d.Set("account", team.Account)
+	if err != nil {
+		return err
+	}
+
+	err = d.Set("tags", team.Tags)
+	if err != nil {
+		return err
+	}
+
+	err = d.Set("users", flattenTeamUsers(team.Users))
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func flattenTeamUsers(users []cfClient.TeamUser) []map[string]interface{} {
+	var res = make([]map[string]interface{}, len(users))
+	for i, user := range users {
+		m := make(map[string]interface{})
+		m["name"] = user.UserName
+		m["user_id"] = user.ID
+
+		res[i] = m
+	}
+
+	return res
+}
+
+func mapResourceToTeam(d *schema.ResourceData) *cfClient.Team {
+	tags := d.Get("tags").(*schema.Set).List()
+	team := &cfClient.Team{
+		ID:      d.Id(),
+		Name:    d.Get("name").(string),
+		Type:    d.Get("type").(string),
+		Account: d.Get("account").(string),
+		Tags:    convertStringArr(tags),
+	}
+
+	if _, ok := d.GetOk("users"); ok {
+		users := d.Get("users").([]interface{})
+		for idx := range users {
+			user := cfClient.TeamUser{
+				ID:       d.Get(fmt.Sprintf("users.%v.user_id", idx)).(string),
+				UserName: d.Get(fmt.Sprintf("users.%v.name", idx)).(string),
+			}
+
+			team.Users = append(team.Users, user)
+		}
+	}
+
+	return team
+}