In class/Microsoft.Build.Engine/Microsoft.Build.BuildEngine:
[mono.git] / 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                 // full name -> AssemblyLoadInfo
40                 Dictionary <string, AssemblyLoadInfo>   assemblyInformation;
41                 // full name -> Type
42                 Dictionary <string, Type>               typesByFullName;
43                 // short name -> Type
44                 Dictionary <string, Type>               typesByShortName;
45         
46                 public TaskDatabase ()
47                 {
48                         assemblyInformation = new Dictionary <string, AssemblyLoadInfo> ();
49                         typesByFullName = new Dictionary <string, Type> ();
50                         typesByShortName = new Dictionary <string, Type> ();
51                 }
52                 
53                 public void RegisterTask (string classname, AssemblyLoadInfo assemblyLoadInfo)
54                 {
55                         assemblyInformation.Add (classname, assemblyLoadInfo);
56                         Assembly assembly;
57
58                         if (assemblyLoadInfo.InfoType == LoadInfoType.AssemblyFilename)
59                                 assembly = Assembly.LoadFrom (assemblyLoadInfo.Filename);
60                         else if (assemblyLoadInfo.InfoType == LoadInfoType.AssemblyName)
61                                 assembly = Assembly.Load (assemblyLoadInfo.AssemblyName);
62                         else
63                                 assembly = Assembly.Load (assemblyLoadInfo.AssemblyNameString);
64                         
65                         Type type = assembly.GetType (classname);
66                         typesByFullName.Add (classname, type);
67                         typesByShortName.Add (GetShortName (classname), type);
68                 }
69                 
70                 public Type GetTypeFromClassName (string classname)
71                 {
72                         if (!typesByFullName.ContainsKey (classname)) {
73                                 if (!typesByShortName.ContainsKey (classname))
74                                         throw new Exception (String.Format ("Not registered task {0}.", classname));
75                                 else
76                                         return typesByShortName [classname];
77                         } else
78                                 return typesByFullName [classname];
79                 }
80                 
81                 public void CopyTasks (TaskDatabase taskDatabase)
82                 {
83                         foreach (KeyValuePair <string, AssemblyLoadInfo> kvp in taskDatabase.assemblyInformation)
84                                 assemblyInformation.Add (kvp.Key, kvp.Value);
85                         foreach (KeyValuePair <string, Type> kvp in taskDatabase.typesByFullName)
86                                 typesByFullName.Add (kvp.Key, kvp.Value);
87                         foreach (KeyValuePair <string, Type> kvp in taskDatabase.typesByShortName)
88                                 typesByShortName.Add (kvp.Key, kvp.Value);
89                 }
90                 
91                 private string GetShortName (string fullname)
92                 {
93                         string[] parts = fullname.Split ('.');
94                         return parts [parts.Length - 1];
95                 }
96         }
97 }
98
99 #endif