o 6i=+@sUdZddlZddlZddlZddlmZddlmZm Z m Z ddl m Z e GdddZ e Gdd d ZGd d d Zdae eed <d efddZdded efddZdS)z Plugin Discovery and Loading System Automatically discovers and loads plugins from the plugins/ directory. Each plugin must be in a subdirectory with a plugin.py file containing a *Plugin class. N)Path)DictAnyOptional) dataclassc@seZdZUdZeed<eed<eed<dZeed<dZe ed<d Z eed <dZ e eed <d Z eed <dZeed <dZeed<dZeed<dZeed<dZeed<dZeed<ddZdS)PluginParameterzODefines a parameter for the plugin, used for both validation and UI generation.nametypelabelFrequiredNdefault placeholderoptions help_textminmax min_length max_lengthpatternpattern_descriptioncCs&|jdur|jdkrg|_dSdSdS)Nselect)rr selfr6/home/byschii/byschiidev/penelope/lib/plugin_loader.py __post_init__%s zPluginParameter.__post_init__)__name__ __module__ __qualname____doc__str__annotations__r boolr rrrlistrrfloatrrintrrrrrrrrrs"            rc@sJeZdZUdZeed<eed<dZeeed<dZee ed<ddZ dS) PluginMetadataz?Plugin metadata describing the plugin's interface and behavior.r descriptionNmethods parameterscCs*|jdur dg|_|jdurg|_dSdS)NPOST)r)r*rrrrr2s   zPluginMetadata.__post_init__) rrrr r!r"r)r$r*rrrrrrr'*s  r'c@seZdZdZddefddZddd Zd ed eddfd d Zde dede efddZ dede e fddZ deee ffddZdeeeffddZdS)PluginRegistryz Central registry for all discovered plugins. Scans the plugins/ directory, loads valid plugin classes, and maintains a mapping of plugin names to plugin instances. plugins plugins_dircCst||_i|_i|_dS)z Initialize the plugin registry. Args: plugins_dir: Path to the directory containing plugin subdirectories N)rr.r- _load_errors)rr.rrr__init__As  zPluginRegistry.__init__returnNcCs2|jstd|jtd|j|jjddddS|js.td|jdSdd|jD}|sDtd |jdStd |jd |D] }|d }|sid |j|j<td|jdqO|||qOtddtdt |j d|jrtdt |jdtdddS)aU Scan plugins directory and load all valid plugins. Expected structure: plugins/ wikipedia_search/ plugin.py # Contains *Plugin class form_filler/ plugin.py Logs success/failure for each plugin discovered. u%⚠️ Plugins directory not found: z Creating directory: T)parentsexist_okNu0❌ Plugins path exists but is not a directory: cSs$g|]}|r|jds|qS)_)is_dirr startswith).0prrr ds$z4PluginRegistry.discover_and_load..u'ℹ️ No plugin directories found in u 🔍 Scanning for plugins in z...z plugin.pyzMissing plugin.py file⊘ z: Missing plugin.py file z<============================================================u✅ Successfully loaded z plugin(s)u❌ Failed to load ) r.existsprintmkdirr5iterdirr/r_load_plugin_from_filelenr-)rplugin_folders plugin_folder plugin_filerrrdiscover_and_loadLs2   z PluginRegistry.discover_and_loadrCrDc szd|jdtj|}|dus|jdurtd|tj|}|tj<|j |fddt |t j D}|sRd|j |j<td|jd WdS|D]}\}}|||}|ru||j |j<td |jd |d |qTz3|} | jj} | |jvrtd |jd |d| dWqT| |j| <td|jd |d| dWqTty} zd| |j |j<td |jd |d| WYd} ~ qTd} ~ wwWdSty} zd| |j |j<td |jd| WYd} ~ dSd} ~ ww)z Load a single plugin from a plugin.py file. Args: plugin_folder: The plugin's directory plugin_file: Path to the plugin.py file zplugins.z.pluginNzCould not load spec for cs,g|]\}}|dr|jkr||fqS)Plugin)endswithr)r7robj module_namerrr9s z9PluginRegistry._load_plugin_from_file..zNo *Plugin class foundr:z%: No class ending with 'Plugin' foundu❌ .z: u⚠️ z: Duplicate plugin name 'z ' (skipping)u✅ u → ''zFailed to instantiate: z: Failed to instantiate: zImport failed: z: Import failed: )r importlibutilspec_from_file_locationloader ImportErrormodule_from_specsysmodules exec_moduleinspect getmembersisclassr/r=_validate_plugin_classmetadatar- Exception) rrCrDspecmoduleplugin_classes class_name plugin_classvalidation_errorplugin_instance plugin_nameerrIrr@~sP           "($z%PluginRegistry._load_plugin_from_filer`r_c Cst|dsdSt|dsdSt|d}t|sdSz.|j}t|dr&|js)WdSt|dr1|js4Wd Std d |jDsFd |jd WSWdSty_}z d|WYd}~Sd}~ww)a" Validate that a plugin class follows the required structure. Args: plugin_class: The class to validate class_name: Name of the class (for error messages) Returns: Error message if invalid, None if valid rZzMissing 'metadata' attributeexecutezMissing 'execute' methodz'execute' method must be asyncrzmetadata.name is requiredr(z metadata.description is requiredcss |] }|p |dvVqdS)z-_N)isalnum)r7crrr sz8PluginRegistry._validate_plugin_class..zmetadata.name 'z?' is not URL-safe (use alphanumeric, hyphens, underscores only)zInvalid metadata: N) hasattrgetattrrViscoroutinefunctionrZrr(allr[)rr`r_execute_methodrZrdrrrrYs*    z%PluginRegistry._validate_plugin_classrcCs |j|S)z Get a plugin instance by name. Args: name: The plugin name (from metadata.name) Returns: Plugin instance or None if not found )r-get)rrrrr get_plugins zPluginRegistry.get_plugincC |jS)z Get all loaded plugins. Returns: Dictionary mapping plugin names to plugin instances )r-copyrrrr list_plugins zPluginRegistry.list_pluginscCrp)z Get all plugin load errors. Returns: Dictionary mapping plugin folder names to error messages )r/rqrrrrget_load_errorsrszPluginRegistry.get_load_errorsr-)r1N)rrrr r!r0rErr@r rrYrrorrrrtrrrrr,9s 2=,  r,_plugin_registryr1cCstdurtatS)zo Get the global plugin registry instance. Returns: The global PluginRegistry instance N)rvr,rrrrget_plugin_registry srwr-r.cCst|attS)z Initialize the plugin system by discovering and loading all plugins. Args: plugins_dir: Path to the plugins directory Returns: The initialized PluginRegistry instance )r,rvrE)r.rrrinitialize_pluginss rxru)r importlib.utilrMrVrSpathlibrtypingrrr dataclassesrrr'r,rvr"rwr!rxrrrrs   O