JW/EMIS.Entities/TableKey.tt

<#@ template language="C#" debug="false" hostspecific="true"#>
<#@ include file="EF.Utility.CS.ttinclude"#>
<#@ assembly name="$(DevEnvDir)Microsoft.Data.Entity.Design.DatabaseGeneration.dll"#>
<#@ import namespace="Microsoft.Data.Entity.Design.DatabaseGeneration" #>
<#@ import namespace="System.Text"#>
<#@ output extension=".cs"#><#
CodeGenerationTools code = new CodeGenerationTools(this);
CodeRegion region = new CodeRegion(this, 1);
MetadataTools ef = new MetadataTools(this);
string inputFile = @"EMISContext.edmx";

var loadResult = LoadMetadata(inputFile);
EdmItemCollection itemCollection = loadResult.EdmItems;
var propertyToColumnMapping = loadResult.PropertyToColumnMapping;
var manyToManyMappings = loadResult.ManyToManyMappings;
var tphMappings = loadResult.TphMappings;
string namespaceName = code.VsNamespaceSuggestion();
EntityFrameworkTemplateFileManager fileManager = EntityFrameworkTemplateFileManager.Create(this);
WriteHeader(fileManager);#>

using System.Collections.Generic;
namespace EMIS.Entities
{
	public class TableKeyDictionary
	{
        private static Dictionary<string, string> items;
        public static Dictionary<string, string> Items
        {
            get { return items; }
        }

