import json
import boto3

sns_client = boto3.client("sns", region_name="ap-northeast-1")
TOPIC_ARN = "arn:aws:sns:ap-northeast-1:123456789012:factory-temp-alert"
THRESHOLD = 80.0

def lambda_handler(event, context):
    device_id = event.get("device_id", "unknown")
    temperature = event.get("temperature")
    source_topic = event.get("source_topic", "")

    if temperature is None:
        return {"statusCode": 400, "body": "缺少temperature欄位"}

    if temperature > THRESHOLD:
        message = (
            f"[ATLANTIS雲端監控告警]\n"
            f"裝置：{device_id}\n"
            f"目前溫度：{temperature}°C（警戒值：{THRESHOLD}°C）\n"
            f"來源主題：{source_topic}"
        )

        sns_client.publish(
            TopicArn=TOPIC_ARN,
            Subject="工廠溫度異常告警",
            Message=message
        )
        print(f"已發送SNS告警：裝置{device_id}溫度{temperature}°C")
    else:
        print(f"裝置 {device_id} 溫度正常：{temperature}°C")

    return {"statusCode": 200, "body": json.dumps("處理完成")}

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Effect": "Allow",
            "Action": "sns:Publish",
            "Resource": "arn:aws:sns:ap-northeast-1:123456789012:factory-temp-alert"
        }
    ]
}