#region Copyright // /************************************************************************ // Copyright (c) 2016 Jamie Rees // File: LayoutModule.cs // Created By: Jamie Rees // // Permission is hereby granted, free of charge, to any person obtaining // a copy of this software and associated documentation files (the // "Software"), to deal in the Software without restriction, including // without limitation the rights to use, copy, modify, merge, publish, // distribute, sublicense, and/or sell copies of the Software, and to // permit persons to whom the Software is furnished to do so, subject to // the following conditions: // // The above copyright notice and this permission notice shall be // included in all copies or substantial portions of the Software. // // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. // ************************************************************************/ #endregion using System; using System.Linq; using System.Threading.Tasks; using Nancy; using NLog; using Ombi.Core; using Ombi.Core.SettingModels; using Ombi.Core.StatusChecker; using Ombi.Helpers; using Ombi.Services.Interfaces; using Ombi.Services.Jobs; using Ombi.UI.Models; using ISecurityExtensions = Ombi.Core.ISecurityExtensions; namespace Ombi.UI.Modules { public class LayoutModule : BaseAuthModule { public LayoutModule(ICacheProvider provider, ISettingsService pr, ISettingsService settings, IJobRecord rec, ISecurityExtensions security) : base("layout", pr, security) { Cache = provider; SystemSettings = settings; Job = rec; Get["/", true] = async (x,ct) => await CheckLatestVersion(); Get["/cacher", true] = async (x,ct) => await CacherRunning(); } private ICacheProvider Cache { get; } private static Logger Log = LogManager.GetCurrentClassLogger(); private ISettingsService SystemSettings { get; } private IJobRecord Job { get; } private async Task CheckLatestVersion() { try { if (!IsAdmin) { return Response.AsJson(new JsonUpdateAvailableModel { UpdateAvailable = false }); } #if DEBUG return Response.AsJson(new JsonUpdateAvailableModel { UpdateAvailable = false }); #endif var checker = new StatusChecker(SystemSettings); var release = await Cache.GetOrSetAsync(CacheKeys.LastestProductVersion, async() => await checker.GetStatus(), 30); return Response.AsJson(release.UpdateAvailable ? new JsonUpdateAvailableModel { UpdateAvailable = true} : new JsonUpdateAvailableModel { UpdateAvailable = false }); } catch (Exception e) { Log.Warn("Exception Thrown when attempting to check the status"); Log.Warn(e); return Response.AsJson(new JsonUpdateAvailableModel { UpdateAvailable = false }); } } private async Task CacherRunning() { try { var jobs = await Job.GetJobsAsync(); // Check to see if any are running var runningJobs = jobs.Where(x => x.Running); // We only want the cachers var cacherJobs = runningJobs.Where(x => x.Name.Equals(JobNames.CpCacher) || x.Name.Equals(JobNames.EpisodeCacher) || x.Name.Equals(JobNames.PlexChecker) || x.Name.Equals(JobNames.SonarrCacher) || x.Name.Equals(JobNames.SrCacher) || x.Name.Equals(JobNames.PlexCacher) || x.Name.Equals(JobNames.WatcherCacher)); return Response.AsJson(cacherJobs.Any() ? new { CurrentlyRunning = true, IsAdmin} : new { CurrentlyRunning = false, IsAdmin }); } catch (Exception e) { Log.Warn("Exception Thrown when attempting to check the status"); Log.Warn(e); return Response.AsJson(new { CurrentlyRunning = false, IsAdmin }); } } } }