From a20bbeb9f871dd9efdcf6c07687f141d5b82219a Mon Sep 17 00:00:00 2001
From: Marco D'Aleo <m.daleo404@gmail.com>
Date: Sun, 9 Nov 2025 09:48:14 +0000
Subject: [PATCH] feat: add support for restoring multiple files with --restore

---
 resrm/src/resrm/core.py | 39 +++++++++++++++++++++++++++++++++++++--
 1 file changed, 37 insertions(+), 2 deletions(-)

diff --git a/resrm/src/resrm/core.py b/resrm/src/resrm/core.py
index 11c30f8..e397117 100644
--- a/resrm/src/resrm/core.py
+++ b/resrm/src/resrm/core.py
@@ -116,6 +116,41 @@ def find_candidates(identifier: str) -> List[Dict]:
     if id_matches:
         return id_matches
 
+def restore_many(identifiers: List[str]):
+    """Restore multiple files, prompting when needed."""
+    for identifier in identifiers:
+        candidates = find_candidates(identifier)
+
+        if not candidates:
+            print(f"No match found for '{identifier}'")
+            continue
+
+        # Only one match - restore immediately
+        if len(candidates) == 1:
+            restore_one(candidates[0])
+            continue
+
+        # Multiple matches - prompt user
+        print(f"Multiple matches for '{identifier}':")
+        for i, entry in enumerate(candidates, start=1):
+            print(f"{i}) {short_id(entry['id'])}  {entry['orig_path']}  ({entry['timestamp']})")
+
+        try:
+            choice = input("Choose number to restore (or skip): ").strip()
+        except KeyboardInterrupt:
+            print("\nAborted.")
+            return
+
+        if not choice.isdigit():
+            print("Skipped.")
+            continue
+
+        idx = int(choice) - 1
+        if 0 <= idx < len(candidates):
+            restore_one(candidates[idx])
+        else:
+            print("Invalid selection. Skipped.")
+
 def restore_one(entry: Dict) -> bool:
     src = TRASH_DIR / entry["id"]
     dest = Path(entry["orig_path"])
@@ -273,7 +308,7 @@ def main(argv: Optional[List[str]] = None):
     parser.add_argument("-f", action="store_true", help="force")
     parser.add_argument("-i", action="store_true", help="interactive")
     parser.add_argument("--perma", action="store_true", help="permanent delete")
-    parser.add_argument("--restore", nargs=1, help="restore by id or basename")
+    parser.add_argument("--restore", nargs="+", help="restore by id or basename")
     parser.add_argument("-l", action="store_true", help="list trash")
     parser.add_argument("--empty", action="store_true", help="empty the trash permanently")
     parser.add_argument("--help", action="store_true", help="show help")
@@ -293,7 +328,7 @@ def main(argv: Optional[List[str]] = None):
         return
 
     if args.restore:
-        restore(args.restore[0])
+        restore_many(args.restore)
         return
 
     if not args.paths: