using System.Globalization;


using Amazon.DynamoDBv2;
using Amazon.DynamoDBv2.Model;


using ServerCore; using ServerBase;
using ServerCommon;



namespace GameServer;

public enum CaliumStorageType
{
    None = 0,
    Converter = 1,
    Operator = 2
}

public class DBQStorageCaliumUpdate : QueryExecutorBase
{
    private CaliumStorageType m_storage_type { get; set; } = CaliumStorageType.None;
    private double m_delta { get; set; }
    private bool m_is_update_last_date { get; set; }
    private bool m_is_pivot { get; set; }
    
    private PrimaryKey m_primary_key = new();
    
    //=====================================================================================
    // 0 < deltaCount => 증가
    // 0 > deltaCount => 감소
    //=====================================================================================

    public DBQStorageCaliumUpdate(CaliumStorageType storageType, double deltaCalium, bool isUpdateLastDate = false, bool isPivot = false) : base(nameof(DBQStorageCaliumUpdate))
    {
        m_storage_type = storageType;
        m_delta = deltaCalium;
        m_is_update_last_date = isUpdateLastDate;
        m_is_pivot = isPivot;
    }
    
    //=====================================================================================
    // DB 쿼리 직전에 준비해야 할 로직들을 작성한다.
    //=====================================================================================
    public override async Task<Result> onPrepareQuery()
    {
        var owner = getOwner();

        var (result, make_primary_key) = await DynamoDBDocBaseHelper.makePrimaryKey<CaliumStorageDoc>(string.Empty);
        if (result.isFail())
        {
            return result;
        }
        NullReferenceCheckHelper.throwIfNull(make_primary_key, () => $"make_primary_key is null !!! - {owner.toBasicString()}");

        m_primary_key = make_primary_key;

        return result;
    }
    
    //=====================================================================================
    // onPrepareQuery()를 성공할 경우 호출된다.
    //=====================================================================================
    public override async Task<Result> onQuery()
    {
        var owner = getOwner();

        var query_batch = getQueryBatch() as QueryBatch<QueryRunnerWithItemRequest>;
        NullReferenceCheckHelper.throwIfNull(query_batch, () => $"query_batch is null !!! - {owner.toBasicString()}");

        var query_runner_with_item_request = query_batch.getQueryRunner();
        NullReferenceCheckHelper.throwIfNull(query_runner_with_item_request, () => $"query_runner_with_item_request is null !!! - {owner.toBasicString()}");

        var db_connector = query_batch.getDynamoDbConnector();
        var update = makeUpdateItemRequestChangeConverterTotalCalium(db_connector, m_primary_key.toKeyWithAttributeValue(), convertCaliumStorageTypeToValueName());
        if (update.result.isFail()) return update.result;
        NullReferenceCheckHelper.throwIfNull(update.updateRequest, () => $"update_request is null !!! - {owner.toBasicString()}");

        var exception_handler = new DynamoDbQueryExceptionNotifier.ExceptionHandler(DynamoDbQueryExceptionNotifier.ConditionalCheckFailed, ServerErrorCode.LackOfTotalCalium);
        
        var query_context = update.updateRequest.createItemRequestQueryContext(QueryType.Update, exception_handler);
        var result = await query_runner_with_item_request.tryRegisterQueryContext(query_context);

        return result;
    }
    
