MikaltoResort/ManagementApp/Dapper.AExtentions/TableMapper.cs

using System.Collections.Concurrent;
using System.Data;

namespace ManagementApp.Dapper.AExtentions
{
    public class TableMapper
    {
        private static readonly ConcurrentDictionary<RuntimeTypeHandle, string> TableNames = new();

        private static string _prefix = string.Empty;

        /// <summary>
        /// Used to setup custom table conventions.
        /// </summary>
        /// <param name="tablePrefix">table name prefix</param>
        /// <param name="tableSuffix">table name suffix</param>
        // ReSharper disable once UnusedMember.Global
        public static void SetupConvention(string tablePrefix)
        {
            if (!TableNames.IsEmpty)
            {
                throw new InvalidConstraintException("TableMapper.SetupConvention called after usage.");
            }

            _prefix = tablePrefix;

            TableNames.Clear();
        }

        internal static string GetTableName(Type type)
        {
            if (TableNames.TryGetValue(type.TypeHandle, out var name))
            {
                return name;
            }

            var tableAttr = type.GetCustomAttributes(false).SingleOrDefault(attr => attr.GetType().Name == "TableAttribute") as dynamic;
            if (tableAttr != null)
            {
                name = tableAttr.Name;

                if (tableAttr.Schema != null)
                {
                    name = tableAttr.Schema + "." + tableAttr.Name;
                }
            }
            else
            {
                name = type.IsInterface && type.Name.StartsWith("I") ? type.Name.Substring(1) : type.Name;
                name = $"{_prefix}{name}";
            }

            TableNames[type.TypeHandle] = name;
            return name;
        }
    }
}