{
	"Version": "2012-10-17",
	"Statement": [
		{
			"Effect": "Allow",
			"Action": "ce:GetCostAndUsage",
			"Resource": "*"
		}
	]
}

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

import os
import json
from datetime import datetime, timedelta
import boto3
from linebot import LineBotApi, WebhookHandler
from linebot.exceptions import LineBotApiError
from linebot.models import TextSendMessage

# 環境変数からLINE Messaging APIのアクセストークンとシークレットを取得
LINE_ACCESS_TOKEN = os.getenv('YOUR_CHANNEL_ACCESS_TOKEN')
LINE_CHANNEL_SECRET = os.getenv('YOUR_CHANNEL_SECRET')

# アクセストークンとシークレットが環境変数に設定されていない場合のエラーハンドリング
if not LINE_ACCESS_TOKEN or not LINE_CHANNEL_SECRET:
    raise ValueError("LINEアクセストークンまたはシークレットが環境変数に設定されていません。")

line_bot_api = LineBotApi(LINE_ACCESS_TOKEN)
handler = WebhookHandler(LINE_CHANNEL_SECRET)

def get_billing_info(message):
    """
    AWS料金情報を取得する。
    - メッセージに応じて期間や表示形式を切り替える。
    """
    client = boto3.client('ce', region_name='us-east-1')
    today = datetime.today()

    # 料金期間と出力形式を設定
    if message == "今月":
        start_date = today.replace(day=1).strftime('%Y-%m-%d')
        end_date = (today + timedelta(days=1)).strftime('%Y-%m-%d')
        detail = False
    elif message == "先月":
        first_day_last_month = (today.replace(day=1) - timedelta(days=1)).replace(day=1)
        start_date = first_day_last_month.strftime('%Y-%m-%d')
        end_date = today.replace(day=1).strftime('%Y-%m-%d')
        detail = False
    elif message == "今月の詳細":
        start_date = today.replace(day=1).strftime('%Y-%m-%d')
        end_date = (today + timedelta(days=1)).strftime('%Y-%m-%d')
        detail = True
    elif message == "先月の詳細":
        first_day_last_month = (today.replace(day=1) - timedelta(days=1)).replace(day=1)
        start_date = first_day_last_month.strftime('%Y-%m-%d')
        end_date = today.replace(day=1).strftime('%Y-%m-%d')
        detail = True
    else:
        return "サポートされていないコマンドです。"

    try:
        # AWSの料金情報を取得
        response = client.get_cost_and_usage(
            TimePeriod={'Start': start_date, 'End': end_date},
            Granularity='MONTHLY',
            Metrics=['AmortizedCost'],
            GroupBy=[{'Type': 'DIMENSION', 'Key': 'SERVICE'}]
        )

        if detail:
            # サービスごとの詳細情報を整形
            details = [
                f"{group['Keys'][0]}: {group['Metrics']['AmortizedCost']['Amount']} USD"
                for group in response['ResultsByTime'][0]['Groups']
            ]
            return f"{start_date}から{end_date}までのサービスごとの料金詳細:\n" + "\n".join(details)
        else:
            # 合計料金を取得
            total_billing = response['ResultsByTime'][0]['Total']['AmortizedCost']['Amount']
            return f"{start_date}から{end_date}までの総料金: {total_billing} USD"

    except boto3.exceptions.Boto3Error as e:
        # AWSサービス関連のエラー処理
        print(f"AWS Error: {str(e)}")
        return "料金情報の取得に失敗しました。AWSサービスに接続できませんでした。"
    except Exception as e:
        # その他の予期しないエラー処理
        print(f"Unexpected Error: {str(e)}")
        return "料金情報の取得に失敗しました。予期しないエラーが発生しました。"

def lambda_handler(event, context):
    """
    Lambda関数のエントリーポイント。
    - LINEからのWebhookイベントを処理し、返信を送信。
    """
    try:
        # イベントデータのログ出力（デバッグ用）
        print("Event received:", event)

        # イベントのボディをパース
        body = json.loads(event['body'])
        message_text = body['events'][0]['message']['text']
        reply_token = body['events'][0]['replyToken']

        # 無効なリプライトークンの場合、ログ出力して終了
        if reply_token == "dummyReplyToken":
            print("Dummy reply token detected. Skipping reply.")
            return {'statusCode': 200, 'body': 'Dummy token, no reply sent'}

        # メッセージに基づいてAWS料金情報を取得
        billing_info = get_billing_info(message_text)

        # LINE Messaging APIで返信
        line_bot_api.reply_message(
            reply_token,
            TextSendMessage(text=billing_info)
        )
        return {'statusCode': 200, 'body': 'Success'}

    except LineBotApiError as e:
        # LINE Messaging APIのエラー処理
        print(f"LineBotApiError: {e.status_code}, {e.error.message}")
        return {'statusCode': 400, 'body': 'LineBotApiError occurred'}

    except Exception as e:
        # その他の例外処理
        print(f"Unexpected Error: {str(e)}")
        return {'statusCode': 500, 'body': 'Internal Server Error'}