PM: Solution-Package Update

Author: Dragan

src/AddIns/Misc/PackageManagement/Project/Src/PackageViewModel.cs

@@ -280,20 +280,6 @@ where project.IsSelected
 			}
 		}
 
-		void SolutionPackageGetInstallOperations(
-			ISolutionPackageRepository solutionRepository, 
-			InstallPackageAction installAction,
-			IPackageFromRepository installPackage) {
-
-			var resolverFactory = new PackageOperationsResolverFactory();
-			var resolver = resolverFactory.CreateInstallPackageOperationResolver(
-				solutionRepository.Repository, 
-				installPackage.Repository, 
-				logger, 
-				installAction);
-			packageOperations = resolver.ResolveOperations(installPackage);
-		}
-
 		IPackageManagementProject GetSingleProjectSelected()
 		{
 			return selectedProjects.GetSingleProjectSelected(package.Repository);
@@ -350,6 +336,20 @@ protected virtual void TrySolutionPackageInstall()
 			}
 		}		
 
+		void SolutionPackageGetInstallOperations(
+			ISolutionPackageRepository solutionRepository, 
+			InstallPackageAction installAction,
+			IPackageFromRepository installPackage) {
+
+			var resolverFactory = new PackageOperationsResolverFactory();
+			var resolver = resolverFactory.CreateInstallPackageOperationResolver(
+				solutionRepository.Repository, 
+				installPackage.Repository, 
+				logger, 
+				installAction);
+			packageOperations = resolver.ResolveOperations(installPackage);
+		}
+
 		void SolutionPackageInstall(
 			ISolutionPackageRepository solutionRepository, 
 			InstallPackageAction installAction,
@@ -366,6 +366,56 @@ void SolutionPackageInstall(
 				installAction.AllowPrereleaseVersions);
 		}
 
+		protected virtual void TrySolutionPackageUpdate()
+		{
+			try {
+				var solutionRepository = selectedProjects.Solution.CreateSolutionPackageRepository();
+				var updateAction = new UpdatePackageAction(null, packageManagementEvents);
+				updateAction.Package = package;
+				updateAction.AllowPrereleaseVersions = parent.IncludePrerelease;
+				SolutionPackageGetUpdateOperations(solutionRepository, updateAction, package);
+				if (LicensesAccepted()) {
+					SolutionPackageUpdate(solutionRepository, updateAction, package);
+					IEnumerable<IPackage> updatedPackages = ( 
+						from packageOperation in packageOperations 
+						where packageOperation.Package != null 
+						select packageOperation.Package).Distinct();
+					packageManagementEvents.OnParentPackagesUpdated(updatedPackages);
+				}
+			} catch (Exception ex) {
+				ReportError(ex);
+				logger.LogError(ex);
+			}
+		}		
+
+		void SolutionPackageGetUpdateOperations(
+			ISolutionPackageRepository solutionRepository, 
+			UpdatePackageAction updateAction,
+			IPackageFromRepository updatePackage) {
+
+			var resolverFactory = new PackageOperationsResolverFactory();
+			var resolver = resolverFactory.CreateUpdatePackageOperationResolver(
+				solutionRepository.Repository, 
+				updatePackage.Repository, 
+				logger, 
+				updateAction);
+			packageOperations = resolver.ResolveOperations(updatePackage);
+		}
+
+		void SolutionPackageUpdate(
+			ISolutionPackageRepository solutionRepository, 
+			UpdatePackageAction installAction,
+			IPackageFromRepository updatePackage) {
+			var packageManager = new PackageManager(
+				updatePackage.Repository, 
+				solutionRepository.PackagePathResolver, 
+				solutionRepository.FileSystem, 
+				solutionRepository.Repository);
+			packageManager.UpdatePackage(
+				updatePackage.Id, true, 
+				installAction.AllowPrereleaseVersions);
+		}
+
 		void TryProjectPackageInstall()
 		{
 			try {

src/AddIns/Misc/PackageManagement/Project/Src/UpdatedPackageViewModel.cs

@@ -44,12 +44,8 @@ public UpdatedPackageViewModel(
 		SelectedProjectsForUpdatedPackages selectedProjects;
 
 		protected override void TrySolutionPackageInstall()
-		{
-			var repository = selectedProjects.Solution.CreateSolutionPackageRepository();
-			foreach (var packageToUpdate in repository.Repository.FindPackagesById(this.Id)) {
-				TrySolutionPackageUninstall(packageToUpdate);
-			}
-			base.TrySolutionPackageInstall();
+		{	// Modify PackageViewModel logic
+			base.TrySolutionPackageUpdate();
 		}
 
 		protected override ProcessPackageAction CreatePackageManageAction(IPackageManagementSelectedProject selectedProject)