Adding delete all layer versions functionality for layer_version resource

Author: Vandita2020

apis/v1alpha1/ack-generate-metadata.yaml

@@ -1,13 +1,13 @@
 ack_generate_info:
-  build_date: "2022-12-13T23:07:54Z"
+  build_date: "2023-01-05T21:17:54Z"
   build_hash: 12246c7da82841b351ec7a9e1f139f9338f2784b
   go_version: go1.19
   version: v0.20.1-14-g12246c7
 api_directory_checksum: 90c9337a64415662f698cf6cd270706cd560bd16
 api_version: v1alpha1
 aws_sdk_go_version: v1.44.93
 generator_config_info:
-  file_checksum: e007d88ecf6ad1c45cb6451f5fda9401af9d7305
+  file_checksum: e3b0b7baf90c92126b784c7147765f29847c303b
   original_file_name: generator.yaml
 last_modification:
   reason: API generation

apis/v1alpha1/generator.yaml

@@ -153,6 +153,9 @@ resources:
             operation: ReadOne
     tags:
       ignore: true
+    hooks:
+      sdk_delete_pre_build_request:
+        template_path: hooks/layerversion/sdk_delete_pre_build_request.go.tpl
     renames:
       operations:
         GetLayerVersion:

generator.yaml

@@ -153,6 +153,9 @@ resources:
             operation: ReadOne
     tags:
       ignore: true
+    hooks:
+      sdk_delete_pre_build_request:
+        template_path: hooks/layerversion/sdk_delete_pre_build_request.go.tpl
     renames:
       operations:
         GetLayerVersion:

pkg/resource/layer_version/hooks.go

@@ -0,0 +1,88 @@
+// Copyright Amazon.com Inc. or its affiliates. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License"). You may
+// not use this file except in compliance with the License. A copy of the
+// License is located at
+//
+//     http://aws.amazon.com/apache2.0/
+//
+// or in the "license" file accompanying this file. This file is distributed
+// on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+// express or implied. See the License for the specific language governing
+// permissions and limitations under the License.
+
+package layer_version
+
+import (
+	"context"
+	"fmt"
+	"sort"
+
+	// "github.com/Jeffail/gabs/v2"
+	svcapitypes "github.com/aws-controllers-k8s/lambda-controller/apis/v1alpha1"
+	ackrtlog "github.com/aws-controllers-k8s/runtime/pkg/runtime/log"
+	svcsdk "github.com/aws/aws-sdk-go/service/lambda"
+)
+
+var (
+	_ = &svcsdk.Lambda{}
+	_ = &svcapitypes.LayerVersion{}
+)
+
+// customPreDelete deletes all the previous versions of a
+//LayerVersion except the latest version
+
+// This function is used as a sdk_delete hook, to delete
+// all the previous versions of a LayerVersion when delete API call is made
+func customPreDelete(
+	r *resource,
+	rm *resourceManager,
+	ctx context.Context,
+) error {
+	var response *svcsdk.ListLayerVersionsOutput
+	var err error
+	var version_list []*svcsdk.LayerVersionsListItem
+	var totalVersion int
+	var _ *svcsdk.DeleteLayerVersionOutput
+
+	// Getting the list of all the versions of a LayerVersion
+	input := &svcsdk.ListLayerVersionsInput{
+		LayerName: r.ko.Spec.LayerName,
+	}
+	response, err = rm.sdkapi.ListLayerVersionsWithContext(ctx, input)
+	if err != nil {
+		return err
+	}
+
+	log := ackrtlog.FromContext(ctx)
+	log.Info("Deleting previous versions of LayerVersion")
+	// The above API call returns output containing list of versions as
+	//LayerVersions and a pagination token as NextMarker
+
+	// Extracting the list of versions and assigning it to a new variable
+	version_list = response.LayerVersions
+	totalVersion = len(version_list) // Number of total versions
+
+	// sorting the list in ascending order
+	sort.Slice(version_list, func(i, j int) bool {
+		return *version_list[i].Version < *version_list[j].Version
+	})
+
+	for i := 0; i < (totalVersion - 1); i++ {
+		input := &svcsdk.DeleteLayerVersionInput{
+			LayerName:     r.ko.Spec.LayerName,
+			VersionNumber: version_list[i].Version,
+		}
+		// Delete API call to delete the versions one by one
+		// // log := ackrtlog.FromContext(ctx)
+		logMessage := fmt.Sprintf("Deleting version %v of %v", *input.VersionNumber, *input.LayerName)
+		log.Info(logMessage)
+
+		_, err = rm.sdkapi.DeleteLayerVersionWithContext(ctx, input)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}

pkg/resource/layer_version/sdk.go

@@ -0,0 +1,5 @@
+err = customPreDelete(r,rm,ctx)
+if err != nil{
+    return nil, err
+}
+    

templates/hooks/layerversion/sdk_delete_pre_build_request.go.tpl

@@ -103,6 +103,10 @@ def test_smoke(self, lambda_client):
         _, deleted = k8s.delete_custom_resource(ref)
         assert deleted is True
 
+        layer_name = cr['spec']['layerName']
+        list = lambda_validator.list_layer_version(layer_name)
+        assert len(list) == 0
+
         time.sleep(DELETE_WAIT_AFTER_SECONDS)
 
         # Check layer version doesn't exist