Added ability to resolve optional dependencies

Author: Ben Davis

src/fastsandpm/dependencies/provider.py

@@ -285,7 +285,7 @@ def narrow_requirement_selection(
 """Type alias for the resolution reporter."""
 
 
-def resolve(manifest: Manifest) -> dict[str, Candidate]:
+def resolve(manifest: Manifest, optional_deps: list[str] | None = None) -> dict[str, Candidate]:
     """Resolve all dependencies for a manifest.
 
     Creates a FastSandProvider with the manifest's registries and runs the
@@ -294,6 +294,7 @@ def resolve(manifest: Manifest) -> dict[str, Candidate]:
 
     Args:
         manifest: The manifest containing dependencies to resolve.
+        optional_deps: Optional dependency groups to include in the library.
 
     Returns:
         A dictionary mapping package names to their resolved Candidate objects.
@@ -313,5 +314,12 @@ def resolve(manifest: Manifest) -> dict[str, Candidate]:
     reporter: FastSandReporter = resolvelib.BaseReporter()
 
     resolver = resolvelib.Resolver(provider, reporter)
-    result = resolver.resolve(manifest.dependencies)
+
+    dependencies = [dep for dep in manifest.dependencies]
+    if optional_deps:
+        for group in optional_deps:
+            if group in manifest.optional_dependencies:
+                dependencies.extend(manifest.optional_dependencies[group])
+
+    result = resolver.resolve(dependencies)
     return result.mapping

src/fastsandpm/install.py

@@ -89,14 +89,7 @@ def library_from_manifest(
     Raises:
         resolvelib.ResolutionImpossible: If no compatible resolution exists.
     """
-    # TODO: Support optional_deps parameter once resolve() is updated to accept it
-    if optional_deps:
-        raise NotImplementedError(
-            "Optional dependencies are not yet supported. "
-            "The resolve() function needs to be updated to accept optional_deps."
-        )
-
-    library = resolve(manifest)
+    library = resolve(manifest, optional_deps)
     build_library(library, dest, clean)