perf: fast path single-token repository search

src/poetry/repositories/link_sources/base.py

@@ -132,11 +132,12 @@ class SimpleRepositoryRootPage:
     """
 
     def search(self, query: str | list[str]) -> list[str]:
-        results: list[str] = []
-        tokens = query if isinstance(query, list) else [query]
+        if isinstance(query, str):
+            return [name for name in self.package_names if query in name]
 
+        results: list[str] = []
         for name in self.package_names:
-            if any(token in name for token in tokens):
+            if any(token in name for token in query):
                 results.append(name)
 
         return results

src/poetry/repositories/repository.py

@@ -74,11 +74,12 @@ def add_package(self, package: Package) -> None:
         self._packages.append(package)
 
     def search(self, query: str | list[str]) -> list[Package]:
-        results: list[Package] = []
-        tokens = query if isinstance(query, list) else [query]
+        if isinstance(query, str):
+            return [package for package in self.packages if query in package.name]
 
+        results: list[Package] = []
         for package in self.packages:
-            if any(token in package.name for token in tokens):
+            if any(token in package.name for token in query):
                 results.append(package)
 
         return results

tests/repositories/test_repository.py

@@ -108,4 +108,5 @@ def test_search() -> None:
 
     assert repo.search("foo") == [package_foo1, package_foo2, package_foobar]
     assert repo.search("bar") == [package_foobar]
+    assert repo.search(["foo", "bar"]) == [package_foo1, package_foo2, package_foobar]
     assert repo.search("nothing") == []