1 /* 2 * Copyright 2013-2015, Haiku, Inc. All Rights Reserved. 3 * Distributed under the terms of the MIT License. 4 * 5 * Authors: 6 * Axel Dörfler <axeld@pinc-software.de> 7 * Rene Gollent <rene@gollent.com> 8 * Ingo Weinhold <ingo_weinhold@gmx.de> 9 */ 10 11 12 #include <StringForSize.h> 13 #include <StringForRate.h> 14 // Must be first, or the BPrivate namespaces are confused 15 16 #include "PackageManager.h" 17 18 #include <InterfaceDefs.h> 19 20 #include <sys/ioctl.h> 21 #include <unistd.h> 22 23 #include <package/CommitTransactionResult.h> 24 #include <package/DownloadFileRequest.h> 25 #include <package/RefreshRepositoryRequest.h> 26 #include <package/solver/SolverPackage.h> 27 #include <package/solver/SolverProblem.h> 28 #include <package/solver/SolverProblemSolution.h> 29 30 #include "pkgman.h" 31 32 33 using namespace BPackageKit::BPrivate; 34 35 36 PackageManager::PackageManager(BPackageInstallationLocation location, 37 bool interactive) 38 : 39 BPackageManager(location, &fClientInstallationInterface, this), 40 BPackageManager::UserInteractionHandler(), 41 fDecisionProvider(interactive), 42 fClientInstallationInterface(), 43 fInteractive(interactive) 44 { 45 } 46 47 48 PackageManager::~PackageManager() 49 { 50 } 51 52 53 void 54 PackageManager::SetInteractive(bool interactive) 55 { 56 fInteractive = interactive; 57 fDecisionProvider.SetInteractive(interactive); 58 } 59 60 61 void 62 PackageManager::JobFailed(BSupportKit::BJob* job) 63 { 64 BString error = job->ErrorString(); 65 if (error.Length() > 0) { 66 error.ReplaceAll("\n", "\n*** "); 67 fprintf(stderr, "%s", error.String()); 68 } 69 } 70 71 72 void 73 PackageManager::JobAborted(BSupportKit::BJob* job) 74 { 75 DIE(job->Result(), "aborted"); 76 } 77 78 79 void 80 PackageManager::HandleProblems() 81 { 82 printf("Encountered problems:\n"); 83 84 int32 problemCount = fSolver->CountProblems(); 85 for (int32 i = 0; i < problemCount; i++) { 86 // print problem and possible solutions 87 BSolverProblem* problem = fSolver->ProblemAt(i); 88 printf("problem %" B_PRId32 ": %s\n", i + 1, 89 problem->ToString().String()); 90 91 int32 solutionCount = problem->CountSolutions(); 92 for (int32 k = 0; k < solutionCount; k++) { 93 const BSolverProblemSolution* solution = problem->SolutionAt(k); 94 printf(" solution %" B_PRId32 ":\n", k + 1); 95 int32 elementCount = solution->CountElements(); 96 for (int32 l = 0; l < elementCount; l++) { 97 const BSolverProblemSolutionElement* element 98 = solution->ElementAt(l); 99 printf(" - %s\n", element->ToString().String()); 100 } 101 } 102 103 if (!fInteractive) 104 continue; 105 106 // let the user choose a solution 107 printf("Please select a solution, skip the problem for now or quit.\n"); 108 for (;;) { 109 if (solutionCount > 1) 110 printf("select [1...%" B_PRId32 "/s/q]: ", solutionCount); 111 else 112 printf("select [1/s/q]: "); 113 114 char buffer[32]; 115 if (fgets(buffer, sizeof(buffer), stdin) == NULL 116 || strcmp(buffer, "q\n") == 0) { 117 exit(1); 118 } 119 120 if (strcmp(buffer, "s\n") == 0) 121 break; 122 123 char* end; 124 long selected = strtol(buffer, &end, 0); 125 if (end == buffer || *end != '\n' || selected < 1 126 || selected > solutionCount) { 127 printf("*** invalid input\n"); 128 continue; 129 } 130 131 status_t error = fSolver->SelectProblemSolution(problem, 132 problem->SolutionAt(selected - 1)); 133 if (error != B_OK) 134 DIE(error, "failed to set solution"); 135 break; 136 } 137 } 138 139 if (problemCount > 0 && !fInteractive) 140 exit(1); 141 } 142 143 144 void 145 PackageManager::ConfirmChanges(bool fromMostSpecific) 146 { 147 printf("The following changes will be made:\n"); 148 149 int32 count = fInstalledRepositories.CountItems(); 150 if (fromMostSpecific) { 151 for (int32 i = count - 1; i >= 0; i--) 152 _PrintResult(*fInstalledRepositories.ItemAt(i)); 153 } else { 154 for (int32 i = 0; i < count; i++) 155 _PrintResult(*fInstalledRepositories.ItemAt(i)); 156 } 157 158 if (!fDecisionProvider.YesNoDecisionNeeded(BString(), "Continue?", "yes", 159 "no", "yes")) { 160 exit(1); 161 } 162 } 163 164 165 void 166 PackageManager::Warn(status_t error, const char* format, ...) 167 { 168 va_list args; 169 va_start(args, format); 170 vfprintf(stderr, format, args); 171 va_end(args); 172 173 if (error == B_OK) 174 printf("\n"); 175 else 176 printf(": %s\n", strerror(error)); 177 } 178 179 180 void 181 PackageManager::ProgressPackageDownloadStarted(const char* packageName) 182 { 183 fLastBytes = 0; 184 fLastRateCalcTime = system_time(); 185 fDownloadRate = 0; 186 printf(" 0%%"); 187 } 188 189 190 void 191 PackageManager::ProgressPackageDownloadActive(const char* packageName, 192 float completionPercentage, off_t bytes, off_t totalBytes) 193 { 194 if (bytes == totalBytes) 195 fLastBytes = totalBytes; 196 if (!fInteractive) 197 return; 198 199 // Do not update if nothing changed in the last 500ms 200 if (bytes <= fLastBytes || (system_time() - fLastRateCalcTime) < 500000) 201 return; 202 203 const bigtime_t time = system_time(); 204 if (time != fLastRateCalcTime) { 205 fDownloadRate = (bytes - fLastBytes) * 1000000 206 / (time - fLastRateCalcTime); 207 } 208 fLastRateCalcTime = time; 209 fLastBytes = bytes; 210 211 // Build the current file progress percentage and size string 212 BString leftStr; 213 BString rightStr; 214 215 int width = 70; 216 struct winsize winSize; 217 if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &winSize) == 0) 218 width = std::min(winSize.ws_col - 2, 78); 219 220 if (width < 30) { 221 // Not much space for anything, just draw a percentage 222 leftStr.SetToFormat("%3d%%", (int)(completionPercentage * 100)); 223 } else { 224 leftStr.SetToFormat("%3d%% %s", (int)(completionPercentage * 100), 225 packageName); 226 227 char byteBuffer[32]; 228 char totalBuffer[32]; 229 char rateBuffer[32]; 230 rightStr.SetToFormat("%s/%s %s ", 231 string_for_size(bytes, byteBuffer, sizeof(byteBuffer)), 232 string_for_size(totalBytes, totalBuffer, sizeof(totalBuffer)), 233 fDownloadRate == 0 ? "--.-" : 234 string_for_rate(fDownloadRate, rateBuffer, sizeof(rateBuffer))); 235 236 if (leftStr.CountChars() + rightStr.CountChars() >= width) 237 { 238 // The full string does not fit! Try to make a shorter one. 239 leftStr.ReplaceLast(".hpkg", ""); 240 leftStr.TruncateChars(width - rightStr.CountChars() - 2); 241 leftStr.Append(B_UTF8_ELLIPSIS " "); 242 } 243 244 int extraSpace = width - leftStr.CountChars() - rightStr.CountChars(); 245 246 leftStr.Append(' ', extraSpace); 247 leftStr.Append(rightStr); 248 } 249 250 const int progChars = leftStr.CountBytes(0, 251 (int)(width * completionPercentage)); 252 253 // Set bg to green, fg to white, and print progress bar. 254 // Then reset colors and print rest of text 255 // And finally remove any stray chars at the end of the line 256 printf("\r\x1B[42;37m%.*s\x1B[0m%s\x1B[K", progChars, leftStr.String(), 257 leftStr.String() + progChars); 258 259 // Force terminal to update when the line is complete, to avoid flickering 260 // because of updates at random times 261 fflush(stdout); 262 } 263 264 265 void 266 PackageManager::ProgressPackageDownloadComplete(const char* packageName) 267 { 268 if (fInteractive) { 269 // Erase the line, return to the start, and reset colors 270 printf("\r\33[2K\r\x1B[0m"); 271 } 272 273 char byteBuffer[32]; 274 printf("100%% %s [%s]\n", packageName, 275 string_for_size(fLastBytes, byteBuffer, sizeof(byteBuffer))); 276 fflush(stdout); 277 } 278 279 280 void 281 PackageManager::ProgressPackageChecksumStarted(const char* title) 282 { 283 printf("%s...", title); 284 } 285 286 287 void 288 PackageManager::ProgressPackageChecksumComplete(const char* title) 289 { 290 printf("done.\n"); 291 } 292 293 294 void 295 PackageManager::ProgressStartApplyingChanges(InstalledRepository& repository) 296 { 297 printf("[%s] Applying changes ...\n", repository.Name().String()); 298 } 299 300 301 void 302 PackageManager::ProgressTransactionCommitted(InstalledRepository& repository, 303 const BCommitTransactionResult& result) 304 { 305 const char* repositoryName = repository.Name().String(); 306 307 int32 issueCount = result.CountIssues(); 308 for (int32 i = 0; i < issueCount; i++) { 309 const BTransactionIssue* issue = result.IssueAt(i); 310 if (issue->PackageName().IsEmpty()) { 311 printf("[%s] warning: %s\n", repositoryName, 312 issue->ToString().String()); 313 } else { 314 printf("[%s] warning: package %s: %s\n", repositoryName, 315 issue->PackageName().String(), issue->ToString().String()); 316 } 317 } 318 319 printf("[%s] Changes applied. Old activation state backed up in \"%s\"\n", 320 repositoryName, result.OldStateDirectory().String()); 321 printf("[%s] Cleaning up ...\n", repositoryName); 322 } 323 324 325 void 326 PackageManager::ProgressApplyingChangesDone(InstalledRepository& repository) 327 { 328 printf("[%s] Done.\n", repository.Name().String()); 329 } 330 331 332 void 333 PackageManager::_PrintResult(InstalledRepository& installationRepository) 334 { 335 if (!installationRepository.HasChanges()) 336 return; 337 338 printf(" in %s:\n", installationRepository.Name().String()); 339 340 PackageList& packagesToActivate 341 = installationRepository.PackagesToActivate(); 342 PackageList& packagesToDeactivate 343 = installationRepository.PackagesToDeactivate(); 344 345 BStringList upgradedPackages; 346 BStringList upgradedPackageVersions; 347 for (int32 i = 0; 348 BSolverPackage* installPackage = packagesToActivate.ItemAt(i); 349 i++) { 350 for (int32 j = 0; 351 BSolverPackage* uninstallPackage = packagesToDeactivate.ItemAt(j); 352 j++) { 353 if (installPackage->Info().Name() == uninstallPackage->Info().Name()) { 354 upgradedPackages.Add(installPackage->Info().Name()); 355 upgradedPackageVersions.Add(uninstallPackage->Info().Version().ToString()); 356 break; 357 } 358 } 359 } 360 361 for (int32 i = 0; BSolverPackage* package = packagesToActivate.ItemAt(i); 362 i++) { 363 BString repository; 364 if (dynamic_cast<MiscLocalRepository*>(package->Repository()) != NULL) 365 repository = "local file"; 366 else 367 repository.SetToFormat("repository %s", package->Repository()->Name().String()); 368 369 int position = upgradedPackages.IndexOf(package->Info().Name()); 370 if (position >= 0) { 371 printf(" upgrade package %s-%s to %s from %s\n", 372 package->Info().Name().String(), 373 upgradedPackageVersions.StringAt(position).String(), 374 package->Info().Version().ToString().String(), 375 repository.String()); 376 } else { 377 printf(" install package %s-%s from %s\n", 378 package->Info().Name().String(), 379 package->Info().Version().ToString().String(), 380 repository.String()); 381 } 382 } 383 384 for (int32 i = 0; BSolverPackage* package = packagesToDeactivate.ItemAt(i); 385 i++) { 386 if (upgradedPackages.HasString(package->Info().Name())) 387 continue; 388 printf(" uninstall package %s\n", package->VersionedName().String()); 389 } 390 // TODO: Print file/download sizes. Unfortunately our package infos don't 391 // contain the file size. Which is probably correct. The file size (and possibly 392 // other information) should, however, be provided by the repository cache in 393 // some way. Extend BPackageInfo? Create a BPackageFileInfo? 394 } 395