// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: MIT-0

package aws.proserve.bcs.dr.lambda.network;

import aws.proserve.bcs.dr.lambda.annotation.Source;
import aws.proserve.bcs.dr.lambda.annotation.Target;
import com.amazonaws.services.ec2.AmazonEC2;
import com.amazonaws.services.ec2.model.DeleteRouteRequest;
import com.amazonaws.services.ec2.model.RouteTable;
import com.amazonaws.services.lambda.runtime.Context;

import javax.inject.Inject;
import javax.inject.Singleton;

public class DeletePeerRoute extends PeerRouteBase {

    @Override
    public void handleRequest(PeerRouteBase.Request request, Context context) {
        final var credential = VpcComponent.getCredentialByProject(request.getProjectId());
        VpcComponent.build(request.getSourceRegion(), request.getTargetRegion(), credential)
                .deletePeerRoute()
                .handle(request);
    }

    @Singleton
    static class Worker extends PeerRouteBase.Worker {

        @Inject
        Worker(@Source AmazonEC2 sourceEc2,
               @Target AmazonEC2 targetEc2) {
            super(sourceEc2, targetEc2, DeletePeerRoute.Worker::deleteRoute);
        }

        private static void deleteRoute(AmazonEC2 ec2, RouteTable table, String cidr, String peerId) {
            final var tableId = table.getRouteTableId();

            log.info("Check table {} for peer ID {} with CIDR {}", tableId, peerId, cidr);

            if (table.getRoutes().stream()
                    .anyMatch(r -> peerId.equals(r.getVpcPeeringConnectionId()))) {
                log.info("Delete route to peer VPC on route table {}.", tableId);

                ec2.deleteRoute(new DeleteRouteRequest()
                        .withRouteTableId(tableId)
                        .withDestinationCidrBlock(cidr));
            }
        }
    }
}