[UX] Template windowed editor

Author: BAndysc

WDE.DatabaseEditors.Avalonia/Views/Template/TemplateDbTableEditorToolBar.axaml

@@ -12,7 +12,7 @@
              IsEnabled="{Binding TaskInProgress, Converter={x:Static converters:InverseBoolConverter.Instance}}"
              x:Class="WDE.DatabaseEditors.Avalonia.Views.Template.TemplateDbTableEditorToolBar">
     <controls:ToolbarPanel>
-        <components:ButtonImage Command="{CompiledBinding AddNewCommand, Mode=OneTime}" Text="Add new" Image="Icons/icon_add.png" />
+        <components:ButtonImage IsVisible="{CompiledBinding AllowMultipleKeys}" Command="{CompiledBinding AddNewCommand, Mode=OneTime}" Text="Add new" Image="Icons/icon_add.png" />
         <ItemsPresenter Items="{CompiledBinding Commands}">
             <ItemsPresenter.ItemsPanel>
                 <ItemsPanelTemplate>

WDE.DatabaseEditors/QueryGenerators/ICustomQueryGenerator.cs

@@ -0,0 +1,14 @@
+using System.Collections.Generic;
+using WDE.Common.Services;
+using WDE.DatabaseEditors.Models;
+using WDE.Module.Attributes;
+using WDE.SqlQueryGenerator;
+
+namespace WDE.DatabaseEditors.QueryGenerators;
+
+[NonUniqueProvider]
+public interface ICustomQueryGenerator
+{
+    string TableName { get; }
+    IQuery Generate(IReadOnlyList<DatabaseKey> keys, IReadOnlyList<DatabaseKey>? deletedKeys, IDatabaseTableData data);
+}

WDE.DatabaseEditors/QueryGenerators/QueryGenerator.cs

@@ -26,15 +26,22 @@ public class QueryGenerator : IQueryGenerator
         private readonly IQueryGenerator<ConditionDeleteModel> conditionDeleteGenerator;
         private readonly IConditionQueryGenerator conditionQueryGenerator;
 
+        private Dictionary<string, ICustomQueryGenerator> customQueryGenerators = new();
+
         public QueryGenerator(ICreatureStatCalculatorService calculatorService,
             IParameterFactory parameterFactory,
             IQueryGenerator<ConditionDeleteModel> conditionDeleteGenerator,
-            IConditionQueryGenerator conditionQueryGenerator)
+            IConditionQueryGenerator conditionQueryGenerator,
+            IEnumerable<ICustomQueryGenerator> customQueryGenerators)
         {
             this.calculatorService = calculatorService;
             this.parameterFactory = parameterFactory;
             this.conditionDeleteGenerator = conditionDeleteGenerator;
             this.conditionQueryGenerator = conditionQueryGenerator;
+            foreach (var gen in customQueryGenerators)
+            {
+                this.customQueryGenerators[gen.TableName] = gen;
+            }
         }
 
         public IQuery GenerateQuery(IReadOnlyList<DatabaseKey> keys, IReadOnlyList<DatabaseKey>? deletedKeys, IDatabaseTableData tableData)
@@ -65,6 +72,11 @@ public IQuery GenerateSingleRecordQuery(IReadOnlyList<DatabaseKey> keys, IReadOn
                     query.Add(GenerateUpdateQuery(tableData.TableDefinition, entity));
             }
 
+            if (customQueryGenerators.TryGetValue(tableData.TableDefinition.TableName, out var customQueryGenerator))
+            {
+                query.Add(customQueryGenerator.Generate(keys, deletedKeys, tableData));
+            }
+            
             return query.Close();
         }
 

WDE.DatabaseEditors/Services/TableEditorPickerService.cs

@@ -17,6 +17,7 @@
 using WDE.DatabaseEditors.ViewModels.MultiRow;
 using WDE.DatabaseEditors.ViewModels.OneToOneForeignKey;
 using WDE.DatabaseEditors.ViewModels.SingleRow;
+using WDE.DatabaseEditors.ViewModels.Template;
 using WDE.Module.Attributes;
 using WDE.MVVM;
 using WDE.MVVM.Observable;
@@ -156,9 +157,6 @@ public async Task ShowTable(string table, string? condition, DatabaseKey? defaul
         if (definition == null)
             throw new UnsupportedTableException(table);
 
-        if (definition.RecordMode == RecordMode.Template)
-            throw new Exception("TemplateMode not (yet?) supported");
-
         ViewModelBase viewModelBase;
         bool openIsNoSaveMode = false;
         if (definition.RecordMode == RecordMode.SingleRow)
@@ -174,7 +172,7 @@ public async Task ShowTable(string table, string? condition, DatabaseKey? defaul
                 singleRow.FilterViewModel.ApplyFilter.Execute(null);
             }
         }
-        else
+        else if (definition.RecordMode == RecordMode.MultiRecord)
         {
             Debug.Assert(definition.RecordMode == RecordMode.MultiRecord);
             Debug.Assert(condition == null);
@@ -185,6 +183,17 @@ public async Task ShowTable(string table, string? condition, DatabaseKey? defaul
             multiRow.AllowMultipleKeys = false;
             viewModelBase = multiRow;
         }
+        else
+        {
+            Debug.Assert(definition.RecordMode == RecordMode.Template);
+            Debug.Assert(condition == null);
+            Debug.Assert(defaultPartialKey.HasValue);
+            var solutionItem = new DatabaseTableSolutionItem(defaultPartialKey.Value, true, false, definition.Id, false);
+            openIsNoSaveMode = await CheckIfItemIsOpened(solutionItem, definition);
+            var template = containerProvider.Resolve<TemplateDbTableEditorViewModel>((typeof(DatabaseTableSolutionItem), solutionItem));
+            template.AllowMultipleKeys = false;
+            viewModelBase = template;
+        }
 
         var viewModel = containerProvider.Resolve<RowPickerViewModel>((typeof(ViewModelBase), viewModelBase), (typeof(bool), openIsNoSaveMode));
         viewModel.DisablePicking = true;

WDE.DatabaseEditors/ViewModels/Template/TemplateDbTableEditorViewModel.cs

@@ -60,6 +60,8 @@ public partial class TemplateDbTableEditorViewModel : ViewModelBase
         public IObservable<Func<DatabaseRowViewModel, bool>> CurrentFilter { get; }
         public SourceList<DatabaseRowViewModel> Rows { get; } = new();
         private HashSet<(DatabaseKey key, string columnName)> forceUpdateCells = new HashSet<(DatabaseKey, string)>();
+
+        [Notify] private bool allowMultipleKeys = true;
 
         public AsyncAutoCommand<DatabaseCellViewModel?> RemoveTemplateCommand { get; }
         public AsyncAutoCommand<DatabaseCellViewModel?> RevertCommand { get; }
@@ -143,6 +145,9 @@ protected override void UpdateSolutionItem()
 
         private async Task AddNewEntity()
         {
+            if (!allowMultipleKeys)
+                return;
+            
             var parameter = parameterFactory.Factory(tableDefinition.Picker);
             var selected = await itemFromListProvider.GetItemFromList(parameter.Items, false);
             if (!selected.HasValue)