add in db user and grant endpoints

sitehostnz · yakmoose · Mar 12, 2024 · Mar 12, 2024 · Mar 12, 2024 · Mar 12, 2024
commit decac33b4cc2bdd20aea3f8a35ae544c0011ad66
diff --git a/go.sum b/go.sum
diff --git a/pkg/api/cloud/db/grant/add.go b/pkg/api/cloud/db/grant/add.go
@@ -0,0 +1,43 @@
+package grant
+
+import (
+	"context"
+	"net/url"
+
+	"github.com/sitehostnz/gosh/pkg/utils"
+)
+
+// Add creates a new grant for the given database/host/user
+func (s *Client) Add(ctx context.Context, request AddRequest) (response AddResponse, err error) {
+	uri := "cloud/db/grant/add.json"
+	keys := []string{
+		"client_id",
+		"server_name",
+		"mysql_host",
+		"database",
+		"username",
+		"grants[]",
+	}
+
+	values := url.Values{}
+	values.Add("client_id", s.client.ClientID)
+	values.Add("server_name", request.ServerName)
+	values.Add("mysql_host", request.MySQLHost)
+	values.Add("username", request.Username)
+	values.Add("database", request.Database)
+
+	for _, grant := range request.Grants {
+		values.Add("grants[]", grant)
+	}
+
+	req, err := s.client.NewRequest("POST", uri, utils.Encode(values, keys))
+	if err != nil {
+		return response, err
+	}
+
+	if err := s.client.Do(ctx, req, &response); err != nil {
+		return response, err
+	}
+
+	return response, nil
+}
diff --git a/pkg/api/cloud/db/grant/delete.go b/pkg/api/cloud/db/grant/delete.go
@@ -0,0 +1,38 @@
+package grant
+
+import (
+	"context"
+	"net/url"
+
+	"github.com/sitehostnz/gosh/pkg/utils"
+)
+
+// Delete deletes the grant from the specific database/host/user
+func (s *Client) Delete(ctx context.Context, request DeleteRequest) (response DeleteResponse, err error) {
+	uri := "cloud/db/grant/delete.json"
+	keys := []string{
+		"client_id",
+		"server_name",
+		"mysql_host",
+		"database",
+		"username",
+	}
+
+	values := url.Values{}
+	values.Add("client_id", s.client.ClientID)
+	values.Add("server_name", request.ServerName)
+	values.Add("mysql_host", request.MySQLHost)
+	values.Add("username", request.Username)
+	values.Add("database", request.Database)
+
+	req, err := s.client.NewRequest("POST", uri, utils.Encode(values, keys))
+	if err != nil {
+		return response, err
+	}
+
+	if err := s.client.Do(ctx, req, &response); err != nil {
+		return response, err
+	}
+
+	return response, nil
+}
diff --git a/pkg/api/cloud/db/grant/models.go b/pkg/api/cloud/db/grant/models.go
@@ -0,0 +1,19 @@
+package grant
+
+import (
+	"github.com/sitehostnz/gosh/pkg/api"
+)
+
+type (
+	// Client is a Service to work with API Jobs.
+	Client struct {
+		client *api.Client
+	}
+)
+
+// New is an initialisation function.
+func New(c *api.Client) *Client {
+	return &Client{
+		client: c,
+	}
+}
diff --git a/pkg/api/cloud/db/grant/request.go b/pkg/api/cloud/db/grant/request.go
@@ -0,0 +1,26 @@
+package grant
+
+type (
+	AddRequest struct {
+		ServerName string   `json:"server_name"`
+		MySQLHost  string   `json:"mysql_host"`
+		Username   string   `json:"username"`
+		Database   string   `json:"database"`
+		Grants     []string `json:"grants"`
+	}
+
+	UpdateRequest struct {
+		ServerName string   `json:"server_name"`
+		MySQLHost  string   `json:"mysql_host"`
+		Username   string   `json:"username"`
+		Database   string   `json:"database"`
+		Grants     []string `json:"grants"`
+	}
+
+	DeleteRequest struct {
+		ServerName string `json:"server_name"`
+		MySQLHost  string `json:"mysql_host"`
+		Username   string `json:"username"`
+		Database   string `json:"database"`
+	}
+)
diff --git a/pkg/api/cloud/db/grant/response.go b/pkg/api/cloud/db/grant/response.go
@@ -0,0 +1,24 @@
+package grant
+
+import "github.com/sitehostnz/gosh/pkg/models"
+
+type (
+	AddResponse struct {
+		Return struct {
+			JobID string `json:"job_id"`
+		} `json:"return"`
+		models.APIResponse
+	}
+	UpdateResponse struct {
+		Return struct {
+			JobID string `json:"job_id"`
+		} `json:"return"`
+		models.APIResponse
+	}
+	DeleteResponse struct {
+		Return struct {
+			JobID string `json:"job_id"`
+		} `json:"return"`
+		models.APIResponse
+	}
+)
diff --git a/pkg/api/cloud/db/grant/update.go b/pkg/api/cloud/db/grant/update.go
@@ -0,0 +1,43 @@
+package grant
+
+import (
+	"context"
+	"net/url"
+
+	"github.com/sitehostnz/gosh/pkg/utils"
+)
+
+// Update updates the grants for a specific db/user/host
+func (s *Client) Update(ctx context.Context, request UpdateRequest) (response UpdateResponse, err error) {
+	uri := "cloud/db/grant/update.json"
+	keys := []string{
+		"client_id",
+		"server_name",
+		"mysql_host",
+		"database",
+		"username",
+		"grants[]",
+	}
+
+	values := url.Values{}
+	values.Add("client_id", s.client.ClientID)
+	values.Add("server_name", request.ServerName)
+	values.Add("mysql_host", request.MySQLHost)
+	values.Add("username", request.Username)
+	values.Add("database", request.Database)
+
+	for _, grant := range request.Grants {
+		values.Add("grants[]", grant)
+	}
+
+	req, err := s.client.NewRequest("POST", uri, utils.Encode(values, keys))
+	if err != nil {
+		return response, err
+	}
+
+	if err := s.client.Do(ctx, req, &response); err != nil {
+		return response, err
+	}
+
+	return response, nil
+}
diff --git a/pkg/api/cloud/db/user/add.go b/pkg/api/cloud/db/user/add.go
@@ -0,0 +1,46 @@
+package user
+
+import (
+	"context"
+	"net/url"
+
+	"github.com/sitehostnz/gosh/pkg/utils"
+)
+
+// Add creates a new cloud database.
+func (s *Client) Add(ctx context.Context, request AddRequest) (response AddResponse, err error) {
+	uri := "cloud/db/user/add.json"
+	keys := []string{
+		"client_id",
+		"server_name",
+		"mysql_host",
+		"username",
+		"password",
+	}
+
+	values := url.Values{}
+	values.Add("client_id", s.client.ClientID)
+	values.Add("server_name", request.ServerName)
+	values.Add("mysql_host", request.MySQLHost)
+	values.Add("username", request.Username)
+	values.Add("password", request.Password)
+
+	// suspect these are required at the same time, so lets not deal with this here?
+	// feels cleaner to handle a grant independantly
+	//if request.Database != "" {
+	//	values.Add("database", request.Database)
+	//}
+	//if request.Grants != nil {
+	//}
+
+	req, err := s.client.NewRequest("POST", uri, utils.Encode(values, keys))
+	if err != nil {
+		return response, err
+	}
+
+	if err := s.client.Do(ctx, req, &response); err != nil {
+		return response, err
+	}
+
+	return response, nil
+}
diff --git a/pkg/api/cloud/db/user/delete.go b/pkg/api/cloud/db/user/delete.go
@@ -0,0 +1,36 @@
+package user
+
+import (
+	"context"
+	"net/url"
+
+	"github.com/sitehostnz/gosh/pkg/utils"
+)
+
+// Delete removes the cloud database user
+func (s *Client) Delete(ctx context.Context, request DeleteRequest) (response DeleteResponse, err error) {
+	uri := "cloud/db/user/delete.json"
+	keys := []string{
+		"client_id",
+		"server_name",
+		"mysql_host",
+		"username",
+	}
+
+	values := url.Values{}
+	values.Add("client_id", s.client.ClientID)
+	values.Add("server_name", request.ServerName)
+	values.Add("mysql_host", request.MySQLHost)
+	values.Add("username", request.Username)
+
+	req, err := s.client.NewRequest("POST", uri, utils.Encode(values, keys))
+	if err != nil {
+		return response, err
+	}
+
+	if err := s.client.Do(ctx, req, &response); err != nil {
+		return response, err
+	}
+
+	return response, nil
+}
diff --git a/pkg/api/cloud/db/user/get.go b/pkg/api/cloud/db/user/get.go
@@ -0,0 +1,39 @@
+package user
+
+import (
+	"context"
+
+	"github.com/sitehostnz/gosh/pkg/utils"
+)
+
+// Get fetches a cloud db.
+func (s *Client) Get(ctx context.Context, request GetRequest) (response GetResponse, err error) {
+	uri := "cloud/db/user/get.json"
+	keys := []string{
+		"apikey",
+		"client_id",
+		"server_name",
+		"mysql_host",
+		"username",
+	}
+
+	req, err := s.client.NewRequest("GET", uri, "")
+	if err != nil {
+		return response, err
+	}
+
+	v := req.URL.Query()
+	v.Add("api_key", s.client.APIKey)
+	v.Add("client_id", s.client.ClientID)
+	v.Add("server_name", request.ServerName)
+	v.Add("mysql_host", request.MySQLHost)
+	v.Add("username", request.Username)
+
+	req.URL.RawQuery = utils.Encode(v, keys)
+
+	if err := s.client.Do(ctx, req, &response); err != nil {
+		return response, err
+	}
+
+	return response, nil
+}
diff --git a/pkg/api/cloud/db/user/list.go b/pkg/api/cloud/db/user/list.go
@@ -0,0 +1,27 @@
+package user
+
+import (
+	"context"
+	"github.com/sitehostnz/gosh/pkg/utils"
+)
+
+// List returns a list of cloud databases, specific to the customer.
+func (s *Client) List(ctx context.Context, opt ListOptions) (response ListResponse, err error) {
+	uri := "cloud/db/user/list_all.json"
+
+	path, err := utils.AddOptions(uri, opt)
+	if err != nil {
+		return response, err
+	}
+
+	req, err := s.client.NewRequest("GET", path, "")
+	if err != nil {
+		return response, err
+	}
+
+	if err := s.client.Do(ctx, req, &response); err != nil {
+		return response, err
+	}
+
+	return response, nil
+}
diff --git a/pkg/api/cloud/db/user/models.go b/pkg/api/cloud/db/user/models.go
@@ -0,0 +1,19 @@
+package user
+
+import (
+	"github.com/sitehostnz/gosh/pkg/api"
+)
+
+type (
+	// Client is a Service to work with API Jobs.
+	Client struct {
+		client *api.Client
+	}
+)
+
+// New is an initialisation function.
+func New(c *api.Client) *Client {
+	return &Client{
+		client: c,
+	}
+}