Reuse existing fleet server hosts

internal/kibana/fleet.go

@@ -22,9 +22,12 @@ type FleetOutput struct {
 }
 
 type FleetServerHost struct {
-	URLs      []string `json:"host_urls"`
-	Name      string   `json:"name"`
-	IsDefault bool     `json:"is_default"`
+	ID   string   `json:"id,omitempty"`
+	URLs []string `json:"host_urls"`
+	Name string   `json:"name"`
+
+	// TODO: Avoid using is_default, so a cluster can be used for multiple environments.
+	IsDefault bool `json:"is_default"`
 }
 
 type AgentSSL struct {
@@ -198,6 +201,31 @@ func (c *Client) AddFleetServerHost(ctx context.Context, host FleetServerHost) e
 	return nil
 }
 
+func (c *Client) UpdateFleetServerHost(ctx context.Context, host FleetServerHost) error {
+	if host.ID == "" {
+		return fmt.Errorf("host id required when updating fleet server host")
+	}
+
+	// Payload should not contain the ID, it is set in the URL.
+	id := host.ID
+	host.ID = ""
+	reqBody, err := json.Marshal(host)
+	if err != nil {
+		return fmt.Errorf("could not convert fleet server host to JSON: %w", err)
+	}
+
+	statusCode, respBody, err := c.put(ctx, fmt.Sprintf("%s/fleet_server_hosts/%s", FleetAPI, id), reqBody)
+	if err != nil {
+		return fmt.Errorf("could not update fleet server host: %w", err)
+	}
+
+	if statusCode != http.StatusOK {
+		return fmt.Errorf("could not update fleet server host; API status code = %d; response body = %s", statusCode, respBody)
+	}
+
+	return nil
+}
+
 // CreateFleetServiceToken creates a service token for Fleet, to be used when enrolling Fleet Servers.
 func (c *Client) CreateFleetServiceToken(ctx context.Context) (string, error) {
 	statusCode, respBody, err := c.post(ctx, fmt.Sprintf("%s/service_tokens", FleetAPI), nil)

internal/stack/environment.go

@@ -53,7 +53,7 @@ func (p *environmentProvider) BootUp(ctx context.Context, options Options) error
 	// TODO: Migrate from serverless variables.
 	config.Parameters[ParamServerlessLocalStackVersion] = options.StackVersion
 
-	config, err = p.setupFleet(ctx, config, options.StackVersion)
+	config, err = p.setupFleet(ctx, config, options)
 	if err != nil {
 		return fmt.Errorf("failed to setup Fleet: %w", err)
 	}
@@ -141,7 +141,7 @@ func (p *environmentProvider) initClients() error {
 	return nil
 }
 
-func (p environmentProvider) setupFleet(ctx context.Context, config Config, stackVersion string) (Config, error) {
+func (p environmentProvider) setupFleet(ctx context.Context, config Config, options Options) (Config, error) {
 	const localFleetServerURL = "https://fleet-server:8220"
 
 	fleetServerURL, err := p.kibana.DefaultFleetServerURL(ctx)
@@ -151,17 +151,23 @@ func (p environmentProvider) setupFleet(ctx context.Context, config Config, stac
 		config.Parameters[paramFleetServerManaged] = "true"
 
 		host := kibana.FleetServerHost{
+			ID:        fleetServerHostID(options.Profile.ProfileName),
 			URLs:      []string{fleetServerURL},
 			IsDefault: true,
 			Name:      "elastic-package-managed-fleet-server",
 		}
-		// TODO: Check if it is already there to avoid creating many of them.
 		err := p.kibana.AddFleetServerHost(ctx, host)
-		if err != nil && !errors.Is(err, kibana.ErrConflict) {
+		if errors.Is(err, kibana.ErrConflict) {
+			err = p.kibana.UpdateFleetServerHost(ctx, host)
+			if err != nil {
+				return config, fmt.Errorf("failed to update existing Fleet Server host (id: %s): %w", host.ID, err)
+			}
+		}
+		if err != nil {
 			return config, fmt.Errorf("failed to add Fleet Server host: %w", err)
 		}
 
-		_, err = createFleetServerPolicy(ctx, p.kibana, stackVersion)
+		_, err = createFleetServerPolicy(ctx, p.kibana, options.StackVersion, options.Profile.ProfileName)
 		if err != nil {
 			return config, fmt.Errorf("failed to create agent policy for Fleet Server: %w", err)
 		}
@@ -178,6 +184,10 @@ func (p environmentProvider) setupFleet(ctx context.Context, config Config, stac
 	return config, nil
 }
 
+func fleetServerHostID(namespace string) string {
+	return "elastic-package-" + namespace
+}
+
 func isFleetServerReachable(ctx context.Context, address string) bool {
 	status, err := fleetserver.NewClient(address).Status(ctx)
 	return err == nil && strings.ToLower(status.Status) == "healthy"

internal/stack/fleetserverpolicy.go

@@ -21,7 +21,7 @@ const (
 
 // createFleetServerPolicy creates an agent policy with the initial configuration used for
 // agents managed by elastic-package.
-func createFleetServerPolicy(ctx context.Context, kibanaClient *kibana.Client, stackVersion string) (*kibana.Policy, error) {
+func createFleetServerPolicy(ctx context.Context, kibanaClient *kibana.Client, stackVersion string, namespace string) (*kibana.Policy, error) {
 	policy := kibana.Policy{
 		Name:                 "Fleet Server (elastic-package)",
 		ID:                   managedFleetServerPolicyID,