[Table Editor] Added 'Generate insert query' to the context menu

BAndysc · BAndysc · commit 5e03a590803d · 2022-03-23T13:45:19.000+01:00
diff --git a/AvaloniaStyles/Controls/FastTableView/ColorsDark.axaml b/AvaloniaStyles/Controls/FastTableView/ColorsDark.axaml
@@ -19,5 +19,6 @@
         <Pen Thickness="1" Brush="#0FFFFFFF" x:Key="FastTableView.ButtonBackgroundPen" />
         <Pen Thickness="1" Brush="#15FFFFFF" x:Key="FastTableView.ButtonBackgroundHoverPen" />
         <Pen Thickness="1" Brush="#08FFFFFF" x:Key="FastTableView.ButtonBackgroundPressedPen" />
+        <Pen Thickness="1" Brush="#10FFFFFF" x:Key="FastTableView.ButtonBackgroundDisabledPen" />
     </Styles.Resources>
 </Styles>
diff --git a/AvaloniaStyles/Controls/FastTableView/ColorsLight.axaml b/AvaloniaStyles/Controls/FastTableView/ColorsLight.axaml
@@ -19,5 +19,6 @@
         <Pen Thickness="1" Brush="White" x:Key="FastTableView.ButtonBackgroundPen" />
         <Pen Thickness="1" Brush="#F5F5F5" x:Key="FastTableView.ButtonBackgroundHoverPen" />
         <Pen Thickness="1" Brush="#D7D7D7" x:Key="FastTableView.ButtonBackgroundPressedPen" />
+        <Pen Thickness="1" Brush="#D2D2D2" x:Key="FastTableView.ButtonBackgroundDisabledPen" />
     </Styles.Resources>
 </Styles>
diff --git a/WDE.DatabaseEditors.Avalonia/Views/SingleRow/CustomCellDrawer.cs b/WDE.DatabaseEditors.Avalonia/Views/SingleRow/CustomCellDrawer.cs
@@ -27,12 +27,11 @@ public class CustomCellDrawer : CustomCellDrawerInteractorBase, ICustomCellDrawe
     private static IPen ButtonBackgroundPen = new Pen(new SolidColorBrush(Colors.White), 0);
     private static IPen ButtonBackgroundHoverPen = new Pen(new SolidColorBrush(Color.FromRgb(245, 245, 245)), 0);
     private static IPen ButtonBackgroundPressedPen = new Pen(new SolidColorBrush(Color.FromRgb(215, 215, 215)), 0);
-
+    private static IPen ButtonBackgroundDisabledPen = new Pen(new SolidColorBrush(Color.FromRgb(170, 170, 170)), 0);
+    
     private Point mouseCursor;
     private bool leftPressed;
     
-    
-
     static CustomCellDrawer()
     {
         ModifiedCellPen.GetResource("FastTableView.ModifiedCellPen", ModifiedCellPen, out ModifiedCellPen);
@@ -42,6 +41,7 @@ static CustomCellDrawer()
         ButtonBackgroundPen.GetResource("FastTableView.ButtonBackgroundPen", ButtonBackgroundPen, out ButtonBackgroundPen);
         ButtonBackgroundHoverPen.GetResource("FastTableView.ButtonBackgroundHoverPen", ButtonBackgroundHoverPen, out ButtonBackgroundHoverPen);
         ButtonBackgroundPressedPen.GetResource("FastTableView.ButtonBackgroundPressedPen", ButtonBackgroundPressedPen, out ButtonBackgroundPressedPen);
+        ButtonBackgroundDisabledPen.GetResource("FastTableView.ButtonBackgroundDisabledPen", ButtonBackgroundDisabledPen, out ButtonBackgroundDisabledPen);
     }
 
     public void DrawRow(DrawingContext context, ITableRow r, Rect rect)
@@ -65,13 +65,13 @@ public bool UpdateCursor(Point point, bool leftPressed)
         return true;
     }
 