		static TableKeyDictionary()
        {
			items = new Dictionary<string, string>();
<#foreach (EntityType entity in itemCollection.GetItems<EntityType>().OrderBy(e => e.Name))
{
	string hasKey;
	hasKey = string.Format("items.Add(\"{0}\", \"{1}\");", entity.Name, entity.KeyMembers[0].Name);#>
			<#=hasKey#>
<#}
#>
		}

		public static string GetKeyName<TEntity>(TEntity entity)
		{
			string tableName = entity.GetType().Name;

			return items[tableName];
		}

		public static string GetKeyName<TEntity>()
		{
			string tableName = typeof(TEntity).Name;

			return items[tableName];
		}
	}
}
<#+
string GetResourceString(string resourceName)
{
	if(_resourceManager2 == null)
	{
		_resourceManager2 = new System.Resources.ResourceManager("System.Data.Entity.Design", typeof(System.Data.Entity.Design.MetadataItemCollectionFactory).Assembly);
	}
	
    return _resourceManager2.GetString(resourceName, null);
}
System.Resources.ResourceManager _resourceManager2;

void WriteHeader(EntityFrameworkTemplateFileManager fileManager)
{
    fileManager.StartHeader();
#>
//------------------------------------------------------------------------------
// <auto-generated>
//    This code was generated from a template.
//
//    Manual changes to this file may cause unexpected behavior in your application.
//    Manual changes to this file will be overwritten if the code is regenerated.
// </auto-generated>
//------------------------------------------------------------------------------

<#+
    fileManager.EndBlock();
}

void BeginNamespace(string namespaceName, CodeGenerationTools code)
{
    CodeRegion region = new CodeRegion(this);
    if (!String.IsNullOrEmpty(namespaceName))
    {
#>
namespace <#=code.EscapeNamespace(namespaceName)#>
{
<#+
        PushIndent(CodeRegion.GetIndent(1));
    }
}


void EndNamespace(string namespaceName)
{
    if (!String.IsNullOrEmpty(namespaceName))
    {
        PopIndent();
#>
}
<#+
    }
}
string ToTable(EntitySet entitySet)
{
	var toTable = entitySet.Name;
	string schema = entitySet.GetSchemaName();
	if(!string.IsNullOrWhiteSpace(schema) && schema != "dbo")	
		toTable += "\", \"" + schema;
	return toTable;
}

bool VerifyTypesAreCaseInsensitiveUnique(EdmItemCollection itemCollection)
{
    var alreadySeen = new Dictionary<string, bool>(StringComparer.OrdinalIgnoreCase);
    foreach(var type in itemCollection.GetItems<StructuralType>())
    {
        if (!(type is EntityType || type is ComplexType))
        {
            continue;
        }

        if (alreadySeen.ContainsKey(type.FullName))
        {
            Error(String.Format(CultureInfo.CurrentCulture, "This template does not support types that differ only by case, the types {0} are not supported", type.FullName));
            return false;
        }
        else
        {
            alreadySeen.Add(type.FullName, true);
        }
    }

    return true;
}

string GetGenerationOption(EdmProperty property, EntityType entity)
{
	string result = "";
	bool isPk = entity.KeyMembers.Contains(property);
	MetadataProperty storeGeneratedPatternProperty = null;
	string storeGeneratedPatternPropertyValue = "None";
	
	if(property.MetadataProperties.TryGetValue(MetadataConsts.EDM_ANNOTATION_09_02 + ":StoreGeneratedPattern", false, out storeGeneratedPatternProperty))	
		storeGeneratedPatternPropertyValue = storeGeneratedPatternProperty.Value.ToString();		
	
	PrimitiveType edmType = (PrimitiveType) property.TypeUsage.EdmType; 
	if (storeGeneratedPatternPropertyValue == "Computed")
	{
	    result = ".HasDatabaseGeneratedOption(DatabaseGeneratedOption.Computed)";
	}
	else if ((edmType.ClrEquivalentType == typeof(int)) || (edmType.ClrEquivalentType == typeof(short)) || (edmType.ClrEquivalentType == typeof(long)))
	{
	    if (isPk && (storeGeneratedPatternPropertyValue != "Identity"))
	        result = ".HasDatabaseGeneratedOption(DatabaseGeneratedOption.None)";
	    else if ((!isPk || (entity.KeyMembers.Count > 1)) && (storeGeneratedPatternPropertyValue == "Identity"))
	        result = ".HasDatabaseGeneratedOption(DatabaseGeneratedOption.Identity)";
	}	
	return result;    
}

MetadataLoadResult LoadMetadata(string inputFile)
{       
	var loader = new MetadataLoader(this);
	bool loaded = false;
	EdmItemCollection edmItemCollection = loader.CreateEdmItemCollection(inputFile);      
	StoreItemCollection storeItemCollection = null;
	if (loader.TryCreateStoreItemCollection(inputFile, out storeItemCollection))
	{
		StorageMappingItemCollection storageMappingItemCollection;
		if (loader.TryCreateStorageMappingItemCollection(inputFile, edmItemCollection, storeItemCollection, out storageMappingItemCollection))
			loaded = true;
	}

	if(loaded == false)
		throw new Exception("Cannot load a metadata from the file " + inputFile);

	var mappingMetadata = LoadMappingMetadata(inputFile);
	var mappingNode		= mappingMetadata.Item1;
	var nsmgr			= mappingMetadata.Item2;

	var allEntitySets = storeItemCollection.GetAllEntitySets();		

	return new MetadataLoadResult
	       	{
	       		EdmItems = edmItemCollection,
	       		PropertyToColumnMapping = BuildEntityMappings(mappingNode, nsmgr, edmItemCollection.GetItems<EntityType>(), edmItemCollection.GetAllEntitySets(), allEntitySets),
	       		ManyToManyMappings = BuildManyToManyMappings(mappingNode, nsmgr, edmItemCollection.GetAllAssociationSets(), allEntitySets),
				TphMappings=BuildTPHMappings(mappingNode, nsmgr, edmItemCollection.GetItems<EntityType>(), edmItemCollection.GetAllEntitySets(), allEntitySets)
	       	};
}

private Tuple<XmlNode, XmlNamespaceManager> LoadMappingMetadata(string inputFile)
{
	var xmlDoc = new XmlDocument();
     xmlDoc.Load(Host.ResolvePath(inputFile));

	 var schemaConstantsList = new SchemaConsts[]
					{						
						MetadataConsts.V2_SCHEMA_CONSTANTS,
						MetadataConsts.V1_SCHEMA_CONSTANTS
					};
	foreach (var schemaConstants in schemaConstantsList)
	{
		var nsmgr = new XmlNamespaceManager(xmlDoc.NameTable);
		nsmgr.AddNamespace("ef", schemaConstants.MslNamespace);
		nsmgr.AddNamespace("edmx", schemaConstants.EdmxNamespace);
		var mappingNode = xmlDoc.DocumentElement.SelectSingleNode("./*/edmx:Mappings", nsmgr);

		if(mappingNode != null)
			return Tuple.Create(mappingNode, nsmgr);
	}     

	throw new Exception(GetResourceString("Template_UnsupportedSchema"));
}

private Dictionary<EntityType, Dictionary<EntityType, Dictionary<EdmProperty, string>>> BuildTPHMappings(XmlNode mappingNode, XmlNamespaceManager nsmgr, IEnumerable<EntityType> entityTypes, IEnumerable<EntitySet> entitySets, IEnumerable<EntitySet> tableSets)
{
	var dictionary = new Dictionary<EntityType, Dictionary<EntityType, Dictionary<EdmProperty, string>>>();	
	foreach (EntitySet set in entitySets)
	{
		XmlNodeList nodes = mappingNode.SelectNodes(string.Format(".//ef:EntitySetMapping[@Name=\"{0}\"]/ef:EntityTypeMapping/ef:MappingFragment", set.Name), nsmgr);
		foreach(XmlNode node in nodes)
		{
			string typeName=node.ParentNode.Attributes["TypeName"].Value;
			if(typeName.StartsWith("IsTypeOf("))
				typeName=typeName.Substring("IsTypeOf(".Length, typeName.Length-"IsTypeOf()".Length);
			EntityType type=entityTypes.Single(z=>z.FullName==typeName);
			string tableName = node.Attributes["StoreEntitySet"].Value;
			EntitySet set2 = tableSets.Single(entitySet => entitySet.Name == tableName);
			var entityMap = new Dictionary<EdmProperty, string>();
			
			XmlNodeList propertyNodes = node.SelectNodes("./ef:Condition", nsmgr);
			if(propertyNodes.Count==0) continue;
			foreach(XmlNode propertyNode in propertyNodes)
			{
				string str = propertyNode.Attributes["ColumnName"].Value;
				EdmProperty property2 = set2.ElementType.Properties[str];
				string val=propertyNode.Attributes["Value"].Value;
				entityMap.Add(property2, val);
			}
			EntityType baseType=(EntityType)(type.BaseType??type);
			if(!dictionary.Keys.Contains(baseType))
			{
			var entityMappings=new Dictionary<EntityType, Dictionary<EdmProperty, string>>();
			//entityMappings.Add(type,entityMap);
			dictionary.Add(baseType, entityMappings);
			}
			dictionary[baseType].Add(type,entityMap);
		}
	}
	return dictionary;
}		

private Dictionary<EntityType, Tuple<EntitySet, Dictionary<EdmProperty, EdmProperty>>> BuildEntityMappings(XmlNode mappingNode, XmlNamespaceManager nsmgr, IEnumerable<EntityType> entityTypes, IEnumerable<EntitySet> entitySets, IEnumerable<EntitySet> tableSets)
{
	var dictionary = new Dictionary<EntityType, Tuple<EntitySet, Dictionary<EdmProperty, EdmProperty>>>();

	foreach (EntitySet set in entitySets)
	{
		XmlNodeList nodes = mappingNode.SelectNodes(string.Format(".//ef:EntitySetMapping[@Name=\"{0}\"]/ef:EntityTypeMapping/ef:MappingFragment", set.Name), nsmgr);
		foreach(XmlNode node in nodes)
		{
			string typeName=node.ParentNode.Attributes["TypeName"].Value;
			if(typeName.StartsWith("IsTypeOf("))
				typeName=typeName.Substring("IsTypeOf(".Length, typeName.Length-"IsTypeOf()".Length);
			EntityType type=entityTypes.Single(z=>z.FullName==typeName);
			string tableName = node.Attributes["StoreEntitySet"].Value;
			EntitySet set2 = tableSets.Single(entitySet => entitySet.Name == tableName);
			var entityMap = new Dictionary<EdmProperty, EdmProperty>();
			foreach (EdmProperty property in type.Properties)
			{
				XmlNode propertyNode = node.SelectSingleNode(string.Format("./ef:ScalarProperty[@Name=\"{0}\"]", property.Name), nsmgr);
				if(propertyNode == null) continue;
				string str = propertyNode.Attributes["ColumnName"].Value;
				EdmProperty property2 = set2.ElementType.Properties[str];
				entityMap.Add(property, property2);
			}
			dictionary.Add(type, Tuple.Create(set2, entityMap));
		}
	}
	return dictionary;
}

Dictionary<AssociationType, Tuple<EntitySet, Dictionary<RelationshipEndMember, Dictionary<EdmMember, string>>>> BuildManyToManyMappings(XmlNode mappingNode, XmlNamespaceManager nsmgr, IEnumerable<AssociationSet> associationSets, IEnumerable<EntitySet> tableSets)
{
	var dictionary = new Dictionary<AssociationType, Tuple<EntitySet, Dictionary<RelationshipEndMember, Dictionary<EdmMember, string>>>>();
	foreach (AssociationSet associationSet in associationSets.Where(set => set.ElementType.IsManyToMany()))
	{
		
		XmlNode node = mappingNode.SelectSingleNode(string.Format("//ef:AssociationSetMapping[@Name=\"{0}\"]", associationSet.Name), nsmgr);
		string tableName = node.Attributes["StoreEntitySet"].Value;
		EntitySet entitySet = tableSets.Single(s => s.Name == tableName);
		
		var relationEndMap = new Dictionary<RelationshipEndMember, Dictionary<EdmMember, string>>();
		foreach (AssociationSetEnd end in associationSet.AssociationSetEnds)
		{
			var map = new Dictionary<EdmMember, string>();
			foreach (XmlNode endProperty in node.SelectSingleNode(string.Format("./ef:EndProperty[@Name=\"{0}\"]", end.Name), nsmgr).ChildNodes)
			{
				string str = endProperty.Attributes["Name"].Value;
				EdmProperty key = end.EntitySet.ElementType.Properties[str];
				string str2 = endProperty.Attributes["ColumnName"].Value;
				map.Add(key, str2);
			}
			relationEndMap.Add(end.CorrespondingAssociationEndMember, map);
		}
		dictionary.Add(associationSet.ElementType, Tuple.Create(entitySet, relationEndMap));
	}
	return dictionary;
}

public class MetadataLoadResult
{
	public EdmItemCollection EdmItems { get; set; }
	public Dictionary<EntityType, Tuple<EntitySet, Dictionary<EdmProperty, EdmProperty>>> PropertyToColumnMapping { get; set; }
	public Dictionary<AssociationType, Tuple<EntitySet, Dictionary<RelationshipEndMember, Dictionary<EdmMember, string>>>> ManyToManyMappings { get; set; }
	public Dictionary<EntityType, Dictionary<EntityType, Dictionary<EdmProperty, string>>> TphMappings { get; set; }
}

/// <summary>
/// Responsible for encapsulating the constants defined in Metadata
/// </summary>
public static class MetadataConsts
{
    public const string CSDL_EXTENSION = ".csdl";

