import os
import datetime
import boto3
import io
import re
import csv
import logging
from abc import *
from zoneinfo import ZoneInfo
import traceback

# 環境変数
CONFIG_BUCKET_NAME = os.environ["CONFIG_BUCKET_NAME"]
RECEIVE_MONTHLY_FILE_NAME_LIST_PATH = os.environ["RECEIVE_MONTHLY_FILE_NAME_LIST_PATH"]
NOTICE_MAIL_TITLE_TEMPLATE_PATH = os.environ["NOTICE_MAIL_TITLE_TEMPLATE_PATH"]
NOTICE_MAIL_BODY_TEMPLATE_PATH = os.environ["NOTICE_MAIL_BODY_TEMPLATE_PATH"]
MBJ_SAP_NOTICE_TOPIC = os.environ["MBJ_SAP_NOTICE_TOPIC"]
MAIL_BODY_REPLACE_SYMBOL = os.environ["MAIL_BODY_REPLACE_SYMBOL"]
NDS_NOTICE_TOPIC = os.environ["NDS_NOTICE_TOPIC"]
NDS_NOTICE_TITLE = os.environ["NDS_NOTICE_TITLE"]
LOG_LEVEL = os.environ["LOG_LEVEL"]

# 定数
INDEX_REGEX = 0
INDEX_DATA_NAME = 1

# メール本文に出力する不足ファイル名一覧のインデント
MAIL_INDENT = '　　'

# 変数
s3_client = boto3.client('s3')
s3_resource = boto3.resource('s3')
sns_client = boto3.client('sns')

# logger設定
logger = logging.getLogger()
def custome_time(*arg):
    return datetime.datetime.now(ZoneInfo("Asia/Tokyo")).timetuple()
formatter = logging.Formatter(
    '[%(levelname)s]\t%(asctime)s\t%(message)s\n',
    '%Y-%m-%d %H:%M:%S'
)
formatter.converter = custome_time
for handler in logger.handlers:
    handler.setFormatter(formatter)

level = logging.getLevelName(LOG_LEVEL)
if not isinstance(level, int):
    level = logging.INFO
logger.setLevel(level)


