mcs/class/Microsoft.Build.Engine/Microsoft.Build.BuildEngine/TaskDatabase.cs

   1 //
   2 // TaskDatabase.cs: Provides information about tasks.
   3 //
   4 // Author:
   5 //   Marek Sieradzki (marek.sieradzki@gmail.com)
   6 //
   7 // (C) 2005 Marek Sieradzki
   8 //
   9 // Permission is hereby granted, free of charge, to any person obtaining
  10 // a copy of this software and associated documentation files (the
  11 // "Software"), to deal in the Software without restriction, including
  12 // without limitation the rights to use, copy, modify, merge, publish,
  13 // distribute, sublicense, and/or sell copies of the Software, and to
  14 // permit persons to whom the Software is furnished to do so, subject to
  15 // the following conditions:
  16 //
  17 // The above copyright notice and this permission notice shall be
  18 // included in all copies or substantial portions of the Software.
  19 //
  20 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  21 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  22 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  23 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  24 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  25 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  26 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  27
  28 #if NET_2_0
  29
  30 using System;
  31 using System.Collections.Generic;
  32 using System.Reflection;
  33 using Microsoft.Build.Framework;
  34 using Mono.XBuild.Framework;
  35
  36 namespace Microsoft.Build.BuildEngine {
  37         internal class TaskDatabase {
  38
  39                 Dictionary<string, UsingTaskInfo> usingTasksByFullName;
  40                 // full name -> AssemblyLoadInfo
  41                 Dictionary <string, AssemblyLoadInfo>   assemblyInformation;
  42                 // full name -> Type
  43                 Dictionary <string, Type>               typesByFullName;
  44                 // short name -> Type
  45                 Dictionary <string, Type>               typesByShortName;
  46
  47                 public TaskDatabase ()
  48                 {
  49                         assemblyInformation = new Dictionary <string, AssemblyLoadInfo> ();
  50                         typesByFullName = new Dictionary <string, Type> (StringComparer.OrdinalIgnoreCase);
  51                         typesByShortName = new Dictionary <string, Type> (StringComparer.OrdinalIgnoreCase);
  52                         usingTasksByFullName = new Dictionary <string, UsingTaskInfo> (StringComparer.OrdinalIgnoreCase);
  53                 }
  54
  55                 public void RegisterTask (string classname, AssemblyLoadInfo assemblyLoadInfo)
  56                 {
  57                         assemblyInformation.Add (classname, assemblyLoadInfo);
  58                         Assembly assembly;
  59
  60                         if (assemblyLoadInfo.InfoType == LoadInfoType.AssemblyFilename)
  61                                 assembly = Assembly.LoadFrom (assemblyLoadInfo.Filename);
  62                         else if (assemblyLoadInfo.InfoType == LoadInfoType.AssemblyName)
  63                                 assembly = Assembly.Load (assemblyLoadInfo.AssemblyName);
  64                         else
  65                                 assembly = Assembly.Load (assemblyLoadInfo.AssemblyNameString);
  66
  67                         Type type = assembly.GetType (classname);
  68                         if (type == null) {
  69                                 // search for matching class in case namespace was not used
  70                                 foreach (Type exportedType in assembly.GetExportedTypes()) {
  71                                         if (exportedType.Name == classname) {
  72                                                 type = exportedType;
  73                                                 break;
  74                                         }
  75                                 }
  76                         }
  77                         typesByFullName.Add (classname, type);
  78                         typesByShortName [GetShortName (classname)] = type;
  79                 }
  80
  81                 public void RegisterUsingTask (UsingTask ut)
  82                 {
  83                         usingTasksByFullName [ut.TaskName] = new UsingTaskInfo (GetShortName (ut.TaskName), ut);
  84                 }
  85
  86                 public Type GetTypeFromClassName (string classname)
  87                 {
  88                         Type ret = GetTypeFromClassNameInternal (classname);
  89                         if (ret == null) {
  90                                 // Task not already loaded,
  91                                 // Check list of pending UsingTasks
  92                                 bool is_shortname = classname.IndexOf ('.') < 0;
  93                                 UsingTaskInfo info = new UsingTaskInfo (String.Empty, null);
  94                                 if (is_shortname) {
  95                                         // Linear search UsingTaskInfo objects for short name match
  96                                         foreach (UsingTaskInfo ut_info in usingTasksByFullName.Values) {
  97                                                 if (String.Compare (ut_info.ShortName, classname, true) == 0) {
  98                                                         info = ut_info;
  99                                                         break;
 100                                                 }
 101                                         }
 102
 103                                         if (info.Task == null)
 104                                                 ThrowTaskNotRegistered (classname);
 105                                 } else {
 106                                         // Look for full name match
 107                                         if (!usingTasksByFullName.TryGetValue (classname, out info))
 108                                                 ThrowTaskNotRegistered (classname);
 109                                 }
 110
 111                                 usingTasksByFullName.Remove (info.Task.TaskName);
 112                                 info.Task.Load (this);
 113                                 ret = GetTypeFromClassNameInternal (classname);
 114                         }
 115
 116                         if (ret == null)
 117                                 ThrowTaskNotRegistered (classname);
 118                         return ret;
 119                 }
 120
 121                 Type GetTypeFromClassNameInternal (string classname)
 122                 {
 123                         if (!typesByFullName.ContainsKey (classname)) {
 124                                 if (!typesByShortName.ContainsKey (classname))
 125                                         return null;
 126                                 else
 127                                         return typesByShortName [classname];
 128                         } else
 129                                 return typesByFullName [classname];
 130                 }
 131
 132                 public void CopyTasks (TaskDatabase taskDatabase)
 133                 {
 134                         foreach (KeyValuePair <string, AssemblyLoadInfo> kvp in taskDatabase.assemblyInformation)
 135                                 assemblyInformation.Add (kvp.Key, kvp.Value);
 136                         foreach (KeyValuePair <string, Type> kvp in taskDatabase.typesByFullName)
 137                                 typesByFullName.Add (kvp.Key, kvp.Value);
 138                         foreach (KeyValuePair <string, Type> kvp in taskDatabase.typesByShortName)
 139                                 typesByShortName.Add (kvp.Key, kvp.Value);
 140                         foreach (KeyValuePair <string, UsingTaskInfo> kvp in taskDatabase.usingTasksByFullName)
 141                                 usingTasksByFullName.Add (kvp.Key, kvp.Value);
 142                 }
 143
 144                 private string GetShortName (string fullname)
 145                 {
 146                         string[] parts = fullname.Split ('.');
 147                         return parts [parts.Length - 1];
 148                 }
 149
 150                 void ThrowTaskNotRegistered (string classname)
 151                 {
 152                         throw new Exception (String.Format ("Not registered task {0}.", classname));
 153                 }
 154         }
 155
 156         struct UsingTaskInfo {
 157                 public string ShortName;
 158                 public UsingTask Task;
 159
 160                 public UsingTaskInfo (string shortname, UsingTask task)
 161                 {
 162                         ShortName = shortname;
 163                         Task = task;
 164                 }
 165         }
 166 }
 167
 168 #endif