You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
|
|
|
using System;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using Autofac;
|
|
|
|
|
using Autofac.Core;
|
|
|
|
|
using NLog;
|
|
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Instrumentation
|
|
|
|
|
{
|
|
|
|
|
public class LogInjectionModule : Module
|
|
|
|
|
{
|
|
|
|
|
protected override void AttachToComponentRegistration(IComponentRegistry registry, IComponentRegistration registration)
|
|
|
|
|
{
|
|
|
|
|
registration.Preparing += OnComponentPreparing;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static void OnComponentPreparing(object sender, PreparingEventArgs e)
|
|
|
|
|
{
|
|
|
|
|
e.Parameters = e.Parameters.Union(new[]
|
|
|
|
|
{
|
|
|
|
|
new ResolvedParameter((p, i) => p.ParameterType == typeof(Logger), (p,i)=> GetLogger(p.Member.DeclaringType))
|
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static object GetLogger(Type type)
|
|
|
|
|
{
|
|
|
|
|
const string STRING_TO_REMOVE = "NzbDrone";
|
|
|
|
|
|
|
|
|
|
var loggerName = type.FullName;
|
|
|
|
|
if (loggerName.StartsWith(STRING_TO_REMOVE))
|
|
|
|
|
{
|
|
|
|
|
loggerName = loggerName.Substring(STRING_TO_REMOVE.Length + 1);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return LogManager.GetLogger(loggerName);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|