extras/MonoDevelop.Database/MonoDevelop.Database.Designer/Widgets/UniqueConstraintEditorWidget.cs

   1 //
   2 // Authors:
   3 //    Ben Motmans  <ben.motmans@gmail.com>
   4 //
   5 // Copyright (c) 2007 Ben Motmans
   6 //
   7 // Permission is hereby granted, free of charge, to any person obtaining a copy
   8 // of this software and associated documentation files (the "Software"), to deal
   9 // in the Software without restriction, including without limitation the rights
  10 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  11 // copies of the Software, and to permit persons to whom the Software is
  12 // furnished to do so, subject to the following conditions:
  13 //
  14 // The above copyright notice and this permission notice shall be included in
  15 // all copies or substantial portions of the Software.
  16 //
  17 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  18 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  19 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  20 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  21 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  22 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  23 // THE SOFTWARE.
  24 //
  25
  26 using Gtk;
  27 using System;
  28 using System.Text;
  29 using System.Collections.Generic;
  30 using MonoDevelop.Core;
  31 using MonoDevelop.Core.Gui;
  32 using MonoDevelop.Components;
  33 using MonoDevelop.Database.Sql;
  34 using MonoDevelop.Database.Components;
  35
  36 namespace MonoDevelop.Database.Designer
  37 {
  38         [System.ComponentModel.Category("widget")]
  39         [System.ComponentModel.ToolboxItem(true)]
  40         public partial class UniqueConstraintEditorWidget : Gtk.Bin
  41         {
  42                 public event EventHandler ContentChanged;
  43
  44                 private ISchemaProvider schemaProvider;
  45                 private ColumnSchemaCollection columns;
  46                 private ConstraintSchemaCollection constraints;
  47                 private TableSchema table;
  48
  49                 private SchemaActions action;
  50
  51                 private ListStore store;
  52
  53                 private const int colNameIndex = 0;
  54                 private const int colIsColumnConstraintIndex = 1;
  55                 private const int colColumnsIndex = 2;
  56                 private const int colObjIndex = 3;
  57
  58                 public UniqueConstraintEditorWidget (ISchemaProvider schemaProvider, SchemaActions action)
  59                 {
  60                         if (schemaProvider == null)
  61                                 throw new ArgumentNullException ("schemaProvider");
  62
  63                         this.schemaProvider = schemaProvider;
  64                         this.action = action;
  65
  66                         this.Build();
  67
  68                         store = new ListStore (typeof (string), typeof (bool), typeof (string), typeof (object));
  69                         listUnique.Model = store;
  70                         listUnique.Selection.Changed += new EventHandler (SelectionChanged);
  71                         columnSelecter.ColumnToggled += new EventHandler (ColumnToggled);
  72
  73                         TreeViewColumn colName = new TreeViewColumn ();
  74                         TreeViewColumn colIsColConstraint = new TreeViewColumn ();
  75
  76                         colName.Title = AddinCatalog.GetString ("Name");
  77                         colIsColConstraint.Title = AddinCatalog.GetString ("Column Constraint");
  78
  79                         CellRendererText nameRenderer = new CellRendererText ();
  80                         CellRendererToggle toggleRenderer = new CellRendererToggle ();
  81
  82                         nameRenderer.Editable = true;
  83                         nameRenderer.Edited += new EditedHandler (NameEdited);
  84
  85                         toggleRenderer.Activatable = true;
  86                         toggleRenderer.Toggled += new ToggledHandler (IsColumnConstraintToggled);
  87
  88                         colName.PackStart (nameRenderer, true);
  89                         colIsColConstraint.PackStart (toggleRenderer, true);
  90
  91                         colName.AddAttribute (nameRenderer, "text", colNameIndex);
  92                         colIsColConstraint.AddAttribute (toggleRenderer, "active", colIsColumnConstraintIndex);
  93
  94                         listUnique.AppendColumn (colName);
  95                         listUnique.AppendColumn (colIsColConstraint);
  96
  97                         ShowAll ();
  98                 }
  99
 100                 public void Initialize (TableSchema table, ColumnSchemaCollection columns, ConstraintSchemaCollection constraints)
 101                 {
 102                         if (columns == null)
 103                                 throw new ArgumentNullException ("columns");
 104                         if (table == null)
 105                                 throw new ArgumentNullException ("table");
 106                         if (constraints == null)
 107                                 throw new ArgumentNullException ("constraints");
 108
 109                         this.table = table;
 110                         this.columns = columns;
 111                         this.constraints = constraints;
 112
 113                         columnSelecter.Initialize (columns);
 114
 115                         foreach (UniqueConstraintSchema uni in constraints.GetConstraints (ConstraintType.Unique))
 116                                 AddConstraint (uni);
 117                         //TODO: also col constraints
 118                 }
 119
 120                 private void NameEdited (object sender, EditedArgs args)
 121                 {
 122                         TreeIter iter;
 123                         if (store.GetIterFromString (out iter, args.Path)) {
 124                                 if (!string.IsNullOrEmpty (args.NewText) && !constraints.Contains (args.NewText)) {
 125                                         store.SetValue (iter, colNameIndex, args.NewText);
 126                                         EmitContentChanged ();
 127                                 } else {
 128                                         string oldText = store.GetValue (iter, colNameIndex) as string;
 129                                         (sender as CellRendererText).Text = oldText;
 130                                 }
 131                         }
 132                 }
 133
 134                 private void IsColumnConstraintToggled (object sender, ToggledArgs args)
 135                 {
 136                         TreeIter iter;
 137                         if (store.GetIterFromString (out iter, args.Path)) {
 138                                 bool val = (bool) store.GetValue (iter, colIsColumnConstraintIndex);
 139                                 store.SetValue (iter, colIsColumnConstraintIndex, !val);
 140                                 SetSelectionFromIter (iter);
 141                                 EmitContentChanged ();
 142                         }
 143                 }
 144
 145                 private void SelectionChanged (object sender, EventArgs args)
 146                 {
 147                         columnSelecter.DeselectAll ();
 148
 149                         TreeIter iter;
 150                         if (listUnique.Selection.GetSelected (out iter)) {
 151                                 columnSelecter.Sensitive = true;
 152                                 SetSelectionFromIter (iter);
 153                         } else {
 154                                 columnSelecter.Sensitive = false;
 155                         }
 156                 }
 157
 158                 private void SetSelectionFromIter (TreeIter iter)
 159                 {
 160                         bool iscolc = (bool)store.GetValue (iter, colIsColumnConstraintIndex);
 161                         columnSelecter.SingleCheck = iscolc;
 162
 163                         string colstr = store.GetValue (iter, colColumnsIndex) as string;
 164                         string[] cols = colstr.Split (',');
 165                         foreach (string col in cols)
 166                                 columnSelecter.Select (col);
 167                 }
 168
 169                 private void ColumnToggled (object sender, EventArgs args)
 170                 {
 171                         TreeIter iter;
 172                         if (listUnique.Selection.GetSelected (out iter)) {
 173                                 store.SetValue (iter, colColumnsIndex, GetColumnsString (columnSelecter.CheckedColumns));
 174                                 EmitContentChanged ();
 175                         }
 176                 }
 177
 178                 protected virtual void AddClicked (object sender, EventArgs e)
 179                 {
 180                         UniqueConstraintSchema uni = schemaProvider.CreateUniqueConstraintSchema ("uni_new");
 181                         int index = 1;
 182                         while (constraints.Contains (uni.Name))
 183                                 uni.Name = "uni_new" + (index++);
 184                         constraints.Add (uni);
 185                         AddConstraint (uni);
 186                         EmitContentChanged ();
 187                 }
 188
 189                 protected virtual void RemoveClicked (object sender, EventArgs e)
 190                 {
 191                         TreeIter iter;
 192                         if (listUnique.Selection.GetSelected (out iter)) {
 193                                 UniqueConstraintSchema uni = store.GetValue (iter, colObjIndex) as UniqueConstraintSchema;
 194
 195                                 if (MessageService.Confirm (
 196                                         AddinCatalog.GetString ("Are you sure you want to remove constraint '{0}'?", uni.Name),
 197                                         AlertButton.Remove
 198                                 )) {
 199                                         store.Remove (ref iter);
 200                                         constraints.Remove (uni);
 201                                         EmitContentChanged ();
 202                                 }
 203                         }
 204                 }
 205
 206                 private void AddConstraint (UniqueConstraintSchema uni)
 207                 {
 208                         string colstr = GetColumnsString (uni.Columns);
 209                         store.AppendValues (uni.Name, uni.IsColumnConstraint, colstr, uni);
 210                 }
 211
 212                 private string GetColumnsString (IEnumerable<ColumnSchema> collection)
 213                 {
 214                         bool first = true;
 215                         StringBuilder sb = new StringBuilder ();
 216                         foreach (ColumnSchema column in collection) {
 217                                 if (first)
 218                                         first = false;
 219                                 else
 220                                         sb.Append (',');
 221
 222                                 sb.Append (column.Name);
 223                         }
 224                         return sb.ToString ();
 225                 }
 226
 227                 public virtual bool ValidateSchemaObjects (out string msg)
 228                 {
 229                         TreeIter iter;
 230                         if (store.GetIterFirst (out iter)) {
 231                                 do {
 232                                         string name = store.GetValue (iter, colNameIndex) as string;
 233                                         string columns = store.GetValue (iter, colColumnsIndex) as string;
 234
 235                                         if (String.IsNullOrEmpty (columns)) {
 236                                                 msg = AddinCatalog.GetString ("Unique Key constraint '{0}' must be applied to one or more columns.", name);
 237                                                 return false;
 238                                         }
 239                                 } while (store.IterNext (ref iter));
 240                         }
 241                         msg = null;
 242                         return true;
 243                 }
 244
 245                 public virtual void FillSchemaObjects ()
 246                 {
 247                         TreeIter iter;
 248                         if (store.GetIterFirst (out iter)) {
 249                                 do {
 250                                         UniqueConstraintSchema uni = store.GetValue (iter, colObjIndex) as UniqueConstraintSchema;
 251
 252                                         uni.Name = store.GetValue (iter, colNameIndex) as string;
 253                                         uni.IsColumnConstraint = (bool)store.GetValue (iter, colIsColumnConstraintIndex);
 254
 255                                         string colstr = store.GetValue (iter, colColumnsIndex) as string;
 256                                         string[] cols = colstr.Split (',');
 257                                         foreach (string col in cols) {
 258                                                 ColumnSchema column = columns.Search (col);
 259                                                 uni.Columns.Add (column);
 260                                         }
 261
 262                                         table.Constraints.Add (uni);
 263                                 } while (store.IterNext (ref iter));
 264                         }
 265                 }
 266
 267                 protected virtual void EmitContentChanged ()
 268                 {
 269                         if (ContentChanged != null)
 270                                 ContentChanged (this, EventArgs.Empty);
 271                 }
 272         }
 273 }