using ServerCommon;
using ServerCore; using ServerBase;

namespace GameServer
{
    public class DBQMailReadAll : QueryExecutorBase
    {
        private string m_combination_key_for_pk = string.Empty;

        private string m_sent_pk = string.Empty;
        private string m_received_pk = string.Empty;

        private readonly Dictionary<MailType, List<MailDoc>> m_to_read_mail_docs = new();

        public DBQMailReadAll(string combinationKeyForPK)
            : base(typeof(DBQMailReadAll).Name)
        {
            m_combination_key_for_pk = combinationKeyForPK;

            m_to_read_mail_docs.Add(MailType.ReceivedMail, new List<MailDoc>());
            m_to_read_mail_docs.Add(MailType.SentMail, new List<MailDoc>());
        }

        //=====================================================================================
        // DB 쿼리 직전에 준비해야 할 로직들을 작성한다.
        //=====================================================================================
        public override Task<Result> onPrepareQuery()
        {
            var result = new Result();
            var err_msg = string.Empty;

            var owner = getOwner();
            NullReferenceCheckHelper.throwIfNull(owner, () => $"owner is null !!!");

            var sent_doc = new SentMailDoc();
            sent_doc.setCombinationKeyForPK(m_combination_key_for_pk);

            var error_code = sent_doc.onApplyPKSK();
            if (error_code.isFail())
            {
                err_msg = $"Failed to onApplyPKSK() !!! : {error_code.toBasicString()} - {toBasicString()}, {owner.toBasicString()}";
                result.setFail(error_code, err_msg);
                Log.getLogger().error(result.toBasicString());
                return Task.FromResult(result);
            }

            m_sent_pk = sent_doc.getPK();
            var received_doc = new ReceivedMailDoc();
            received_doc.setCombinationKeyForPK(m_combination_key_for_pk);

            error_code = received_doc.onApplyPKSK();
            if (error_code.isFail())
            {
                err_msg = $"Failed to onApplyPKSK() !!! : {error_code.toBasicString()} - {toBasicString()}, {owner.toBasicString()}";
                result.setFail(error_code, err_msg);
                Log.getLogger().error(result.toBasicString());
                return Task.FromResult(result);
            }

            m_received_pk = received_doc.getPK();
            return Task.FromResult(result);
        }

        //=====================================================================================
        // onPrepareQuery()를 성공할 경우 호출된다.
        //=====================================================================================
        public override async Task<Result> onQuery()
        {
            var result = new Result();
            var err_msg = string.Empty;

            var owner = getOwner();
            NullReferenceCheckHelper.throwIfNull(owner, () => $"owner is null !!!");

            var query_batch = getQueryBatch();
            NullReferenceCheckHelper.throwIfNull(query_batch, () => $"query_batch is null !!!");

            var db_connector = query_batch.getDynamoDbConnector();
            var sent_query_config = db_connector.makeQueryConfigForReadByPKOnly(m_sent_pk);

            var mail_action = owner.getEntityAction<MailAction>();
            NullReferenceCheckHelper.throwIfNull(mail_action, () => $"mail_action is null !!!");

            (result, var sent_read_docs) = await db_connector.simpleQueryDocTypesWithQueryOperationConfig<SentMailDoc>(sent_query_config, eventTid: query_batch.getTransId());
            if (result.isFail())
            {
                return result;
            }

            var received_query_config = db_connector.makeQueryConfigForReadByPKOnly(m_received_pk);

            (result, var received_read_docs) = await db_connector.simpleQueryDocTypesWithQueryOperationConfig<ReceivedMailDoc>(received_query_config, eventTid: query_batch.getTransId());
            if (result.isFail())
            {
                return result;
            }

            result = await mail_action.AddMailFromDocs(sent_read_docs);
            if (result.isFail()) return result;

            if (m_to_read_mail_docs.TryGetValue(MailType.SentMail, out var sentMailList))
            {
                sentMailList.AddRange(sent_read_docs);
            }

            result = await mail_action.AddMailFromDocs(received_read_docs);
            if (result.isFail()) return result;

            if (m_to_read_mail_docs.TryGetValue(MailType.ReceivedMail, out var receivedMailList))
            {
                receivedMailList.AddRange(received_read_docs);
            }

            return await Task.FromResult(result);
        }

        public Dictionary<MailType, List<MailDoc>> getToReadItemDocs() => m_to_read_mail_docs;

        //=====================================================================================
        // DB 쿼리를 성공하고, doFnCommit()가 QueryResultType.NotCalledQueryFunc를 반환할 경우 호출된다.
        //=====================================================================================
        public override async Task onQueryResponseCommit()
        {
            await Task.CompletedTask;
            return;
        }

        //=====================================================================================
        // DB 쿼리를 실패하고, doFnRollback()가 QueryResultType.NotCalledQueryFunc를 반환할 경우 호출된다.
        //=====================================================================================
        public override async Task onQueryResponseRollback(Result errorResult)
        {
            await Task.CompletedTask;
            return;
        }

        public new Player? getOwner() => getQueryBatch()?.getLogActor() as Player;
    }
}