    public const string CSDL_EDMX_SECTION_NAME = "ConceptualModels";
    public const string CSDL_ROOT_ELEMENT_NAME = "Schema";
    public const string EDM_ANNOTATION_09_02 = "http://schemas.microsoft.com/ado/2009/02/edm/annotation";

    public const string SSDL_EXTENSION = ".ssdl";

    public const string SSDL_EDMX_SECTION_NAME = "StorageModels";
    public const string SSDL_ROOT_ELEMENT_NAME = "Schema";

    public const string MSL_EXTENSION = ".msl";

    public const string MSL_EDMX_SECTION_NAME = "Mappings";
    public const string MSL_ROOT_ELEMENT_NAME = "Mapping";

    public const string TT_TEMPLATE_NAME = "TemplateName";
    public const string TT_TEMPLATE_VERSION = "TemplateVersion";
    public const string TT_MINIMUM_ENTITY_FRAMEWORK_VERSION = "MinimumEntityFrameworkVersion";

    public const string DEFAULT_TEMPLATE_VERSION = "4.0";

    public static readonly SchemaConsts V1_SCHEMA_CONSTANTS = new SchemaConsts(
        "http://schemas.microsoft.com/ado/2007/06/edmx",
        "http://schemas.microsoft.com/ado/2006/04/edm",
        "http://schemas.microsoft.com/ado/2006/04/edm/ssdl",
        "urn:schemas-microsoft-com:windows:storage:mapping:CS",
        new Version("3.5"));

