#2 add project management API

Author: flambchap

go/api/client.go

@@ -1,6 +1,7 @@
 package api
 
 import (
+	"encoding/json"
 	"errors"
 	"fmt"
 	"log/slog"
@@ -35,6 +36,20 @@ query createTeam($teamId: ID!) {
 query deleteTeam($teamId: ID!, $force: Boolean) {
   deleteTeam(teamId: $teamId, force: $force)
 }
+`
+
+	createProjectMutation = `
+mutation RmCreateProject($projectName: String!) {
+    rmCreateProject(projectName: $projectName) {
+        id
+    }
+}
+`
+
+	deleteProjectMutation = `
+mutation RmDeleteProject($projectId: ID!) {
+    rmDeleteProject(projectId: $projectId)
+}
 `
 )
 
@@ -43,7 +58,7 @@ type Client struct {
 	Endpoint      string
 }
 
-func (client Client) sendRequest(operationName, query string, variables map[string]any) error {
+func (client Client) sendRequest(operationName, query string, variables graphql.Object) (json.RawMessage, error) {
 	request := graphql.Request{Query: query, Variables: variables}
 	slog.Info("--> GraphQL call [" + operationName + "]")
 	start := time.Now()
@@ -53,32 +68,77 @@ func (client Client) sendRequest(operationName, query string, variables map[stri
 	}()
 	response, err := client.GraphQLClient.Execute(client.Endpoint, request)
 	if err != nil {
-		return lib.WrapError("error while sending an api request", err)
+		return json.RawMessage{}, lib.WrapError("error while sending an api request", err)
 	}
 	if response.Errors != nil {
-		return errors.New("error recieved when executing an API call: \n" + string(response.Errors))
+		err = errors.New("error recieved when executing an API call: \n" + string(response.Errors))
 	}
-	return nil
+	return response.Data, err
+}
+
+func (client Client) sendRequestDiscardingData(operationName, query string, variables graphql.Object) error {
+	_, err := client.sendRequest(operationName, query, variables)
+	return err
 }
 
 func (client Client) Auth(token string) error {
 	variables := map[string]any{
 		"token": token,
 	}
-	return client.sendRequest("auth", authQuery, variables)
+	return client.sendRequestDiscardingData("auth", authQuery, variables)
 }
 
 func (client Client) CreateTeam(teamId string) error {
 	variables := map[string]any{
 		"teamId": teamId,
 	}
-	return client.sendRequest(fmt.Sprintf("create team '%s'", teamId), createTeamQuery, variables)
+	return client.sendRequestDiscardingData(
+		fmt.Sprintf("create team '%s'", teamId),
+		createTeamQuery,
+		variables)
 }
 
 func (client Client) DeleteTeam(teamId string, force bool) error {
 	variables := map[string]any{
 		"teamId": teamId,
 		"force":  force,
 	}
-	return client.sendRequest(fmt.Sprintf("delete team '%s'", teamId), deleteTeamQuery, variables)
+	return client.sendRequestDiscardingData(
+		fmt.Sprintf("delete team '%s'", teamId),
+		deleteTeamQuery,
+		variables)
+}
+
+func (client Client) CreateProject(projectName string) (id string, err error) {
+	variables := map[string]any{
+		"projectName": projectName,
+	}
+	rawData, err := client.sendRequest(
+		fmt.Sprintf("create project with name '%s'", projectName),
+		createProjectMutation,
+		variables)
+	if err != nil {
+		return id, err
+	}
+	var rmCreateProjectResponse RmCreateProjectResponse
+	err = json.Unmarshal(rawData, &rmCreateProjectResponse)
+	if err != nil {
+		return id, lib.WrapError("unable to unmarshal rmCreateProjectResponse", err)
+	}
+	return rmCreateProjectResponse.RmCreateProject.Id, err
+}
+
+func (client Client) DeleteProject(projectId string) error {
+	variables := map[string]any{
+		"projectId": projectId,
+	}
+	rawData, err := client.sendRequest(
+		fmt.Sprintf("delete project with id '%s'", projectId),
+		deleteProjectMutation,
+		variables)
+	if err != nil {
+		return err
+	}
+	slog.Debug(string(rawData))
+	return nil
 }

go/api/types.go

@@ -0,0 +1,9 @@
+package api
+
+type RmCreateProjectResponse struct {
+	RmCreateProject RmCreateProject
+}
+
+type RmCreateProject struct {
+	Id string
+}

go/main.go

@@ -12,6 +12,12 @@ import (
 	"github.com/dbeaver/cloudbeaver-graphql-examples/go/lib"
 )
 
+const (
+	objectPrefix = "cloudbeaver-graqhql-examples-go-"
+	teamId       = objectPrefix + "team"
+	projectName  = objectPrefix + "project"
+)
+
 func main() {
 	if err := main0(); err != nil {
 		slog.Error(err.Error())
@@ -25,12 +31,7 @@ func main0() error {
 	if err != nil {
 		return lib.WrapError("error while reading variables", err)
 	}
-	cookieJar, err := cookiejar.New(nil)
-	if err != nil {
-		return lib.WrapError("unable to create a cookie jar", err)
-	}
-	graphQLClient := graphql.Client{HttpClient: &http.Client{Jar: cookieJar}}
-	apiClient := api.Client{GraphQLClient: graphQLClient, Endpoint: env.GraphqlEndpoint()}
+	apiClient := initClient(env.GraphqlEndpoint())
 
 	// Auth
 	err = apiClient.Auth(env.Token)
@@ -39,18 +40,41 @@ func main0() error {
 		return err
 	}
 
-	// Create a team
-	teamId := "exampleTeamId"
+	// Creation / deletion of a team
 	err = apiClient.CreateTeam(teamId)
 	if err != nil {
 		return err
 	}
+	defer cleanup("delete team "+teamId, func() error {
+		return apiClient.DeleteTeam(teamId, true)
+	})
 
-	// Delete a team
-	err = apiClient.DeleteTeam(teamId, true)
+	// Creation of a project
+	projectId, err := apiClient.CreateProject(projectName)
 	if err != nil {
 		return err
 	}
+	defer cleanup("delete project "+projectId, func() error {
+		return apiClient.DeleteProject(projectId)
+	})
 
 	return nil
 }
+
+func initClient(endpoint string) api.Client {
+	cookieJar, err := cookiejar.New(nil)
+	if err != nil {
+		// Invariant: the method that creates cookie jar with no options never returns non-nil err
+		panic("encountered error while creating a cookie jar! " + err.Error())
+	}
+	graphQLClient := graphql.Client{HttpClient: &http.Client{Jar: cookieJar}}
+	return api.Client{GraphQLClient: graphQLClient, Endpoint: endpoint}
+}
+
+func cleanup(callDescription string, apiCall func() error) {
+	err := apiCall()
+	if err != nil {
+		slog.Warn("unable to " + callDescription)
+		slog.Warn(err.Error())
+	}
+}