Added bitbucketserver_banner resource and bitbucketserver_cluster data source

bitbucket/data_cluster.go

@@ -0,0 +1,121 @@
+package bitbucket
+
+import (
+	"encoding/json"
+	"github.com/hashicorp/terraform/helper/schema"
+	"io/ioutil"
+)
+
+type ClusterNode struct {
+	ID      string `json:"id,omitempty"`
+	Name    string `json:"name,omitempty"`
+	Address struct {
+		Hostname string `json:"hostName,omitempty"`
+		Port     int    `json:"port,omitempty"`
+	} `json:"address,omitempty"`
+	Local bool `json:"local,omitempty"`
+}
+
+type Cluster struct {
+	LocalNode ClusterNode   `json:"localNode,omitempty"`
+	Nodes     []ClusterNode `json:"nodes,omitempty"`
+	Running   bool          `json:"running,omitempty"`
+}
+
+func dataSourceCluster() *schema.Resource {
+	return &schema.Resource{
+		Read: dataSourceClusterRead,
+
+		Schema: map[string]*schema.Schema{
+			"local_node": {
+				Type:     schema.TypeList,
+				Computed: true,
+				Elem:     clusterNodeResourceSchema(),
+			},
+			"nodes": {
+				Type:     schema.TypeList,
+				Computed: true,
+				Elem:     clusterNodeResourceSchema(),
+			},
+			"running": {
+				Type:     schema.TypeBool,
+				Computed: true,
+			},
+		},
+	}
+}
+
+func clusterNodeResourceSchema() *schema.Resource {
+	return &schema.Resource{
+		Schema: map[string]*schema.Schema{
+			"id": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"name": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"hostname": {
+				Type:     schema.TypeString,
+				Computed: true,
+			},
+			"port": {
+				Type:     schema.TypeInt,
+				Computed: true,
+			},
+			"local": {
+				Type:     schema.TypeBool,
+				Computed: true,
+			},
+		},
+	}
+}
+
+func dataSourceClusterRead(d *schema.ResourceData, m interface{}) error {
+	client := m.(*BitbucketServerProvider).BitbucketClient
+	req, err := client.Get("/rest/api/1.0/admin/cluster")
+
+	if err != nil {
+		return err
+	}
+
+	var cluster Cluster
+
+	body, readErr := ioutil.ReadAll(req.Body)
+	if readErr != nil {
+		return readErr
+	}
+
+	decodeErr := json.Unmarshal(body, &cluster)
+	if decodeErr != nil {
+		return decodeErr
+	}
+
+	d.SetId("cluster")
+	_ = d.Set("running", cluster.Running)
+
+	var nodes []interface{}
+	for _, node := range cluster.Nodes {
+		n := make(map[string]interface{})
+		n["id"] = node.ID
+		n["name"] = node.Name
+		n["hostname"] = node.Address.Hostname
+		n["port"] = node.Address.Port
+		n["local"] = node.Local
+		nodes = append(nodes, n)
+	}
+	_ = d.Set("nodes", nodes)
+
+	var localNode []interface{}
+	n := make(map[string]interface{})
+	n["id"] = cluster.LocalNode.ID
+	n["name"] = cluster.LocalNode.Name
+	n["hostname"] = cluster.LocalNode.Address.Hostname
+	n["port"] = cluster.LocalNode.Address.Port
+	n["local"] = cluster.LocalNode.Local
+	localNode = append(localNode, n)
+	_ = d.Set("local_node", localNode)
+
+	return nil
+}

bitbucket/data_cluster_test.go

@@ -0,0 +1,36 @@
+package bitbucket
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform/helper/resource"
+)
+
+func TestAccBitbucketDataCluster(t *testing.T) {
+	config := `
+		data "bitbucketserver_cluster" "main" {}
+	`
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: config,
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttr("data.bitbucketserver_cluster.main", "running", "true"),
+					resource.TestCheckResourceAttr("data.bitbucketserver_cluster.main", "local_node.#", "1"),
+					resource.TestCheckResourceAttrSet("data.bitbucketserver_cluster.main", "local_node.0.id"),
+					resource.TestCheckResourceAttrSet("data.bitbucketserver_cluster.main", "local_node.0.hostname"),
+					resource.TestCheckResourceAttrSet("data.bitbucketserver_cluster.main", "local_node.0.port"),
+					resource.TestCheckResourceAttr("data.bitbucketserver_cluster.main", "local_node.0.local", "true"),
+					resource.TestCheckResourceAttr("data.bitbucketserver_cluster.main", "nodes.#", "1"),
+					resource.TestCheckResourceAttrSet("data.bitbucketserver_cluster.main", "nodes.0.id"),
+					resource.TestCheckResourceAttrSet("data.bitbucketserver_cluster.main", "nodes.0.hostname"),
+					resource.TestCheckResourceAttrSet("data.bitbucketserver_cluster.main", "nodes.0.port"),
+					resource.TestCheckResourceAttr("data.bitbucketserver_cluster.main", "nodes.0.local", "true"),
+				),
+			},
+		},
+	})
+}

bitbucket/provider.go