def lambda_handler(event, context):
    try:
        # ① 処理開始ログを出力する
        logger.info('I-01-01 処理開始　SAP_finデータ月次I/F受領通知処理')
        execute_date = datetime.date.today().strftime('%Y/%m/%d')
        logger.info(f'I-01-02 処理稼働日：{execute_date}')
        mail_msg = ''

        # ② 処理開始時に受け取ったイベント情報の以下内容をログに出力しメモリに保持する
        logger.info('I-02-01 イベント情報出力処理')
        s3_event = event["Records"][0]["s3"]
        event_bucket_name = s3_event["bucket"]["name"]
        event_file_path = s3_event["object"]["key"]
        event_file_name = os.path.basename(event_file_path)
        logger.info(f'I-02-02 バケット名：{event_bucket_name}')
        logger.info(f'I-02-03 ファイルパス：{event_file_path}')

        # ③ 設定ファイル[SAP_finI/Fファイルネーム設定ファイル（月次）]を読み込む
        try:
            logger.info(f'I-03-01 月次I/Fファイルネーム設定ファイル読込　読込元：{CONFIG_BUCKET_NAME}/{RECEIVE_MONTHLY_FILE_NAME_LIST_PATH}')
            receive_monthly_file_name_obj = s3_resource.Object(CONFIG_BUCKET_NAME, RECEIVE_MONTHLY_FILE_NAME_LIST_PATH)
            receive_monthly_file_name_response = receive_monthly_file_name_obj.get()
            logger.info('I-03-02 月次I/Fファイルネーム設定ファイルを読み込みました')
        except Exception as e:
            logger.error(f'E-03-01 月次I/Fファイルネーム設定ファイルの読み込みに失敗しました　エラー内容：{e}')
            raise FileReadException('E-03-01', e)

        # ④ 月次I/Fファイル受領通知処理を行う
        logger.info('I-04-01 月次I/Fファイル受領通知処理開始')
        logger.info(f'I-04-02 受領したファイル名：{event_file_name}')
        logger.info('I-04-03 受領したファイル名と月次I/Fファイルネーム設定ファイルの突き合わせを開始します')
        file_exists = False
        receive_monthly_file_name_body = io.TextIOWrapper(io.BytesIO(receive_monthly_file_name_response["Body"].read()), encoding='utf-8')
        for row in csv.reader(receive_monthly_file_name_body, delimiter='\t'):
            match_result = re.fullmatch(row[INDEX_REGEX], event_file_name)
            if match_result is not None:
                file_exists = True
                break
        if file_exists == True:
            logger.info(f'I-04-04 月次I/Fを受領しました　ファイル名：{row[INDEX_DATA_NAME]}　{event_file_name}')
            mail_msg += f'{MAIL_INDENT}{row[INDEX_DATA_NAME]}　{event_file_name}\n'
        else:
            logger.info('I-04-05 受領したファイルは月次I/Fではありませんでした')

        # ⑤ 「①」でメモリ保持しているメール挿入用文言に出力内容が存在するか確認する
        logger.info('I-05-01 メール送信処理開始')
        if len(mail_msg) > 0:
            # 1.存在した場合
            logger.info('I-05-02 月次I/Fファイルを受領したため、メール送信処理を開始します')
            try:
                logger.info(f'I-05-03 通知メール（タイトル）テンプレートファイル読込　読込元：{CONFIG_BUCKET_NAME}/{NOTICE_MAIL_TITLE_TEMPLATE_PATH}')
                mail_title_obj = s3_client.get_object(Bucket=CONFIG_BUCKET_NAME, Key=NOTICE_MAIL_TITLE_TEMPLATE_PATH)
                mail_title = mail_title_obj['Body'].read().decode('utf-8')
                logger.info('I-05-04 通知メール（タイトル）テンプレートファイルを読み込みました')
            except Exception as e:
                logger.error(f'E-05-01 通知メール（タイトル）テンプレートファイルの読み込みに失敗しました　エラー内容：{e}')
                raise FileReadException('E-05-01', e)

            try:
                logger.info(f'I-05-05 通知メール（本文）テンプレートファイル読込　読込元：{CONFIG_BUCKET_NAME}/{NOTICE_MAIL_BODY_TEMPLATE_PATH}')
                mail_body_obj = s3_client.get_object(Bucket=CONFIG_BUCKET_NAME, Key=NOTICE_MAIL_BODY_TEMPLATE_PATH)
                mail_body_response = mail_body_obj['Body'].read().decode('utf-8')
                mail_body = mail_body_response.replace(MAIL_BODY_REPLACE_SYMBOL, mail_msg)
                logger.info('I-05-06 通知メール（本文）テンプレートファイルを読み込みました')
            except Exception as e:
                logger.error(f'E-05-02 通知メール（本文）テンプレートファイルの読み込みに失敗しました　エラー内容：{e}')
                raise FileReadException('E-05-02', e)

            logger.info(f'I-05-07 メール送信指示をします　送信先トピック：{MBJ_SAP_NOTICE_TOPIC}')
            params = {
                'TopicArn': MBJ_SAP_NOTICE_TOPIC,
                'Subject': mail_title.rstrip('\n'),
                'Message': mail_body
            }
            sns_client.publish(**params)
            logger.info('I-05-08 メール送信指示をしました')
        else:
            # 2.存在しない場合
            logger.info('I-05-09 受領したファイルは月次I/Fファイルではないため、メール送信処理をスキップします')

        # ⑥ 処理終了ログを出力する
        logger.info('I-06-01 処理終了　SAP_finデータ月次I/F受領通知処理')
    except CustomException as e:
        traceback.print_exc()
        error_notice(e.id, e.arg)
    except Exception as e:
        logger.error(f'E-99 想定外のエラーが発生しました　エラー内容：{e}')
        traceback.print_exc()
        error_notice('E-99', e)
    return


# 保守要員チーム通知
def error_notice(error_log_id, exception) -> None:
    try:
        error_msg = f'{error_log_id} のエラーが発生しました。ご確認ください\n詳細：{exception}'
        params = {
            'TopicArn': NDS_NOTICE_TOPIC,
            'Subject': NDS_NOTICE_TITLE,
            'Message': error_msg
        }
        sns_client.publish(**params)
        logger.error(f'E-ERR-01 処理異常通知の送信指示をしました　通知先トピック：{NDS_NOTICE_TOPIC}')
    except Exception as e:
        logger.error(f'E-98 処理異常通知の送信指示に失敗しました　エラー内容：{e}')
        traceback.print_exc()
    return


# カスタムExceptionクラス
class CustomException(Exception, metaclass=ABCMeta):
    def __init__(self, id, arg):
        self.arg = arg
        self.id = id


class FileReadException(CustomException):
    pass