WIP: CRD replacement

provider/pkg/await/await.go

@@ -354,7 +354,7 @@ func Update(c UpdateConfig) (*unstructured.Unstructured, error) {
 	// - [ ] Support server-side apply, when it comes out.
 	//
 
-	client, err := c.ClientSet.ResourceClient(c.Inputs.GroupVersionKind(), c.Inputs.GetNamespace())
+	client, err := c.ClientSet.ResourceClientForObject(c.Inputs)
 	if err != nil {
 		return nil, err
 	}
@@ -366,21 +366,28 @@ func Update(c UpdateConfig) (*unstructured.Unstructured, error) {
 		return nil, err
 	}
 
-	// Create merge patch (prefer strategic merge patch, fall back to JSON merge patch).
-	patch, patchType, _, err := openapi.PatchForResourceUpdate(c.Resources, c.Previous, c.Inputs, liveOldObj)
-	if err != nil {
-		return nil, err
-	}
+	// TODO: May just need to GET, set resourceVersion, and then immediately Update
+	var currentOutputs *unstructured.Unstructured
+	if clients.IsCRD(c.Inputs) {
+		c.Inputs.SetResourceVersion(liveOldObj.GetResourceVersion())
+		currentOutputs, err = client.Update(context.TODO(), c.Inputs, metav1.UpdateOptions{})
+	} else {
+		// Create merge patch (prefer strategic merge patch, fall back to JSON merge patch).
+		patch, patchType, _, err := openapi.PatchForResourceUpdate(c.Resources, c.Previous, c.Inputs, liveOldObj)
+		if err != nil {
+			return nil, err
+		}
 
-	var options metav1.PatchOptions
-	if c.DryRun {
-		options.DryRun = []string{metav1.DryRunAll}
-	}
+		var options metav1.PatchOptions
+		if c.DryRun {
+			options.DryRun = []string{metav1.DryRunAll}
+		}
 
-	// Issue patch request.
-	// NOTE: We can use the same client because if the `kind` changes, this will cause
-	// a replace (i.e., destroy and create).
-	currentOutputs, err := client.Patch(context.TODO(), c.Inputs.GetName(), patchType, patch, options)
+		// Issue patch request.
+		// NOTE: We can use the same client because if the `kind` changes, this will cause
+		// a replace (i.e., destroy and create).
+		currentOutputs, err = client.Patch(context.TODO(), c.Inputs.GetName(), patchType, patch, options)
+	}
 	if err != nil {
 		return nil, err
 	}