SharpDevelop/src/AddIns/Misc/TextTemplating/Project/Src/TextTemplatingFilePreprocessor.cs

// Copyright (c) AlphaSierraPapa for the SharpDevelop Team (for details please see \doc\copyright.txt)
// This code is distributed under the GNU LGPL (for details please see \doc\license.txt)

using System;
using System.CodeDom.Compiler;
using System.IO;
using System.Text;

using ICSharpCode.Core;
using ICSharpCode.SharpDevelop;
using ICSharpCode.SharpDevelop.Project;
using Microsoft.CSharp;

namespace ICSharpCode.TextTemplating
{
	public class TextTemplatingFilePreprocessor : TextTemplatingFileProcessor, ITextTemplatingFilePreprocessor
	{
		public TextTemplatingFilePreprocessor(
			ITextTemplatingHost host,
			FileProjectItem templateFile,
			ITextTemplatingCustomToolContext context)
			: base(host, templateFile, context)
		{
		}
		
		public void PreprocessTemplate()
		{
			ClearTasksExceptCommentTasks();
			
			string outputFileName = GetOutputFileName();
			string classNamespace = GetClassNamespace();
			
			SetNamespaceHint(classNamespace);
			if (TryGenerateOutputFileForTemplate(outputFileName, classNamespace)) {
				AddOutputFileToProjectIfRequired(outputFileName);
			}
			AddAnyErrorsToTaskList();
			BringErrorsToFrontIfRequired();
		}
		
		string GetOutputFileName()
		{
			string extension = GetFileExtensionForProject();
			return Path.ChangeExtension(TemplateFile.FileName, extension);
		}
		
		string GetFileExtensionForProject()
		{
			if (TemplateFile.Project.Language == "VB") {
				return ".vb";
			}
			return ".cs";
		}
		
		string GetClassNamespace()
		{
			var hint = new NamespaceHint(TemplateFile);
			return hint.ToString();
		}
		
		bool TryGenerateOutputFileForTemplate(string outputFileName, string classNamespace)
		{
			string language = null;
			string[] references = null;
			
			string className = GetClassName();
			
			string inputFileName = TemplateFile.FileName;
			
			try {
				return Host.PreprocessTemplate(
					inputFileName,
					className,
					classNamespace,
					outputFileName,
					Encoding.UTF8,
					out language,
					out references);
			} catch (Exception ex) {
				AddCompilerErrorToTemplatingHost(ex, inputFileName);
				DebugLogException(ex, inputFileName);
			}
			return false;
		}
		
		string GetClassName()
		{
			string className = Path.GetFileNameWithoutExtension(TemplateFile.FileName);
			return CreateValidClassName(className);
		}
		
		string CreateValidClassName(string className)
		{
			return CreateCodeDomProvider().CreateValidIdentifier(className);
		}
		
		CodeDomProvider CreateCodeDomProvider()
		{
			CodeDomProvider provider = TemplateFile.Project.CreateCodeDomProvider();
			if (provider != null) {
				return provider;
			}
			AddMissingCodeDomProviderTask();
			return new CSharpCodeProvider();
		}
		
		void AddMissingCodeDomProviderTask()
		{
			string message = "Project does not provide a CodeDomProvider. Using C# provider by default";
			Context.AddTask(new SDTask(null, message, 0, 0, TaskType.Warning));
		}
	}
}