#region Copyright // /************************************************************************ // Copyright (c) 2016 Jamie Rees // File: ApplicationTesterModule.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.IO; using Nancy; using Nancy.ModelBinding; using Nancy.Security; using Nancy.Validation; using NLog; using Ombi.Api.Interfaces; using Ombi.Core; using Ombi.Core.SettingModels; using Ombi.UI.Helpers; using Ombi.UI.Models; using ISecurityExtensions = Ombi.Core.ISecurityExtensions; namespace Ombi.UI.Modules { public class ApplicationTesterModule : BaseAuthModule { public ApplicationTesterModule(ICouchPotatoApi cpApi, ISonarrApi sonarrApi, IPlexApi plexApi, ISickRageApi srApi, IHeadphonesApi hpApi, ISettingsService pr, ISecurityExtensions security, IWatcherApi watcherApi) : base("test", pr, security) { this.RequiresAuthentication(); CpApi = cpApi; SonarrApi = sonarrApi; PlexApi = plexApi; SickRageApi = srApi; HeadphonesApi = hpApi; WatcherApi = watcherApi; Post["/cp"] = _ => CouchPotatoTest(); Post["/sonarr"] = _ => SonarrTest(); Post["/plex"] = _ => PlexTest(); Post["/sickrage"] = _ => SickRageTest(); Post["/headphones"] = _ => HeadphonesTest(); Post["/plexdb"] = _ => TestPlexDb(); Post["/watcher"] = _ => WatcherTest(); } private static readonly Logger Log = LogManager.GetCurrentClassLogger(); private ISonarrApi SonarrApi { get; } private ICouchPotatoApi CpApi { get; } private IPlexApi PlexApi { get; } private ISickRageApi SickRageApi { get; } private IHeadphonesApi HeadphonesApi { get; } private IWatcherApi WatcherApi { get; } private Response CouchPotatoTest() { var couchPotatoSettings = this.Bind(); var valid = this.Validate(couchPotatoSettings); if (!valid.IsValid) { return Response.AsJson(valid.SendJsonError()); } try { var status = CpApi.GetStatus(couchPotatoSettings.FullUri, couchPotatoSettings.ApiKey); return status.success ? Response.AsJson(new JsonResponseModel { Result = true, Message = "Connected to CouchPotato successfully!" }) : Response.AsJson(new JsonResponseModel { Result = false, Message = "Could not connect to CouchPotato, please check your settings." }); } catch (Exception e) // Exceptions are expected if we cannot connect so we will just log and swallow them. { Log.Warn("Exception thrown when attempting to get CP's status: "); Log.Warn(e); var message = $"Could not connect to CouchPotato, please check your settings. Exception Message: {e.Message}"; if (e.InnerException != null) { message = $"Could not connect to CouchPotato, please check your settings. Exception Message: {e.InnerException.Message}"; } return Response.AsJson(new JsonResponseModel { Result = false, Message = message }); } } private Response WatcherTest() { var settings = this.Bind(); var valid = this.Validate(settings); if (!valid.IsValid) { return Response.AsJson(valid.SendJsonError()); } try { var status = WatcherApi.ListMovies(settings.ApiKey, settings.FullUri); return !status.Error ? Response.AsJson(new JsonResponseModel { Result = true, Message = "Connected to Watcher successfully!" }) : Response.AsJson(new JsonResponseModel { Result = false, Message = $"Could not connect to Watcher, Error: {status.ErrorMessage}" }); } catch (Exception e) // Exceptions are expected if we cannot connect so we will just log and swallow them. { Log.Warn("Exception thrown when attempting to test Watcher "); Log.Warn(e); var message = $"Could not connect to Watcher, please check your settings. Exception Message: {e.Message}"; if (e.InnerException != null) { message = $"Could not connect to Watcher, please check your settings. Exception Message: {e.InnerException.Message}"; } return Response.AsJson(new JsonResponseModel { Result = false, Message = message }); } } private Response SonarrTest() { var sonarrSettings = this.Bind(); var valid = this.Validate(sonarrSettings); if (!valid.IsValid) { return Response.AsJson(valid.SendJsonError()); } try { var status = SonarrApi.SystemStatus(sonarrSettings.ApiKey, sonarrSettings.FullUri); return status?.version != null ? Response.AsJson(new JsonResponseModel { Result = true, Message = "Connected to Sonarr successfully!" }) : Response.AsJson(new JsonResponseModel { Result = false, Message = "Could not connect to Sonarr, please check your settings." }); } catch (Exception e) // Exceptions are expected, if we cannot connect so we will just log and swallow them. { Log.Warn("Exception thrown when attempting to get Sonarr's status: "); Log.Warn(e); var message = $"Could not connect to Sonarr, please check your settings. Exception Message: {e.Message}"; if (e.InnerException != null) { message = $"Could not connect to Sonarr, please check your settings. Exception Message: {e.InnerException.Message}"; } return Response.AsJson(new JsonResponseModel { Result = false, Message = message }); } } private Response PlexTest() { var plexSettings = this.Bind(); var valid = this.Validate(plexSettings); if (!valid.IsValid) { return Response.AsJson(valid.SendJsonError()); } if (plexSettings?.PlexAuthToken == null) { return Response.AsJson(new JsonResponseModel { Result = false, Message = "Plex is not setup yet, you need to update your Authentication settings" }); } try { var status = PlexApi.GetStatus(plexSettings.PlexAuthToken, plexSettings.FullUri); return status != null ? Response.AsJson(new JsonResponseModel { Result = true, Message = "Connected to Plex successfully!" }) : Response.AsJson(new JsonResponseModel { Result = false, Message = "Could not connect to Plex, please check your settings." }); } catch (Exception e) // Exceptions are expected, if we cannot connect so we will just log and swallow them. { Log.Warn("Exception thrown when attempting to get Plex's status: "); Log.Warn(e); var message = $"Could not connect to Plex, please check your settings. Exception Message: {e.Message}"; if (e.InnerException != null) { message = $"Could not connect to Plex, please check your settings. Exception Message: {e.InnerException.Message}"; } return Response.AsJson(new JsonResponseModel { Result = false, Message = message }); } } private Response SickRageTest() { var sickRageSettings = this.Bind(); var valid = this.Validate(sickRageSettings); if (!valid.IsValid) { return Response.AsJson(valid.SendJsonError()); } try { var status = SickRageApi.Ping(sickRageSettings.ApiKey, sickRageSettings.FullUri); return status?.result == "success" ? Response.AsJson(new JsonResponseModel { Result = true, Message = "Connected to SickRage successfully!" }) : Response.AsJson(new JsonResponseModel { Result = false, Message = "Could not connect to SickRage, please check your settings." }); } catch (Exception e) // Exceptions are expected, if we cannot connect so we will just log and swallow them. { Log.Warn("Exception thrown when attempting to get SickRage's status: "); Log.Warn(e); var message = $"Could not connect to SickRage, please check your settings. Exception Message: {e.Message}"; if (e.InnerException != null) { message = $"Could not connect to SickRage, please check your settings. Exception Message: {e.InnerException.Message}"; } return Response.AsJson(new JsonResponseModel { Result = false, Message = message }); } } private Response HeadphonesTest() { var settings = this.Bind(); var valid = this.Validate(settings); if (!valid.IsValid) { return Response.AsJson(valid.SendJsonError()); } try { var result = HeadphonesApi.GetVersion(settings.ApiKey, settings.FullUri); if (!string.IsNullOrEmpty(result.latest_version)) { return Response.AsJson(new JsonResponseModel { Result = true, Message = "Connected to Headphones successfully!" }); } return Response.AsJson(new JsonResponseModel { Result = false, Message = "Could not connect to Headphones, please check your settings." }); } catch (Exception e) { Log.Warn("Exception thrown when attempting to get Headphones's status: "); Log.Warn(e); var message = $"Could not connect to Headphones, please check your settings. Exception Message: {e.Message}"; if (e.InnerException != null) { message = $"Could not connect to Headphones, please check your settings. Exception Message: {e.InnerException.Message}"; } return Response.AsJson(new JsonResponseModel { Result = false, Message = message }); ; } } private Response TestPlexDb() { var settings = this.Bind(); var valid = this.Validate(settings); if (!valid.IsValid) { return Response.AsJson(valid.SendJsonError()); } try { var location = string.Empty; if (string.IsNullOrEmpty(settings.PlexDatabaseLocationOverride)) { if (Type.GetType("Mono.Runtime") != null) { // Mono location = Path.Combine("/var/lib/plexmediaserver/Library/Application Support/", "Plex Media Server", "Plug-in Support", "Databases", "com.plexapp.plugins.library.db"); } else { // Default Windows location = Path.Combine(Environment.ExpandEnvironmentVariables("%LOCALAPPDATA%"), "Plex Media Server", "Plug-in Support", "Databases", "com.plexapp.plugins.library.db"); } } else { location = Path.Combine(settings.PlexDatabaseLocationOverride, "Plug-in Support", "Databases", "com.plexapp.plugins.library.db"); } if (File.Exists(location)) { return Response.AsJson(new JsonResponseModel { Result = true, Message = "Found the database!" }); } return Response.AsJson(new JsonResponseModel { Result = false, Message = $"Could not find the database at the following full location : {location}" }); } catch (Exception e) { Log.Warn("Exception thrown when attempting to find the plex database: "); Log.Warn(e); var message = $"Could not find Plex's DB, please check your settings. Exception Message: {e.Message}"; if (e.InnerException != null) { message = $"Could not find Plex's DB, please check your settings. Exception Message: {e.InnerException.Message}"; } return Response.AsJson(new JsonResponseModel { Result = false, Message = message }); ; } } } }