Support parameter to override builder region (#59)

superfly · May 8, 2024 · 3883e36 · 3883e36
1 parent 9c32a87
commit 3883e36
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/resource_remote_builders.go b/resource_remote_builders.go
@@ -2,7 +2,7 @@ package fly
 
 import "context"
 
-func (client *Client) EnsureRemoteBuilder(ctx context.Context, orgID, appName string) (*GqlMachine, *App, error) {
+func (client *Client) EnsureRemoteBuilder(ctx context.Context, orgID, appName, region string) (*GqlMachine, *App, error) {
 	query := `
 		mutation($input: EnsureMachineRemoteBuilderInput!) {
 			ensureMachineRemoteBuilder(input: $input) {
@@ -31,16 +31,16 @@ func (client *Client) EnsureRemoteBuilder(ctx context.Context, orgID, appName st
 	req := client.NewRequest(query)
 	ctx = ctxWithAction(ctx, "ensure_remote_builder")
 
+	input := EnsureRemoteBuilderInput{}
+	if region != "" {
+		input.Region = StringPointer(region)
+	}
 	if orgID != "" {
-		req.Var("input", EnsureRemoteBuilderInput{
-			OrganizationID: StringPointer(orgID),
-		})
+		input.OrganizationID = StringPointer(orgID)
 	} else {
-		req.Var("input", EnsureRemoteBuilderInput{
-			AppName: StringPointer(appName),
-		})
-
+		input.AppName = StringPointer(appName)
 	}
+	req.Var("input", input)
 
 	data, err := client.RunWithContext(ctx, req)
 	if err != nil {

diff --git a/types.go b/types.go
@@ -742,6 +742,7 @@ type PostgresEnableConsulPayload struct {
 type EnsureRemoteBuilderInput struct {
 	AppName        *string `json:"appName"`
 	OrganizationID *string `json:"organizationId"`
+	Region         *string `json:"region"`
 }
 
 type PostgresClusterAttachment struct {