Fix upgrademenu crashing on non installed package

Fixes #347
Morganamilo · Apr 29, 2021 · de26909 · de26909
1 parent 28528e2
commit de26909
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/upgrade.rs b/src/upgrade.rs
@@ -285,6 +285,8 @@ pub async fn get_upgrades<'a, 'b>(
         );
     }
 
+
+    let (_, dbs) = repo::repo_aur_dbs(config);
     for (n, pkg) in aur_upgrades.iter().rev().enumerate().rev() {
         let remote = dbs
             .pkg(pkg.local.name())
@@ -309,14 +311,19 @@ pub async fn get_upgrades<'a, 'b>(
             .map(|p| p.db().unwrap().name())
             .map(|p| format!("{}-devel", p));
         let remote = remote.as_deref().unwrap_or("devel");
+        let current = dbs
+            .pkg(pkg.as_str())
+            .or_else(|_| db.pkg(pkg.as_str()))
+            .unwrap();
+        let ver = current.version();
         print_upgrade(
             config,
             n + 1,
             n_max,
             pkg,
             remote,
             db_pkg_max,
-            db.pkg(pkg.as_str()).unwrap().version(),
+            ver,
             old_max,
             "latest-commit",
         );