@@ -31,6 +31,7 @@ func Provider() terraform.ResourceProvider {
 		ConfigureFunc: providerConfigure,
 		DataSourcesMap: map[string]*schema.Resource{
 			"bitbucketserver_application_properties":        dataSourceApplicationProperties(),
+			"bitbucketserver_cluster":                       dataSourceCluster(),
 			"bitbucketserver_global_permissions_groups":     dataSourceGlobalPermissionsGroups(),
 			"bitbucketserver_global_permissions_users":      dataSourceGlobalPermissionsUsers(),
 			"bitbucketserver_groups":                        dataSourceGroups(),
@@ -42,6 +43,7 @@ func Provider() terraform.ResourceProvider {
 			"bitbucketserver_repository_permissions_users":  dataSourceRepositoryPermissionsUsers(),
 		},
 		ResourcesMap: map[string]*schema.Resource{
+			"bitbucketserver_banner":                       resourceBanner(),
 			"bitbucketserver_global_permissions_group":     resourceGlobalPermissionsGroup(),
 			"bitbucketserver_global_permissions_user":      resourceGlobalPermissionsUser(),
 			"bitbucketserver_group":                        resourceGroup(),

bitbucket/resource_banner.go

@@ -0,0 +1,128 @@
+package bitbucket
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"github.com/hashicorp/terraform/helper/schema"
+	"github.com/hashicorp/terraform/helper/validation"
+	"io/ioutil"
+)
+
+type Banner struct {
+	Message  string `json:"message,omitempty"`
+	Audience string `json:"audience,omitempty"`
+	Enabled  bool   `json:"enabled,omitempty"`
+}
+
+func resourceBanner() *schema.Resource {
+	return &schema.Resource{
+		Create: resourceBannerCreate,
+		Update: resourceBannerUpdate,
+		Read:   resourceBannerRead,
+		Exists: resourceBannerExists,
+		Delete: resourceBannerDelete,
+		Importer: &schema.ResourceImporter{
+			State: schema.ImportStatePassthrough,
+		},
+
+		Schema: map[string]*schema.Schema{
+			"message": {
+				Type:     schema.TypeString,
+				Required: true,
+			},
+			"audience": {
+				Type:         schema.TypeString,
+				Optional:     true,
+				Default:      "ALL",
+				ValidateFunc: validation.StringInSlice([]string{"ALL", "AUTHENTICATED"}, false),
+			},
+			"enabled": {
+				Type:     schema.TypeBool,
+				Optional: true,
+				Default:  true,
+			},
+		},
+	}
+}
+
+func newBannerFromResource(d *schema.ResourceData) *Banner {
+	banner := &Banner{
+		Message:  d.Get("message").(string),
+		Audience: d.Get("audience").(string),
+		Enabled:  d.Get("enabled").(bool),
+	}
+
+	return banner
+}
+
+func resourceBannerUpdate(d *schema.ResourceData, m interface{}) error {
+	client := m.(*BitbucketServerProvider).BitbucketClient
+	banner := newBannerFromResource(d)
+
+	bytedata, err := json.Marshal(banner)
+
+	if err != nil {
+		return err
+	}
+
+	_, err = client.Put("/rest/api/1.0/admin/banner", bytes.NewBuffer(bytedata))
+	if err != nil {
+		return err
+	}
+
+	d.SetId("banner")
+	return resourceBannerRead(d, m)
+}
+
+func resourceBannerCreate(d *schema.ResourceData, m interface{}) error {
+	return resourceBannerUpdate(d, m)
+}
+
+func resourceBannerRead(d *schema.ResourceData, m interface{}) error {
+
+	client := m.(*BitbucketServerProvider).BitbucketClient
+	req, err := client.Get("/rest/api/1.0/admin/banner")
+
+	if err != nil {
+		return err
+	}
+
+	var banner Banner
+
+	body, readErr := ioutil.ReadAll(req.Body)
+	if readErr != nil {
+		return readErr
+	}
+
+	decodeErr := json.Unmarshal(body, &banner)
+	if decodeErr != nil {
+		return decodeErr
+	}
+
+	_ = d.Set("message", banner.Message)
+	_ = d.Set("audience", banner.Audience)
+	_ = d.Set("enabled", banner.Enabled)
+
+	return nil
+}
+
+func resourceBannerExists(d *schema.ResourceData, m interface{}) (bool, error) {
+	client := m.(*BitbucketServerProvider).BitbucketClient
+	repoReq, err := client.Get("/rest/api/1.0/admin/banner")
+	if err != nil {
+		return false, fmt.Errorf("failed to get banner from bitbucket: %+v", err)
+	}
+
+	if repoReq.StatusCode == 200 {
+		return true, nil
+	} else {
+		return false, nil
+	}
+}
+
+func resourceBannerDelete(d *schema.ResourceData, m interface{}) error {
+	client := m.(*BitbucketServerProvider).BitbucketClient
+	_, err := client.Delete("/rest/api/1.0/admin/banner")
+	return err
+}

bitbucket/resource_banner_test.go

@@ -0,0 +1,53 @@
+package bitbucket
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform/helper/resource"
+)
+
+func TestAccBitbucketBanner_basic(t *testing.T) {
+	testAccBitbucketBannerConfig := `
+		resource "bitbucketserver_banner" "test" {
+			message = "Test Banner\n*bold*"
+		}
+	`
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccBitbucketBannerConfig,
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttr("bitbucketserver_banner.test", "message", "Test Banner\n*bold*"),
+					resource.TestCheckResourceAttr("bitbucketserver_banner.test", "enabled", "true"),
+					resource.TestCheckResourceAttr("bitbucketserver_banner.test", "audience", "ALL"),
+				),
+			},
+		},
+	})
+}
+
+func TestAccBitbucketBanner_authenticated(t *testing.T) {
+	testAccBitbucketBannerConfig := `
+		resource "bitbucketserver_banner" "test" {
+			message  = "Test Banner\n*bold*"
+			audience = "AUTHENTICATED"
+		}
+	`
+
+	resource.Test(t, resource.TestCase{
+		PreCheck:  func() { testAccPreCheck(t) },
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccBitbucketBannerConfig,
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttr("bitbucketserver_banner.test", "message", "Test Banner\n*bold*"),
+					resource.TestCheckResourceAttr("bitbucketserver_banner.test", "audience", "AUTHENTICATED"),
+				),
+			},
+		},
+	})
+}