DbUp ScriptProvider

DBScriptProvider.cs

using DbUp.Engine;
using System;
using System.Collections.Generic;
using DbUp.Engine.Transactions;
using System.Reflection;
using System.Linq;
using System.ComponentModel;

namespace DevDBScriptRunner
{
	public class DBScriptProvider : IScriptProvider
	{
		private readonly Assembly assembly;
		private readonly Func<string, bool> filter;
		private readonly object[] arguments;

		public DBScriptProvider(Assembly assembly, Func<string, bool> filter, params object[] arguments)
		{
			this.assembly = assembly;
			this.filter = filter;
			this.arguments = arguments;
		}

		public IEnumerable<SqlScript> GetScripts(IConnectionManager connectionManager)
		{
			List<SqlScript> scripts = new List<SqlScript>();

			scripts.AddRange(
				this.assembly.GetManifestResourceNames()
				.Where(this.filter).ToArray()
				.Select(resource =>
					SqlScript.FromStream(resource, this.assembly.GetManifestResourceStream(resource))
				)
			);

			var script = typeof(IScript);

			scripts.AddRange(
				connectionManager.ExecuteCommandsWithManagedConnection(
					dbCommandFactory =>
					{
						var _types = this.assembly.GetTypes()
						.Where(type => type.IsClass && script.IsAssignableFrom(type))
						.Select(t2 => new
						{
							Type = t2,
							Description = string.Join(", ", t2.CustomAttributes?.Where(t3 => t3.AttributeType == typeof(DescriptionAttribute)).SelectMany(ca => ca.ConstructorArguments.Select(c => c.Value))) ?? ""
						});

						var onlyDescription = _types.Select(t => t.Description).Where(this.filter);

						return _types.Where(i => onlyDescription.Contains(i.Description)).Select(s =>
							(SqlScript)new LazySqlScript(
								s.Type.FullName + ".cs",
								() => ((IScript)Activator.CreateInstance(s.Type, this.arguments)).ProvideScript(dbCommandFactory)
							)
						);
					}
				)
			);

			return scripts.OrderBy(s => s.Name);
		}
	}
}