path: root/ThirdParty/TribalMedia/TribalMedia.Framework.Data/FieldMapper.cs

/*
* Copyright (c) Tribal Media AB, http://tribalmedia.se/
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*     * Redistributions of source code must retain the above copyright
*       notice, this list of conditions and the following disclaimer.
*     * Redistributions in binary form must reproduce the above copyright
*       notice, this list of conditions and the following disclaimer in the
*       documentation and/or other materials provided with the distribution.
*     * The name of Tribal Media AB may not be used to endorse or promote products
*       derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE DEVELOPERS ``AS IS'' AND ANY
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL THE CONTRIBUTORS BE LIABLE FOR ANY
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
* 
*/

using System;
using System.Collections.Generic;
using System.Data.Common;

namespace TribalMedia.Framework.Data
{
    public delegate TField RowMapperGetAccessor<TRowMapper, TField>(TRowMapper rowMapper);

    public delegate void RowMapperSetAccessor<TRowMapper, TField>(TRowMapper rowMapper, TField value);

    public delegate TField ObjectGetAccessor<TObj, TField>(TObj obj);

    public delegate void ObjectSetAccessor<TObj, TField>(TObj obj, TField value);

    public abstract class FieldMapper
    {
        private readonly TableMapper m_tableMapper;
        private readonly string m_fieldName;

        public string FieldName
        {
            get { return m_fieldName; }
        }

        protected Type m_valueType;

        public Type ValueType
        {
            get { return m_valueType; }
        }

        public abstract object GetParamValue(object obj);

        public FieldMapper( TableMapper tableMapper, string fieldName, Type valueType)
        {
            m_fieldName = fieldName;
            m_valueType = valueType;
            m_tableMapper = tableMapper;
        }

        public abstract void SetPropertyFromReader(object mapper, DataReader reader);

        public void RawAddParam(DbCommand command, List<string> fieldNames, string fieldName, object value)
        {
            string paramName = m_tableMapper.CreateParamName(fieldName);
            fieldNames.Add(fieldName);

            DbParameter param = command.CreateParameter();
            param.ParameterName = paramName;
            param.Value = value;

            command.Parameters.Add(param);
        }

        public void ExpandField<TObj>(TObj obj, DbCommand command, List<string> fieldNames)
        {
            string fieldName = FieldName;
            object value = GetParamValue(obj);

            //if (ValueType == typeof (Vector3))
            //{
            //    Vector3 vector = (Vector3) value;

            //    RawAddParam(command, fieldNames, fieldName + "X", vector.X);
            //    RawAddParam(command, fieldNames, fieldName + "Y", vector.Y);
            //    RawAddParam(command, fieldNames, fieldName + "Z", vector.Z);
            //}
            //else if (ValueType == typeof (Quaternion))
            //{
            //    Quaternion quaternion = (Quaternion) value;

            //    RawAddParam(command, fieldNames, fieldName + "X", quaternion.X);
            //    RawAddParam(command, fieldNames, fieldName + "Y", quaternion.Y);
            //    RawAddParam(command, fieldNames, fieldName + "Z", quaternion.Z);
            //    RawAddParam(command, fieldNames, fieldName + "W", quaternion.W);
            //}
            //else
            //{
                RawAddParam(command, fieldNames, fieldName, ConvertToDbType(value));
            //}
        }

        protected object GetValue(DataReader reader)
        {
            object value;
            //if (ValueType == typeof (Vector3))
            //{
            //    value = reader.GetVector(m_fieldName);
            //}
            //else if (ValueType == typeof (Quaternion))
            //{
            //    value = reader.GetQuaternion(m_fieldName);
            //}
            //else 
            //if (ValueType == typeof(UID))
            //{
            //    Guid guid = reader.GetGuid(m_fieldName);
            //    value = new UID(guid);
            //}
            //else 
            if (ValueType == typeof(Guid))
            {
                value = reader.GetGuid(m_fieldName);
            }
            else if (ValueType == typeof (bool))
            {
                uint boolVal = reader.GetUShort(m_fieldName);
                value = (boolVal == 1);
            }
            else 
            if (ValueType == typeof (byte))
            {
                value = reader.GetByte(m_fieldName);
            }
            else if (ValueType == typeof (sbyte))
            {
                value = reader.GetSByte(m_fieldName);
            }
            else if (ValueType == typeof (ushort))
            {
                value = reader.GetUShort(m_fieldName);
            }
            else if (ValueType == typeof (byte[]))
            {
                value = reader.GetBytes(m_fieldName);
            }
            else
            {
                value = reader.Get(m_fieldName);
            }

            if (value is DBNull)
            {
                value = default(ValueType);
            }

            return value;
        }

        public static object ConvertToDbType(object value)
        {
            //if (value is UID)
            //{
            //    return (value as UID).UUID.ToString();
            //}

            return value;
        }
    }

    public class RowMapperField<TRowMapper, TField> : FieldMapper
        where TRowMapper : RowMapper
    {
        private readonly RowMapperGetAccessor<TRowMapper, TField> m_fieldGetAccessor;
        private readonly RowMapperSetAccessor<TRowMapper, TField> m_fieldSetAccessor;

        public override object GetParamValue(object obj)
        {
            return m_fieldGetAccessor((TRowMapper) obj);
        }

        public override void SetPropertyFromReader(object mapper, DataReader reader)
        {
            object value;

            value = GetValue(reader);

            if (value == null)
            {
                m_fieldSetAccessor((TRowMapper) mapper, default(TField));
            }
            else
            {
                m_fieldSetAccessor((TRowMapper) mapper, (TField) value);
            }
        }


        public RowMapperField(TableMapper tableMapper, string fieldName, RowMapperGetAccessor<TRowMapper, TField> rowMapperGetAccessor,
                              RowMapperSetAccessor<TRowMapper, TField> rowMapperSetAccessor)
            : base(tableMapper, fieldName, typeof(TField))
        {
            m_fieldGetAccessor = rowMapperGetAccessor;
            m_fieldSetAccessor = rowMapperSetAccessor;
        }
    }

    public class ObjectField<TObject, TField> : FieldMapper
    {
        private readonly ObjectGetAccessor<TObject, TField> m_fieldGetAccessor;
        private readonly ObjectSetAccessor<TObject, TField> m_fieldSetAccessor;

        public override object GetParamValue(object obj)
        {
            return m_fieldGetAccessor((TObject) obj);
        }

        public override void SetPropertyFromReader(object obj, DataReader reader)
        {
            object value;

            value = GetValue(reader);

            if (value == null)
            {
                m_fieldSetAccessor((TObject) obj, default(TField));
            }
            else
            {
                m_fieldSetAccessor((TObject) obj, (TField) value);
            }
        }


        public ObjectField(TableMapper tableMapper, string fieldName, ObjectGetAccessor<TObject, TField> rowMapperGetAccessor,
                           ObjectSetAccessor<TObject, TField> rowMapperSetAccessor)
            : base(tableMapper, fieldName, typeof (TField))
        {
            m_fieldGetAccessor = rowMapperGetAccessor;
            m_fieldSetAccessor = rowMapperSetAccessor;
        }
    }
}