2013-05-05 00:09:25 +03:00
|
|
|
using System.Collections.Generic;
|
2013-05-06 00:24:33 +03:00
|
|
|
using Microsoft.AspNet.SignalR;
|
|
|
|
using NzbDrone.Api.SignalR;
|
2013-05-11 02:53:50 +03:00
|
|
|
using NzbDrone.Common.Composition;
|
2013-05-05 00:09:25 +03:00
|
|
|
using Owin;
|
|
|
|
|
2013-08-07 08:32:22 +03:00
|
|
|
namespace NzbDrone.Host.Owin.MiddleWare
|
2013-05-05 00:09:25 +03:00
|
|
|
{
|
|
|
|
public class SignalRMiddleWare : IOwinMiddleWare
|
|
|
|
{
|
2013-05-06 00:24:33 +03:00
|
|
|
private readonly IEnumerable<NzbDronePersistentConnection> _persistentConnections;
|
2013-05-05 00:09:25 +03:00
|
|
|
|
2013-05-06 00:24:33 +03:00
|
|
|
public int Order { get { return 0; } }
|
|
|
|
|
2013-05-11 02:53:50 +03:00
|
|
|
public SignalRMiddleWare(IEnumerable<NzbDronePersistentConnection> persistentConnections, IContainer container)
|
2013-05-05 00:09:25 +03:00
|
|
|
{
|
2013-05-06 00:24:33 +03:00
|
|
|
_persistentConnections = persistentConnections;
|
|
|
|
|
|
|
|
SignalrDependencyResolver.Register(container);
|
2013-05-05 00:09:25 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
public void Attach(IAppBuilder appBuilder)
|
|
|
|
{
|
2013-05-06 00:24:33 +03:00
|
|
|
foreach (var nzbDronePersistentConnection in _persistentConnections)
|
|
|
|
{
|
2013-05-06 03:33:43 +03:00
|
|
|
var url = string.Format("signalr/{0}", nzbDronePersistentConnection.Resource.Trim('/'));
|
|
|
|
appBuilder.MapConnection(url, nzbDronePersistentConnection.GetType(), new ConnectionConfiguration { EnableCrossDomain = true });
|
2013-05-06 00:24:33 +03:00
|
|
|
}
|
|
|
|
|
2013-05-05 00:09:25 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|