Add hooks for attaching VPC

Author: Nicholas Thomson

apis/v1alpha1/ack-generate-metadata.yaml

@@ -1,13 +1,13 @@
 ack_generate_info:
-  build_date: "2022-03-30T20:20:28Z"
+  build_date: "2022-03-30T21:05:10Z"
   build_hash: 8a55c536d84c0a798df5be9f1cbdebdef51f3355
   go_version: go1.17.8
   version: v0.18.1
 api_directory_checksum: 35ef0e4da69ded8c1fa7a4a6029510864a1069af
 api_version: v1alpha1
 aws_sdk_go_version: v1.42.0
 generator_config_info:
-  file_checksum: c3e88eb4352e0de787dbd97dd9ac48762b633798
+  file_checksum: 8bcbefb7a9f69e907e36562fc8d2e9c11e74cf28
   original_file_name: generator.yaml
 last_modification:
   reason: API generation

apis/v1alpha1/generator.yaml

@@ -200,6 +200,15 @@ resources:
         references:
           resource: VPC
           path: Status.VPCID
+    hooks:
+      sdk_create_post_set_output:
+        template_path: hooks/internet_gateway/sdk_create_post_set_output.go.tpl
+      sdk_read_many_post_set_output:
+        template_path: hooks/internet_gateway/sdk_read_many_post_set_output.go.tpl
+      sdk_delete_pre_build_request:
+        template_path: hooks/internet_gateway/sdk_delete_pre_build_request.go.tpl
+    update_operation:
+      custom_method_name: customUpdateInternetGateway
   NatGateway:
     fields:
       AllocationId:

generator.yaml

@@ -200,6 +200,15 @@ resources:
         references:
           resource: VPC
           path: Status.VPCID
+    hooks:
+      sdk_create_post_set_output:
+        template_path: hooks/internet_gateway/sdk_create_post_set_output.go.tpl
+      sdk_read_many_post_set_output:
+        template_path: hooks/internet_gateway/sdk_read_many_post_set_output.go.tpl
+      sdk_delete_pre_build_request:
+        template_path: hooks/internet_gateway/sdk_delete_pre_build_request.go.tpl
+    update_operation:
+      custom_method_name: customUpdateInternetGateway
   NatGateway:
     fields:
       AllocationId:

pkg/resource/internet_gateway/hook.go

@@ -12,3 +12,121 @@
 // permissions and limitations under the License.
 
 package internet_gateway
+
+import (
+	"context"
+
+	ackcompare "github.com/aws-controllers-k8s/runtime/pkg/compare"
+	ackrtlog "github.com/aws-controllers-k8s/runtime/pkg/runtime/log"
+	svcsdk "github.com/aws/aws-sdk-go/service/ec2"
+)
+
+func (rm *resourceManager) customUpdateInternetGateway(
+	ctx context.Context,
+	desired *resource,
+	latest *resource,
+	delta *ackcompare.Delta,
+) (updated *resource, err error) {
+	rlog := ackrtlog.FromContext(ctx)
+	exit := rlog.Trace("rm.customUpdateInternetGateway")
+	defer func(err error) {
+		exit(err)
+	}(err)
+
+	ko := desired.ko.DeepCopy()
+	rm.setStatusDefaults(ko)
+
+	if delta.DifferentAt("Spec.VPC") {
+		if latest.ko.Spec.VPC != nil {
+			if err = rm.detachFromVPC(ctx, *latest.ko.Spec.VPC, *latest.ko.Status.InternetGatewayID); err != nil {
+				return nil, err
+			}
+		}
+		if desired.ko.Spec.VPC != nil {
+			if err = rm.attachToVPC(ctx, desired); err != nil {
+				return nil, err
+			}
+		}
+	}
+
+	return &resource{ko}, nil
+}
+
+// getAttachedVPC will attempt to find the VPCID for any VPC that the
+// InternetGateway is currently attached to. If it is not attached, or is
+// actively being detached, then it will return nil.
+func (rm *resourceManager) getAttachedVPC(
+	ctx context.Context,
+	latest *resource,
+) (vpcID *string, err error) {
+	rlog := ackrtlog.FromContext(ctx)
+	exit := rlog.Trace("rm.getAttachedVPC")
+	defer func(err error) {
+		exit(err)
+	}(err)
+
+	// InternetGateways can only be attached to a single VPC at a time - even
+	// though attachments is a slice. Attaching is almost instant, but if the
+	// request returns that it is in `Attaching` status still, we can assume
+	// that it will be attached in the near future and does not need to be
+	// updated.
+	for _, att := range latest.ko.Status.Attachments {
+		// There is no `AttachmentStatusAvailable` - so we can just check by
+		// using negative logic with the constants we have, instead
+		if *att.State != svcsdk.AttachmentStatusDetached &&
+			*att.State != svcsdk.AttachmentStatusDetaching {
+			return att.VPCID, nil
+		}
+	}
+
+	return nil, nil
+}
+
+func (rm *resourceManager) attachToVPC(
+	ctx context.Context,
+	desired *resource,
+) (err error) {
+	rlog := ackrtlog.FromContext(ctx)
+	exit := rlog.Trace("rm.attachToVPC")
+	defer func(err error) {
+		exit(err)
+	}(err)
+
+	input := &svcsdk.AttachInternetGatewayInput{
+		InternetGatewayId: desired.ko.Status.InternetGatewayID,
+		VpcId:             desired.ko.Spec.VPC,
+	}
+
+	_, err = rm.sdkapi.AttachInternetGatewayWithContext(ctx, input)
+	rm.metrics.RecordAPICall("UPDATE", "AttachInternetGateway", err)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (rm *resourceManager) detachFromVPC(
+	ctx context.Context,
+	vpcID string,
+	igwID string,
+) (err error) {
+	rlog := ackrtlog.FromContext(ctx)
+	exit := rlog.Trace("rm.detachFromVPC")
+	defer func(err error) {
+		exit(err)
+	}(err)
+
+	input := &svcsdk.DetachInternetGatewayInput{
+		InternetGatewayId: &igwID,
+		VpcId:             &vpcID,
+	}
+
+	_, err = rm.sdkapi.DetachInternetGatewayWithContext(ctx, input)
+	rm.metrics.RecordAPICall("UPDATE", "DetachInternetGateway", err)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}

pkg/resource/internet_gateway/sdk.go

@@ -0,0 +1,5 @@
+	if ko.Spec.VPC != nil {
+		if err = rm.attachToVPC(ctx, &resource{ko}); err != nil {
+			return nil, err
+		}
+	}

templates/hooks/internet_gateway/sdk_create_post_set_output.go.tpl

@@ -0,0 +1,5 @@
+    if r.ko.Spec.VPC != nil && r.ko.Status.InternetGatewayID != nil {
+		if err = rm.detachFromVPC(ctx, *r.ko.Spec.VPC, *r.ko.Status.InternetGatewayID); err != nil {
+			return nil, err
+		}
+	}

templates/hooks/internet_gateway/sdk_delete_pre_build_request.go.tpl

@@ -0,0 +1,6 @@
+	vpcID, err := rm.getAttachedVPC(ctx, &resource{ko})
+	if err != nil {
+		return nil, err
+	} else {
+		ko.Spec.VPC = vpcID
+	}