Fixing upgrading issues

--HG--
branch : 1.x

src/Orchard.Web/Modules/Upgrade/Controllers/ContentPickerController.cs

@@ -1,5 +1,6 @@
 using System;
 using System.Linq;
+using System.Transactions;
 using System.Web.Mvc;
 using Orchard;
 using Orchard.Environment.Features;
@@ -41,7 +42,9 @@ namespace Upgrade.Controllers {
         public ActionResult IndexPOST() {
             if (!_orchardServices.Authorizer.Authorize(StandardPermissions.SiteOwner, T("Not allowed to migrate Orchard.ContentPicker.")))
                 return new HttpUnauthorizedResult();
+
             try {
+                _upgradeService.ExecuteReader("DELETE FROM " + _upgradeService.GetPrefixedTableName("Orchard_ContentPicker_ContentMenuItemPartRecord"), null);
                 _upgradeService.CopyTable("Navigation_ContentMenuItemPartRecord", "Orchard_ContentPicker_ContentMenuItemPartRecord", new string[0]);
                 
                 _orchardServices.Notifier.Information(T("Content Picker menu items were migrated successfully."));

src/Orchard.Web/Modules/Upgrade/Controllers/MediaController.cs

@@ -2,10 +2,10 @@
 using System.Collections.Generic;
 using System.IO;
 using System.Linq;
-using System.Net;
 using System.Web.Mvc;
 using Orchard;
 using Orchard.Environment.Features;
+using Orchard.FileSystems.Media;
 using Orchard.Localization;
 using Orchard.Logging;
 using Orchard.Media.Models;
@@ -16,7 +16,6 @@ using Orchard.Security;
 using Orchard.UI.Admin;
 using Orchard.UI.Notify;
 using MediaFolder = Orchard.Media.Models.MediaFolder;
-using Orchard.Mvc.Extensions;
 
 namespace Upgrade.Controllers {
     [Admin]
@@ -25,16 +24,19 @@ namespace Upgrade.Controllers {
         private readonly IFeatureManager _featureManager;
         private readonly IMediaService _mediaService;
         private readonly IMediaLibraryService _mediaLibraryService;
+        private readonly IStorageProvider _storageProvider;
 
         public MediaController(
             IOrchardServices orchardServices,
             IFeatureManager featureManager,
             IMediaService mediaService,
-            IMediaLibraryService mediaLibraryService) {
+            IMediaLibraryService mediaLibraryService,
+            IStorageProvider storageProvider) {
             _orchardServices = orchardServices;
             _featureManager = featureManager;
             _mediaService = mediaService;
             _mediaLibraryService = mediaLibraryService;
+            _storageProvider = storageProvider;
 
             Logger = NullLogger.Instance;
         }
@@ -95,13 +97,14 @@ namespace Upgrade.Controllers {
             }
             
             try {
-                _orchardServices.Notifier.Information(T("Importing {0}.", mediaFile.MediaPath));
+                var prefix = _mediaService.GetPublicUrl("foo.$$$");
-                var url = _mediaService.GetPublicUrl(mediaFile.MediaPath);
+                var trim = prefix.IndexOf("foo.$$$");
-                url = Url.MakeAbsolute(url);
 
-                var buffer = new WebClient().DownloadData(url);
+                _orchardServices.Notifier.Information(T("Importing {0}.", mediaFile.MediaPath));
-                var stream = new MemoryStream(buffer);
+                var storageFile = _storageProvider.GetFile(mediaFile.MediaPath.Substring(trim));
-                var mediaPart = _mediaLibraryService.ImportStream(mediaLibraryFolder.TermId, stream, Path.GetFileName(url));
+                using (var stream = storageFile.OpenRead()) {
+                    _mediaLibraryService.ImportStream(mediaLibraryFolder.TermId, stream, Path.GetFileName(mediaFile.MediaPath));
+                }
             }
             catch(Exception e) {
                 _orchardServices.Notifier.Error(T("Error while importing {0}. Please check the logs", mediaFile.MediaPath));

src/Orchard.Web/Modules/Upgrade/Controllers/TaxonomyController.cs

@@ -1,10 +1,8 @@
 using System;
-using System.Collections.Generic;
 using System.Linq;
+using System.Transactions;
 using System.Web.Mvc;
 using Orchard;
-using Orchard.ContentManagement;
-using Orchard.ContentManagement.MetaData;
 using Orchard.Environment.Features;
 using Orchard.Localization;
 using Orchard.Logging;
@@ -12,23 +10,19 @@ using Orchard.Security;
 using Orchard.UI.Admin;
 using Orchard.UI.Notify;
 using Upgrade.Services;
-using Upgrade.ViewModels;
 
 namespace Upgrade.Controllers {
     [Admin]
     public class TaxonomyController : Controller {
         private readonly IUpgradeService _upgradeService;
-        private readonly IContentDefinitionManager _contentDefinitionManager;
         private readonly IOrchardServices _orchardServices;
         private readonly IFeatureManager _featureManager;
 
         public TaxonomyController(
             IUpgradeService upgradeService,
-            IContentDefinitionManager contentDefinitionManager,
             IOrchardServices orchardServices,
             IFeatureManager featureManager) {
             _upgradeService = upgradeService;
-            _contentDefinitionManager = contentDefinitionManager;
             _orchardServices = orchardServices;
             _featureManager = featureManager;
         }
@@ -40,6 +34,16 @@ namespace Upgrade.Controllers {
             if(_featureManager.GetEnabledFeatures().All(x => x.Id != "Orchard.Taxonomies")) {
                 _orchardServices.Notifier.Warning(T("You need to enable Orchard.Taxonomies in order to migrate Contrib.Taxonomies to Orchard.Taxonomies."));
             }
+            else {
+                var flag = false;
+                _upgradeService.ExecuteReader("SELECT * FROM " + _upgradeService.GetPrefixedTableName("Orchard_Taxonomies_TermContentItem"), (reader, conn) => {
+                    flag = true;
+                });
+
+                if (flag) {
+                    _orchardServices.Notifier.Warning(T("This migration step might have been done already."));
+                }
+            }
 
             return View();
         }

src/Orchard.Web/Modules/Upgrade/Services/UpgradeService.cs

@@ -94,8 +94,10 @@ namespace Upgrade.Services {
 
             while (reader != null && reader.Read()) {
                 try {
+                    if (action != null) {
                         action(reader, session.Connection);
                     }
+                }
                 catch (Exception e) {
                     Logger.Error(e, "Error while executing custom SQL Statement in Upgrade.");
                 }