From 004fde73c063ad2dd4d408153007444f1cc54dfb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bartosz=20Korczy=C5=84ski?= <b.andysc@gmail.com>
Date: Sat, 8 May 2021 13:43:40 +0100
Subject: [PATCH] [Database editor] Support for conditions in database table
 editors

---
 .../Helpers/FieldValueTemplateSelector.cs     |  3 +
 .../MultiRow/MultiRowDbTableEditorView.axaml  |  5 ++
 .../Helpers/FieldValueTemplateSelector.cs     |  3 +
 .../MultiRow/MultiRowDbTableEditorView.xaml   |  5 ++
 .../Data/DatabaseTableModelGenerator.cs       | 13 +++-
 .../IDatabaseTableModelGenerator.cs           |  1 +
 .../Data/Structs/DatabaseColumnJson.cs        |  3 +
 .../Structs/DatabaseTableDefinitionJson.cs    | 30 ++++++++
 .../DbDefinitions/gossip_menu_option.json     | 20 ++++++
 ...aseEntityConditionsChangedHistoryAction.cs | 37 ++++++++++
 .../Loaders/DatabaseTableDataProvider.cs      | 46 +++++++++++-
 WDE.DatabaseEditors/Models/DatabaseEntity.cs  | 43 +++++++++--
 .../QueryGenerators/QueryGenerator.cs         | 67 ++++++++++++++++-
 .../MultiRow/DatabaseCellViewModel.cs         | 24 ++++++-
 .../MultiRow/DatabaseColumnHeaderViewModel.cs |  6 ++
 .../MultiRowDbTableEditorViewModel.cs         | 71 +++++++++++++------
 16 files changed, 345 insertions(+), 32 deletions(-)
 create mode 100644 WDE.DatabaseEditors/History/DatabaseEntityConditionsChangedHistoryAction.cs

