adds "active" attribute

Author: benzaita

Makefile

@@ -3,7 +3,7 @@ HOSTNAME=registry.terraform.io
 NAMESPACE=benzaita
 NAME=chaossearch
 BINARY=terraform-provider-${NAME}
-VERSION=0.4.0
+VERSION=0.5.0
 OS_ARCH=linux_amd64
 
 default: install

chaossearch/client/model.go

@@ -25,6 +25,7 @@ type ReadObjectGroupResponse struct {
 	PartitionBy      string
 	SourceBucket     string
 	IndexRetention   int
+	Active           bool
 }
 
 type CreateObjectGroupRequest struct {
@@ -39,6 +40,11 @@ type CreateObjectGroupRequest struct {
 	IndexRetention   int
 }
 
+type SetActiveRequest struct {
+	ObjectGroupName string
+	Active          bool
+}
+
 type DeleteObjectGroupRequest struct {
 	Name string
 }

chaossearch/client/operation_read_object_group.go

@@ -64,6 +64,10 @@ func mapBucketTaggingToResponse(tagging *s3.GetBucketTaggingOutput, v *ReadObjec
 		return err
 	}
 
+	if err := readJSONTagValue(tagging, "cs3.modeled", &v.Active); err != nil {
+		return err
+	}
+
 	var filterObject struct {
 		Type string `json:"_type"`
 	}

chaossearch/client/operation_set_active.go

@@ -0,0 +1,54 @@
+package client
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/http"
+)
+
+func (client *Client) SetActive(ctx context.Context, req *SetActiveRequest) error {
+	method := "POST"
+	url := fmt.Sprintf("%s/Bucket/model", client.config.URL)
+
+	bodyAsBytes, err := marshalSetActiveRequest(req)
+	if err != nil {
+		return err
+	}
+
+	httpReq, err := http.NewRequestWithContext(ctx, method, url, bytes.NewReader(bodyAsBytes))
+	if err != nil {
+		return fmt.Errorf("Failed to create request: %s", err)
+	}
+
+	httpResp, err := client.signAndDo(httpReq, bodyAsBytes)
+	if err != nil {
+		return fmt.Errorf("Failed to %s to %s: %s", method, url, err)
+	}
+	defer httpResp.Body.Close()
+
+	return nil
+}
+
+func marshalSetActiveRequest(req *SetActiveRequest) ([]byte, error) {
+	var modelMode int
+
+	if req.Active {
+		modelMode = 0
+	} else {
+		modelMode = -1
+	}
+
+	body := map[string]interface{}{
+		"BucketName": req.ObjectGroupName,
+		"ModelMode":  modelMode,
+	}
+
+	bodyAsBytes, err := json.Marshal(body)
+	if err != nil {
+		return nil, err
+	}
+
+	return bodyAsBytes, nil
+}

chaossearch/resource_object_group.go

@@ -79,6 +79,12 @@ func resourceObjectGroup() *schema.Resource {
 				Optional:    true,
 				ForceNew:    false,
 			},
+			"active": {
+				Type:        schema.TypeBool,
+				Description: "Whether the live indexing should be running or not",
+				Required:    true,
+				ForceNew:    false,
+			},
 
 			// Workaround. Otherwise Terraform fails with "All fields are ForceNew or Computed w/out Optional, Update is superfluous"
 			"description": {
@@ -93,7 +99,7 @@ func resourceObjectGroup() *schema.Resource {
 func resourceObjectGroupCreate(ctx context.Context, data *schema.ResourceData, meta interface{}) diag.Diagnostics {
 	c := meta.(*ProviderMeta).Client
 
-	request := &client.CreateObjectGroupRequest{
+	createObjectGroupRequest := &client.CreateObjectGroupRequest{
 		Name:             data.Get("name").(string),
 		SourceBucket:     data.Get("source_bucket").(string),
 		FilterJSON:       data.Get("filter_json").(string),
@@ -105,8 +111,16 @@ func resourceObjectGroupCreate(ctx context.Context, data *schema.ResourceData, m
 		IndexRetention:   data.Get("index_retention").(int),
 	}
 
-	err := c.CreateObjectGroup(ctx, request)
-	if err != nil {
+	if err := c.CreateObjectGroup(ctx, createObjectGroupRequest); err != nil {
+		return diag.FromErr(err)
+	}
+
+	setActiveRequest := &client.SetActiveRequest{
+		ObjectGroupName: data.Get("name").(string),
+		Active:          data.Get("active").(bool),
+	}
+
+	if err := c.SetActive(ctx, setActiveRequest); err != nil {
 		return diag.FromErr(err)
 	}
 
@@ -133,6 +147,7 @@ func resourceObjectGroupRead(ctx context.Context, data *schema.ResourceData, met
 	data.Set("format", resp.Format)
 	data.Set("live_events_sqs_arn", resp.LiveEventsSqsArn)
 	data.Set("index_retention", resp.IndexRetention)
+	data.Set("active", resp.Active)
 
 	// When the object in an Object Group use no compression, you need to create it with
 	// `compression = ""`. However, when querying an Object Group whose object are not
@@ -155,13 +170,20 @@ func resourceObjectGroupRead(ctx context.Context, data *schema.ResourceData, met
 func resourceObjectGroupUpdate(ctx context.Context, data *schema.ResourceData, meta interface{}) diag.Diagnostics {
 	c := meta.(*ProviderMeta).Client
 
-	req := &client.UpdateObjectGroupRequest{
+	setActiveRequest := &client.SetActiveRequest{
+		ObjectGroupName: data.Get("name").(string),
+		Active:          data.Get("active").(bool),
+	}
+	if err := c.SetActive(ctx, setActiveRequest); err != nil {
+		return diag.FromErr(err)
+	}
+
+	updateObjectGroupRequest := &client.UpdateObjectGroupRequest{
 		Name:           data.Get("name").(string),
 		IndexRetention: data.Get("index_retention").(int),
 	}
 
-	err := c.UpdateObjectGroup(ctx, req)
-	if err != nil {
+	if err := c.UpdateObjectGroup(ctx, updateObjectGroupRequest); err != nil {
 		return diag.FromErr(err)
 	}
 
@@ -171,12 +193,19 @@ func resourceObjectGroupUpdate(ctx context.Context, data *schema.ResourceData, m
 func resourceObjectGroupDelete(ctx context.Context, data *schema.ResourceData, meta interface{}) diag.Diagnostics {
 	c := meta.(*ProviderMeta).Client
 
-	req := &client.DeleteObjectGroupRequest{
+	stopIndexingRequest := &client.SetActiveRequest{
+		ObjectGroupName: data.Get("name").(string),
+		Active:          false,
+	}
+	if err := c.SetActive(ctx, stopIndexingRequest); err != nil {
+		return diag.FromErr(err)
+	}
+
+	deleteObjectGroupRequest := &client.DeleteObjectGroupRequest{
 		Name: data.Get("name").(string),
 	}
 
-	err := c.DeleteObjectGroup(ctx, req)
-	if err != nil {
+	if err := c.DeleteObjectGroup(ctx, deleteObjectGroupRequest); err != nil {
 		return diag.FromErr(err)
 	}