DocSpace-buildtools/common/ASC.Data.Backup.Core/Core/DbHelper.cs

namespace ASC.Data.Backup
{
    [Scope]
    public class DbHelper : IDisposable
    {
        private readonly DbProviderFactory factory;
        private readonly DbConnection connect;
        private readonly DbCommandBuilder builder;
        private readonly DataTable columns;
        private readonly bool mysql;
        private readonly IDictionary<string, string> whereExceptions = new Dictionary<string, string>();
        private readonly ILog log;
        private readonly TenantDbContext tenantDbContext;
        private readonly CoreDbContext coreDbContext;

        public DbHelper(IOptionsMonitor<ILog> options, ConnectionStringSettings connectionString, TenantDbContext tenantDbContext, CoreDbContext coreDbContext)
        {
            log = options.CurrentValue;
            this.tenantDbContext = tenantDbContext;
            this.coreDbContext = coreDbContext;
            var file = connectionString.ElementInformation.Source;
            if ("web.connections.config".Equals(Path.GetFileName(file), StringComparison.InvariantCultureIgnoreCase))
            {
                file = CrossPlatform.PathCombine(Path.GetDirectoryName(file), "Web.config");
            }
            var xconfig = XDocument.Load(file);
            var provider = xconfig.XPathSelectElement("/configuration/system.data/DbProviderFactories/add[@invariant='" + connectionString.ProviderName + "']");
            factory = (DbProviderFactory)Activator.CreateInstance(Type.GetType(provider.Attribute("type").Value, true));
            builder = factory.CreateCommandBuilder();
            connect = factory.CreateConnection();
            connect.ConnectionString = connectionString.ConnectionString;
            connect.Open();

            mysql = connectionString.ProviderName.Contains("mysql", StringComparison.OrdinalIgnoreCase);
            if (mysql)
            {
                CreateCommand("set @@session.sql_mode = concat(@@session.sql_mode, ',NO_AUTO_VALUE_ON_ZERO')").ExecuteNonQuery();
            }

            columns = connect.GetSchema("Columns");

            whereExceptions["calendar_calendar_item"] = " where calendar_id in (select id from calendar_calendars where tenant = {0}) ";
            whereExceptions["calendar_calendar_user"] = " where calendar_id in (select id from calendar_calendars where tenant = {0}) ";
            whereExceptions["calendar_event_item"] = " inner join calendar_events on calendar_event_item.event_id = calendar_events.id where calendar_events.tenant = {0} ";
            whereExceptions["calendar_event_user"] = " inner join calendar_events on calendar_event_user.event_id = calendar_events.id where calendar_events.tenant = {0} ";
            whereExceptions["crm_entity_contact"] = " inner join crm_contact on crm_entity_contact.contact_id = crm_contact.id where crm_contact.tenant_id = {0} ";
            whereExceptions["crm_entity_tag"] = " inner join crm_tag on crm_entity_tag.tag_id = crm_tag.id where crm_tag.tenant_id = {0} ";
            whereExceptions["files_folder_tree"] = " inner join files_folder on folder_id = id where tenant_id = {0} ";
            whereExceptions["forum_answer_variant"] = " where answer_id in (select id from forum_answer where tenantid = {0})";
            whereExceptions["forum_topic_tag"] = " where topic_id in (select id from forum_topic where tenantid = {0})";
            whereExceptions["forum_variant"] = " where question_id in (select id from forum_question where tenantid = {0})";
            whereExceptions["projects_project_participant"] = " inner join projects_projects on projects_project_participant.project_id = projects_projects.id where projects_projects.tenant_id = {0} ";
            whereExceptions["projects_following_project_participant"] = " inner join projects_projects on projects_following_project_participant.project_id = projects_projects.id where projects_projects.tenant_id = {0} ";
            whereExceptions["projects_project_tag"] = " inner join projects_projects on projects_project_tag.project_id = projects_projects.id where projects_projects.tenant_id = {0} ";
            whereExceptions["tenants_tenants"] = " where id = {0}";
            whereExceptions["core_acl"] = " where tenant = {0} or tenant = -1";
            whereExceptions["core_subscription"] = " where tenant = {0} or tenant = -1";
            whereExceptions["core_subscriptionmethod"] = " where tenant = {0} or tenant = -1";
        }

        public List<string> GetTables()
        {
            var allowTables = new List<string>
                {
                    "blogs_",
                    "bookmarking_",
                    "calendar_",
                    "core_",
                    "crm_",
                    "events_",
                    "files_",
                    "forum_",
                    "photo_",
                    "projects_",
                    "tenants_",
                    "webstudio_",
                    "wiki_",
                };

            var disallowTables = new List<string>
                {
                    "core_settings",
                    "webstudio_uservisit",
                    "webstudio_useractivity",
                    "tenants_forbiden",
                };

            IEnumerable<string> tables;

            if (mysql)
            {
                tables = ExecuteList(CreateCommand("show tables"));
            }
            else
            {
                tables = connect
                    .GetSchema("Tables")
                    .Select(@"TABLE_TYPE <> 'SYSTEM_TABLE'")
                    .Select(row => (string)row["TABLE_NAME"]);
            }

            return tables
                .Where(t => allowTables.Any(a => t.StartsWith(a)) && !disallowTables.Any(d => t.StartsWith(d)))
                .ToList();
        }

