add Registry Lookup

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 {

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 {