import boto3
import json
import time
import os
 
latest_version = os.environ["ec2_launch_v2"]
ec2_launch_sns_topic = os.environ["ec2_launch_sns_topic"]
 
def get_ec2_instances(instance_id, command_id):
    command_list = []
    final_res_list = []
    result_list = []
    print("entered into get ec2 instances")
    ssm_client = boto3.client('ssm')
    response = ssm_client.describe_instance_information()
    print("complete ec2 instance response", response)
    instances = response.get('InstanceInformationList', [])
    platformtype = [item.get('PlatformType') for item in instances]
    print("platformtype", platformtype)
    instance_ids = [item.get('InstanceId') for item in instances]
    print("Instance id's list", instance_ids)
    print("list of instances", instances)
    for item in instances:
        if 'PlatformType' in item:
            print("Platform Type:", item['PlatformType'])
            print("Instance id's list", instance_ids)
            result_item = {'InstanceId': item['InstanceId'], 'PlatformType': item['PlatformType']}
            result_list.append(result_item)
            print("ec2 instance id and platform type", result_list)
        else:
            print("No Platform Type found for this item")
    print("ec2 instance id and platform type", result_list)
 
    windows_instances = [instance for instance in result_list if instance['PlatformType'] == 'Windows']
 
    print("windows instances", windows_instances)
    windows_instance_dict = {}
    if not windows_instances:
        print("No Windows instances found.")
    else:
        print("Windows Instances:")
        print(windows_instances)
        windows_instance_ids = [instance['InstanceId'] for instance in windows_instances]
        print("windows instance id's", windows_instance_ids)
 
        for instance_id in windows_instance_ids:
            print("windows instances in loop", instance_id)
            document_name = 'Abrigo-EC2LaunchV2Info'
            response = run_ssm_command(instance_id, document_name)
            command_list.append(response)
            time.sleep(10)
            final_res = get_command_invocation(instance_id, response)
            # final_res_list.append(final_res)
            windows_instance_dict[instance_id] = final_res
            print("final_res EC2LaunchV2Info installation", final_res)
        print("&&&&&&&", command_list)
        print("windows instance id's and command id's", windows_instance_dict)
 
        # AWS SNS configurations
    sns_topic_arn = os.environ["ec2_launch_sns_topic"]
    email_subject = 'EC2LaunchV2Info installation details'
    email_body = f"Please find the below details for EC2LaunchV2Info installation on EC2 instances.\n\n"
    if len(windows_instance_dict) > 0:
        email_body += f"Windows Instances List :\n{json.dumps(windows_instance_dict, indent=2)}\n\n"
 
    if len(windows_instance_dict) == 0:
        email_body += f"No Instances found with SSM enabled"
    print("email_body type", type(email_body))
 
    # Create an SNS client
    sns_client = boto3.client('sns', region_name='us-east-1')
 
    # Publish the email message to the SNS topic
    response = sns_client.publish(
        TopicArn=sns_topic_arn,
        Subject=email_subject,
        Message=email_body
    )
 
    print("Message published successfully. MessageId:", response['MessageId'])
    return windows_instance_dict, response['MessageId']
 
 
def run_ssm_command(instance_id, document_name, parameters=None):
    print("entered into function run ssm command")
    """
    Run an SSM command on a specific EC2 instance.
 
    :param instance_id: The ID of the target EC2 instance.
    :param document_name: The name of the SSM document to execute.
    :param parameters: Optional parameters for the SSM command.
    :return: The command execution ID.
    """
    ssm_client = boto3.client('ssm')
    print("entered into boto3 ssm command")
    # Build the command parameters
    command_params = {
        'DocumentName': document_name,
        'Targets': [
            {
                'Key': 'instanceids',
                'Values': [instance_id]
            }
        ]
    }
    print("entered into command_params ssm command")
    if parameters:
        print("inside parameters")
        command_params['Parameters'] = parameters
 
    # Execute the SSM command
    print("entering into ssm send command")
    response = ssm_client.send_command(**command_params)
    print("printing ssm-run-response", response)
 
    # Get the command execution ID
    command_id = response['Command']['CommandId']
 
    # instance_id = instance_id  # Replace with your instance ID
    document_name = 'ssmdocument'  # Replace with your SSM document name
    # parameters = {'param_name': ['param_value']}  # Replace with your SSM command parameters if any
 
    # command_id = run_ssm_command(instance_id, document_name)
 
    print(f"SSM Command executed successfully. Command ID: {command_id}")
    return command_id
 
 
