Implemented maratonctl package rm for #2

Author: ashwanthkumar

cmd/package_rm.go

@@ -0,0 +1,27 @@
+package cmd
+
+import (
+  "errors"
+  "github.com/ashwanthkumar/marathonctl/packages"
+  "github.com/spf13/cobra"
+)
+
+var PackageRm = &cobra.Command{
+  Use:   "rm <repository>",
+  Short: "Remove a package repository from local cache",
+  Long:  "Remove a package repository from local cache",
+  Run: AttachHandler(rmPackageCache),
+}
+
+func rmPackageCache(args []string) (err error) {
+  if len(args) > 0 {
+    repository := args[0]
+    return packages.Remove(repository)
+  }
+
+  return errors.New("We need a repository name to remove")
+}
+
+func init() {
+  Package.AddCommand(PackageRm)
+}

packages/packages.go

@@ -46,6 +46,17 @@ func Add(name, location string) error {
   return WritePackageMetadata()
 }
 
+func Remove(name string) (err error) {
+  AllRepos.Remove(name)
+  packageRepoPath := config.GetPackageCachePath() + "/" + name
+  fmt.Printf("Removing %s from %s\n", name, packageRepoPath)
+  err = os.RemoveAll(packageRepoPath)
+  if err != nil {
+    return err
+  }
+  return WritePackageMetadata()
+}
+
 func Update(name string) (err error) {
   repository := AllRepos.Get(name)
   if repository == nil {

packages/repositories.go

@@ -40,6 +40,18 @@ func (r *Repositories) Add(repo Repository) *Repositories {
   return r
 }
 
+func (r *Repositories) Remove(repo string) *Repositories {
+  var newRepositories Repositories
+  for _, repository := range *r {
+    if repository.Name != repo {
+      newRepositories = append(newRepositories, repository)
+    }
+  }
+
+  *r = newRepositories
+  return r
+}
+
 func (r *Repositories) Serialize() ([]byte, error) {
   return json.Marshal(r)
 }

packages/repositories_test.go

@@ -25,6 +25,15 @@ func TestRepositoriesAdd(t *testing.T) {
   assert.Equal(t, repos.Exists("r4"), true)
 }
 
+func TestRepositoriesRemove(t *testing.T) {
+  repos := newRepositories([]Repository {
+    repo("r1", "loc1"), repo("r2", "loc2"),
+  })
+  assert.Equal(t, true, repos.Exists("r2"))
+  repos.Remove("r2")
+  assert.Equal(t, false, repos.Exists("r2"))
+}
+
 func TestRepositoriesGet(t *testing.T) {
   repos := newRepositories([]Repository {
     repo("r1", "loc1"), repo("r2", "loc2"),