h

cmdlet/add_file.py

@@ -554,6 +554,14 @@ class Add_File(Cmdlet):
                 media_path, file_hash, temp_dir_to_cleanup = self._resolve_source(
                     item, path_arg, pipe_obj, config, store_instance=storage_registry
                 )
+                if not media_path and provider_name:
+                    media_path, file_hash, temp_dir_to_cleanup = Add_File._download_provider_source(
+                        pipe_obj, config, storage_registry
+                    )
+                    if media_path:
+                        debug(
+                            f"[add-file] Provider source downloaded: {media_path}"
+                        )
                 debug(
                     f"[add-file] RESOLVED source: path={media_path}, hash={file_hash  if file_hash else 'N/A'}..."
                 )
@@ -1071,6 +1079,28 @@ class Add_File(Cmdlet):
                 pass
         return None, None
 
+    @staticmethod
+    def _resolve_backend_by_name(store: Any, backend_name: str) -> Optional[Any]:
+        if not store or not backend_name:
+            return None
+        try:
+            return store[backend_name]
+        except Exception:
+            pass
+        target = str(backend_name or "").strip().lower()
+        if not target:
+            return None
+        try:
+            for candidate in store.list_backends():
+                if isinstance(candidate, str) and candidate.strip().lower() == target:
+                    try:
+                        return store[candidate]
+                    except Exception:
+                        continue
+        except Exception:
+            pass
+        return None
+
     @staticmethod
     def _resolve_source(
         result: Any,
@@ -1111,15 +1141,12 @@ class Add_File(Cmdlet):
                 if not store:
                     store = Store(config)
                 
-                if r_store in store.list_backends():
-                    backend = store[r_store]
-                    # Try direct access (Path)
+                backend = Add_File._resolve_backend_by_name(store, r_store)
+                if backend is not None:
                     mp = backend.get_file(r_hash)
                     if isinstance(mp, Path) and mp.exists():
                         pipe_obj.path = str(mp)
                         return mp, str(r_hash), None
-                    
-                    # Try download to temp
                     if isinstance(mp, str) and mp.strip():
                         dl_path, tmp_dir = Add_File._maybe_download_backend_file(
                             backend, str(r_hash), pipe_obj
@@ -1162,6 +1189,41 @@ class Add_File(Cmdlet):
         log("File path could not be resolved")
         return None, None, None
 
+    @staticmethod
+    def _download_provider_source(
+        pipe_obj: models.PipeObject,
+        config: Dict[str, Any],
+        store_instance: Optional[Any],
+    ) -> Tuple[Optional[Path], Optional[str], Optional[Path]]:
+        r_hash = str(getattr(pipe_obj, "hash", None) or getattr(pipe_obj, "file_hash", None) or "").strip()
+        r_store = str(getattr(pipe_obj, "store", None) or "").strip()
+        if not (r_hash and r_store):
+            return None, None, None
+
+        try:
+            store = store_instance or Store(config)
+        except Exception:
+            store = None
+        backend = Add_File._resolve_backend_by_name(store, r_store) if store is not None else None
+        if backend is None:
+            return None, None, None
+
+        try:
+            source = backend.get_file(r_hash.lower())
+            if isinstance(source, Path) and source.exists():
+                pipe_obj.path = str(source)
+                return source, str(r_hash), None
+            if isinstance(source, str) and source.strip():
+                dl_path, tmp_dir = Add_File._maybe_download_backend_file(
+                    backend, str(r_hash), pipe_obj
+                )
+                if dl_path and dl_path.exists():
+                    return dl_path, str(r_hash), tmp_dir
+        except Exception:
+            pass
+
+        return None, None, None
+
     @staticmethod
     def _scan_directory_for_files(directory: Path, compute_hash: bool = True) -> List[Dict[str, Any]]:
         """Scan a directory for supported media files and return list of file info dicts.