bug fixes

1 files changed, 12 insertions, 4 deletions

diff --git a/package/package.go b/package/package.go
index 35c6f5a..8681717 100644
--- a/package/package.go
+++ b/package/package.go
@@ -165,7 +165,14 @@ func Install(name string, args []string, destDir string) error {
 				if n == pkg.Name || n == pkg.Repo + "/" + pkg.Name {
 					return errors.New("broken depends in package " + n)
 				}
-				fmt.Printf("%v. %s - %v", i, n, database.GetPackageVersion(n, destDir))
+				
+				depPkgs := FindPackages(n, destDir)
+				v := "not found"
+			    if len(depPkgs) > 0 {
+					v = fmt.Sprintf("%v", depPkgs[0].Data[1])
+				}
+				
+				fmt.Printf("%v. %s - %v\n", i+1, n, v)
 			}
 		}
 	}
@@ -234,9 +241,10 @@ func Remove(name string, args []string, destDir string) error {
 	
 	err = database.UnRegPackage(pkg.Repo + "/" + pkg.Name, destDir)
 	if err != nil {
-		return errors.New("failed to unregister package in installed database: " + err.Error())
+		return err
 	}
-	
-	fmt.Println(":: " + pkg.Repo + "/" + pkg.Name + " (" + database.GetPackageVersion(pkg.Repo + "/" + pkg.Name, destDir) + ") removed")
+
+	v := helpers.ToString(pkg.Data[1])
+	fmt.Println(":: " + pkg.Repo + "/" + pkg.Name + " (" + v + ") removed")
 	return nil 
 }