def get_command_invocation(instance_id, command_id):
    print("entered into get command invocation function")
    print("command id", command_id)
    ssm_client = boto3.client('ssm')
 
    try:
        response = ssm_client.get_command_invocation(
            InstanceId=instance_id,
            CommandId=command_id
        )
 
        print("!!!!!!!!!!!!!!!!!!!!!!!!!!!!response!!!!!!!!!!!!!!!!!!!!!!!!", response)
        final_response = response
        ec2_instance_id = instance_id
        standard_output_content = final_response.get('StandardOutputContent', '')
 
        # Check conditions based on StandardOutputContent
        if latest_version in standard_output_content:
            compliance_type = 'COMPLIANT'
            print(f"EC2Launch V2 is installed and with version {latest_version} and compliance status: {compliance_type}")
            return f"EC2LaunchV2 is installed with version {latest_version} and  compliance status: {compliance_type}"
        else:
            compliance_type = 'NON_COMPLIANT'
            print(f"$$$EC2Launch V2 is not installed.  compliance status is: {compliance_type} ; current verison : {standard_output_content} Expected version: {latest_version} $$$")
            return f"EC2Launch V2 is not installed. compliance status is: {compliance_type} {standard_output_content} Expected version: {latest_version} "
 
    except Exception as e:
        print(f"Error retrieving command invocation details: {e}")
        return None
 
if __name__ == "__main__":
    get_ec2_instances(" ", " ")