    public static readonly SchemaConsts V2_SCHEMA_CONSTANTS = new SchemaConsts(
        "http://schemas.microsoft.com/ado/2008/10/edmx",
        "http://schemas.microsoft.com/ado/2008/09/edm",
        "http://schemas.microsoft.com/ado/2009/02/edm/ssdl",
        "http://schemas.microsoft.com/ado/2008/09/mapping/cs",
        new Version("4.0"));

    public static readonly SchemaConsts V3_SCHEMA_CONSTANTS = new SchemaConsts(
        "http://schemas.microsoft.com/ado/2009/11/edmx",
        "http://schemas.microsoft.com/ado/2009/11/edm",
        "http://schemas.microsoft.com/ado/2009/11/edm/ssdl",
        "http://schemas.microsoft.com/ado/2009/11/mapping/cs",
        new Version("4.5"));
}

public struct SchemaConsts
{
    public SchemaConsts(string edmxNamespace, string csdlNamespace, string ssdlNamespace, string mslNamespace, Version minimumTemplateVersion) : this()
    {
        EdmxNamespace = edmxNamespace;
        CsdlNamespace = csdlNamespace;
        SsdlNamespace = ssdlNamespace;
        MslNamespace = mslNamespace;
        MinimumTemplateVersion = minimumTemplateVersion;
    }

   public string EdmxNamespace { get; private set; }
    public string CsdlNamespace { get; private set; }
    public string SsdlNamespace { get; private set; }
    public string MslNamespace { get; private set; }
    public Version MinimumTemplateVersion { get; private set; }
}
#>