main/src/addins/MonoDevelop.XmlEditor/MonoDevelop.XmlEditor.Completion/InferredXmlCompletionProvider.cs

   1 //
   2 // InferredXmlCompletionProvider.cs
   3 //
   4 // Author:
   5 //       Michael Hutchinson <mhutchinson@novell.com>
   6 //
   7 // Copyright (c) 2009 Novell, Inc. (http://www.novell.com)
   8 //
   9 // Permission is hereby granted, free of charge, to any person obtaining a copy
  10 // of this software and associated documentation files (the "Software"), to deal
  11 // in the Software without restriction, including without limitation the rights
  12 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  13 // copies of the Software, and to permit persons to whom the Software is
  14 // furnished to do so, subject to the following conditions:
  15 //
  16 // The above copyright notice and this permission notice shall be included in
  17 // all copies or substantial portions of the Software.
  18 //
  19 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  20 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  21 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  22 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  23 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  24 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  25 // THE SOFTWARE.
  26
  27 using System;
  28 using System.Collections.Generic;
  29 using MonoDevelop.Projects.Gui.Completion;
  30 using MonoDevelop.Xml.StateEngine;
  31
  32 namespace MonoDevelop.XmlEditor.Completion
  33 {
  34
  35
  36         public class InferredXmlCompletionProvider : IXmlCompletionProvider
  37         {
  38                 Dictionary<string,HashSet<string>> elementCompletions = new Dictionary<string,HashSet<string>> ();
  39                 Dictionary<string,HashSet<string>> attributeCompletions = new Dictionary<string,HashSet<string>> ();
  40
  41                 public DateTime TimeStamp { get; set; }
  42                 public int ErrorCount { get; set; }
  43
  44                 public InferredXmlCompletionProvider ()
  45                 {
  46                 }
  47
  48                 //TODO: respect namespaces
  49                 public void Populate (XDocument doc)
  50                 {
  51                         foreach (XNode node in doc.AllDescendentNodes) {
  52                                 XElement el = node as XElement;
  53                                 if (el == null)
  54                                         continue;
  55                                 string parentName = "";
  56                                 XElement parentEl = el.Parent as XElement;
  57                                 if (parentEl != null)
  58                                         parentName = parentEl.Name.Name;
  59
  60                                 HashSet<string> map;
  61                                 if (!elementCompletions.TryGetValue (parentName, out map)) {
  62                                         map = new HashSet<string> ();
  63                                         elementCompletions.Add (parentName, map);
  64                                 }
  65                                 map.Add (el.Name.Name);
  66
  67                                 if (!attributeCompletions.TryGetValue (el.Name.Name, out map)) {
  68                                         map = new HashSet<string> ();
  69                                         attributeCompletions.Add (el.Name.Name, map);
  70                                 }
  71                                 foreach (XAttribute att in el.Attributes)
  72                                         map.Add (att.Name.Name);
  73                         }
  74                 }
  75
  76                 public ICompletionData[] GetElementCompletionData ()
  77                 {
  78                         return GetChildElementCompletionData ("");
  79                 }
  80
  81                 public ICompletionData[] GetElementCompletionData (string namespacePrefix)
  82                 {
  83                         return new ICompletionData[0];
  84                 }
  85
  86                 public ICompletionData[] GetChildElementCompletionData (XmlElementPath path)
  87                 {
  88                         return GetCompletions (elementCompletions, path, XmlCompletionData.DataType.XmlElement);
  89                 }
  90
  91                 public ICompletionData[] GetAttributeCompletionData (XmlElementPath path)
  92                 {
  93                         return GetCompletions (attributeCompletions, path, XmlCompletionData.DataType.XmlAttribute);
  94                 }
  95
  96                 public ICompletionData[] GetAttributeValueCompletionData (XmlElementPath path, string name)
  97                 {
  98                         return new ICompletionData [0];
  99                 }
 100
 101                 public ICompletionData[] GetChildElementCompletionData (string tagName)
 102                 {
 103                         return GetCompletions (elementCompletions, tagName, XmlCompletionData.DataType.XmlElement);
 104                 }
 105
 106                 public ICompletionData[] GetAttributeCompletionData (string tagName)
 107                 {
 108                         return GetCompletions (attributeCompletions, tagName, XmlCompletionData.DataType.XmlAttribute);
 109                 }
 110
 111                 public ICompletionData[] GetAttributeValueCompletionData (string tagName, string name)
 112                 {
 113                         return new ICompletionData [0];
 114                 }
 115
 116                 static ICompletionData[] GetCompletions (Dictionary<string,HashSet<string>> map, string tagName, XmlCompletionData.DataType type)
 117                 {
 118                         HashSet<string> values;
 119                         if (!map.TryGetValue (tagName, out values))
 120                                 return new ICompletionData [0];
 121                         ICompletionData[] data = new ICompletionData[values.Count];
 122                         int i = 0;
 123                         foreach (string s in values)
 124                                 data[i++] = new XmlCompletionData (s, type);
 125                         return data;
 126                 }
 127
 128                 static ICompletionData[] GetCompletions (Dictionary<string,HashSet<string>> map, XmlElementPath path, XmlCompletionData.DataType type)
 129                 {
 130                         if (path == null || path.Elements.Count == 0)
 131                                 return new ICompletionData[0];
 132                         return GetCompletions (map, path.Elements[path.Elements.Count - 1].Name, type);
 133                 }
 134         }
 135 }