-    private void DrawButton(DrawingContext context, Rect rect, string text, int margin)
+    private void DrawButton(DrawingContext context, Rect rect, string text, int margin, bool enabled = true)
     {
         rect = rect.Deflate(margin);
 
         bool isOver = rect.Contains(mouseCursor);
             
-        context.DrawRectangle((isOver ? (leftPressed ? ButtonBackgroundPressedPen : ButtonBackgroundHoverPen) : ButtonBackgroundPen).Brush, ButtonBorderPen, rect, 4, 4);
+        context.DrawRectangle((!enabled ? ButtonBackgroundDisabledPen : isOver ? (leftPressed ? ButtonBackgroundPressedPen : ButtonBackgroundHoverPen) : ButtonBackgroundPen).Brush, ButtonBorderPen, rect, 4, 4);
 
         var state = context.PushClip(rect);
         var ft = new FormattedText
@@ -92,7 +92,7 @@ public bool Draw(DrawingContext context, ref Rect rect, ITableCell c)
 
         if (cell.ActionCommand != null)
         {
-            DrawButton(context, rect, cell.ActionLabel, 3);
+            DrawButton(context, rect, cell.ActionLabel, 3, cell.ActionCommand.CanExecute(null));
             return true;
         }
         
diff --git a/WDE.DatabaseEditors.Avalonia/Views/SingleRow/CustomCellInteractor.cs b/WDE.DatabaseEditors.Avalonia/Views/SingleRow/CustomCellInteractor.cs
@@ -30,9 +30,10 @@ public bool SpawnEditorFor(string? initialText, Visual parent, Rect rect, ITable
         if (c is not SingleRecordDatabaseCellViewModel cell)
             return false;
 
-        if (cell.ActionCommand != null && cell.ActionCommand.CanExecute(null))
+        if (cell.ActionCommand != null)
         {
-            cell.ActionCommand.Execute(null);
+            if (cell.ActionCommand.CanExecute(null))
+                cell.ActionCommand.Execute(null);
             return true;
         }
         
diff --git a/WDE.DatabaseEditors.Avalonia/Views/SingleRow/SingleRowDbTableEditorView.axaml b/WDE.DatabaseEditors.Avalonia/Views/SingleRow/SingleRowDbTableEditorView.axaml
@@ -180,6 +180,8 @@
                             <MenuItem Header="-"></MenuItem>
                             <MenuItem Header="Delete row" Command="{CompiledBinding DeleteRowSelectedCommand}" />
                             <MenuItem Header="Duplicate row"  Command="{CompiledBinding DuplicateSelectedCommand}"/>
+                            <MenuItem Header="-"></MenuItem>
+                            <MenuItem Header="Generate insert query"  Command="{CompiledBinding GenerateInsertQueryForSelectedCommand}"/>
                         </MenuFlyout>
                     </fastTableView:VeryFastTableView.ContextFlyout>
                 </fastTableView:VeryFastTableView>
diff --git a/WDE.DatabaseEditors/Loaders/DatabaseTableDataProvider.cs b/WDE.DatabaseEditors/Loaders/DatabaseTableDataProvider.cs
@@ -120,6 +120,8 @@ public async Task<long> GetCount(string definitionId, string? customWhere, IEnum
             try
             {
                 var result = await sqlExecutor.ExecuteSelectSql(sql);
+                if (result.Count == 0)
+                    return 0;
                 return Convert.ToInt64(result[0]["num"].Item2);
             }
             catch (IMySqlExecutor.CannotConnectToDatabaseException)
diff --git a/WDE.DatabaseEditors/QueryGenerators/IQueryGenerator.cs b/WDE.DatabaseEditors/QueryGenerators/IQueryGenerator.cs
@@ -8,6 +8,7 @@ namespace WDE.DatabaseEditors.QueryGenerators
 {
     public interface IQueryGenerator
     {
+        public IQuery GenerateInsertQuery(IReadOnlyList<DatabaseKey> keys, IDatabaseTableData tableData);
         public IQuery GenerateQuery(IReadOnlyList<DatabaseKey> keys, IReadOnlyList<DatabaseKey>? deletedKeys, IDatabaseTableData tableData);
         public IQuery GenerateUpdateFieldQuery(DatabaseTableDefinitionJson table, DatabaseEntity entity, IDatabaseField field);
         public IQuery GenerateDeleteQuery(DatabaseTableDefinitionJson table, DatabaseEntity entity);
diff --git a/WDE.DatabaseEditors/QueryGenerators/QueryGenerator.cs b/WDE.DatabaseEditors/QueryGenerators/QueryGenerator.cs
@@ -101,7 +101,10 @@ private static void GeneratePrimaryKeyDeletion(DatabaseTableDefinitionJson defin
                 {
                     table = query.Table(foreign.TableName);
                     if (keys.Count > 1 && keys[0].Count == 1)
-                        table.WhereIn(foreign.ForeignKeys[0], keys.Select(k => k[0])).Delete();
+                    {
+                        foreach (var chunk in keys.Select(k => k[0]).Chunk(128))
+                            table.WhereIn(foreign.ForeignKeys[0], chunk).Delete();
+                    }
                     else
                     {
                         foreach (var key in keys)
@@ -112,7 +115,10 @@ private static void GeneratePrimaryKeyDeletion(DatabaseTableDefinitionJson defin
             
             table = query.Table(definition.TableName);
             if (keys.Count > 1 && keys[0].Count == 1)
-                table.WhereIn(definition.TablePrimaryKeyColumnName, keys.Select(k => k[0])).Delete();
+            {
+                foreach (var chunk in keys.Select(k => k[0]).Chunk(128))
+                    table.WhereIn(definition.TablePrimaryKeyColumnName, chunk).Delete();
+            }
             else
             {
                 foreach (var key in keys)
@@ -180,7 +186,7 @@ public int Compare(DatabaseEntity? x, DatabaseEntity? y)
             }
         }
 
-        private IQuery GenerateInsertQuery(IReadOnlyList<DatabaseKey> keys, IDatabaseTableData tableData)
+        public IQuery GenerateInsertQuery(IReadOnlyList<DatabaseKey> keys, IDatabaseTableData tableData)
         {
             if (keys.Count == 0)
                 return Queries.Empty();
diff --git a/WDE.DatabaseEditors/ViewModels/RowPickerViewModel.cs b/WDE.DatabaseEditors/ViewModels/RowPickerViewModel.cs
@@ -151,6 +151,11 @@ public JustQuerySolutionItem(string query)
     {
         Query = query;
     }
+    
+    public JustQuerySolutionItem(IQuery query)
+    {
+        Query = query.QueryString;
+    }
 }
 
 [AutoRegister]
diff --git a/WDE.DatabaseEditors/ViewModels/SingleRow/SingleRowDbTableEditorViewModel.cs b/WDE.DatabaseEditors/ViewModels/SingleRow/SingleRowDbTableEditorViewModel.cs
@@ -17,6 +17,7 @@
 using WDE.Common;
 using WDE.Common.Database;
 using WDE.Common.Disposables;
+using WDE.Common.Events;
 using WDE.Common.History;
 using WDE.Common.Managers;
 using WDE.Common.Parameters;
@@ -161,6 +162,7 @@ public bool SplitVertical
         
         public AsyncAutoCommand DeleteRowSelectedCommand { get; }
         public DelegateCommand SetNullSelectedCommand { get; }
+        public DelegateCommand GenerateInsertQueryForSelectedCommand { get; }
         public AsyncAutoCommand DuplicateSelectedCommand { get; }
         public AsyncAutoCommand RevertSelectedCommand { get; }
         
@@ -257,7 +259,16 @@ public SingleRowDbTableEditorViewModel(DatabaseTableSolutionItem solutionItem,
                 ForceInsertEntity(duplicate, FocusedRowIndex + 1);
             }, () => FocusedRow != null);
             AutoDispose(this.ToObservable(() => FocusedRow).Subscribe(_ => DuplicateSelectedCommand.RaiseCanExecuteChanged()));
-                
+
+            GenerateInsertQueryForSelectedCommand = new DelegateCommand(() =>
+            {
+                var selectedKeys = MultiSelection.All().Select(index => Entities[index].GenerateKey(TableDefinition)).ToList();
+                var tableData = new DatabaseTableData(TableDefinition, Entities);
+                var query = queryGenerator.GenerateInsertQuery(selectedKeys, tableData);
+                var item = new MetaSolutionSQL(new JustQuerySolutionItem(query));
+                eventAggregator.GetEvent<EventRequestOpenItem>().Publish(item);
+            }, () => !MultiSelection.Empty).ObservesProperty(() => FocusedRow);
+            
             foreach (var key in solutionItem.Entries.Select(x => x.Key))
                 keys.Add(key);
             
diff --git a/WoWDatabaseEditor.Common/WDE.Common/Utils/MultiIndexContainer.cs b/WoWDatabaseEditor.Common/WDE.Common/Utils/MultiIndexContainer.cs
@@ -14,6 +14,7 @@ public interface IMultiIndexContainer
     IEnumerable<int> AllReversed();
     IEfficientContainsIterator ContainsIterator { get; }
     bool MoreThanOne { get; }
+    bool Empty { get; }
 }
 
 public interface IEfficientContainsIterator
@@ -25,8 +26,7 @@ public interface IEfficientContainsIterator
 public class MultiIndexContainer : IMultiIndexContainer
 {
     private List<(int start, int end)> ranges = new List<(int start, int end)>();
-    //(2,5) (7, 10)
-
+ 
     public IReadOnlyList<(int start, int end)> DebugRanges => ranges;
 
     private int FindFirstHigherThanEnd(int index)
@@ -222,6 +222,7 @@ public IEnumerable<int> AllReversed()
     public IEfficientContainsIterator ContainsIterator => new EfficientContainsIterator(this);
 
     public bool MoreThanOne => ranges.Count > 1 || (ranges.Count == 1 && ranges[0].start != ranges[0].end);
+    public bool Empty => ranges.Count == 0;
 
     private class EfficientContainsIterator : IEfficientContainsIterator
     {

Original file line number	Diff line number	Diff line change
`@@ -30,9 +30,10 @@ public bool SpawnEditorFor(string? initialText, Visual parent, Rect rect, ITable`
`30`	`30`	`if (c is not SingleRecordDatabaseCellViewModel cell)`
`31`	`31`	`return false;`
`32`	`32`
`33`		`- if (cell.ActionCommand != null && cell.ActionCommand.CanExecute(null))`
	`33`	`+ if (cell.ActionCommand != null)`
`34`	`34`	`{`
`35`		`- cell.ActionCommand.Execute(null);`
	`35`	`+ if (cell.ActionCommand.CanExecute(null))`
	`36`	`+ cell.ActionCommand.Execute(null);`
`36`	`37`	`return true;`
`37`	`38`	`}`
`38`	`39`
Original file line number	Diff line number	Diff line change
`@@ -120,6 +120,8 @@ public async Task<long> GetCount(string definitionId, string? customWhere, IEnum`
`120`	`120`	`try`
`121`	`121`	`{`
`122`	`122`	`var result = await sqlExecutor.ExecuteSelectSql(sql);`
	`123`	`+ if (result.Count == 0)`
	`124`	`+ return 0;`
`123`	`125`	`return Convert.ToInt64(result[0]["num"].Item2);`
`124`	`126`	`}`
`125`	`127`	`catch (IMySqlExecutor.CannotConnectToDatabaseException)`
Original file line number	Diff line number	Diff line change
`@@ -8,6 +8,7 @@ namespace WDE.DatabaseEditors.QueryGenerators`
`8`	`8`	`{`
`9`	`9`	`public interface IQueryGenerator`
`10`	`10`	`{`
	`11`	`+ public IQuery GenerateInsertQuery(IReadOnlyList<DatabaseKey> keys, IDatabaseTableData tableData);`
`11`	`12`	`public IQuery GenerateQuery(IReadOnlyList<DatabaseKey> keys, IReadOnlyList<DatabaseKey>? deletedKeys, IDatabaseTableData tableData);`
`12`	`13`	`public IQuery GenerateUpdateFieldQuery(DatabaseTableDefinitionJson table, DatabaseEntity entity, IDatabaseField field);`
`13`	`14`	`public IQuery GenerateDeleteQuery(DatabaseTableDefinitionJson table, DatabaseEntity entity);`
Original file line number	Diff line number	Diff line change
`@@ -101,7 +101,10 @@ private static void GeneratePrimaryKeyDeletion(DatabaseTableDefinitionJson defin`
`101`	`101`	`{`
`102`	`102`	`table = query.Table(foreign.TableName);`
`103`	`103`	`if (keys.Count > 1 && keys[0].Count == 1)`
`104`		`- table.WhereIn(foreign.ForeignKeys[0], keys.Select(k => k[0])).Delete();`
	`104`	`+ {`
	`105`	`+ foreach (var chunk in keys.Select(k => k[0]).Chunk(128))`
	`106`	`+ table.WhereIn(foreign.ForeignKeys[0], chunk).Delete();`
	`107`	`+ }`
`105`	`108`	`else`
`106`	`109`	`{`
`107`	`110`	`foreach (var key in keys)`
`@@ -112,7 +115,10 @@ private static void GeneratePrimaryKeyDeletion(DatabaseTableDefinitionJson defin`
`112`	`115`
`113`	`116`	`table = query.Table(definition.TableName);`
`114`	`117`	`if (keys.Count > 1 && keys[0].Count == 1)`
`115`		`- table.WhereIn(definition.TablePrimaryKeyColumnName, keys.Select(k => k[0])).Delete();`
	`118`	`+ {`
	`119`	`+ foreach (var chunk in keys.Select(k => k[0]).Chunk(128))`
	`120`	`+ table.WhereIn(definition.TablePrimaryKeyColumnName, chunk).Delete();`
	`121`	`+ }`
`116`	`122`	`else`
`117`	`123`	`{`
`118`	`124`	`foreach (var key in keys)`
`@@ -180,7 +186,7 @@ public int Compare(DatabaseEntity? x, DatabaseEntity? y)`
`180`	`186`	`}`
`181`	`187`	`}`
`182`	`188`
`183`		`- private IQuery GenerateInsertQuery(IReadOnlyList<DatabaseKey> keys, IDatabaseTableData tableData)`
	`189`	`+ public IQuery GenerateInsertQuery(IReadOnlyList<DatabaseKey> keys, IDatabaseTableData tableData)`
`184`	`190`	`{`
`185`	`191`	`if (keys.Count == 0)`
`186`	`192`	`return Queries.Empty();`
Original file line number	Diff line number	Diff line change
`@@ -151,6 +151,11 @@ public JustQuerySolutionItem(string query)`
`151`	`151`	`{`
`152`	`152`	`Query = query;`
`153`	`153`	`}`
	`154`	`+`
	`155`	`+ public JustQuerySolutionItem(IQuery query)`
	`156`	`+ {`
	`157`	`+ Query = query.QueryString;`
	`158`	`+ }`
`154`	`159`	`}`
`155`	`160`
`156`	`161`	`[AutoRegister]`