Lambda 関数 + SES を利用して EC2 インスタンスのスクリーンショットをメールアドレスに送信してみた

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Effect": "Allow",
            "Principal": {
                "Service": "lambda.amazonaws.com"
            },
            "Action": "sts:AssumeRole"
        }
    ]
}

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Effect": "Allow",
            "Action": "ec2:GetConsoleScreenshot",
            "Resource": "arn:aws:ec2:*:*:instance/*"
        },
        {
            "Effect": "Allow",
            "Action": [
                "ses:SendEmail",
                "ses:SendRawEmail"
            ],
            "Resource": "*"
        },
        {
            "Effect": "Allow",
            "Action": [
                "logs:CreateLogGroup",
                "logs:CreateLogStream",
                "logs:PutLogEvents"
            ],
            "Resource": "arn:aws:logs:*:*:*"
        }
    ]
}

import boto3
import base64
import os
import email.mime.multipart
import email.mime.text
import email.mime.application
from datetime import datetime

def lambda_handler(event, context):
    # 環境変数から設定を取得
    instance_id = os.environ.get('INSTANCE_ID')
    recipient_email = os.environ.get('RECIPIENT_EMAIL')
    sender_email = os.environ.get('SENDER_EMAIL')
    aws_region = os.environ.get('AWS_REGION')

    # イベントからパラメータを上書き（オプション）
    if 'instance_id' in event:
        instance_id = event['instance_id']
    if 'email' in event:
        recipient_email = event['email']

    # 必須パラメータの検証
    if not instance_id:
        return {
            'statusCode': 400,
            'body': 'Missing required environment variable: INSTANCE_ID'
        }

    if not recipient_email:
        return {
            'statusCode': 400,
            'body': 'Missing required environment variable: RECIPIENT_EMAIL'
        }

    if not sender_email:
        return {
            'statusCode': 400,
            'body': 'Missing required environment variable: SENDER_EMAIL'
        }

    try:
        # EC2クライアントを作成
        ec2_client = boto3.client('ec2')

        # スクリーンショットを取得
        response = ec2_client.get_console_screenshot(
            InstanceId=instance_id,
            WakeUp=True  # コンソールをウェイクアップ
        )

        # Base64エンコードされた画像をデコード
        image_data = base64.b64decode(response['ImageData'])

        # 現在の日時を取得してファイル名に使用
        timestamp = datetime.now().strftime("%Y%m%d-%H%M%S")
        filename = f"ec2-screenshot-{instance_id}-{timestamp}.png"

        # SESクライアントを作成
        ses_client = boto3.client('ses', region_name=aws_region)

        # メールの作成
        msg = email.mime.multipart.MIMEMultipart()
        msg['Subject'] = f'EC2 Instance {instance_id} Screenshot - {timestamp}'
        msg['From'] = sender_email
        msg['To'] = recipient_email

        # メール本文
        body = email.mime.text.MIMEText(
            f'EC2インスタンス {instance_id} のスクリーンショットです。\n'
            f'取得日時: {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}\n\n'
            '自動送信メールです。'
        )
        msg.attach(body)

        # スクリーンショットを添付
        attachment = email.mime.application.MIMEApplication(image_data)
        attachment.add_header('Content-Disposition', 'attachment', filename=filename)
        msg.attach(attachment)

        # メールを送信
        ses_client.send_raw_email(
            Source=sender_email,
            Destinations=[recipient_email],
            RawMessage={'Data': msg.as_string()}
        )

        print(f"スクリーンショットを {recipient_email} に送信しました")

        return {
            'statusCode': 200,
            'body': f'Screenshot of instance {instance_id} sent to {recipient_email}'
        }

    except Exception as e:
        print(f"エラーが発生しました: {str(e)}")
        return {
            'statusCode': 500,
            'body': f'Error: {str(e)}'
        }