Moved source code under src folder - massive change

2026-04-27 22:57:09 -04:00 · 2013-10-02 18:01:32 -07:00
parent 2fc8123d6b
commit 5bf0e197ec
1499 changed files with 1054 additions and 1444 deletions
@@ -0,0 +1,67 @@
+using NLog.Config;
+using NLog;
+using NLog.Targets;
+using NzbDrone.Core.Lifecycle;
+using NzbDrone.Core.Messaging.Commands;
+using NzbDrone.Core.Messaging.Commands.Tracking;
+using NzbDrone.Core.Messaging.Events;
+
+namespace NzbDrone.Core.ProgressMessaging
+{
+
+    public class ProgressMessageTarget : Target, IHandle<ApplicationStartedEvent>
+    {
+        private readonly IEventAggregator _eventAggregator;
+        private readonly ITrackCommands _trackCommands;
+        private static LoggingRule _rule;
+
+        public ProgressMessageTarget(IEventAggregator eventAggregator, ITrackCommands trackCommands)
+        {
+            _eventAggregator = eventAggregator;
+            _trackCommands = trackCommands;
+        }
+
+        protected override void Write(LogEventInfo logEvent)
+        {
+            var command = GetCurrentCommand();
+
+            if (IsClientMessage(logEvent, command))
+            {
+                command.SetMessage(logEvent.FormattedMessage);
+                _eventAggregator.PublishEvent(new CommandUpdatedEvent(command));
+            }
+        }
+
+
+        private Command GetCurrentCommand()
+        {
+            var commandId = MappedDiagnosticsContext.Get("CommandId");
+
+            if (string.IsNullOrWhiteSpace(commandId))
+            {
+                return null;
+            }
+
+            return _trackCommands.GetById(commandId);
+        }
+
+        private bool IsClientMessage(LogEventInfo logEvent, Command command)
+        {
+            if (command == null || !command.SendUpdatesToClient)
+            {
+                return false;
+            }
+
+            return logEvent.Properties.ContainsKey("Status");
+        }
+
+        public void Handle(ApplicationStartedEvent message)
+        {
+            _rule = new LoggingRule("*", LogLevel.Trace, this);
+
+            LogManager.Configuration.AddTarget("ProgressMessagingLogger", this);
+            LogManager.Configuration.LoggingRules.Add(_rule);
+            LogManager.ReconfigExistingLoggers();
+        }
+    }
+}