Merge pull request #18 from bugout-dev/resource-holders

Support of resource holder permissions

Author: kompotkot

cmd/bugout/brood/resources.go

@@ -8,6 +8,7 @@ import (
 
 	"github.com/bugout-dev/bugout-go/cmd/bugout/cmdutils"
 	bugout "github.com/bugout-dev/bugout-go/pkg"
+	"github.com/bugout-dev/bugout-go/pkg/brood"
 )
 
 func GenerateResourcesCommand() *cobra.Command {
@@ -21,11 +22,28 @@ func GenerateResourcesCommand() *cobra.Command {
 	resourcesDeleteCmd := GenerateResourceDeleteCommand()
 	resourcesGetCmd := GenerateResourcesGetCommand()
 
-	resourcesCmd.AddCommand(resourcesCreateCmd, resourcesUpdateCmd, resourcesDeleteCmd, resourcesGetCmd)
+	resourceHoldersCmd := GenerateResourceHoldersCommand()
+
+	resourcesCmd.AddCommand(resourcesCreateCmd, resourcesUpdateCmd, resourcesDeleteCmd, resourcesGetCmd, resourceHoldersCmd)
 
 	return resourcesCmd
 }
 
+func GenerateResourceHoldersCommand() *cobra.Command {
+	holdersCmd := &cobra.Command{
+		Use:   "holders",
+		Short: "Bugout resource holders operations",
+	}
+
+	resourceHoldersGetCmd := GenerateResourceHoldersGetCommand()
+	resourceHoldersAddCmd := GenerateResourceHoldersAddCommand()
+	resourceHoldersDeleteCmd := GenerateResourceHoldersDeleteCommand()
+
+	holdersCmd.AddCommand(resourceHoldersGetCmd, resourceHoldersAddCmd, resourceHoldersDeleteCmd)
+
+	return holdersCmd
+}
+
 func GenerateResourceCreateCommand() *cobra.Command {
 	var token, applicationId string
 	resourceCreateCmd := &cobra.Command{
@@ -76,7 +94,7 @@ func GenerateResourceUpdateCommand() *cobra.Command {
 		PreRunE: cmdutils.TokenArgPopulator,
 		RunE: func(cmd *cobra.Command, args []string) error {
 			if len(args) != 1 {
-				return fmt.Errorf(`One resource argument as json string in format {"update": {"key": "value"}, drop_keys: ["key"]} must be specified`)
+				return fmt.Errorf(`Resource argument as json string in format {"update": {"key": "value"}, "drop_keys": ["key"]} must be specified`)
 			}
 			var resourceUpdateRaw resourceUpdateArg
 			err := json.Unmarshal([]byte(args[0]), &resourceUpdateRaw)
@@ -162,3 +180,105 @@ func GenerateResourcesGetCommand() *cobra.Command {
 
 	return resourcesGetCmd
 }
+
+func GenerateResourceHoldersGetCommand() *cobra.Command {
+	var token, resourceId string
+	resourceHoldersGetCmd := &cobra.Command{
+		Use:     "get",
+		Short:   "Get resource holders",
+		PreRunE: cmdutils.TokenArgPopulator,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			client, clientErr := bugout.ClientFromEnv()
+			if clientErr != nil {
+				return clientErr
+			}
+
+			resourceHolders, err := client.Brood.GetResourceHolders(token, resourceId)
+			if err != nil {
+				return err
+			}
+
+			encodeErr := json.NewEncoder(cmd.OutOrStdout()).Encode(&resourceHolders)
+			return encodeErr
+		},
+	}
+
+	resourceHoldersGetCmd.Flags().StringVarP(&token, "token", "t", "", "Bugout access token to use for the request")
+	resourceHoldersGetCmd.Flags().StringVarP(&resourceId, "resource_id", "r", "", "Resource ID")
+
+	return resourceHoldersGetCmd
+}
+
+func GenerateResourceHoldersAddCommand() *cobra.Command {
+	var token, resourceId string
+	resourceHoldersAddCmd := &cobra.Command{
+		Use:     "add [holder]",
+		Short:   "Add resource holders",
+		PreRunE: cmdutils.TokenArgPopulator,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			if len(args) != 1 {
+				return fmt.Errorf(`Resource holder argument as json string in format {"holder_id": "user_or_group_uuid", "holder_type": "user_or_group", "permissions": ["admin", "create", "read", "update", "delete"]} must be specified`)
+			}
+			var resourceHolderPermissionsRaw brood.ResourceHolder
+			err := json.Unmarshal([]byte(args[0]), &resourceHolderPermissionsRaw)
+			if err != nil {
+				return err
+			}
+
+			client, clientErr := bugout.ClientFromEnv()
+			if clientErr != nil {
+				return clientErr
+			}
+
+			resource, err := client.Brood.AddResourceHolderPermissions(token, resourceId, resourceHolderPermissionsRaw)
+			if err != nil {
+				return err
+			}
+
+			encodeErr := json.NewEncoder(cmd.OutOrStdout()).Encode(&resource)
+			return encodeErr
+		},
+	}
+
+	resourceHoldersAddCmd.Flags().StringVarP(&token, "token", "t", "", "Bugout access token to use for the request")
+	resourceHoldersAddCmd.Flags().StringVarP(&resourceId, "resource_id", "r", "", "Resource ID")
+
+	return resourceHoldersAddCmd
+}
+
+func GenerateResourceHoldersDeleteCommand() *cobra.Command {
+	var token, resourceId string
+	resourceHoldersDeleteCmd := &cobra.Command{
+		Use:     "delete [holder]",
+		Short:   "Delete resource holders",
+		PreRunE: cmdutils.TokenArgPopulator,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			if len(args) != 1 {
+				return fmt.Errorf(`Resource holder argument as json string in format {"holder_id": "user_or_group_uuid", "holder_type": "user_or_group", "permissions": ["admin", "create", "read", "update", "delete"]} must be specified`)
+			}
+			var resourceHolderPermissionsRaw brood.ResourceHolder
+			err := json.Unmarshal([]byte(args[0]), &resourceHolderPermissionsRaw)
+			if err != nil {
+				return err
+			}
+
+			client, clientErr := bugout.ClientFromEnv()
+			if clientErr != nil {
+				return clientErr
+			}
+
+			resource, err := client.Brood.DeleteResourceHolderPermissions(token, resourceId, resourceHolderPermissionsRaw)
+			if err != nil {
+				return err
+			}
+
+			encodeErr := json.NewEncoder(cmd.OutOrStdout()).Encode(&resource)
+			return encodeErr
+		},
+	}
+
+	resourceHoldersDeleteCmd.Flags().StringVarP(&token, "token", "t", "", "Bugout access token to use for the request")
+	resourceHoldersDeleteCmd.Flags().StringVarP(&resourceId, "resource_id", "r", "", "Resource ID")
+
+	return resourceHoldersDeleteCmd
+}

pkg/brood/client.go

@@ -36,6 +36,9 @@ type BroodCaller interface {
 	UpdateResource(token, resourceId string, update interface{}, dropKeys []string) (Resource, error)
 	GetResources(token, applicationId string, queryParameters map[string]string) (Resources, error)
 	DeleteResource(token, resourceId string) (Resource, error)
+	GetResourceHolders(token, resourceId string) (ResourceHolders, error)
+	AddResourceHolderPermissions(token, resourceId string, resourceHolder ResourceHolder) (ResourceHolders, error)
+	DeleteResourceHolderPermissions(token, resourceId string, resourceHolder ResourceHolder) (ResourceHolders, error)
 	CreateApplication(token, groupId, name, description string) (Application, error)
 	GetApplication(token, applicationId string) (Application, error)
 	ListApplications(token, groupId string) (ApplicationsList, error)

pkg/brood/data.go

@@ -81,3 +81,14 @@ type resourceUpdateRequest struct {
 	Update   interface{} `json:"update"`
 	DropKeys []string    `json:"drop_keys"`
 }
+
+type ResourceHolder struct {
+	Id          string   `json:"holder_id"`
+	HolderType  string   `json:"holder_type"`
+	Permissions []string `json:"permissions"`
+}
+
+type ResourceHolders struct {
+	ResourceId string           `json:"resource_id"`
+	Holders    []ResourceHolder `json:"holders"`
+}

pkg/brood/resources.go

@@ -141,3 +141,90 @@ func (client BroodClient) DeleteResource(token, resourceId string) (Resource, er
 
 	return resource, decodeErr
 }
+
+func (client BroodClient) GetResourceHolders(token, resourceId string) (ResourceHolders, error) {
+	resourcesRoute := fmt.Sprintf("%s/%s/holders", client.Routes.Resources, resourceId)
+	request, requestErr := http.NewRequest("GET", resourcesRoute, nil)
+	if requestErr != nil {
+		return ResourceHolders{}, requestErr
+	}
+	request.Header.Add("Authorization", fmt.Sprintf("Bearer %s", token))
+	request.Header.Add("Accept", "application/json")
+
+	response, responseErr := client.HTTPClient.Do(request)
+	if responseErr != nil {
+		return ResourceHolders{}, responseErr
+	}
+	defer response.Body.Close()
+
+	statusErr := utils.HTTPStatusCheck(response)
+	if statusErr != nil {
+		return ResourceHolders{}, statusErr
+	}
+
+	var resourceHolders ResourceHolders
+	decodeErr := json.NewDecoder(response.Body).Decode(&resourceHolders)
+	return resourceHolders, decodeErr
+}
+
+func (client BroodClient) AddResourceHolderPermissions(token, resourceId string, resourceHolder ResourceHolder) (ResourceHolders, error) {
+	requestBody := ResourceHolder(resourceHolder)
+	requestBuffer := new(bytes.Buffer)
+	encodeErr := json.NewEncoder(requestBuffer).Encode(requestBody)
+	if encodeErr != nil {
+		return ResourceHolders{}, encodeErr
+	}
+	resourcesRoute := fmt.Sprintf("%s/%s/holders", client.Routes.Resources, resourceId)
+	request, requestErr := http.NewRequest("POST", resourcesRoute, requestBuffer)
+	if requestErr != nil {
+		return ResourceHolders{}, requestErr
+	}
+	request.Header.Add("Authorization", fmt.Sprintf("Bearer %s", token))
+	request.Header.Add("Accept", "application/json")
+
+	response, responseErr := client.HTTPClient.Do(request)
+	if responseErr != nil {
+		return ResourceHolders{}, responseErr
+	}
+	defer response.Body.Close()
+
+	statusErr := utils.HTTPStatusCheck(response)
+	if statusErr != nil {
+		return ResourceHolders{}, statusErr
+	}
+
+	var resourceHolders ResourceHolders
+	decodeErr := json.NewDecoder(response.Body).Decode(&resourceHolders)
+	return resourceHolders, decodeErr
+}
+
+func (client BroodClient) DeleteResourceHolderPermissions(token, resourceId string, resourceHolder ResourceHolder) (ResourceHolders, error) {
+	requestBody := ResourceHolder(resourceHolder)
+	requestBuffer := new(bytes.Buffer)
+	encodeErr := json.NewEncoder(requestBuffer).Encode(requestBody)
+	if encodeErr != nil {
+		return ResourceHolders{}, encodeErr
+	}
+	resourcesRoute := fmt.Sprintf("%s/%s/holders", client.Routes.Resources, resourceId)
+	request, requestErr := http.NewRequest("DELETE", resourcesRoute, requestBuffer)
+	if requestErr != nil {
+		return ResourceHolders{}, requestErr
+	}
+	request.Header.Add("Authorization", fmt.Sprintf("Bearer %s", token))
+	request.Header.Add("Accept", "application/json")
+
+	response, responseErr := client.HTTPClient.Do(request)
+	if responseErr != nil {
+		return ResourceHolders{}, responseErr
+	}
+	defer response.Body.Close()
+
+	statusErr := utils.HTTPStatusCheck(response)
+	if statusErr != nil {
+		return ResourceHolders{}, statusErr
+	}
+
+	var resourceHolders ResourceHolders
+	decodeErr := json.NewDecoder(response.Body).Decode(&resourceHolders)
+	return resourceHolders, decodeErr
+}

pkg/version.go

@@ -1,3 +1,3 @@
 package bugout
 
-const Version string = "0.4.1"
+const Version string = "0.4.2"