Emby.Server.Implementations/EntryPoints/KeepServerAwake.cs


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

using MediaBrowser.Controller;
using MediaBrowser.Controller.Plugins;
using MediaBrowser.Controller.Session;
using Microsoft.Extensions.Logging;
using System;
using System.Linq;
using MediaBrowser.Model.System;
using MediaBrowser.Model.Threading;

namespace Emby.Server.Implementations.EntryPoints
{
    public class KeepServerAwake : IServerEntryPoint
    {
        private readonly ISessionManager _sessionManager;
        private readonly ILogger _logger;
        private ITimer _timer;
        private readonly IServerApplicationHost _appHost;
        private readonly ITimerFactory _timerFactory;
        private readonly IPowerManagement _powerManagement;

        public KeepServerAwake(ISessionManager sessionManager, ILogger logger, IServerApplicationHost appHost, ITimerFactory timerFactory, IPowerManagement powerManagement)
        {
            _sessionManager = sessionManager;
            _logger = logger;
            _appHost = appHost;
            _timerFactory = timerFactory;
            _powerManagement = powerManagement;
        }

        public void Run()
        {
            _timer = _timerFactory.Create(OnTimerCallback, null, TimeSpan.FromMinutes(5), TimeSpan.FromMinutes(5));
        }

        private void OnTimerCallback(object state)
        {
            var now = DateTime.UtcNow;

            try
            {
                if (_sessionManager.Sessions.Any(i => (now - i.LastActivityDate).TotalMinutes < 15))
                {
                    _powerManagement.PreventSystemStandby();
                }
                else
                {
                    _powerManagement.AllowSystemStandby();
                }
            }
            catch (Exception ex)
            {
                _logger.LogError(ex, "Error resetting system standby timer");
            }
        }

        public void Dispose()
        {
            if (_timer != null)
            {
                _timer.Dispose();
                _timer = null;
            }
        }
    }
}