aW1wb3J0IGJvdG8zCmltcG9ydCBqc29uCmltcG9ydCB0aW1lCmltcG9ydCBvcwoKbGF0ZXN0X3ZlcnNpb24gPSBvcy5lbnZpcm9uWyJlYzJfbGF1bmNoX3YyIl0KZWMyX2xhdW5jaF9zbnNfdG9waWMgPSBvcy5lbnZpcm9uWyJlYzJfbGF1bmNoX3Nuc190b3BpYyJdCgpkZWYgZ2V0X2VjMl9pbnN0YW5jZXMoaW5zdGFuY2VfaWQsIGNvbW1hbmRfaWQpOgogICAgY29tbWFuZF9saXN0ID0gW10KICAgIGZpbmFsX3Jlc19saXN0ID0gW10KICAgIHJlc3VsdF9saXN0ID0gW10KICAgIHByaW50KCJlbnRlcmVkIGludG8gZ2V0IGVjMiBpbnN0YW5jZXMiKQogICAgc3NtX2NsaWVudCA9IGJvdG8zLmNsaWVudCgnc3NtJykKICAgIHJlc3BvbnNlID0gc3NtX2NsaWVudC5kZXNjcmliZV9pbnN0YW5jZV9pbmZvcm1hdGlvbigpCiAgICBwcmludCgiY29tcGxldGUgZWMyIGluc3RhbmNlIHJlc3BvbnNlIiwgcmVzcG9uc2UpCiAgICBpbnN0YW5jZXMgPSByZXNwb25zZS5nZXQoJ0luc3RhbmNlSW5mb3JtYXRpb25MaXN0JywgW10pCiAgICBwbGF0Zm9ybXR5cGUgPSBbaXRlbS5nZXQoJ1BsYXRmb3JtVHlwZScpIGZvciBpdGVtIGluIGluc3RhbmNlc10KICAgIHByaW50KCJwbGF0Zm9ybXR5cGUiLCBwbGF0Zm9ybXR5cGUpCiAgICBpbnN0YW5jZV9pZHMgPSBbaXRlbS5nZXQoJ0luc3RhbmNlSWQnKSBmb3IgaXRlbSBpbiBpbnN0YW5jZXNdCiAgICBwcmludCgiSW5zdGFuY2UgaWQncyBsaXN0IiwgaW5zdGFuY2VfaWRzKQogICAgcHJpbnQoImxpc3Qgb2YgaW5zdGFuY2VzIiwgaW5zdGFuY2VzKQogICAgZm9yIGl0ZW0gaW4gaW5zdGFuY2VzOgogICAgICAgIGlmICdQbGF0Zm9ybVR5cGUnIGluIGl0ZW06CiAgICAgICAgICAgIHByaW50KCJQbGF0Zm9ybSBUeXBlOiIsIGl0ZW1bJ1BsYXRmb3JtVHlwZSddKQogICAgICAgICAgICBwcmludCgiSW5zdGFuY2UgaWQncyBsaXN0IiwgaW5zdGFuY2VfaWRzKQogICAgICAgICAgICByZXN1bHRfaXRlbSA9IHsnSW5zdGFuY2VJZCc6IGl0ZW1bJ0luc3RhbmNlSWQnXSwgJ1BsYXRmb3JtVHlwZSc6IGl0ZW1bJ1BsYXRmb3JtVHlwZSddfQogICAgICAgICAgICByZXN1bHRfbGlzdC5hcHBlbmQocmVzdWx0X2l0ZW0pCiAgICAgICAgICAgIHByaW50KCJlYzIgaW5zdGFuY2UgaWQgYW5kIHBsYXRmb3JtIHR5cGUiLCByZXN1bHRfbGlzdCkKICAgICAgICBlbHNlOgogICAgICAgICAgICBwcmludCgiTm8gUGxhdGZvcm0gVHlwZSBmb3VuZCBmb3IgdGhpcyBpdGVtIikKICAgIHByaW50KCJlYzIgaW5zdGFuY2UgaWQgYW5kIHBsYXRmb3JtIHR5cGUiLCByZXN1bHRfbGlzdCkKCiAgICB3aW5kb3dzX2luc3RhbmNlcyA9IFtpbnN0YW5jZSBmb3IgaW5zdGFuY2UgaW4gcmVzdWx0X2xpc3QgaWYgaW5zdGFuY2VbJ1BsYXRmb3JtVHlwZSddID09ICdXaW5kb3dzJ10KCiAgICBwcmludCgid2luZG93cyBpbnN0YW5jZXMiLCB3aW5kb3dzX2luc3RhbmNlcykKICAgIHdpbmRvd3NfaW5zdGFuY2VfZGljdCA9IHt9CiAgICBpZiBub3Qgd2luZG93c19pbnN0YW5jZXM6CiAgICAgICAgcHJpbnQoIk5vIFdpbmRvd3MgaW5zdGFuY2VzIGZvdW5kLiIpCiAgICBlbHNlOgogICAgICAgIHByaW50KCJXaW5kb3dzIEluc3RhbmNlczoiKQogICAgICAgIHByaW50KHdpbmRvd3NfaW5zdGFuY2VzKQogICAgICAgIHdpbmRvd3NfaW5zdGFuY2VfaWRzID0gW2luc3RhbmNlWydJbnN0YW5jZUlkJ10gZm9yIGluc3RhbmNlIGluIHdpbmRvd3NfaW5zdGFuY2VzXQogICAgICAgIHByaW50KCJ3aW5kb3dzIGluc3RhbmNlIGlkJ3MiLCB3aW5kb3dzX2luc3RhbmNlX2lkcykKCiAgICAgICAgZm9yIGluc3RhbmNlX2lkIGluIHdpbmRvd3NfaW5zdGFuY2VfaWRzOgogICAgICAgICAgICBwcmludCgid2luZG93cyBpbnN0YW5jZXMgaW4gbG9vcCIsIGluc3RhbmNlX2lkKQogICAgICAgICAgICBkb2N1bWVudF9uYW1lID0gJ0Ficmlnby1FQzJMYXVuY2hWMkluZm8nCiAgICAgICAgICAgIHJlc3BvbnNlID0gcnVuX3NzbV9jb21tYW5kKGluc3RhbmNlX2lkLCBkb2N1bWVudF9uYW1lKQogICAgICAgICAgICBjb21tYW5kX2xpc3QuYXBwZW5kKHJlc3BvbnNlKQogICAgICAgICAgICB0aW1lLnNsZWVwKDEwKQogICAgICAgICAgICBmaW5hbF9yZXMgPSBnZXRfY29tbWFuZF9pbnZvY2F0aW9uKGluc3RhbmNlX2lkLCByZXNwb25zZSkKICAgICAgICAgICAgIyBmaW5hbF9yZXNfbGlzdC5hcHBlbmQoZmluYWxfcmVzKQogICAgICAgICAgICB3aW5kb3dzX2luc3RhbmNlX2RpY3RbaW5zdGFuY2VfaWRdID0gZmluYWxfcmVzCiAgICAgICAgICAgIHByaW50KCJmaW5hbF9yZXMgRUMyTGF1bmNoVjJJbmZvIGluc3RhbGxhdGlvbiIsIGZpbmFsX3JlcykKICAgICAgICBwcmludCgiJiYmJiYmJiIsIGNvbW1hbmRfbGlzdCkKICAgICAgICBwcmludCgid2luZG93cyBpbnN0YW5jZSBpZCdzIGFuZCBjb21tYW5kIGlkJ3MiLCB3aW5kb3dzX2luc3RhbmNlX2RpY3QpCgogICAgICAgICMgQVdTIFNOUyBjb25maWd1cmF0aW9ucwogICAgc25zX3RvcGljX2FybiA9IG9zLmVudmlyb25bImVjMl9sYXVuY2hfc25zX3RvcGljIl0KICAgIGVtYWlsX3N1YmplY3QgPSAnRUMyTGF1bmNoVjJJbmZvIGluc3RhbGxhdGlvbiBkZXRhaWxzJwogICAgZW1haWxfYm9keSA9IGYiUGxlYXNlIGZpbmQgdGhlIGJlbG93IGRldGFpbHMgZm9yIEVDMkxhdW5jaFYySW5mbyBpbnN0YWxsYXRpb24gb24gRUMyIGluc3RhbmNlcy5cblxuIgogICAgaWYgbGVuKHdpbmRvd3NfaW5zdGFuY2VfZGljdCkgPiAwOgogICAgICAgIGVtYWlsX2JvZHkgKz0gZiJXaW5kb3dzIEluc3RhbmNlcyBMaXN0IDpcbntqc29uLmR1bXBzKHdpbmRvd3NfaW5zdGFuY2VfZGljdCwgaW5kZW50PTIpfVxuXG4iCgogICAgaWYgbGVuKHdpbmRvd3NfaW5zdGFuY2VfZGljdCkgPT0gMDoKICAgICAgICBlbWFpbF9ib2R5ICs9IGYiTm8gSW5zdGFuY2VzIGZvdW5kIHdpdGggU1NNIGVuYWJsZWQiCiAgICBwcmludCgiZW1haWxfYm9keSB0eXBlIiwgdHlwZShlbWFpbF9ib2R5KSkKCiAgICAjIENyZWF0ZSBhbiBTTlMgY2xpZW50CiAgICBzbnNfY2xpZW50ID0gYm90bzMuY2xpZW50KCdzbnMnLCByZWdpb25fbmFtZT0ndXMtZWFzdC0xJykKCiAgICAjIFB1Ymxpc2ggdGhlIGVtYWlsIG1lc3NhZ2UgdG8gdGhlIFNOUyB0b3BpYwogICAgcmVzcG9uc2UgPSBzbnNfY2xpZW50LnB1Ymxpc2goCiAgICAgICAgVG9waWNBcm49c25zX3RvcGljX2FybiwKICAgICAgICBTdWJqZWN0PWVtYWlsX3N1YmplY3QsCiAgICAgICAgTWVzc2FnZT1lbWFpbF9ib2R5CiAgICApCgogICAgcHJpbnQoIk1lc3NhZ2UgcHVibGlzaGVkIHN1Y2Nlc3NmdWxseS4gTWVzc2FnZUlkOiIsIHJlc3BvbnNlWydNZXNzYWdlSWQnXSkKICAgIHJldHVybiB3aW5kb3dzX2luc3RhbmNlX2RpY3QsIHJlc3BvbnNlWydNZXNzYWdlSWQnXQoKCmRlZiBydW5fc3NtX2NvbW1hbmQoaW5zdGFuY2VfaWQsIGRvY3VtZW50X25hbWUsIHBhcmFtZXRlcnM9Tm9uZSk6CiAgICBwcmludCgiZW50ZXJlZCBpbnRvIGZ1bmN0aW9uIHJ1biBzc20gY29tbWFuZCIpCiAgICAiIiIKICAgIFJ1biBhbiBTU00gY29tbWFuZCBvbiBhIHNwZWNpZmljIEVDMiBpbnN0YW5jZS4KCiAgICA6cGFyYW0gaW5zdGFuY2VfaWQ6IFRoZSBJRCBvZiB0aGUgdGFyZ2V0IEVDMiBpbnN0YW5jZS4KICAgIDpwYXJhbSBkb2N1bWVudF9uYW1lOiBUaGUgbmFtZSBvZiB0aGUgU1NNIGRvY3VtZW50IHRvIGV4ZWN1dGUuCiAgICA6cGFyYW0gcGFyYW1ldGVyczogT3B0aW9uYWwgcGFyYW1ldGVycyBmb3IgdGhlIFNTTSBjb21tYW5kLgogICAgOnJldHVybjogVGhlIGNvbW1hbmQgZXhlY3V0aW9uIElELgogICAgIiIiCiAgICBzc21fY2xpZW50ID0gYm90bzMuY2xpZW50KCdzc20nKQogICAgcHJpbnQoImVudGVyZWQgaW50byBib3RvMyBzc20gY29tbWFuZCIpCiAgICAjIEJ1aWxkIHRoZSBjb21tYW5kIHBhcmFtZXRlcnMKICAgIGNvbW1hbmRfcGFyYW1zID0gewogICAgICAgICdEb2N1bWVudE5hbWUnOiBkb2N1bWVudF9uYW1lLAogICAgICAgICdUYXJnZXRzJzogWwogICAgICAgICAgICB7CiAgICAgICAgICAgICAgICAnS2V5JzogJ2luc3RhbmNlaWRzJywKICAgICAgICAgICAgICAgICdWYWx1ZXMnOiBbaW5zdGFuY2VfaWRdCiAgICAgICAgICAgIH0KICAgICAgICBdCiAgICB9CiAgICBwcmludCgiZW50ZXJlZCBpbnRvIGNvbW1hbmRfcGFyYW1zIHNzbSBjb21tYW5kIikKICAgIGlmIHBhcmFtZXRlcnM6CiAgICAgICAgcHJpbnQoImluc2lkZSBwYXJhbWV0ZXJzIikKICAgICAgICBjb21tYW5kX3BhcmFtc1snUGFyYW1ldGVycyddID0gcGFyYW1ldGVycwoKICAgICMgRXhlY3V0ZSB0aGUgU1NNIGNvbW1hbmQKICAgIHByaW50KCJlbnRlcmluZyBpbnRvIHNzbSBzZW5kIGNvbW1hbmQiKQogICAgcmVzcG9uc2UgPSBzc21fY2xpZW50LnNlbmRfY29tbWFuZCgqKmNvbW1hbmRfcGFyYW1zKQogICAgcHJpbnQoInByaW50aW5nIHNzbS1ydW4tcmVzcG9uc2UiLCByZXNwb25zZSkKCiAgICAjIEdldCB0aGUgY29tbWFuZCBleGVjdXRpb24gSUQKICAgIGNvbW1hbmRfaWQgPSByZXNwb25zZVsnQ29tbWFuZCddWydDb21tYW5kSWQnXQoKICAgICMgaW5zdGFuY2VfaWQgPSBpbnN0YW5jZV9pZCAgIyBSZXBsYWNlIHdpdGggeW91ciBpbnN0YW5jZSBJRAogICAgZG9jdW1lbnRfbmFtZSA9ICdzc21kb2N1bWVudCcgICMgUmVwbGFjZSB3aXRoIHlvdXIgU1NNIGRvY3VtZW50IG5hbWUKICAgICMgcGFyYW1ldGVycyA9IHsncGFyYW1fbmFtZSc6IFsncGFyYW1fdmFsdWUnXX0gICMgUmVwbGFjZSB3aXRoIHlvdXIgU1NNIGNvbW1hbmQgcGFyYW1ldGVycyBpZiBhbnkKCiAgICAjIGNvbW1hbmRfaWQgPSBydW5fc3NtX2NvbW1hbmQoaW5zdGFuY2VfaWQsIGRvY3VtZW50X25hbWUpCgogICAgcHJpbnQoZiJTU00gQ29tbWFuZCBleGVjdXRlZCBzdWNjZXNzZnVsbHkuIENvbW1hbmQgSUQ6IHtjb21tYW5kX2lkfSIpCiAgICByZXR1cm4gY29tbWFuZF9pZAoKCmRlZiBnZXRfY29tbWFuZF9pbnZvY2F0aW9uKGluc3RhbmNlX2lkLCBjb21tYW5kX2lkKToKICAgIHByaW50KCJlbnRlcmVkIGludG8gZ2V0IGNvbW1hbmQgaW52b2NhdGlvbiBmdW5jdGlvbiIpCiAgICBwcmludCgiY29tbWFuZCBpZCIsIGNvbW1hbmRfaWQpCiAgICBzc21fY2xpZW50ID0gYm90bzMuY2xpZW50KCdzc20nKQoKICAgIHRyeToKICAgICAgICByZXNwb25zZSA9IHNzbV9jbGllbnQuZ2V0X2NvbW1hbmRfaW52b2NhdGlvbigKICAgICAgICAgICAgSW5zdGFuY2VJZD1pbnN0YW5jZV9pZCwKICAgICAgICAgICAgQ29tbWFuZElkPWNvbW1hbmRfaWQKICAgICAgICApCgogICAgICAgIHByaW50KCIhISEhISEhISEhISEhISEhISEhISEhISEhISEhcmVzcG9uc2UhISEhISEhISEhISEhISEhISEhISEhISEiLCByZXNwb25zZSkKICAgICAgICBmaW5hbF9yZXNwb25zZSA9IHJlc3BvbnNlCiAgICAgICAgZWMyX2luc3RhbmNlX2lkID0gaW5zdGFuY2VfaWQKICAgICAgICBzdGFuZGFyZF9vdXRwdXRfY29udGVudCA9IGZpbmFsX3Jlc3BvbnNlLmdldCgnU3RhbmRhcmRPdXRwdXRDb250ZW50JywgJycpCgogICAgICAgICMgQ2hlY2sgY29uZGl0aW9ucyBiYXNlZCBvbiBTdGFuZGFyZE91dHB1dENvbnRlbnQKICAgICAgICBpZiBsYXRlc3RfdmVyc2lvbiBpbiBzdGFuZGFyZF9vdXRwdXRfY29udGVudDoKICAgICAgICAgICAgY29tcGxpYW5jZV90eXBlID0gJ0NPTVBMSUFOVCcKICAgICAgICAgICAgcHJpbnQoZiJFQzJMYXVuY2ggVjIgaXMgaW5zdGFsbGVkIGFuZCB3aXRoIHZlcnNpb24ge2xhdGVzdF92ZXJzaW9ufSBhbmQgY29tcGxpYW5jZSBzdGF0dXM6IHtjb21wbGlhbmNlX3R5cGV9IikKICAgICAgICAgICAgcmV0dXJuIGYiRUMyTGF1bmNoVjIgaXMgaW5zdGFsbGVkIHdpdGggdmVyc2lvbiB7bGF0ZXN0X3ZlcnNpb259IGFuZCAgY29tcGxpYW5jZSBzdGF0dXM6IHtjb21wbGlhbmNlX3R5cGV9IgogICAgICAgIGVsc2U6CiAgICAgICAgICAgIGNvbXBsaWFuY2VfdHlwZSA9ICdOT05fQ09NUExJQU5UJwogICAgICAgICAgICBwcmludChmIiQkJEVDMkxhdW5jaCBWMiBpcyBub3QgaW5zdGFsbGVkLiAgY29tcGxpYW5jZSBzdGF0dXMgaXM6IHtjb21wbGlhbmNlX3R5cGV9IDsgY3VycmVudCB2ZXJpc29uIDoge3N0YW5kYXJkX291dHB1dF9jb250ZW50fSBFeHBlY3RlZCB2ZXJzaW9uOiB7bGF0ZXN0X3ZlcnNpb259ICQkJCIpCiAgICAgICAgICAgIHJldHVybiBmIkVDMkxhdW5jaCBWMiBpcyBub3QgaW5zdGFsbGVkLiBjb21wbGlhbmNlIHN0YXR1cyBpczoge2NvbXBsaWFuY2VfdHlwZX0ge3N0YW5kYXJkX291dHB1dF9jb250ZW50fSBFeHBlY3RlZCB2ZXJzaW9uOiB7bGF0ZXN0X3ZlcnNpb259ICIKCiAgICBleGNlcHQgRXhjZXB0aW9uIGFzIGU6CiAgICAgICAgcHJpbnQoZiJFcnJvciByZXRyaWV2aW5nIGNvbW1hbmQgaW52b2NhdGlvbiBkZXRhaWxzOiB7ZX0iKQogICAgICAgIHJldHVybiBOb25lCgppZiBfX25hbWVfXyA9PSAiX19tYWluX18iOgogICAgZ2V0X2VjMl9pbnN0YW5jZXMoIiAiLCAiICIp