Optimized package search with bisect-based index #42

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged

ericbsd merged 15 commits into ghostbsd:master from xcrsz:master

Jun 12, 2025

README.md

            
                      Original file line number
                      Diff line number
                      Diff line change
                  
    @@ -1,4 +1,4 @@
  
    software-station

    ================

    Software Station is the future software manager using pkgng.

    Software Station is the software manager for GhostBSD.

pkg_info.py

-Original file line number
+Diff line change
@@ -0,0 +1,92 @@
+    """Package information loader for installed and available FreeBSD packages.
+    This module defines the PkgInfo class, which loads installed and available packages
+    using the system's `pkg` tool, and allows basic search/filtering operations.
+    """
+    import subprocess
+    from software_station.search_index import Package, PkgSearchIndex
+    class PkgInfo:
+        """Handles retrieval of installed and available packages on the system."""
+        def __init__(self):
+            """Initializes the package lists and loads data from the package system."""
+            self.available: list[Package] = []
+            self.installed_names: set[str] = set()
+            self.index = None
+            self.load()
+        def load(self):
+            """Loads both installed and available package data."""
+            self.load_installed()
+            self.load_available()
+        def load_installed(self):
+            """Retrieves the names of currently installed packages."""
+            try:
+                result = subprocess.run(
+                    ['pkg', 'query', '%n'],
+                    stdout=subprocess.PIPE,
+                    stderr=subprocess.DEVNULL,
+                    check=True,
+                    text=True,
+                    timeout=10
+                )
+                self.installed_names = {name.strip() for name in result.stdout.splitlines()}
+            except (subprocess.CalledProcessError, FileNotFoundError, subprocess.TimeoutExpired):
+                self.installed_names = set()
+        def load_available(self):
+            """Retrieves all available packages with name, version, and description."""
+            try:
+                result = subprocess.run(
+                    ['pkg', 'query', '-a', '%n %v %e'],
+                    stdout=subprocess.PIPE,
+                    stderr=subprocess.DEVNULL,
+                    check=True,
+                    text=True,
+                    timeout=10
+                )
+                self.available = []
+                for line in result.stdout.splitlines():
+                    parts = line.strip().split(' ', 2)
+                    if len(parts) >= 2:
+                        name, version = parts[0], parts[1]
+                        desc = parts[2] if len(parts) == 3 else ''
+                        self.available.append(Package(
+                            name=name,
+                            version=version,
+                            description=desc,
+                            installed=name in self.installed_names
+                        ))
+                self.index = PkgSearchIndex(self.available)
+            except (subprocess.CalledProcessError, FileNotFoundError, subprocess.TimeoutExpired):
+                self.available = []
+                self.index = None
+        def search(self, prefix: str) -> list[Package]:
+            """Searches available packages by prefix.
+            Args:
+                prefix (str): The string to match at the beginning of package names.
+            Returns:
+                list[Package]: Matching package list.
+            """
+            return self.available if self.index is None else self.index.search_prefix(prefix)
+        def get_installed(self) -> list[Package]:
+            """Returns all packages that are currently installed.
+            Returns:
+                list[Package]: Installed packages.
+            """
+            return [pkg for pkg in self.available if pkg.installed]

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Optimized package search with bisect-based index #42

Uh oh!

Diff view

Diff view

There are no files selected for viewing

Uh oh!