platform_search.py•5.83 kB
"""Platform-specific search implementations with dedicated parameter models."""
from typing import Optional, List, Dict, Any
from pydantic import BaseModel, Field
from enum import Enum
import platform
class BaseSearchQuery(BaseModel):
"""Base search parameters common to all platforms."""
query: str = Field(
description="Search query string. See platform-specific documentation for syntax details."
)
max_results: int = Field(
default=100,
ge=1,
le=1000,
description="Maximum number of results to return (1-1000)"
)
class MacSpecificParams(BaseModel):
"""macOS-specific search parameters for mdfind."""
live_updates: bool = Field(
default=False,
description="Provide live updates to search results"
)
search_directory: Optional[str] = Field(
default=None,
description="Limit search to specific directory (-onlyin parameter)"
)
literal_query: bool = Field(
default=False,
description="Treat query as literal string without interpretation"
)
interpret_query: bool = Field(
default=False,
description="Interpret query as if typed in Spotlight menu"
)
class LinuxSpecificParams(BaseModel):
"""Linux-specific search parameters for locate."""
ignore_case: bool = Field(
default=True,
description="Ignore case distinctions (-i parameter)"
)
regex_search: bool = Field(
default=False,
description="Use regular expressions in patterns (-r parameter)"
)
existing_files: bool = Field(
default=True,
description="Only output existing files (-e parameter)"
)
count_only: bool = Field(
default=False,
description="Only display count of matches (-c parameter)"
)
class WindowsSortOption(int, Enum):
"""Sort options for Windows Everything search."""
NAME_ASC = 1
NAME_DESC = 2
PATH_ASC = 3
PATH_DESC = 4
SIZE_ASC = 5
SIZE_DESC = 6
EXT_ASC = 7
EXT_DESC = 8
CREATED_ASC = 11
CREATED_DESC = 12
MODIFIED_ASC = 13
MODIFIED_DESC = 14
class WindowsSpecificParams(BaseModel):
"""Windows-specific search parameters for Everything SDK."""
match_path: bool = Field(
default=False,
description="Match against full path instead of filename only"
)
match_case: bool = Field(
default=False,
description="Enable case-sensitive search"
)
match_whole_word: bool = Field(
default=False,
description="Match whole words only"
)
match_regex: bool = Field(
default=False,
description="Enable regex search"
)
sort_by: WindowsSortOption = Field(
default=WindowsSortOption.NAME_ASC,
description="Sort order for results"
)
class UnifiedSearchQuery(BaseSearchQuery):
"""Combined search parameters model."""
mac_params: Optional[MacSpecificParams] = None
linux_params: Optional[LinuxSpecificParams] = None
windows_params: Optional[WindowsSpecificParams] = None
@classmethod
def get_schema_for_platform(cls) -> Dict[str, Any]:
"""Get the appropriate schema based on the current platform."""
system = platform.system().lower()
schema = {
"type": "object",
"properties": {
"base": BaseSearchQuery.model_json_schema()
},
"required": ["base"]
}
# Add platform-specific parameters
if system == "darwin":
schema["properties"]["mac_params"] = MacSpecificParams.model_json_schema()
elif system == "linux":
schema["properties"]["linux_params"] = LinuxSpecificParams.model_json_schema()
elif system == "windows":
schema["properties"]["windows_params"] = WindowsSpecificParams.model_json_schema()
return schema
def get_platform_params(self) -> Optional[BaseModel]:
"""Get the parameters specific to the current platform."""
system = platform.system().lower()
if system == "darwin":
return self.mac_params
elif system == "linux":
return self.linux_params
elif system == "windows":
return self.windows_params
return None
def build_search_command(query: UnifiedSearchQuery) -> List[str]:
"""Build the appropriate search command based on platform and parameters."""
system = platform.system().lower()
platform_params = query.get_platform_params()
if system == "darwin":
cmd = ["mdfind"]
if platform_params:
if platform_params.live_updates:
cmd.append("-live")
if platform_params.search_directory:
cmd.extend(["-onlyin", platform_params.search_directory])
if platform_params.literal_query:
cmd.append("-literal")
if platform_params.interpret_query:
cmd.append("-interpret")
cmd.append(query.query) # Use query directly from UnifiedSearchQuery
return cmd
elif system == "linux":
cmd = ["locate"]
if platform_params:
if platform_params.ignore_case:
cmd.append("-i")
if platform_params.regex_search:
cmd.append("-r")
if platform_params.existing_files:
cmd.append("-e")
if platform_params.count_only:
cmd.append("-c")
cmd.append(query.query) # Use query directly from UnifiedSearchQuery
return cmd
elif system == "windows":
# For Windows, return None as we'll use the Everything SDK directly
return []
raise NotImplementedError(f"Unsupported platform: {system}")