Source code for runway.cfngin.hooks.staticsite.cleanup

"""Replicated Lambda Function cleanup warning."""
from __future__ import annotations

import logging
from typing import TYPE_CHECKING, Any, List

from ..base import HookArgsBaseModel

    from mypy_boto3_cloudformation.type_defs import OutputTypeDef

    from ....context import CfnginContext

LOGGER = logging.getLogger(__name__)

[docs]class HookArgs(HookArgsBaseModel): """Hook arguments.""" stack_relative_name: str """Name of the CloudFormation Stack as defined in the config file (no namespace)."""
[docs]def get_replicated_function_names(outputs: List[OutputTypeDef]) -> List[str]: """Extract replicated function names from CFN outputs.""" function_names: List[str] = [] for i in REPLICATED_FUNCTION_OUTPUTS: function_arn = next( ( output.get("OutputValue") for output in outputs if output.get("OutputKey") == i ), None, ) if function_arn: function_names.append(function_arn.split(":")[-1]) return function_names
[docs]def warn(context: CfnginContext, *__args: Any, **kwargs: Any) -> bool: """Notify the user of Lambda functions to delete. Arguments parsed by :class:`~runway.cfngin.hooks.staticsite.cleanup.HookArgs`. Args: context: The context instance. """ args = HookArgs.parse_obj(kwargs) cfn_client = context.get_session().client("cloudformation") try: describe_response = cfn_client.describe_stacks( StackName=context.namespace + context.namespace_delimiter + args.stack_relative_name ) stack = next( x for x in describe_response.get("Stacks", []) if ( x.get("StackStatus") and x.get("StackStatus") not in STACK_STATUSES_TO_IGNORE ) ) functions = get_replicated_function_names(stack["Outputs"]) if functions: cmd = ( "aws lambda delete-function --function-name $x " f"--region {context.env.aws_region}" ) LOGGER.warning( "About to delete the Static Site stack that contains " "replicated Lambda functions. These functions cannot " "be deleted until AWS automatically deletes their " "replicas. After some time has passed (a day is " "typically sufficient), they can be manually deleted. " "E.g.:" ) LOGGER.warning("On macOS/Linux:") LOGGER.warning("for x in %s; do %s; done", (" ").join(functions), cmd) LOGGER.warning("On Windows:") LOGGER.warning('Foreach ($x in "%s") { %s }', ('","').join(functions), cmd) except Exception: # pylint: disable=broad-except # There's no harm in continuing on in the event of an error # Orphaned functions have no cost pass return True