using ServerCommon;
using ServerCore; using ServerBase;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace GameServer;

public class DBQQuestMailSend : QueryExecutorBase
{
    private string m_combination_key_for_pk = string.Empty;
    private string m_pk = string.Empty;


    private readonly List<QuestMailDoc> m_quest_mail_docs = new();

    public DBQQuestMailSend(string combinationKeyForPK, List<QuestMailDoc> docs) 
        : base(typeof(DBQQuestMailSend).Name)
    {
        m_combination_key_for_pk = combinationKeyForPK;
        m_quest_mail_docs = docs;
    }


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

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

        foreach (var quest_mail_doc in m_quest_mail_docs)
        {
            var mail_attrib = quest_mail_doc.getAttrib<QuestMailAttrib>();
            NullReferenceCheckHelper.throwIfNull(mail_attrib, () => $"mail_attrib is null !!!");

            quest_mail_doc.setCombinationKeyForPK(m_combination_key_for_pk);
            quest_mail_doc.setCombinationKeyForSK(mail_attrib.QuestId.ToString());
            
            var error_code = quest_mail_doc.onApplyPKSK();
            if (error_code.isFail())
            {
                var err_msg = $"Failed to onApplyPKSK() !!! : {error_code.toBasicString()} - {toBasicString()}, {owner.toBasicString()}";
                Log.getLogger().error(err_msg);
                continue;
            }
        }

        return Task.FromResult(result);
    }


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

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

        var db_connector = query_batch.getDynamoDbConnector();
        var query_contexts = new List<DynamoDbDocumentQueryContext>();

        foreach (var doc in m_quest_mail_docs)
        {
            (result, var quest_mail_document) = await doc.onCopyToDocument();
            if(result.isFail())
            {
                err_msg = $"Failed to onCopyToDocument() !!! : {result.toBasicString()}, m_pk:{m_pk} - {toBasicString()}";
                Log.getLogger().error(err_msg);
                continue;
            }
            var docmument_ctx = new DynamoDbDocumentQueryContext(db_connector.getTableFullName(doc.TableName), quest_mail_document, QueryType.Insert);
            query_contexts.Add(docmument_ctx);
        }

        var table = db_connector.getTableByDoc<QuestMailDoc>();
        (result, _) = await table.simpleTransactWriteWithDocument(query_contexts, 2, query_batch.getTransId());
        if (result.isFail())
        {
            err_msg = $"Failed to simpleTransactWrite !!! : {result.toBasicString()}, m_pk:{m_pk}, {toBasicString()}";
            Log.getLogger().error(err_msg);
            return result;
        }

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

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

        foreach (var doc in m_quest_mail_docs)
        {
            await quest_mail_action.setQuestMailFromDoc(doc);
        }
        
        return result;
    }


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


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

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