jellyfin-server/MediaBrowser.Api/ConnectService.cs

using MediaBrowser.Common.Extensions;
using MediaBrowser.Controller.Connect;
using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.Net;
using MediaBrowser.Model.Connect;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using MediaBrowser.Controller.Session;
using MediaBrowser.Model.Services;

namespace MediaBrowser.Api
{
    [Route("/Users/{Id}/Connect/Link", "POST", Summary = "Creates a Connect link for a user")]
    [Authenticated(Roles = "Admin")]
    public class CreateConnectLink : IReturn<UserLinkResult>
    {
        [ApiMember(Name = "Id", Description = "User Id", IsRequired = true, DataType = "string", ParameterType = "query", Verb = "POST")]
        public string Id { get; set; }

        [ApiMember(Name = "ConnectUsername", Description = "Connect username", IsRequired = true, DataType = "string", ParameterType = "query", Verb = "POST")]
        public string ConnectUsername { get; set; }
    }

    [Route("/Users/{Id}/Connect/Link", "DELETE", Summary = "Removes a Connect link for a user")]
    [Authenticated(Roles = "Admin")]
    public class DeleteConnectLink : IReturnVoid
    {
        [ApiMember(Name = "Id", Description = "User Id", IsRequired = true, DataType = "string", ParameterType = "query", Verb = "DELETE")]
        public string Id { get; set; }
    }

    [Route("/Connect/Invite", "POST", Summary = "Creates a Connect link for a user")]
    [Authenticated(Roles = "Admin")]
    public class CreateConnectInvite : IReturn<UserLinkResult>
    {
        [ApiMember(Name = "ConnectUsername", Description = "Connect username", IsRequired = true, DataType = "string", ParameterType = "body", Verb = "POST")]
        public string ConnectUsername { get; set; }

        [ApiMember(Name = "SendingUserId", Description = "Sending User Id", IsRequired = true, DataType = "string", ParameterType = "body", Verb = "POST")]
        public string SendingUserId { get; set; }

        [ApiMember(Name = "EnabledLibraries", Description = "EnabledLibraries", IsRequired = true, DataType = "string", ParameterType = "body", Verb = "POST")]
        public string EnabledLibraries { get; set; }

        [ApiMember(Name = "EnabledChannels", Description = "EnabledChannels", IsRequired = true, DataType = "string", ParameterType = "body", Verb = "POST")]
        public string EnabledChannels { get; set; }

        [ApiMember(Name = "EnableLiveTv", Description = "EnableLiveTv", IsRequired = true, DataType = "string", ParameterType = "body", Verb = "POST")]
        public bool EnableLiveTv { get; set; }
    }


    [Route("/Connect/Pending", "GET", Summary = "Creates a Connect link for a user")]
    [Authenticated(Roles = "Admin")]
    public class GetPendingGuests : IReturn<List<ConnectAuthorization>>
    {
    }


    [Route("/Connect/Pending", "DELETE", Summary = "Deletes a Connect link for a user")]
    [Authenticated(Roles = "Admin")]
    public class DeleteAuthorization : IReturnVoid
    {
        [ApiMember(Name = "Id", Description = "Authorization Id", IsRequired = true, DataType = "string", ParameterType = "query", Verb = "DELETE")]
        public string Id { get; set; }
    }

    [Route("/Connect/Exchange", "GET", Summary = "Gets the corresponding local user from a connect user id")]
    [Authenticated]
    public class GetLocalUser : IReturn<ConnectAuthenticationExchangeResult>
    {
        [ApiMember(Name = "ConnectUserId", Description = "ConnectUserId", IsRequired = true, DataType = "string", ParameterType = "query", Verb = "POST")]
        public string ConnectUserId { get; set; }
    }

    public class ConnectService : BaseApiService
    {
        private readonly IConnectManager _connectManager;
        private readonly ISessionManager _sessionManager;
        private readonly IAuthorizationContext _authContext;

        public ConnectService(IConnectManager connectManager, ISessionManager sessionManager, IAuthorizationContext authContext)
        {
            _connectManager = connectManager;
            _sessionManager = sessionManager;
            _authContext = authContext;
        }

        public object Post(CreateConnectLink request)
        {
            return _connectManager.LinkUser(request.Id, request.ConnectUsername);
        }

        public object Post(CreateConnectInvite request)
        {
            var enabledLibraries = (request.EnabledLibraries ?? string.Empty)
                .Split(',')
                .Where(i => !string.IsNullOrWhiteSpace(i))
                .ToArray();

            var enabledChannels = (request.EnabledChannels ?? string.Empty)
                .Split(',')
                .Where(i => !string.IsNullOrWhiteSpace(i))
                .ToArray();

            return _connectManager.InviteUser(new ConnectAuthorizationRequest
            {
                ConnectUserName = request.ConnectUsername,
                SendingUserId = request.SendingUserId,
                EnabledLibraries = enabledLibraries,
                EnabledChannels = enabledChannels,
                EnableLiveTv = request.EnableLiveTv
            });
        }

        public void Delete(DeleteConnectLink request)
        {
            var task = _connectManager.RemoveConnect(request.Id);

            Task.WaitAll(task);
        }

        public async Task<object> Get(GetPendingGuests request)
        {
            var result = await _connectManager.GetPendingGuests().ConfigureAwait(false);

            return ToOptimizedResult(result);
        }

        public void Delete(DeleteAuthorization request)
        {
            var task = _connectManager.CancelAuthorization(request.Id);

            Task.WaitAll(task);
        }

        public async Task<object> Get(GetLocalUser request)
        {
            var user = await _connectManager.GetLocalUser(request.ConnectUserId).ConfigureAwait(false);

            if (user == null)
            {
                throw new ResourceNotFoundException();
            }

            var auth = _authContext.GetAuthorizationInfo(Request);

            if (string.IsNullOrWhiteSpace(auth.Client))
            {
                return ToOptimizedResult(new ConnectAuthenticationExchangeResult
                {
                    AccessToken = user.ConnectAccessKey,
                    LocalUserId = user.Id.ToString("N")
                });
            }

            var session = await _sessionManager.CreateNewSession(new AuthenticationRequest
            {
                App = auth.Client,
                AppVersion = auth.Version,
                DeviceId = auth.DeviceId,
                DeviceName = auth.Device,
                RemoteEndPoint = Request.RemoteIp,
                Username = user.Name,
                UserId = user.Id.ToString("N")

            }).ConfigureAwait(false);

            return ToOptimizedResult(new ConnectAuthenticationExchangeResult
            {
                AccessToken = session.AccessToken,
                LocalUserId = session.User.Id
            });
        }
    }
}