<#@ template debug="true" hostspecific="true" language="C#" #> <#@ assembly name="System.Core"#> <#@ assembly name="System.Data.Linq"#> <#@ assembly name="EnvDTE"#> <#@ assembly name="System.Xml"#> <#@ assembly name="System.Xml.Linq"#> <#@ assembly name="$(SolutionDir)\packages\EntityFramework.6.1.3\lib\net40\EntityFramework.dll" #> <#@ assembly name="$(SolutionDir)\EMIS.Entities.HRServices\bin\Debug\EMIS.Entities.HRServices.dll" #> <#@ import namespace="System.Diagnostics" #> <#@ import namespace="System"#> <#@ import namespace="System.CodeDom"#> <#@ import namespace="System.CodeDom.Compiler"#> <#@ import namespace="System.Collections.Generic"#> <#@ import namespace="System.Data.Linq"#> <#@ import namespace="System.Data.Linq.Mapping"#> <#@ import namespace="System.IO"#> <#@ import namespace="System.Linq"#> <#@ import namespace="System.Reflection"#> <#@ import namespace="System.Text"#> <#@ import namespace="System.Xml.Linq"#> <#@ output extension="/" #> <#@ import namespace="Microsoft.VisualStudio.TextTemplating"#> <# var manager = Manager.Create(Host, GenerationEnvironment); var blAssembly = Assembly.GetAssembly(typeof(EMIS.Entities.HRServices.HRServiceContextContainer)); var types = blAssembly.GetTypes().Where(x => x.FullName.StartsWith("EMIS.Entities.HRServices") && !x.Name.EndsWith("Mapping") && x.Name != "TableKeyDictionary" && !x.Name.EndsWith("Context") && !x.Name.EndsWith("DbFunctions")); foreach(var type in types) { int startIndex = type.Name.IndexOf("_") + 1; int length = type.Name.Length - startIndex; string classShortName = type.Name.Substring(startIndex, length); string className = classShortName + "Repository"; if (type.Name.StartsWith("VW")) { int secondStartIndex = classShortName.IndexOf("_") + 1; classShortName = "VW" + classShortName.Substring(secondStartIndex, classShortName.Length - secondStartIndex); className = classShortName + "Repository"; } if (classShortName.StartsWith("SS_")) { continue; } manager.StartNewFile(className + ".cs"); #> using System; using System.Collections.Generic; using System.Linq; using System.Text; using EMIS.Entities.HRServices; namespace EMIS.DataLogic.Repositories.HRService { public class <#=className#> : Repository<<#=type.Name#>> { public <#=className#>(HRUnitOfWork unitOfWork) : base(unitOfWork) { } } } <# manager.EndBlock(); } manager.Process(true); #> <#+ // Manager class records the various blocks so it can split them up class Manager { private class Block { public String Name; public int Start, Length; } private Block currentBlock; private List files = new List(); private Block footer = new Block(); private Block header = new Block(); private ITextTemplatingEngineHost host; private StringBuilder template; protected List generatedFileNames = new List(); public static Manager Create(ITextTemplatingEngineHost host, StringBuilder template) { return (host is IServiceProvider) ? new VSManager(host, template) : new Manager(host, template); } public void StartNewFile(String name) { if (name == null) throw new ArgumentNullException("name"); CurrentBlock = new Block { Name = name }; } public void StartFooter() { CurrentBlock = footer; } public void StartHeader() { CurrentBlock = header; } public void EndBlock() { if (CurrentBlock == null) return; CurrentBlock.Length = template.Length - CurrentBlock.Start; if (CurrentBlock != header && CurrentBlock != footer) files.Add(CurrentBlock); currentBlock = null; } public virtual void Process(bool split) { if (split) { EndBlock(); String headerText = template.ToString(header.Start, header.Length); String footerText = template.ToString(footer.Start, footer.Length); String outputPath = Path.GetDirectoryName(host.TemplateFile); files.Reverse(); foreach(Block block in files) { String fileName = Path.Combine(outputPath, block.Name); String content = headerText + template.ToString(block.Start, block.Length) + footerText; generatedFileNames.Add(fileName); CreateFile(fileName, content); template.Remove(block.Start, block.Length); } } } protected virtual void CreateFile(String fileName, String content) { if (IsFileContentDifferent(fileName, content)) File.WriteAllText(fileName, content); } public virtual String GetCustomToolNamespace(String fileName) { return null; } public virtual String DefaultProjectNamespace { get { return null; } } protected bool IsFileContentDifferent(String fileName, String newContent) { return !(File.Exists(fileName) && File.ReadAllText(fileName) == newContent); } private Manager(ITextTemplatingEngineHost host, StringBuilder template) { this.host = host; this.template = template; } private Block CurrentBlock { get { return currentBlock; } set { if (CurrentBlock != null) EndBlock(); if (value != null) value.Start = template.Length; currentBlock = value; } } private class VSManager: Manager { private EnvDTE.ProjectItem templateProjectItem; private EnvDTE.DTE dte; private Action checkOutAction; private Action> projectSyncAction; public override String DefaultProjectNamespace { get { return templateProjectItem.ContainingProject.Properties.Item("DefaultNamespace").Value.ToString(); } } public override String GetCustomToolNamespace(string fileName) { return dte.Solution.FindProjectItem(fileName).Properties.Item("CustomToolNamespace").Value.ToString(); } public override void Process(bool split) { if (templateProjectItem.ProjectItems == null) return; base.Process(split); projectSyncAction.EndInvoke(projectSyncAction.BeginInvoke(generatedFileNames, null, null)); } protected override void CreateFile(String fileName, String content) { if (IsFileContentDifferent(fileName, content)) { CheckoutFileIfRequired(fileName); File.WriteAllText(fileName, content); } } internal VSManager(ITextTemplatingEngineHost host, StringBuilder template) : base(host, template) { var hostServiceProvider = (IServiceProvider) host; if (hostServiceProvider == null) throw new ArgumentNullException("Could not obtain IServiceProvider"); dte = (EnvDTE.DTE) hostServiceProvider.GetService(typeof(EnvDTE.DTE)); if (dte == null) throw new ArgumentNullException("Could not obtain DTE from host"); templateProjectItem = dte.Solution.FindProjectItem(host.TemplateFile); checkOutAction = (String fileName) => dte.SourceControl.CheckOutItem(fileName); projectSyncAction = (IEnumerable keepFileNames) => ProjectSync(templateProjectItem, keepFileNames); } private static void ProjectSync(EnvDTE.ProjectItem templateProjectItem, IEnumerable keepFileNames) { var keepFileNameSet = new HashSet(keepFileNames); var projectFiles = new Dictionary(); var originalFilePrefix = Path.GetFileNameWithoutExtension(templateProjectItem.get_FileNames(0)) + "."; foreach(EnvDTE.ProjectItem projectItem in templateProjectItem.ProjectItems) projectFiles.Add(projectItem.get_FileNames(0), projectItem); // Remove unused items from the project foreach(var pair in projectFiles) if (!keepFileNames.Contains(pair.Key) && !(Path.GetFileNameWithoutExtension(pair.Key) + ".").StartsWith(originalFilePrefix)) pair.Value.Delete(); // Add missing files to the project foreach(String fileName in keepFileNameSet) if (!projectFiles.ContainsKey(fileName)) templateProjectItem.ProjectItems.AddFromFile(fileName); } private void CheckoutFileIfRequired(String fileName) { var sc = dte.SourceControl; if (sc != null && sc.IsItemUnderSCC(fileName) && !sc.IsItemCheckedOut(fileName)) checkOutAction.EndInvoke(checkOutAction.BeginInvoke(fileName, null, null)); } } } #>