using System; using System.Collections.Generic; using System.Linq; using System.Net; using System.Net.Http; using System.Reflection; using System.Text; using System.Web.Http; using Newtonsoft.Json; using Newtonsoft.Json.Linq; using Umbraco.Core; using Umbraco.Core.Configuration; using Umbraco.Core.Logging; using Umbraco.Core.Persistence; using Umbraco.Web.Install.Models; using Umbraco.Web.WebApi; namespace Umbraco.Web.Install.Controllers { [AngularJsonOnlyConfiguration] [HttpInstallAuthorize] public class InstallApiController : ApiController { protected InstallApiController() : this(UmbracoContext.Current) { } protected InstallApiController(UmbracoContext umbracoContext) { if (umbracoContext == null) throw new ArgumentNullException("umbracoContext"); UmbracoContext = umbracoContext; } /// /// Returns the current UmbracoContext /// public UmbracoContext UmbracoContext { get; private set; } public ApplicationContext ApplicationContext { get { return UmbracoContext.Application; } } private InstallHelper _helper; internal InstallHelper InstallHelper { get { return _helper ?? (_helper = new InstallHelper(UmbracoContext)); } } /// /// Gets the install setup /// /// public InstallSetup GetSetup() { var setup = new InstallSetup() { Status = InstallHelper.GetStatus() }; //TODO: Check for user/site token var steps = new List(); steps.AddRange(InstallHelper.GetSteps().Where(x => x.RequiresExecution())); setup.Steps = steps; return setup; } /// /// Checks if the db can be connected to /// /// public HttpResponseMessage PostCheckDbConnection() { throw new NotImplementedException(); } /// /// Checks if the db credentials are correct /// /// public HttpResponseMessage PostCheckDbCredentials() { throw new NotImplementedException(); } public IEnumerable GetPackages() { var r = new org.umbraco.our.Repository(); var modules = r.Modules(); return modules.Select(package => new Package() { Id = package.RepoGuid, Name = package.Text, Thumbnail = package.Thumbnail }); } /// /// Does the install /// /// public HttpResponseMessage PostPerformInstall(InstallInstructions installModel) { if (installModel == null) throw new ArgumentNullException("installModel"); var status = InstallStatusTracker.GetStatus().ToArray(); if (status.Any() == false) { status = InstallStatusTracker.Initialize(installModel.InstallId, InstallHelper.GetSteps()).ToArray(); } foreach (var stepStatus in status) { //if it is not complete, then we need to execute it if (stepStatus.IsComplete == false) { var step = InstallHelper.GetSteps().Single(x => x.Name == stepStatus.Name); JToken instruction = null; if (step.HasUIElement) { //Since this is a UI instruction, we will extract the model from it if (installModel.Instructions.Any(x => x.Key == step.Name) == false) { return Request.CreateValidationErrorResponse("No instruction defined for step: " + step.Name); } instruction = installModel.Instructions[step.Name]; } //If this step doesn't require execution then continue to the next one. if (step.RequiresExecution() == false) { //set this as complete and continue InstallStatusTracker.SetComplete(installModel.InstallId, step.Name, null); continue; } try { var setupData = ExecuteStep(step, instruction); //update the status InstallStatusTracker.SetComplete(installModel.InstallId, step.Name, setupData != null ? setupData.SavedStepData : null); //check if there's a custom view to return for this step if (setupData != null && setupData.View.IsNullOrWhiteSpace() == false) { return Json(new { complete = false, stepCompleted = step.Name, view = setupData.View, model = setupData.ViewModel }, HttpStatusCode.OK); } return Json(new { complete = false, stepCompleted = step.Name }, HttpStatusCode.OK); } catch (Exception ex) { if (ex is TargetInvocationException && ex.InnerException != null) { ex = ex.InnerException; } var installException = ex as InstallException; if (installException != null) { return Json(new { view = installException.View, model = installException.ViewModel, message = installException.Message }, HttpStatusCode.BadRequest); } return Request.CreateValidationErrorResponse("An error occurred executing the step: " + step.Name + ". Error: " + ex.Message); } } } InstallStatusTracker.Reset(); return Json(new { complete = true }, HttpStatusCode.OK); } internal InstallSetupResult ExecuteStep(InstallSetupStep step, JToken instruction) { using (DisposableTimer.TraceDuration("Executing installation step: " + step.Name, "Step completed")) { var model = instruction == null ? null : instruction.ToObject(step.StepType); var genericStepType = typeof(InstallSetupStep<>); Type[] typeArgs = { step.StepType }; var typedStepType = genericStepType.MakeGenericType(typeArgs); try { var method = typedStepType.GetMethods().Single(x => x.Name == "Execute"); return (InstallSetupResult)method.Invoke(step, new object[] { model }); } catch (Exception ex) { LogHelper.Error("Installation step " + step.Name + " failed.", ex); throw; } } } private HttpResponseMessage Json(object jsonObject, HttpStatusCode status) { var response = Request.CreateResponse(status); var json = JObject.FromObject(jsonObject); response.Content = new StringContent(json.ToString(), Encoding.UTF8, "application/json"); return response; } } }