#region Copyright // /************************************************************************ // Copyright (c) 2016 Jamie Rees // File: UserWizardModule.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 Nancy.Authentication.Forms; using Nancy.Extensions; using Nancy.ModelBinding; using Nancy.Responses.Negotiation; using Nancy.Validation; using NLog; using PlexRequests.Api.Interfaces; using PlexRequests.Core; using PlexRequests.Core.SettingModels; using PlexRequests.Helpers; using PlexRequests.Helpers.Analytics; using PlexRequests.UI.Helpers; using PlexRequests.UI.Models; using Action = PlexRequests.Helpers.Analytics.Action; namespace PlexRequests.UI.Modules { public class UserWizardModule : BaseModule { public UserWizardModule(ISettingsService pr, ISettingsService plex, IPlexApi plexApi, ISettingsService auth, ICustomUserMapper m, IAnalytics a) : base("wizard", pr) { PlexSettings = plex; PlexApi = plexApi; PlexRequestSettings = pr; Auth = auth; Mapper = m; Analytics = a; Get["/", true] = async (x, ct) => { a.TrackEventAsync(Category.Wizard, Action.Start, "Started the wizard", Username, CookieHelper.GetAnalyticClientId(Cookies)); var settings = await PlexRequestSettings.GetSettingsAsync(); if (settings.Wizard) { return Context.GetRedirect("~/search"); } return View["Index"]; }; Post["/plexAuth"] = x => PlexAuth(); Post["/plex", true] = async (x, ct) => await Plex(); Post["/plexrequest", true] = async (x, ct) => await PlexRequest(); Post["/auth", true] = async (x, ct) => await Authentication(); Post["/createuser",true] = async (x,ct) => await CreateUser(); } private ISettingsService PlexSettings { get; } private IPlexApi PlexApi { get; } private ISettingsService PlexRequestSettings { get; } private ISettingsService Auth { get; } private ICustomUserMapper Mapper { get; } private IAnalytics Analytics { get; } private static Logger Log = LogManager.GetCurrentClassLogger(); private Response PlexAuth() { var user = this.Bind(); if (string.IsNullOrEmpty(user.username) || string.IsNullOrEmpty(user.password)) { return Response.AsJson(new JsonResponseModel { Result = false, Message = "Please provide a valid username and password" }); } var model = PlexApi.SignIn(user.username, user.password); if (model?.user == null) { return Response.AsJson(new JsonResponseModel { Result = false, Message = "Incorrect username or password!" }); } // Set the auth token in the session so we can use it in the next form Session[SessionKeys.UserWizardPlexAuth] = model.user.authentication_token; var servers = PlexApi.GetServer(model.user.authentication_token); var firstServer = servers.Server.FirstOrDefault(); return Response.AsJson(new { Result = true, firstServer?.Port, Ip = firstServer?.LocalAddresses, firstServer?.Scheme }); } private async Task Plex() { var form = this.Bind(); var valid = this.Validate(form); if (!valid.IsValid) { return Response.AsJson(valid.SendJsonError()); } form.PlexAuthToken = Session[SessionKeys.UserWizardPlexAuth].ToString(); // Set the auth token from the previous form // Get the machine ID from the settings (This could have changed) try { var servers = PlexApi.GetServer(form.PlexAuthToken); var firstServer = servers.Server.FirstOrDefault(x => x.AccessToken == form.PlexAuthToken); Session[SessionKeys.UserWizardMachineId] = firstServer?.MachineIdentifier; } catch (Exception e) { // Probably bad settings, just continue Log.Error(e); } var result = await PlexSettings.SaveSettingsAsync(form); if (result) { return Response.AsJson(new JsonResponseModel { Result = true }); } return Response.AsJson(new JsonResponseModel { Result = false, Message = "Could not save the settings to the database, please try again." }); } private async Task PlexRequest() { var form = this.Bind(); var valid = this.Validate(form); if (!valid.IsValid) { return Response.AsJson(valid.SendJsonError()); } var currentSettings = await PlexRequestSettings.GetSettingsAsync(); currentSettings.SearchForMovies = form.SearchForMovies; currentSettings.SearchForTvShows = form.SearchForTvShows; currentSettings.SearchForMusic = form.SearchForMusic; currentSettings.RequireMovieApproval = form.RequireMovieApproval; currentSettings.RequireTvShowApproval = form.RequireTvShowApproval; currentSettings.RequireMusicApproval = form.RequireMusicApproval; var result = await PlexRequestSettings.SaveSettingsAsync(currentSettings); if (result) { return Response.AsJson(new { Result = true }); } return Response.AsJson(new JsonResponseModel { Result = false, Message = "Could not save the settings to the database, please try again." }); } private async Task Authentication() { var form = this.Bind(); var result = await Auth.SaveSettingsAsync(form); if (result) { return Response.AsJson(new JsonResponseModel { Result = true }); } return Response.AsJson(new JsonResponseModel { Result = false, Message = "Could not save the settings to the database, please try again." }); } private async Task CreateUser() { var username = (string)Request.Form.Username; var userId = Mapper.CreateAdmin(username, Request.Form.Password); Analytics.TrackEventAsync(Category.Wizard, Action.Finish, "Finished the wizard", username, CookieHelper.GetAnalyticClientId(Cookies)); Session[SessionKeys.UsernameKey] = username; // Destroy the Plex Auth Token Session.Delete(SessionKeys.UserWizardPlexAuth); // Update the settings so we know we have been through the wizard var settings = await PlexRequestSettings.GetSettingsAsync(); settings.Wizard = true; await PlexRequestSettings.SaveSettingsAsync(settings); var baseUrl = string.IsNullOrEmpty(settings.BaseUrl) ? string.Empty : $"/{settings.BaseUrl}"; return this.LoginAndRedirect((Guid)userId, fallbackRedirectUrl: $"{baseUrl}/search"); } } }