fix epm

Author: daemitus

generated/fleet/fleet.gen.go

@@ -73,6 +73,7 @@ var transformers = []TransformFunc{
 	transformAddPackagePolicyVars,
 	transformAddPackagePolicySecretReferences,
 	transformFixPackageSearchResult,
+	transformFixGetPackageResult,
 }
 
 // transformFilterPaths filters the paths in a schema down to
@@ -232,16 +233,36 @@ func transformInlinePackageDefinitions(schema *Schema) {
 
 	// Get
 	{
-		props, ok := epmPath.Get.Responses.GetFields("200.content.application/json.schema.allOf.1.properties")
+		respSchema, ok := epmPath.Get.Responses.GetFields("200.content.application/json.schema")
 		if !ok {
 			panic("properties not found")
 		}
 
-		// status needs to be moved to schemes and a ref inserted in its place.
-		value, _ := props.Get("status")
+		// allOf.1 should also be inside "item"
+		_allOf, _ := respSchema.Get("allOf")
+		allOf := _allOf.([]any)
+		respSchema.Delete("allOf")
+
+		list := make([]any, 2)
+		allOf0, _ := Fields(allOf[0].(map[string]any)).GetFields("properties.item")
+		allOf1 := Fields(allOf[1].(map[string]any))
+		list[0] = allOf0
+		list[1] = allOf1
+		respSchema.Set("properties.item.allOf", list)
+
+		// item needs to be moved to schemas and a ref inserted in its place.
+		value, _ := respSchema.Get("properties.item.allOf")
+		respSchema.Delete("properties.item.allOf")
+		schema.Components.Set("schemas.get_package_item.allOf", value)
+		respSchema.Set("properties.item.$ref", "#/components/schemas/get_package_item")
+
+		// status needs to be moved to schemas and a ref inserted in its place.
+		props, _ := schema.Components.GetFields("schemas.get_package_item.allOf.1.properties")
+		value, _ = props.GetFields("status")
 		schema.Components.Set("schemas.package_status", value)
 		props.Delete("status")
 		props.Set("status.$ref", "#/components/schemas/package_status")
+
 	}
 
 	// Post
@@ -369,6 +390,18 @@ func transformFixPackageSearchResult(schema *Schema) {
 	properties.Delete("installationInfo")
 }
 
+// transformFixGetPackageResult removes unneeded fields from the
+// GetPackageResult struct. These fields are also causing parsing errors.
+func transformFixGetPackageResult(schema *Schema) {
+	properties, ok := schema.Components.GetFields("schemas.package_info.properties")
+	if !ok {
+		panic("properties not found")
+	}
+	properties.Delete("assets")
+	properties.Delete("icons")
+	properties.Delete("installationInfo")
+}
+
 // downloadFile will download a file from url and return the
 // bytes. If the request fails, or a non 200 error code is
 // observed in the response, an error is returned instead.
@@ -462,11 +495,15 @@ func (f Fields) Get(key string) (any, bool) {
 			if !ok {
 				return nil, false
 			}
-			if m, isMap := slicedValue.(map[string]any); ok && isMap {
+
+			switch m := slicedValue.(type) {
+			case map[string]any:
 				return Fields(m).Get(postSliceKeys)
+			case Fields:
+				return m.Get(postSliceKeys)
+			default:
+				return slicedValue, true
 			}
-			return slicedValue, true
-
 		default:
 			rootKey = key
 		}

generated/fleet/getschema.go

@@ -4,8 +4,8 @@ import (
 	"context"
 	"errors"
 	"fmt"
-	"io"
 	"net/http"
+	"strings"
 
 	fleetapi "github.com/elastic/terraform-provider-elasticstack/generated/fleet"
 	"github.com/elastic/terraform-provider-elasticstack/internal/utils"
@@ -333,27 +333,21 @@ func DeletePackagePolicy(ctx context.Context, client *Client, id string, force b
 }
 
 // ReadPackage reads a specific package from the API.
-func ReadPackage(ctx context.Context, client *Client, name, version string) diag.Diagnostics {
+func ReadPackage(ctx context.Context, client *Client, name, version string) (*fleetapi.GetPackageItem, diag.Diagnostics) {
 	params := fleetapi.GetPackageParams{}
 
-	resp, err := client.API.GetPackage(ctx, name, version, &params)
+	resp, err := client.API.GetPackageWithResponse(ctx, name, version, &params)
 	if err != nil {
-		return utils.FrameworkDiagFromError(err)
+		return nil, utils.FrameworkDiagFromError(err)
 	}
-	defer resp.Body.Close()
 
-	switch resp.StatusCode {
+	switch resp.StatusCode() {
 	case http.StatusOK:
-		return nil
+		return resp.JSON200.Item, nil
 	case http.StatusNotFound:
-		return utils.FrameworkDiagFromError(ErrPackageNotFound)
+		return nil, utils.FrameworkDiagFromError(ErrPackageNotFound)
 	default:
-		errData, err := io.ReadAll(resp.Body)
-		if err != nil {
-			return utils.FrameworkDiagFromError(err)
-		}
-
-		return reportUnknownError(resp.StatusCode, errData)
+		return nil, reportUnknownError(resp.StatusCode(), resp.Body)
 	}
 }
 
@@ -365,22 +359,16 @@ func InstallPackage(ctx context.Context, client *Client, name, version string, f
 		IgnoreConstraints: nil,
 	}
 
-	resp, err := client.API.InstallPackage(ctx, name, version, &params, body)
+	resp, err := client.API.InstallPackageWithResponse(ctx, name, version, &params, body)
 	if err != nil {
 		return utils.FrameworkDiagFromError(err)
 	}
-	defer resp.Body.Close()
 
-	switch resp.StatusCode {
+	switch resp.StatusCode() {
 	case http.StatusOK:
 		return nil
 	default:
-		errData, err := io.ReadAll(resp.Body)
-		if err != nil {
-			return utils.FrameworkDiagFromError(err)
-		}
-
-		return reportUnknownError(resp.StatusCode, errData)
+		return reportUnknownError(resp.StatusCode(), resp.Body)
 	}
 }
 
@@ -399,6 +387,14 @@ func Uninstall(ctx context.Context, client *Client, name, version string, force
 	switch resp.StatusCode() {
 	case http.StatusOK:
 		return nil
+	case http.StatusBadRequest:
+		msg := resp.JSON400.Message
+		// {"statusCode":400,"error":"Bad Request","message":"some-integration is not installed"}
+		if msg != nil && strings.Contains(*msg, "is not installed") {
+			return nil
+		} else {
+			return reportUnknownError(resp.StatusCode(), resp.Body)
+		}
 	case http.StatusNotFound:
 		return nil
 	default:

internal/clients/fleet/fleet.go

@@ -11,10 +11,10 @@ import (
 )
 
 func (r *integrationResource) Create(ctx context.Context, req resource.CreateRequest, resp *resource.CreateResponse) {
-	r.create(ctx, req.Plan, &resp.State, resp.Diagnostics)
+	r.create(ctx, req.Plan, &resp.State, &resp.Diagnostics)
 }
 
-func (r integrationResource) create(ctx context.Context, plan tfsdk.Plan, state *tfsdk.State, respDiags diag.Diagnostics) {
+func (r integrationResource) create(ctx context.Context, plan tfsdk.Plan, state *tfsdk.State, respDiags *diag.Diagnostics) {
 	var planModel integrationModel
 
 	diags := plan.Get(ctx, &planModel)