wabbajack/Wabbajack.Server/DataLayer/Mappers.cs

using System;
using System.Data;
using Dapper;
using Wabbajack.Common;
using Wabbajack.Lib.AuthorApi;
using Wabbajack.Lib.Downloaders;

namespace Wabbajack.Server.DataLayer
{
    public partial class SqlService
    {
        static SqlService()
        {
            SqlMapper.AddTypeHandler(new HashMapper());
            SqlMapper.AddTypeHandler(new RelativePathMapper());
            SqlMapper.AddTypeHandler(new JsonMapper<AbstractDownloadState>());
            SqlMapper.AddTypeHandler(new JsonMapper<CDNFileDefinition>());
            SqlMapper.AddTypeHandler(new VersionMapper());
            SqlMapper.AddTypeHandler(new GameMapper());
            SqlMapper.AddTypeHandler(new DateTimeHandler());
        }
        
        /// <summary>
        /// Needed to make sure dates are all in UTC format
        /// </summary>
        private class DateTimeHandler : SqlMapper.TypeHandler<DateTime>
        {
            public override void SetValue(IDbDataParameter parameter, DateTime value)
            {
                parameter.Value = value;
            }

            public override DateTime Parse(object value)
            {
                return DateTime.SpecifyKind((DateTime)value, DateTimeKind.Utc);
            }
        }

        class JsonMapper<T> : SqlMapper.TypeHandler<T>
        {
            public override void SetValue(IDbDataParameter parameter, T value)
            {
                parameter.Value = value.ToJson();
            }

            public override T Parse(object value)
            {
                return ((string)value).FromJsonString<T>();
            }
        }

        class RelativePathMapper : SqlMapper.TypeHandler<RelativePath>
        {
            public override void SetValue(IDbDataParameter parameter, RelativePath value)
            {
                parameter.Value = value.ToJson();
            }

            public override RelativePath Parse(object value)
            {
                return (RelativePath)(string)value;
            }
        }
        
        class HashMapper : SqlMapper.TypeHandler<Hash>
        {
            public override void SetValue(IDbDataParameter parameter, Hash value)
            {
                parameter.Value = (long)value;
            }

            public override Hash Parse(object value)
            {
                return Hash.FromLong((long)value);
            }
        }
        
        class VersionMapper : SqlMapper.TypeHandler<Version>
        {
            public override void SetValue(IDbDataParameter parameter, Version value)
            {
                parameter.Value = value.ToString();
            }

            public override Version Parse(object value)
            {
                return Version.Parse((string)value);
            }
        }
        
        class GameMapper : SqlMapper.TypeHandler<Game>
        {
            public override void SetValue(IDbDataParameter parameter, Game value)
            {
                parameter.Value = value.ToString();
            }

            public override Game Parse(object value)
            {
                return GameRegistry.GetByFuzzyName((string)value).Game;
            }
        }
    }
}