aboutsummaryrefslogtreecommitdiff
path: root/MediaBrowser.Api/Sync/SyncJobWebSocketListener.cs
blob: 61a26d160a45546ed1288fc1f6d61de6632aa3dc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
using MediaBrowser.Controller.Net;
using MediaBrowser.Controller.Sync;
using MediaBrowser.Model.Events;
using MediaBrowser.Model.Logging;
using MediaBrowser.Model.Sync;
using System;
using System.Linq;
using System.Threading.Tasks;

namespace MediaBrowser.Api.Sync
{
    /// <summary>
    /// Class SessionInfoWebSocketListener
    /// </summary>
    class SyncJobWebSocketListener : BasePeriodicWebSocketListener<CompleteSyncJobInfo, WebSocketListenerState>
    {
        /// <summary>
        /// Gets the name.
        /// </summary>
        /// <value>The name.</value>
        protected override string Name
        {
            get { return "SyncJob"; }
        }

        private readonly ISyncManager _syncManager;
        private string _jobId;

        public SyncJobWebSocketListener(ILogger logger, ISyncManager syncManager)
            : base(logger)
        {
            _syncManager = syncManager;
            _syncManager.SyncJobCancelled += _syncManager_SyncJobCancelled;
            _syncManager.SyncJobUpdated += _syncManager_SyncJobUpdated;
            _syncManager.SyncJobItemCreated += _syncManager_SyncJobItemCreated;
            _syncManager.SyncJobItemUpdated += _syncManager_SyncJobItemUpdated;
        }

        void _syncManager_SyncJobItemUpdated(object sender, GenericEventArgs<SyncJobItem> e)
        {
            if (string.Equals(e.Argument.Id, _jobId, StringComparison.Ordinal))
            {
                SendData(false);
            }
        }

        void _syncManager_SyncJobItemCreated(object sender, GenericEventArgs<SyncJobItem> e)
        {
            if (string.Equals(e.Argument.Id, _jobId, StringComparison.Ordinal))
            {
                SendData(true);
            }
        }

        protected override void ParseMessageParams(string[] values)
        {
            base.ParseMessageParams(values);

            if (values.Length > 0)
            {
                _jobId = values[0];
            }
        }

        void _syncManager_SyncJobUpdated(object sender, GenericEventArgs<SyncJob> e)
        {
            if (string.Equals(e.Argument.Id, _jobId, StringComparison.Ordinal))
            {
                SendData(false);
            }
        }

        void _syncManager_SyncJobCancelled(object sender, GenericEventArgs<SyncJob> e)
        {
            if (string.Equals(e.Argument.Id, _jobId, StringComparison.Ordinal))
            {
                SendData(true);
            }
        }

        /// <summary>
        /// Gets the data to send.
        /// </summary>
        /// <param name="state">The state.</param>
        /// <returns>Task{SystemInfo}.</returns>
        protected override Task<CompleteSyncJobInfo> GetDataToSend(WebSocketListenerState state)
        {
            var job = _syncManager.GetJob(_jobId);
            var items = _syncManager.GetJobItems(new SyncJobItemQuery
            {
                AddMetadata = true,
                JobId = _jobId
            });

            var info = new CompleteSyncJobInfo
            {
                Job = job,
                JobItems = items.Items.ToList()
            };

            return Task.FromResult(info);
        }

        protected override bool SendOnTimer
        {
            get
            {
                return false;
            }
        }

        protected override void Dispose(bool dispose)
        {
            _syncManager.SyncJobCancelled -= _syncManager_SyncJobCancelled;
            _syncManager.SyncJobUpdated -= _syncManager_SyncJobUpdated;

            base.Dispose(dispose);
        }
    }
}