Merge pull request #139 from cporcellijr/feature/restore-auto-forge-collections

Fix auto forge collections
Fix Booklore shelf assignment during batch-match

Author: cporcellijr

src/services/forge_service.py

@@ -565,6 +565,22 @@ def _auto_forge_background_task(self, abs_id, text_item, title, author, original
             else:
                 logger.error(f"❌ Auto-Forge: Book {abs_id} not found in DB to update!")
 
+            # --- ADD TO COLLECTIONS/SHELVES ---
+            try:
+                abs_collection_name = os.environ.get("ABS_COLLECTION_NAME", "Synced with KOReader")
+                self.abs_client.add_to_collection(abs_id, abs_collection_name)
+
+                if self.booklore_client:
+                    shelf_filename = original_filename if original_filename else target_filename
+                    booklore_shelf_name = os.environ.get("BOOKLORE_SHELF_NAME", "Kobo")
+                    self.booklore_client.add_to_shelf(shelf_filename, booklore_shelf_name)
+
+                if self.storyteller_client:
+                    self.storyteller_client.add_to_collection(target_filename)
+                    
+            except Exception as e:
+                logger.warning(f"⚠️ Auto-Forge: Failed to add to collections/shelves: {e}")
+
             # --- CLEANUP ---
             AUDIO_EXTENSIONS = {'.mp3', '.m4b', '.m4a', '.flac', '.ogg', '.opus', '.wma', '.wav', '.aac'}
             for f in course_dir.iterdir():

src/web_server.py

@@ -1489,7 +1489,8 @@ def batch_match():
 
                 container.abs_client().add_to_collection(item['abs_id'], ABS_COLLECTION_NAME)
                 if container.booklore_client().is_configured():
-                    container.booklore_client().add_to_shelf(ebook_filename, BOOKLORE_SHELF_NAME)
+                    shelf_filename = original_ebook_filename or ebook_filename
+                    container.booklore_client().add_to_shelf(shelf_filename, BOOKLORE_SHELF_NAME)
                 if container.storyteller_client().is_configured():
                     container.storyteller_client().add_to_collection(ebook_filename)