    //=====================================================================================
    // 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;
    }

    private string convertCaliumStorageTypeToValueName()
    {
        return m_storage_type switch
        {
            CaliumStorageType.Converter => JsonHelper.getJsonPropertyName<CaliumConverterInfo>(nameof(CaliumConverterInfo.TotalCalium)),
            CaliumStorageType.Operator => JsonHelper.getJsonPropertyName<CaliumOperatorInfo>(nameof(CaliumOperatorInfo.TotalCalium)),
            _ => string.Empty
        };
    }
    
    private (Result result, UpdateItemRequest? updateRequest) makeUpdateItemRequestChangeConverterTotalCalium( DynamoDbClient dbClient
                                                                                                             , Dictionary<string, AttributeValue> attributeValueWithPrimaryKey
                                                                                                             , string targetAttribName )
    {
        var result = new Result();
        string err_msg;

        if (string.IsNullOrEmpty(targetAttribName))
        {
            err_msg = $"Failed to DynamoDbClientHelper.toAttributeExpressionFromJson() !!! : targetAttribName - {targetAttribName}";
            result.setFail(ServerErrorCode.AttribPathMakeFailed, err_msg);
            Log.getLogger().error(result.toBasicString());

            return (result, null);
        }

        var target_doc = new CaliumStorageDoc();

        var query_builder = new DynamoDbItemRequestHelper.UpdateItemRequestBuilder(dbClient.getTableFullName(target_doc.TableName));
        query_builder.withKeys(attributeValueWithPrimaryKey);

        var attrib_path_json_string = target_doc.toJsonStringOfAttribs();
        (var is_success, var attribute_expression) = DynamoDbClientHelper.toAttributeExpressionFromJson(attrib_path_json_string, targetAttribName);
        if (false == is_success)
        {
            err_msg = $"Failed to DynamoDbClientHelper.toAttributeExpressionFromJson() !!! : attribPath:{attrib_path_json_string}, targetKey:{targetAttribName}";
            result.setFail(ServerErrorCode.AttribPathMakeFailed, err_msg);
            Log.getLogger().error(result.toBasicString());

            return (result, null);
        }

        var attribute_names = DynamoDbClientHelper.toExpressionAttributeNamesFromJson(attrib_path_json_string, targetAttribName);
        query_builder.withExpressionAttributeNames(attribute_names);

        var update_expression = (m_delta >= 0)
            ? $"SET {attribute_expression} = if_not_exists({attribute_expression}, :start) + :changeValue"
            : $"SET {attribute_expression} = if_not_exists({attribute_expression}, :start) - :changeValue";

        query_builder.withUpdateExpression(update_expression);
        
        // 차감시 조건 추가
        if (m_delta < 0)
        {
            query_builder.withConditionExpression($"{attribute_expression} >= :changeValue");
        }

        var expression_attribute_values = new Dictionary<string, AttributeValue>
        {
            { ":changeValue", new AttributeValue { N = Math.Abs(m_delta).ToString(CultureInfo.InvariantCulture) } },
            { ":start", new AttributeValue { N = "0" } }
        };
        query_builder.withExpressionAttributeValues(expression_attribute_values);

        // 마지막 업데이트 날짜 수정
        if (m_is_update_last_date)
        {
            result = updateLastDate(query_builder, attrib_path_json_string);
            if (result.isFail()) return (result, null);
        }

        query_builder.withReturnValues(ReturnValue.ALL_NEW);

        return query_builder.build();
    }

    private Result updateLastDate(DynamoDbItemRequestHelper.UpdateItemRequestBuilder query_builder, string attrib_path_json_string)
    {
        var result = new Result();
        
        var request = query_builder.getItemRequest();
        var attribute_last_date = JsonHelper.getJsonPropertyName<CaliumConverterInfo>(nameof(CaliumConverterInfo.LastFillUpDate));
            
        // 1. add attribute names
        var attribute_last_date_names = DynamoDbClientHelper.toExpressionAttributeNamesFromJson(attrib_path_json_string, attribute_last_date);
        foreach (var name in attribute_last_date_names.Where(name => !request.ExpressionAttributeNames.ContainsKey(name.Key)))
        {
            request.ExpressionAttributeNames.Add(name.Key, name.Value);
        }
            
        // 2. add attribute expression
        var (is_success, last_date_expression) = DynamoDbClientHelper.toAttributeExpressionFromJson(attrib_path_json_string, attribute_last_date);
        if (false == is_success)
        {
            var err_msg = $"Failed to DynamoDbClientHelper.toAttributeExpressionFromJson() !!! : attribPath:{attrib_path_json_string}, targetKey:{attribute_last_date}";
            result.setFail(ServerErrorCode.AttribPathMakeFailed, err_msg);
            Log.getLogger().error(result.toBasicString());

            return result;
        }
            
        request.UpdateExpression += ", ";
        request.UpdateExpression += $"{last_date_expression} = :lastDateValue";
            
        // 3. add values
        request.ExpressionAttributeValues.Add(":lastDateValue", new AttributeValue { S = DateTimeHelper.Current.toString("yyyy-MM-ddTHH:mm:ss")});

        return result;
    }
}