From eef31529a1e0fd9c11cafaf886220f7f6cff7d52 Mon Sep 17 00:00:00 2001
From: Alexis Lopez Zubieta <contact@azubieta.net>
Date: Fri, 5 Jun 2020 06:05:52 -0500
Subject: [PATCH] add Registry Lookup

---
 app/commands/remove.go | 12 +-----------
 app/utils/registry.go  | 10 ++++++++++
 2 files changed, 11 insertions(+), 11 deletions(-)

diff --git a/app/commands/remove.go b/app/commands/remove.go
index 2f26219..b9fc57f 100644
--- a/app/commands/remove.go
+++ b/app/commands/remove.go
@@ -22,7 +22,7 @@ func (cmd *RemoveCmd) Run(*Context) (err error) {
 	}
 	registry.Update()
 
-	fileName, ok := cmd.lookupFileName(registry)
+	fileName, ok := registry.Lookup(cmd.Id)
 	if !ok {
 		fileName = cmd.Id
 	}
@@ -50,16 +50,6 @@ func (cmd *RemoveCmd) Run(*Context) (err error) {
 	return err
 }
 
-func (cmd *RemoveCmd) lookupFileName(registry *utils.Registry) (string, bool) {
-	for fileName, entry := range registry.Entries {
-		if entry.Id == cmd.Id {
-			return fileName, true
-		}
-	}
-
-	return "", false
-}
-
 func uninstallAppImage(filePath string) error {
 	lib, err := dl.Open("libappimage.so", 0)
 	if err != nil {
diff --git a/app/utils/registry.go b/app/utils/registry.go
index 8687a83..5dc4741 100644
--- a/app/utils/registry.go
+++ b/app/utils/registry.go
@@ -111,6 +111,16 @@ func (registry *Registry) Update() {
 	}
 }
 
+func (registry *Registry) Lookup(id string) (string, bool) {
+	for fileName, entry := range registry.Entries {
+		if entry.Id == id {
+			return fileName, true
+		}
+	}
+
+	return "", false
+}
+
 func getFileSha1Checksum(fileName string) (string, error) {
 	applicationsDir, err := MakeApplicationsDirPath()
 	if err != nil {
-- 
GitLab