diff --git a/WDE.DatabaseEditors.Avalonia/Helpers/FieldValueTemplateSelector.cs b/WDE.DatabaseEditors.Avalonia/Helpers/FieldValueTemplateSelector.cs
index 843568958..f51b75d2f 100644
--- a/WDE.DatabaseEditors.Avalonia/Helpers/FieldValueTemplateSelector.cs
+++ b/WDE.DatabaseEditors.Avalonia/Helpers/FieldValueTemplateSelector.cs
@@ -13,9 +13,12 @@ public class FieldValueTemplateSelector : IDataTemplate
     {
         public DataTemplate? GenericTemplate { get; set; }
         public DataTemplate? BoolTemplate { get; set; }
+        public DataTemplate? CommandTemplate { get; set; }
 
         public IControl Build(object param)
         {
+            if (param is ViewModels.MultiRow.DatabaseCellViewModel vm3 && vm3.ActionCommand != null)
+                return CommandTemplate!.Build(param);
             if ((param is DatabaseCellViewModel vm && vm.ParameterValue is ParameterValue<long> holder && holder.Parameter is BoolParameter) ||
                 (param is ViewModels.MultiRow.DatabaseCellViewModel vm2 && vm2.ParameterValue is ParameterValue<long> holder2 && holder2.Parameter is BoolParameter))
                 return BoolTemplate!.Build(param);
diff --git a/WDE.DatabaseEditors.Avalonia/Views/MultiRow/MultiRowDbTableEditorView.axaml b/WDE.DatabaseEditors.Avalonia/Views/MultiRow/MultiRowDbTableEditorView.axaml
index 7ec9f83cb..858093abb 100644
--- a/WDE.DatabaseEditors.Avalonia/Views/MultiRow/MultiRowDbTableEditorView.axaml
+++ b/WDE.DatabaseEditors.Avalonia/Views/MultiRow/MultiRowDbTableEditorView.axaml
@@ -74,6 +74,11 @@
     </UserControl.Styles>
     <UserControl.DataTemplates>
         <helpers:FieldValueTemplateSelector>
+            <helpers:FieldValueTemplateSelector.CommandTemplate>
+                <DataTemplate>
+                    <Button Command="{Binding ActionCommand}" CommandParameter="{Binding}" Content="{Binding ActionLabel}" />
+                </DataTemplate>
+            </helpers:FieldValueTemplateSelector.CommandTemplate>
             <helpers:FieldValueTemplateSelector.BoolTemplate>
                 <DataTemplate>
                     <controls2:FastBoolCellView Height="24" Margin="0,0,10,0"
diff --git a/WDE.DatabaseEditors.WPF/Helpers/FieldValueTemplateSelector.cs b/WDE.DatabaseEditors.WPF/Helpers/FieldValueTemplateSelector.cs
index 73ca3cdc5..ee17ca3a1 100644
--- a/WDE.DatabaseEditors.WPF/Helpers/FieldValueTemplateSelector.cs
+++ b/WDE.DatabaseEditors.WPF/Helpers/FieldValueTemplateSelector.cs
@@ -10,9 +10,12 @@ public class FieldValueTemplateSelector : DataTemplateSelector
     {
         public DataTemplate GenericTemplate { get; set; }
         public DataTemplate BoolTemplate { get; set; }
+        public DataTemplate CommandTemplate { get; set; }
 
         public override DataTemplate SelectTemplate(object param, DependencyObject container)
         {
+            if (param is ViewModels.MultiRow.DatabaseCellViewModel vm3 && vm3.ActionCommand != null)
+                return CommandTemplate;
             if ((param is DatabaseCellViewModel vm && vm.ParameterValue is ParameterValue<long> holder && holder.Parameter is BoolParameter) ||
                 (param is ViewModels.MultiRow.DatabaseCellViewModel vm2 && vm2.ParameterValue is ParameterValue<long> holder2 && holder2.Parameter is BoolParameter))
                 return BoolTemplate;
diff --git a/WDE.DatabaseEditors.WPF/Views/MultiRow/MultiRowDbTableEditorView.xaml b/WDE.DatabaseEditors.WPF/Views/MultiRow/MultiRowDbTableEditorView.xaml
index dbae7610b..3dff94868 100644
--- a/WDE.DatabaseEditors.WPF/Views/MultiRow/MultiRowDbTableEditorView.xaml
+++ b/WDE.DatabaseEditors.WPF/Views/MultiRow/MultiRowDbTableEditorView.xaml
@@ -54,6 +54,11 @@
                                                 </ItemsControl.ItemsPanel>
                                                 <ItemsControl.ItemTemplateSelector>
                                                     <helpers:FieldValueTemplateSelector>
+                                                        <helpers:FieldValueTemplateSelector.CommandTemplate>
+                                                            <DataTemplate>
+                                                                <Button Command="{Binding ActionCommand}" CommandParameter="{Binding}" Content="{Binding ActionLabel}" />
+                                                            </DataTemplate>
+                                                        </helpers:FieldValueTemplateSelector.CommandTemplate>
                                                         <helpers:FieldValueTemplateSelector.BoolTemplate>
                                                             <DataTemplate>
                                                                 <controls:FastBoolCellView Height="24" Margin="0,0,10,0"
diff --git a/WDE.DatabaseEditors/Data/DatabaseTableModelGenerator.cs b/WDE.DatabaseEditors/Data/DatabaseTableModelGenerator.cs
index fb99187a4..4a861a940 100644
--- a/WDE.DatabaseEditors/Data/DatabaseTableModelGenerator.cs
+++ b/WDE.DatabaseEditors/Data/DatabaseTableModelGenerator.cs
@@ -1,6 +1,7 @@
 using System;
 using System.Collections.Generic;
 using System.Linq;
+using WDE.Common.Database;
 using WDE.Common.Services.MessageBox;
 using WDE.DatabaseEditors.Data.Interfaces;
 using WDE.DatabaseEditors.Data.Structs;
@@ -55,7 +56,7 @@ public DatabaseEntity CreateEmptyEntity(DatabaseTableDefinitionJson definition,
                 columns[column.DbColumnName] = databaseFieldFactory.CreateField(column.DbColumnName, valueHolder);
             }
 
-            return new DatabaseEntity(false, key, columns);
+            return new DatabaseEntity(false, key, columns, null);
         }
         
         public IDatabaseTableData? CreateDatabaseTable(DatabaseTableDefinitionJson tableDefinition,
@@ -63,7 +64,8 @@ public DatabaseEntity CreateEmptyEntity(DatabaseTableDefinitionJson definition,
             IList<Dictionary<string, (System.Type type, object value)>> fieldsFromDb)
         {
             HashSet<uint> providedKeys = new();
-            
+
+            IList<IConditionLine>? conditions = null;
             List<DatabaseEntity> rows = new();
             foreach (var entity in fieldsFromDb)
             {
@@ -112,6 +114,11 @@ public DatabaseEntity CreateEmptyEntity(DatabaseTableDefinitionJson definition,
                     {
                         valueHolder = new ValueHolder<long>(column.Value.value is DBNull ? 0 : ((bool)column.Value.value ? 1 : 0));
                     }
+                    else if (column.Value.type == typeof(IList<IConditionLine>))
+                    {
+                        conditions = ((IList<IConditionLine>)column.Value.value);
+                        continue;
+                    }
                     else
                     {
                         throw new NotImplementedException();
@@ -129,7 +136,7 @@ public DatabaseEntity CreateEmptyEntity(DatabaseTableDefinitionJson definition,
                     columns[column.Key] = databaseFieldFactory.CreateField(column.Key, valueHolder);
                 }
                 if (key.HasValue)
-                    rows.Add(new DatabaseEntity(true, key.Value, columns));
+                    rows.Add(new DatabaseEntity(true, key.Value, columns, conditions?.ToList<ICondition>()));
             }
 
             if (!tableDefinition.IsMultiRecord)
diff --git a/WDE.DatabaseEditors/Data/Interfaces/IDatabaseTableModelGenerator.cs b/WDE.DatabaseEditors/Data/Interfaces/IDatabaseTableModelGenerator.cs
index fb38b2b24..7a1d21a94 100644
--- a/WDE.DatabaseEditors/Data/Interfaces/IDatabaseTableModelGenerator.cs
+++ b/WDE.DatabaseEditors/Data/Interfaces/IDatabaseTableModelGenerator.cs
@@ -1,4 +1,5 @@
 using System.Collections.Generic;
+using WDE.Common.Database;
 using WDE.DatabaseEditors.Data.Structs;
 using WDE.DatabaseEditors.Models;
 using WDE.Module.Attributes;
diff --git a/WDE.DatabaseEditors/Data/Structs/DatabaseColumnJson.cs b/WDE.DatabaseEditors/Data/Structs/DatabaseColumnJson.cs
index ac3cb712f..574b9847f 100644
--- a/WDE.DatabaseEditors/Data/Structs/DatabaseColumnJson.cs
+++ b/WDE.DatabaseEditors/Data/Structs/DatabaseColumnJson.cs
@@ -31,6 +31,9 @@ public class DatabaseColumnJson
         [JsonProperty(PropertyName = "can_be_null")]
         public bool CanBeNull { get; set; }
         
+        [JsonProperty(PropertyName = "is_condition")]
+        public bool IsConditionColumn { get; set; }
+        
         [JsonProperty(PropertyName = "preferred_width")]
         public float? PreferredWidth { get; set; }
     }
diff --git a/WDE.DatabaseEditors/Data/Structs/DatabaseTableDefinitionJson.cs b/WDE.DatabaseEditors/Data/Structs/DatabaseTableDefinitionJson.cs
index 5443c5257..a13587364 100644
--- a/WDE.DatabaseEditors/Data/Structs/DatabaseTableDefinitionJson.cs
+++ b/WDE.DatabaseEditors/Data/Structs/DatabaseTableDefinitionJson.cs
@@ -49,6 +49,9 @@ public class DatabaseTableDefinitionJson
         [JsonProperty(PropertyName = "primary_key")]
         public IList<string>? PrimaryKey { get; set; }
         
+        [JsonProperty(PropertyName = "conditions")]
+        public DatabaseConditionReferenceJson? Condition { get; set; }
+
         [JsonProperty(PropertyName = "foreign_tables")]
         public IList<DatabaseForeignTableJson>? ForeignTable { get; set; }
         
@@ -64,4 +67,31 @@ public class DatabaseTableDefinitionJson
         [JsonIgnore] 
         public IDictionary<string, DatabaseForeignTableJson> ForeignTableByName { get; set; } = null!;
     }
+    
+    public class DatabaseConditionReferenceJson
+    {
+        [JsonProperty(PropertyName = "source_type")]
+        public int SourceType { get; set; }
+        
+        [JsonProperty(PropertyName = "source_group")]
+        public string? SourceGroupColumn { get; set; }
+        
+        [JsonProperty(PropertyName = "source_entry")]
+        public string? SourceEntryColumn { get; set; }
+        
+        [JsonProperty(PropertyName = "source_id")]
+        public string? SourceIdColumn { get; set; }
+        
+        [JsonProperty(PropertyName = "targets")]
+        public IList<DatabaseConditionTargetJson>? Targets { get; set; }
+    }
+
+    public class DatabaseConditionTargetJson
+    {
+        [JsonProperty(PropertyName = "id")]
+        public uint Id { get; set; }
+
+        [JsonProperty(PropertyName = "name")] 
+        public string Name { get; set; } = "";
+    }
 }
\ No newline at end of file
diff --git a/WDE.DatabaseEditors/DbDefinitions/gossip_menu_option.json b/WDE.DatabaseEditors/DbDefinitions/gossip_menu_option.json
index cf5bfe660..48630f566 100644
--- a/WDE.DatabaseEditors/DbDefinitions/gossip_menu_option.json
+++ b/WDE.DatabaseEditors/DbDefinitions/gossip_menu_option.json
@@ -17,6 +17,21 @@
     "MenuID",
     "OptionID"
   ],
+  "conditions": {
+    "source_type": 15,
+    "source_group": "MenuID",
+    "source_entry": "OptionID",
+    "targets": [
+      {
+        "id": 0,
+        "name": "player"
+      },
+      {
+        "id": 1,
+        "name": "gossip sender"
+      }
+    ]
+  },
   "groups": [
     {
       "group_name": "group",
@@ -70,6 +85,11 @@
           "db_column_name": "ActionMenuID",
           "value_type": "GossipMenuParameter"
         },
+        {
+          "is_condition": true,
+          "name": "Conditions",
+          "db_column_name": "conditions"
+        },
         {
           "name": "Action Poi ID",
           "db_column_name": "ActionPoiID",
diff --git a/WDE.DatabaseEditors/History/DatabaseEntityConditionsChangedHistoryAction.cs b/WDE.DatabaseEditors/History/DatabaseEntityConditionsChangedHistoryAction.cs
new file mode 100644
index 000000000..681ddf2c5
--- /dev/null
+++ b/WDE.DatabaseEditors/History/DatabaseEntityConditionsChangedHistoryAction.cs
@@ -0,0 +1,37 @@
+using System.Collections.Generic;
+using WDE.Common.Database;
+using WDE.Common.History;
+using WDE.DatabaseEditors.Models;
+
+namespace WDE.DatabaseEditors.History
+{
+    public class DatabaseEntityConditionsChangedHistoryAction : IHistoryAction
+    {
+        private readonly DatabaseEntity entity;
+        private readonly IReadOnlyList<ICondition>? oldConditions;
+        private readonly IReadOnlyList<ICondition>? newConditions;
+
+        public DatabaseEntityConditionsChangedHistoryAction(DatabaseEntity entity,
+            IReadOnlyList<ICondition>? oldConditions, IReadOnlyList<ICondition>? newConditions)
+        {
+            this.entity = entity;
+            this.oldConditions = oldConditions;
+            this.newConditions = newConditions;
+        }
+        
+        public void Undo()
+        {
+            entity.Conditions = oldConditions;
+        }
+
+        public void Redo()
+        {
+            entity.Conditions = newConditions;
+        }
+
+        public string GetDescription()
+        {
+            return $"Entity {entity.Key} conditions changed";
+        }
+    }
+}
\ No newline at end of file
diff --git a/WDE.DatabaseEditors/Loaders/DatabaseTableDataProvider.cs b/WDE.DatabaseEditors/Loaders/DatabaseTableDataProvider.cs
index 6f180cfd0..4d71815ed 100644
--- a/WDE.DatabaseEditors/Loaders/DatabaseTableDataProvider.cs
+++ b/WDE.DatabaseEditors/Loaders/DatabaseTableDataProvider.cs
@@ -18,16 +18,19 @@ public class DatabaseTableDataProvider : IDatabaseTableDataProvider
         private readonly ITableDefinitionProvider tableDefinitionProvider;
         private readonly IMySqlExecutor sqlExecutor;
         private readonly IMessageBoxService messageBoxService;
+        private readonly IDatabaseProvider databaseProvider;
         private readonly IDatabaseTableModelGenerator tableModelGenerator;
         
         public DatabaseTableDataProvider(ITableDefinitionProvider tableDefinitionProvider, 
             IMySqlExecutor sqlExecutor,
             IMessageBoxService messageBoxService,
+            IDatabaseProvider databaseProvider,
             IDatabaseTableModelGenerator tableModelGenerator)
         {
             this.tableDefinitionProvider = tableDefinitionProvider;
             this.sqlExecutor = sqlExecutor;
             this.messageBoxService = messageBoxService;
+            this.databaseProvider = databaseProvider;
             this.tableModelGenerator = tableModelGenerator;
         }
 
@@ -37,6 +40,7 @@ private string BuildSQLQueryFromTableDefinition(in DatabaseTableDefinitionJson t
             var tablePrimaryKey = tableDefinitionJson.TablePrimaryKeyColumnName;
             var columns = tableDefinitionJson.Groups
                 .SelectMany(x => x.Fields)
+                .Where(x => !x.IsConditionColumn)
                 .Select(x => $"`{x.ForeignTable ?? tableName}`.`{x.DbColumnName}`")
                 .Distinct();
             var names = string.Join(",", columns);
@@ -59,14 +63,54 @@ private string BuildSQLQueryFromTableDefinition(in DatabaseTableDefinitionJson t
                 return null;
             
             IList<Dictionary<string, (Type, object)>>? result = null;
+            IDatabaseProvider.ConditionKeyMask keyMask = IDatabaseProvider.ConditionKeyMask.None;
+            if (definition.Condition != null)
+            {
+                if (definition.Condition.SourceEntryColumn != null)
+                    keyMask |= IDatabaseProvider.ConditionKeyMask.SourceEntry;
+                if (definition.Condition.SourceGroupColumn != null)
+                    keyMask |= IDatabaseProvider.ConditionKeyMask.SourceGroup;
+                if (definition.Condition.SourceIdColumn != null)
+                    keyMask |= IDatabaseProvider.ConditionKeyMask.SourceId;
+            }
 
             if (keys.Length > 0)
             {
                 var sqlStatement = BuildSQLQueryFromTableDefinition(definition, keys);
-
                 try
                 {
                     result = await sqlExecutor.ExecuteSelectSql(sqlStatement);
+
+                    if (definition.Condition != null)
+                    {
+                        foreach (var row in result)
+                        {
+                            int? sourceGroup = null, sourceEntry = null, sourceId = null;
+
+                            if (definition.Condition.SourceGroupColumn != null &&
+                                row.TryGetValue(definition.Condition.SourceGroupColumn, out var groupData) &&
+                                int.TryParse(groupData.Item2.ToString(), out var groupInt))
+                                sourceGroup = groupInt;
+
+                            if (definition.Condition.SourceEntryColumn != null &&
+                                row.TryGetValue(definition.Condition.SourceEntryColumn, out var entryData) &&
+                                int.TryParse(entryData.Item2.ToString(), out var entryInt))
+                                sourceEntry = entryInt;
+
+                            if (definition.Condition.SourceIdColumn != null &&
+                                row.TryGetValue(definition.Condition.SourceIdColumn, out var idData) &&
+                                int.TryParse(idData.Item2.ToString(), out var idInt))
+                                sourceId = idInt;
+
+                            var conditionList = await databaseProvider.GetConditionsForAsync(keyMask,
+                                new IDatabaseProvider.ConditionKey(definition.Condition.SourceType, sourceGroup,
+                                    sourceEntry, sourceId));
+                            if (conditionList?.Count > 0)
+                            {
+                                row.Add("conditions", (typeof(IList<IConditionLine>), conditionList));
+                            }
+                        }
+                    }
                 }
                 catch (IMySqlExecutor.CannotConnectToDatabaseException e)
                 {
diff --git a/WDE.DatabaseEditors/Models/DatabaseEntity.cs b/WDE.DatabaseEditors/Models/DatabaseEntity.cs
index 74700cc9e..418bebec6 100644
--- a/WDE.DatabaseEditors/Models/DatabaseEntity.cs
+++ b/WDE.DatabaseEditors/Models/DatabaseEntity.cs
@@ -1,12 +1,32 @@
 using System.Collections.Generic;
+using System.ComponentModel;
+using System.Linq;
+using System.Runtime.CompilerServices;
+using WDE.Common.Annotations;
+using WDE.Common.Database;
 using WDE.Common.History;
+using WDE.DatabaseEditors.History;
 
 namespace WDE.DatabaseEditors.Models
 {
-    public class DatabaseEntity
+    public class DatabaseEntity : INotifyPropertyChanged
     {
         public Dictionary<string, IDatabaseField> Cells { get; }
-        
+
+        private IReadOnlyList<ICondition>? conditions;
+
+        public IReadOnlyList<ICondition>? Conditions
+        {
+            get => conditions;
+            set
+            {
+                var old = conditions;
+                conditions = value;
+                OnAction?.Invoke(new DatabaseEntityConditionsChangedHistoryAction(this, old, value));
+                OnPropertyChanged();
+            }
+        }
+
         public IEnumerable<IDatabaseField> Fields => Cells.Values;
 
         public event System.Action<IHistoryAction>? OnAction;
@@ -15,11 +35,12 @@ public class DatabaseEntity
         
         public uint Key { get; }
         
-        public DatabaseEntity(bool existInDatabase, uint key, Dictionary<string, IDatabaseField> cells)
+        public DatabaseEntity(bool existInDatabase, uint key, Dictionary<string, IDatabaseField> cells, IReadOnlyList<ICondition>? conditions)
         {
             ExistInDatabase = existInDatabase;
             Key = key;
             Cells = cells;
+            Conditions = conditions;
             foreach (var databaseField in Cells)
             {
                 databaseField.Value.OnChanged += action =>
@@ -41,7 +62,21 @@ public DatabaseEntity Clone()
             var fields = new Dictionary<string, IDatabaseField>();
             foreach (var field in Cells)
                 fields[field.Key] = field.Value.Clone();
-            return new DatabaseEntity(ExistInDatabase, Key, fields);
+            
+            return new DatabaseEntity(ExistInDatabase, Key, fields, Conditions == null ? null : CloneConditions(Conditions));
+        }
+
+        private IReadOnlyList<ICondition> CloneConditions(IReadOnlyList<ICondition> conditions)
+        {
+            return conditions.Select(c => new AbstractCondition(c)).ToList<ICondition>();
+        }
+
+        public event PropertyChangedEventHandler? PropertyChanged;
+
+        [NotifyPropertyChangedInvocator]
+        protected virtual void OnPropertyChanged([CallerMemberName] string? propertyName = null)
+        {
+            PropertyChanged?.Invoke(this, new PropertyChangedEventArgs(propertyName));
         }
     }
 }
\ No newline at end of file
diff --git a/WDE.DatabaseEditors/QueryGenerators/QueryGenerator.cs b/WDE.DatabaseEditors/QueryGenerators/QueryGenerator.cs
index ae3409b57..30df7074e 100644
--- a/WDE.DatabaseEditors/QueryGenerators/QueryGenerator.cs
+++ b/WDE.DatabaseEditors/QueryGenerators/QueryGenerator.cs
@@ -2,6 +2,7 @@
 using System.Collections.Generic;
 using System.Linq;
 using System.Text;
+using WDE.Common.Database;
 using WDE.DatabaseEditors.Data.Interfaces;
 using WDE.DatabaseEditors.Data.Structs;
 using WDE.DatabaseEditors.Models;
@@ -15,10 +16,12 @@ namespace WDE.DatabaseEditors.QueryGenerators
     public class QueryGenerator : IQueryGenerator
     {       
         private readonly ITableDefinitionProvider tableDefinitionProvider;
+        private readonly IConditionQueryGenerator conditionQueryGenerator;
 
-        public QueryGenerator(ITableDefinitionProvider tableDefinitionProvider)
+        public QueryGenerator(ITableDefinitionProvider tableDefinitionProvider, IConditionQueryGenerator conditionQueryGenerator)
         {
             this.tableDefinitionProvider = tableDefinitionProvider;
+            this.conditionQueryGenerator = conditionQueryGenerator;
         }
         
         public string GenerateQuery(ICollection<uint> keys, IDatabaseTableData tableData)
@@ -76,10 +79,70 @@ private string GenerateInsertQuery(ICollection<uint> keys, IDatabaseTableData ta
                 foreach (var line in duplicates)
                     query.AppendLine(" -- " + line);
             }
-            
+
+            if (tableData.TableDefinition.Condition != null)
+            {
+                query.AppendLine(BuildConditionsDeleteQuery(keys, tableData));
+                List<IConditionLine> conditions = new();
+                int sourceType = tableData.TableDefinition.Condition.SourceType;
+                    
+                foreach (var entity in tableData.Entities)
+                {
+                    if (entity.Conditions == null)
+                        continue;
+
+                    int sourceGroup = 0;
+                    int sourceEntry = 0;
+                    int sourceId = 0;
+
+                    if (tableData.TableDefinition.Condition.SourceEntryColumn != null &&
+                        entity.GetCell(tableData.TableDefinition.Condition.SourceEntryColumn) is DatabaseField<long>
+                            entryCell)
+                        sourceEntry = (int)entryCell.Current.Value;
+                    
+                    if (tableData.TableDefinition.Condition.SourceGroupColumn != null &&
+                        entity.GetCell(tableData.TableDefinition.Condition.SourceGroupColumn) is DatabaseField<long>
+                            groupCell)
+                        sourceGroup = (int)groupCell.Current.Value;
+                    
+                    if (tableData.TableDefinition.Condition.SourceIdColumn != null &&
+                        entity.GetCell(tableData.TableDefinition.Condition.SourceIdColumn) is DatabaseField<long>
+                            idCell)
+                        sourceId = (int)idCell.Current.Value;
+                    
+                    foreach (var condition in entity.Conditions)
+                        conditions.Add(new AbstractConditionLine(sourceType, sourceGroup, sourceEntry, sourceId, condition));
+                }
+                query.AppendLine(conditionQueryGenerator.BuildInsertQuery(conditions));
+            }
+
             return query.ToString();
         }
 
+        private string BuildConditionsDeleteQuery(ICollection<uint> keys, IDatabaseTableData tableData)
+        {
+            if (tableData.TableDefinition.Condition == null)
+                return "";
+
+            string? columnKey = null;
+
+            if (tableData.TableDefinition.Condition.SourceEntryColumn ==
+                tableData.TableDefinition.TablePrimaryKeyColumnName)
+                columnKey = "SourceEntry";
+            else if (tableData.TableDefinition.Condition.SourceGroupColumn ==
+                     tableData.TableDefinition.TablePrimaryKeyColumnName)
+                columnKey = "SourceGroup";
+            else if (tableData.TableDefinition.Condition.SourceIdColumn ==
+                     tableData.TableDefinition.TablePrimaryKeyColumnName)
+                columnKey = "SourceId";
+
+            if (columnKey == null)
+                throw new Exception("No condition source group/entry/id is table primary key. Unable to generate SQL.");
+            
+            string c = string.Join(", ", keys.Distinct());
+            return $"DELETE FROM `conditions` WHERE `SourceTypeOrReferenceId` = {tableData.TableDefinition.Condition.SourceType} AND `{columnKey}` IN ({c});";
+        }
+
         public string GenerateUpdateFieldQuery(DatabaseTableDefinitionJson table, DatabaseEntity entity,
             IDatabaseField field)
         {
diff --git a/WDE.DatabaseEditors/ViewModels/MultiRow/DatabaseCellViewModel.cs b/WDE.DatabaseEditors/ViewModels/MultiRow/DatabaseCellViewModel.cs
index 8c2e50960..0f5fcd782 100644
--- a/WDE.DatabaseEditors/ViewModels/MultiRow/DatabaseCellViewModel.cs
+++ b/WDE.DatabaseEditors/ViewModels/MultiRow/DatabaseCellViewModel.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Windows.Input;
 using WDE.DatabaseEditors.Data.Structs;
 using WDE.DatabaseEditors.Models;
 using WDE.MVVM;
@@ -10,8 +11,8 @@ public class DatabaseCellViewModel : ObservableBase
     {
         public DatabaseEntityViewModel Parent { get; }
         public DatabaseEntity ParentEntity { get; }
-        public IDatabaseField TableField { get; }
-        public IParameterValue ParameterValue { get; }
+        public IDatabaseField? TableField { get; }
+        public IParameterValue? ParameterValue { get; }
         public bool IsVisible { get; private set; } = true;
         public string? OriginalValueTooltip { get; private set; }
         public bool CanBeNull { get; }
@@ -19,6 +20,8 @@ public class DatabaseCellViewModel : ObservableBase
         public bool CanBeSetToNull => CanBeNull && !IsReadOnly;
         public bool CanBeReverted => !IsReadOnly;
         public int ColumnIndex { get; }
+        public ICommand? ActionCommand { get; }
+        public string ActionLabel { get; set; } = "";
         
         public string ColumnName { get; }
 
@@ -42,6 +45,23 @@ public DatabaseCellViewModel(int columnIndex, DatabaseColumnJson columnDefinitio
             }));
         }
 
+        public DatabaseCellViewModel(int columnIndex, string columnName, ICommand action, DatabaseEntityViewModel parent, DatabaseEntity entity, System.IObservable<string> label)
+        {
+            Parent = parent;
+            ParentEntity = entity;
+            ColumnIndex = columnIndex * 2;
+            CanBeNull = false;
+            IsReadOnly = false;
+            ColumnName = columnName;
+            OriginalValueTooltip = null;
+            ActionCommand = action;
+            AutoDispose(label.SubscribeAction(s =>
+            {
+                ActionLabel = s;
+                RaisePropertyChanged(nameof(ActionLabel));
+            }));
+        }
+
         public bool AsBoolValue
         {
             get => ((ParameterValue as ParameterValue<long>)?.Value ?? 0) != 0;
diff --git a/WDE.DatabaseEditors/ViewModels/MultiRow/DatabaseColumnHeaderViewModel.cs b/WDE.DatabaseEditors/ViewModels/MultiRow/DatabaseColumnHeaderViewModel.cs
index 3b8f2b7ae..705638e7d 100644
--- a/WDE.DatabaseEditors/ViewModels/MultiRow/DatabaseColumnHeaderViewModel.cs
+++ b/WDE.DatabaseEditors/ViewModels/MultiRow/DatabaseColumnHeaderViewModel.cs
@@ -17,5 +17,11 @@ public DatabaseColumnHeaderViewModel(DatabaseColumnJson column)
             Help = column.Help;
             PreferredWidth = column.PreferredWidth;
         }
+
+        public DatabaseColumnHeaderViewModel(string name, string databaseName)
+        {
+            Name = name;
+            DatabaseName = databaseName;
+        }
     }
 }
\ No newline at end of file
diff --git a/WDE.DatabaseEditors/ViewModels/MultiRow/MultiRowDbTableEditorViewModel.cs b/WDE.DatabaseEditors/ViewModels/MultiRow/MultiRowDbTableEditorViewModel.cs
index 5c8dd835c..dd80878a7 100644
--- a/WDE.DatabaseEditors/ViewModels/MultiRow/MultiRowDbTableEditorViewModel.cs
+++ b/WDE.DatabaseEditors/ViewModels/MultiRow/MultiRowDbTableEditorViewModel.cs
@@ -23,6 +23,8 @@
 using WDE.DatabaseEditors.Models;
 using WDE.DatabaseEditors.QueryGenerators;
 using WDE.DatabaseEditors.Solution;
+using WDE.MVVM;
+using WDE.MVVM.Observable;
 
 namespace WDE.DatabaseEditors.ViewModels.MultiRow
 {
@@ -34,6 +36,7 @@ public class MultiRowDbTableEditorViewModel : ViewModelBase
         private readonly IMySqlExecutor mySqlExecutor;
         private readonly IQueryGenerator queryGenerator;
         private readonly IDatabaseTableModelGenerator modelGenerator;
+        private readonly IConditionEditService conditionEditService;
         private readonly IDatabaseTableDataProvider tableDataProvider;
 
         private Dictionary<uint, DatabaseEntitiesGroupViewModel> byEntryGroups = new();
@@ -48,6 +51,7 @@ public class MultiRowDbTableEditorViewModel : ViewModelBase
         public AsyncAutoCommand AddNewCommand { get; }
         public DelegateCommand<DatabaseCellViewModel?> RemoveTemplateCommand { get; }
         public AsyncAutoCommand<DatabaseCellViewModel?> RevertCommand { get; }
+        public AsyncAutoCommand<DatabaseCellViewModel?> EditConditionsCommand { get; }
         public DelegateCommand<DatabaseCellViewModel?> SetNullCommand { get; }
         public DelegateCommand<DatabaseCellViewModel?> DuplicateCommand { get; }
         public DelegateCommand<DatabaseEntitiesGroupViewModel> AddRowCommand { get; }
@@ -63,7 +67,8 @@ public class MultiRowDbTableEditorViewModel : ViewModelBase
             IEventAggregator eventAggregator, ISolutionManager solutionManager, 
             IParameterFactory parameterFactory, ISolutionTasksService solutionTasksService,
             ISolutionItemNameRegistry solutionItemName, IMySqlExecutor mySqlExecutor,
-            IQueryGenerator queryGenerator, IDatabaseTableModelGenerator modelGenerator) : base(history, solutionItem, solutionItemName, 
+            IQueryGenerator queryGenerator, IDatabaseTableModelGenerator modelGenerator,
+            IConditionEditService conditionEditService) : base(history, solutionItem, solutionItemName, 
             solutionManager, solutionTasksService, eventAggregator, 
             queryGenerator, tableDataProvider, messageBoxService, taskRunner, parameterFactory)
         {
@@ -75,12 +80,14 @@ public class MultiRowDbTableEditorViewModel : ViewModelBase
             this.mySqlExecutor = mySqlExecutor;
             this.queryGenerator = queryGenerator;
             this.modelGenerator = modelGenerator;
+            this.conditionEditService = conditionEditService;
 
             OpenParameterWindow = new AsyncAutoCommand<DatabaseCellViewModel>(EditParameter);
             RemoveTemplateCommand = new DelegateCommand<DatabaseCellViewModel?>(RemoveTemplate, vm => vm != null);
-            RevertCommand = new AsyncAutoCommand<DatabaseCellViewModel?>(Revert, cell => cell is DatabaseCellViewModel vm && vm.CanBeReverted && vm.TableField.IsModified);
+            RevertCommand = new AsyncAutoCommand<DatabaseCellViewModel?>(Revert, cell => cell is DatabaseCellViewModel vm && vm.CanBeReverted && (vm.TableField?.IsModified ?? false));
             SetNullCommand = new DelegateCommand<DatabaseCellViewModel?>(SetToNull, vm => vm != null && vm.CanBeSetToNull);
             DuplicateCommand = new DelegateCommand<DatabaseCellViewModel?>(Duplicate, vm => vm != null);
+            EditConditionsCommand = new AsyncAutoCommand<DatabaseCellViewModel?>(EditConditions);
             AddRowCommand = new DelegateCommand<DatabaseEntitiesGroupViewModel>(AddRow);
             AddNewCommand = new AsyncAutoCommand(AddNewEntity);
             
@@ -144,7 +151,7 @@ await messageBoxService.ShowDialog(new MessageBoxFactory<bool>()
         private void SetToNull(DatabaseCellViewModel? view)
         {
             if (view != null && view.CanBeNull && !view.IsReadOnly) 
-                view.ParameterValue.SetNull();
+                view.ParameterValue?.SetNull();
         }
 
         private void Duplicate(DatabaseCellViewModel? view)
@@ -155,13 +162,27 @@ private void Duplicate(DatabaseCellViewModel? view)
                 ForceInsertEntity(duplicate, 0);
             }
         }
+
+        private async Task EditConditions(DatabaseCellViewModel? view)
+        {
+            if (view == null)
+                return;
+            
+            var conditionList = view.ParentEntity.Conditions;
+            
+            var newConditions = await conditionEditService.EditConditions(tableDefinition.Condition!.SourceType, conditionList);
+            if (newConditions == null)
+                return;
+            
+            view.ParentEntity.Conditions = newConditions.ToList();
+        }
         
         private async Task Revert(DatabaseCellViewModel? view)
         {
             if (view == null || view.IsReadOnly)
                 return;
             
-            view.ParameterValue.Revert();
+            view.ParameterValue?.Revert();
         }
 
         private void RemoveTemplate(DatabaseCellViewModel? view)
@@ -199,7 +220,7 @@ protected override async Task InternalLoadData(DatabaseTableData data)
             autoIncrementColumn = columns.FirstOrDefault(c => c.AutoIncrement);
             Columns.Clear();
             Columns.AddRange(columns.Select(c => new DatabaseColumnHeaderViewModel(c)));
-                
+            
             foreach (var entity in solutionItem.Entries)
                 EnsureKey(entity.Key);
 
@@ -299,29 +320,39 @@ public override bool ForceInsertEntity(DatabaseEntity entity, int index)
             int columnIndex = 0;
             foreach (var column in columns)
             {
-                var cell = entity.GetCell(column.DbColumnName);
-                if (cell == null)
-                    throw new Exception("this should never happen");
+                DatabaseCellViewModel cellViewModel;
 
-                IParameterValue parameterValue = null!;
-                if (cell is DatabaseField<long> longParam)
-                {
-                    parameterValue = new ParameterValue<long>(longParam.Current, longParam.Original, parameterFactory.Factory(column.ValueType));
-                }
-                else if (cell is DatabaseField<string> stringParam)
+                if (column.IsConditionColumn)
                 {
-                    parameterValue = new ParameterValue<string>(stringParam.Current, stringParam.Original, StringParameter.Instance);
+                    var label = entity.ToObservable(e => e.Conditions).Select(c => "Edit (" + (c?.Count ?? 0) + ")");
+                    cellViewModel = AutoDispose(new DatabaseCellViewModel(columnIndex, "conditions", EditConditionsCommand, row, entity, label));
                 }
-                else if (cell is DatabaseField<float> floatParameter)
+                else
                 {
-                    parameterValue = new ParameterValue<float>(floatParameter.Current, floatParameter.Original, FloatParameter.Instance);
-                }
+                    var cell = entity.GetCell(column.DbColumnName);
+                    if (cell == null)
+                        throw new Exception("this should never happen");
+
+                    IParameterValue parameterValue = null!;
+                    if (cell is DatabaseField<long> longParam)
+                    {
+                        parameterValue = new ParameterValue<long>(longParam.Current, longParam.Original, parameterFactory.Factory(column.ValueType));
+                    }
+                    else if (cell is DatabaseField<string> stringParam)
+                    {
+                        parameterValue = new ParameterValue<string>(stringParam.Current, stringParam.Original, StringParameter.Instance);
+                    }
+                    else if (cell is DatabaseField<float> floatParameter)
+                    {
+                        parameterValue = new ParameterValue<float>(floatParameter.Current, floatParameter.Original, FloatParameter.Instance);
+                    }
 
-                var cellViewModel = AutoDispose(new DatabaseCellViewModel(columnIndex, column, row, entity, cell, parameterValue));
+                    cellViewModel = AutoDispose(new DatabaseCellViewModel(columnIndex, column, row, entity, cell, parameterValue));
+                }
                 row.Cells.Add(cellViewModel);
                 columnIndex++;
             }
-
+            
             Entities.Insert(index, entity);
             EnsureKey(entity.Key);