From 046a00f4733c7c51275747564993106da9dd5e8e Mon Sep 17 00:00:00 2001
From: Daniel <dhaavi@users.noreply.github.com>
Date: Mon, 25 May 2020 17:20:02 +0200
Subject: [PATCH] Reload indexes when restarting

---
 pmctl/main.go | 26 +++++++++++++++-----------
 pmctl/run.go  |  3 +++
 2 files changed, 18 insertions(+), 11 deletions(-)

diff --git a/pmctl/main.go b/pmctl/main.go
index 23de8f0d..ba92d87c 100644
--- a/pmctl/main.go
+++ b/pmctl/main.go
@@ -194,17 +194,7 @@ func cmdSetup(cmd *cobra.Command, args []string) (err error) {
 			Beta:   false,
 		})
 
-		err = registry.LoadIndexes()
-		if err != nil {
-			return err
-		}
-
-		err = registry.ScanStorage("")
-		if err != nil {
-			log.Printf("WARNING: error during storage scan: %s\n", err)
-		}
-
-		registry.SelectVersions()
+		updateRegistryIndex()
 	}
 
 	// logs and warning
@@ -225,3 +215,17 @@ func cmdSetup(cmd *cobra.Command, args []string) (err error) {
 
 	return nil
 }
+
+func updateRegistryIndex() {
+	err := registry.LoadIndexes()
+	if err != nil {
+		log.Printf("WARNING: error loading indexes: %s\n", err)
+	}
+
+	err = registry.ScanStorage("")
+	if err != nil {
+		log.Printf("WARNING: error during storage scan: %s\n", err)
+	}
+
+	registry.SelectVersions()
+}
diff --git a/pmctl/run.go b/pmctl/run.go
index 432bfa3d..53557d8a 100644
--- a/pmctl/run.go
+++ b/pmctl/run.go
@@ -191,6 +191,9 @@ func run(cmd *cobra.Command, opts *Options) (err error) {
 		case tryAgain && err == nil:
 			// upgrade
 			log.Println("restarting by request...")
+			// update index
+			log.Println("checking versions...")
+			updateRegistryIndex()
 		case !tryAgain && err != nil:
 			// fatal error
 			return err