Fix incorrect file listing for add/remove

durierem · Nov 4, 2020 · b778cc3 · b778cc3
1 parent e5ed7f6
commit b778cc3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/cmd/add.rb b/lib/cmd/add.rb
@@ -12,7 +12,7 @@ class Add
   # files - The String file names to add in the local repository.
   def initialize(*files)
     @mt = MasterTree.new(Config.master_tree_path, Dir.home)
-    @files = files
+    @files = files.flatten
   end
 
   def perform

diff --git a/lib/cmd/remove.rb b/lib/cmd/remove.rb
@@ -15,7 +15,7 @@ class Remove
   # files - The String file names to remove from the local repository.
   def initialize(*files)
     @mt = MasterTree.new(Config.master_tree_path, Dir.home)
-    @files = Config.options[:remove_all] ? @mt.list : files
+    @files = Config.options[:remove_all] ? @mt.list : files.flatten
   end
 
   def perform