Orchard/src/Orchard.Web/Modules/Orchard.Setup/Controllers/SetupController.cs

using System;
using System.IO;
using System.Web.Mvc;
using Orchard.Comments.Models;
using Orchard.ContentManagement;
using Orchard.Core.Common.Models;
using Orchard.Core.Navigation.Models;
using Orchard.Core.Settings.Models;
using Orchard.Data;
using Orchard.Data.Migrations;
using Orchard.Environment;
using Orchard.Environment.Configuration;
using Orchard.Security;
using Orchard.Settings;
using Orchard.Setup.ViewModels;
using Orchard.Localization;
using Orchard.UI.Notify;
using MenuItem=Orchard.Core.Navigation.Models.MenuItem;

namespace Orchard.Setup.Controllers {
    public class SetupController : Controller {
        private readonly INotifier _notifier;
        private readonly IDatabaseMigrationManager _databaseMigrationManager;
        private readonly IOrchardHost _orchardHost;
        private readonly IShellSettingsLoader _shellSettingsLoader;

        public SetupController(
            INotifier notifier,
            IDatabaseMigrationManager databaseMigrationManager,
            IOrchardHost orchardHost,
            IShellSettingsLoader shellSettingsLoader) {
            _notifier = notifier;
            _databaseMigrationManager = databaseMigrationManager;
            _orchardHost = orchardHost;
            _shellSettingsLoader = shellSettingsLoader;
            T = NullLocalizer.Instance;
        }

        private Localizer T { get; set; }

        public ActionResult Index(SetupViewModel model) {
            string message = "";
            if(!CanWriteTo(Server.MapPath("~/App_Data"), out message)) {
                _notifier.Error(
                    T(
                        "Hey, it looks like I can't write to the App_Data folder in the root of this application and that's where I need to save some of the information you're about to enter.\r\n\r\nPlease give me (the machine account this application is running under) write access to App_Data so I can get this app all set up for you.\r\n\r\nThanks!\r\n\r\n----\r\n{0}",
                        message));
            }

            return View(model ?? new SetupViewModel { AdminUsername = "admin" });
        }

        [HttpPost, ActionName("Index")]
        public ActionResult IndexPOST(SetupViewModel model) {
            //HACK: (erikpo) Couldn't get a custom ValidationAttribute to validate two properties
            if (!model.DatabaseOptions && string.IsNullOrEmpty(model.DatabaseConnectionString))
                ModelState.AddModelError("DatabaseConnectionString", "A SQL connection string is required");

            if (!ModelState.IsValid) {
                return Index(model);
            }

            try {
                var shellSettings = new ShellSettings {
                                                          Name = "default",
                                                          DataProvider = model.DatabaseOptions ? "SQLite" : "SqlServer",
                                                          DataConnectionString = model.DatabaseConnectionString
                                                      };

                // initialize the database:
                // provider: SqlServer or SQLite 
                // dataFolder: physical path (map before calling). Builtin database will be created in this location
                // connectionString: optional - if provided the dataFolder is essentially ignored, but should still be passed in
                _databaseMigrationManager.CreateCoordinator(shellSettings.DataProvider, Server.MapPath("~/App_Data"), shellSettings.DataConnectionString);

                // creating a standalone environment. 
                // in theory this environment can be used to resolve any normal components by interface, and those
                // components will exist entirely in isolation - no crossover between the safemode container currently in effect
                using (var finiteEnvironment = _orchardHost.CreateStandaloneEnvironment(shellSettings)) {
                    try {
                        var contentManager = finiteEnvironment.Resolve<IContentManager>();

                        // create superuser
                        var membershipService = finiteEnvironment.Resolve<IMembershipService>();
                        var user =
                            membershipService.CreateUser(new CreateUserParams(model.AdminUsername, model.AdminPassword,
                                                                              String.Empty, String.Empty, String.Empty,
                                                                              true));

                        // set site name and settings
                        var siteService = finiteEnvironment.Resolve<ISiteService>();
                        var siteSettings = siteService.GetSiteSettings().As<SiteSettings>();
                        siteSettings.Record.SiteSalt = Guid.NewGuid().ToString("N");
                        siteSettings.Record.SiteName = model.SiteName;
                        siteSettings.Record.SuperUser = model.AdminUsername;
                        siteSettings.Record.PageTitleSeparator = " - ";

                        // create home page as a CMS page
                        var page = contentManager.Create("page");
                        page.As<BodyAspect>().Text = "Welcome to Orchard";
                        page.As<RoutableAspect>().Slug = "";
                        page.As<RoutableAspect>().Title = model.SiteName;
                        page.As<HasComments>().CommentsShown = false;
                        page.As<CommonAspect>().Owner = user;
                        contentManager.Publish(page);
                        siteSettings.Record.HomePage = "PagesHomePageProvider;" + page.Id;

                        // add a menu item for the shiny new home page
                        var homeMenuItem = contentManager.Create("menuitem");
                        homeMenuItem.As<MenuPart>().MenuPosition = "1";
                        homeMenuItem.As<MenuPart>().MenuText = T("Home").ToString();
                        homeMenuItem.As<MenuPart>().OnMainMenu = true;
                        homeMenuItem.As<MenuItem>().Url = Request.Url.AbsolutePath;

                        // add a menu item for the admin
                        var adminMenuItem = contentManager.Create("menuitem");
                        adminMenuItem.As<MenuPart>().MenuPosition = null; //"2";
                        adminMenuItem.As<MenuPart>().MenuText = T("Admin").ToString();
                        adminMenuItem.As<MenuPart>().OnMainMenu = true;
                        //adminMenuItem.As<MenuItem>().Permissions = new [] {StandardPermissions.AccessAdminPanel};
                        //todo: (heskew) pull "/blogs" once the is a ~/admin
                        adminMenuItem.As<MenuItem>().Url = string.Format("{0}admin/blogs", Request.Url.AbsolutePath);

                        var authenticationService = finiteEnvironment.Resolve<IAuthenticationService>();
                        authenticationService.SignIn(user, true);
                    }
                    catch {
                        finiteEnvironment.Resolve<ITransactionManager>().Cancel();
                        throw;
                    }
                }

                _shellSettingsLoader.SaveSettings(shellSettings);

                _orchardHost.Reinitialize();

                _notifier.Information(T("Setup succeeded"));

                // redirect to the welcome page.
                return Redirect("~/");
            }
            catch (Exception exception) {
                _notifier.Error(T("Setup failed: " + exception.Message));
                return Index(model);
            }
        }

        static bool CanWriteTo(string path, out string message) {
            try {
                var systemCheckPath = Path.Combine(path, "_systemcheck.txt");

                System.IO.File.WriteAllText(systemCheckPath, "Communicator check one two one two");
                System.IO.File.AppendAllText(systemCheckPath, "\r\nThis is Bones McCoy on a line to Sulu");
                System.IO.File.Delete(systemCheckPath);

                message = "";
                return true;
            } catch (Exception ex) {
                message = ex.Message.Replace("_systemcheck.txt", "");
                return false;
            }
        }
    }
}