sodoff/src/Services/KeyValueService.cs

using sodoff.Model;
using sodoff.Schema;

namespace sodoff.Services;
public class KeyValueService {

    private readonly DBContext ctx;

    public KeyValueService(DBContext ctx) {
        this.ctx = ctx;
    }

    public Model.PairData? GetPairData(User? user, Viking? viking, string? userId, int pairId) {
        Dragon? dragon = null;
        return CheckOwnerAndGetPairData(ref user, ref viking, ref dragon, userId, pairId);
    }

    public bool SetPairData(User? user, Viking? viking, string? userId, int pairId, Schema.PairData schemaData) {
        // Get the pair
        Dragon? dragon = null;
        Model.PairData? pair = CheckOwnerAndGetPairData(ref user, ref viking, ref dragon, userId, pairId);

        // Create the pair if it doesn't exist
        bool exists = true;
        if (pair is null) {
            pair = new Model.PairData {
                PairId = pairId,
                UserId = user?.Id,
                VikingId = viking?.Id,
                DragonId = dragon?.Id,
                Pairs = new List<Model.Pair>()
            };
            exists = false;
        }
        if (schemaData is null || schemaData.Pairs is null)
            return true;
        // Update or create the key-value pairs
        foreach (var p in schemaData.Pairs) {
            if (string.IsNullOrEmpty(p.PairValue))
                return false; // NOTE: The real server doesn't update anything if one value is empty
            var pairItem = pair.Pairs.FirstOrDefault(e => e.Key == p.PairKey);
            if (pairItem != null)
                pairItem.Value = p.PairValue;
            else {
                pairItem = new Model.Pair {
                    Key = p.PairKey,
                    Value = p.PairValue
                };
                pair.Pairs.Add(pairItem);
            }
        }

        if (exists)
            ctx.PairData.Update(pair);
        else
            ctx.PairData.Add(pair);
        ctx.SaveChanges();

        return true;
    }

    public Schema.PairData? ModelToSchema(Model.PairData? model) {
        if (model is null) return null;

        // Convert to schema class
        List<Schema.Pair> pairs = new List<Schema.Pair>();
        foreach (var p in model.Pairs) {
            pairs.Add(new Schema.Pair {
                PairKey = p.Key,
                PairValue = p.Value
            });
        }

        return new Schema.PairData { Pairs = pairs.ToArray() };
    } 

    private Model.PairData? CheckOwnerAndGetPairData(ref User? user, ref Viking? viking, ref Dragon dragon, string? userId, int pairId) {
        if (userId != null) {
            Guid userIdGuid = Guid.Parse(userId);
            // find dragon
            dragon = viking?.Dragons.FirstOrDefault(e => e.EntityId == userIdGuid);

            // if not dragon then try viking -> check ID
            if (dragon != null || viking?.Uid != userIdGuid) {
                // if not viking and user not set, then try set user from viking
                if (user is null)
                    user = viking?.User;
                viking = null;
            }

            // if not dragon nor viking then try user -> check ID
            if (viking != null || user?.Id != userIdGuid) user = null;
        }

        // NOTE: only one of (dragon, viking, user) can be not null here

        Model.PairData? pair = null;
        if (viking != null)
            pair = viking.PairData?.FirstOrDefault(e => e.PairId == pairId);
        else if (user != null)
            pair = user.PairData.FirstOrDefault(e => e.PairId == pairId);
        else if (dragon != null)
            pair = dragon.PairData.FirstOrDefault(e => e.PairId == pairId);

        return pair;
    }
}