        public DataTable GetTable(string table, int tenant)
        {
            try
            {
                var dataTable = new DataTable(table);
                var adapter = factory.CreateDataAdapter();
                adapter.SelectCommand = CreateCommand("select " + Quote(table) + ".* from " + Quote(table) + GetWhere(table, tenant));

                log.Debug(adapter.SelectCommand.CommandText);

                adapter.Fill(dataTable);
                return dataTable;
            }
            catch (Exception error)
            {
                log.ErrorFormat("Table {0}: {1}", table, error);
                throw;
            }
        }


        public void SetTable(DataTable table)
        {
            using var tx = connect.BeginTransaction();
            try
            {
                if ("tenants_tenants".Equals(table.TableName, StringComparison.InvariantCultureIgnoreCase))
                {
                    // remove last tenant
                    var tenant = tenantDbContext.Tenants.LastOrDefault();
                    if (tenant != null)
                    {
                        tenantDbContext.Tenants.Remove(tenant);
                        tenantDbContext.SaveChanges();
                    }
                    /*  var tenantid = CreateCommand("select id from tenants_tenants order by id desc limit 1").ExecuteScalar();
                         CreateCommand("delete from tenants_tenants where id = " + tenantid).ExecuteNonQuery();*/
                    if (table.Columns.Contains("mappeddomain"))
                    {
                        foreach (var r in table.Rows.Cast<DataRow>())
                        {
                            r[table.Columns["mappeddomain"]] = null;
                            if (table.Columns.Contains("id"))
                            {
                                var tariff = coreDbContext.Tariffs.FirstOrDefault(t => t.Tenant == tenant.Id);
                                tariff.Tenant = (int)r[table.Columns["id"]];
                                //  CreateCommand("update tenants_tariff set tenant = " + r[table.Columns["id"]] + " where tenant = " + tenantid).ExecuteNonQuery();
                                coreDbContext.Entry(tariff).State = EntityState.Modified;
                                coreDbContext.SaveChanges();
                            }
                        }
                    }
                }

                var sql = new StringBuilder("replace into " + Quote(table.TableName) + "(");

                var tableColumns = GetColumnsFrom(table.TableName)
                    .Intersect(table.Columns.Cast<DataColumn>().Select(c => c.ColumnName), StringComparer.InvariantCultureIgnoreCase)
                    .ToList();

                tableColumns.ForEach(column => sql.Append($"{Quote(column)}, "));
                sql.Replace(", ", ") values (", sql.Length - 2, 2);

                var insert = connect.CreateCommand();
                tableColumns.ForEach(column =>
                {
                    sql.Append($"@{column}, ");
                    var p = insert.CreateParameter();
                    p.ParameterName = "@" + column;
                    insert.Parameters.Add(p);
                });
                sql.Replace(", ", ")", sql.Length - 2, 2);
                insert.CommandText = sql.ToString();

                foreach (var r in table.Rows.Cast<DataRow>())
                {
                    foreach (var c in tableColumns)
                    {
                        ((IDbDataParameter)insert.Parameters["@" + c]).Value = r[c];
                    }
                    insert.ExecuteNonQuery();
                }

                tx.Commit();
            }
            catch (Exception e)
            {
                log.ErrorFormat("Table {0}: {1}", table, e);
            }
        }

        public void Dispose()
        {
            builder.Dispose();
            connect.Dispose();
        }

        public DbCommand CreateCommand(string sql)
        {
            var command = connect.CreateCommand();
            command.CommandText = sql;
            return command;
        }

        public List<string> ExecuteList(DbCommand command)
        {
            var list = new List<string>();
            using (var result = command.ExecuteReader())
            {
                while (result.Read())
                {
                    list.Add(result.GetString(0));
                }
            }
            return list;
        }

        private string Quote(string identifier)
        {
            return identifier;
        }

        private IEnumerable<string> GetColumnsFrom(string table)
        {
            if (mysql)
            {
                return ExecuteList(CreateCommand("show columns from " + Quote(table)));
            }
            else
            {
                return columns.Select($"TABLE_NAME = '{table}'")
                    .Select(r => r["COLUMN_NAME"].ToString());
            }
        }

        private string GetWhere(string tableName, int tenant)
        {
            if (tenant == -1) return string.Empty;

            if (whereExceptions.TryGetValue(tableName.ToLower(), out var exc))
            {
                return string.Format(exc, tenant);
            }
            var tenantColumn = GetColumnsFrom(tableName).FirstOrDefault(c => c.StartsWith("tenant", StringComparison.OrdinalIgnoreCase));
            return tenantColumn != null ?
                " where " + Quote(tenantColumn) + " = " + tenant :
                " where 1 = 0";
        }
    }
}