204 files changed
+51505
No files matched your search
@@ -0,0 +1,183 @@
|
||||
import importlib
|
||||
import importlib.util
|
||||
import os
|
||||
from typing import Dict, List, Type
|
||||
|
||||
from modules.launcher.plugin_base import PluginBase
|
||||
|
||||
|
||||
class PluginManager:
|
||||
"""
|
||||
Manages launcher plugins.
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.plugins: Dict[str, PluginBase] = {}
|
||||
self.plugin_classes: Dict[str, Type[PluginBase]] = {}
|
||||
self.active_plugins: List[str] = []
|
||||
|
||||
# Load built-in plugins
|
||||
self._load_builtin_plugins()
|
||||
|
||||
# Load external plugins
|
||||
self._load_external_plugins()
|
||||
|
||||
# Activate default plugins
|
||||
self._activate_default_plugins()
|
||||
|
||||
def _load_builtin_plugins(self):
|
||||
"""Load built-in plugins from the plugins directory."""
|
||||
plugins_dir = os.path.join(os.path.dirname(__file__), "plugins")
|
||||
|
||||
if not os.path.exists(plugins_dir):
|
||||
return
|
||||
|
||||
for filename in os.listdir(plugins_dir):
|
||||
if filename.endswith(".py") and not filename.startswith("_"):
|
||||
plugin_name = filename[:-3] # Remove .py extension
|
||||
self._load_plugin_from_file(plugins_dir, plugin_name)
|
||||
|
||||
def _load_external_plugins(self):
|
||||
"""Load external plugins from user directory."""
|
||||
# Could be implemented to load from ~/.config/launcher/plugins/
|
||||
pass
|
||||
|
||||
def _load_plugin_from_file(self, plugins_dir: str, plugin_name: str):
|
||||
"""Load a plugin from a Python file."""
|
||||
try:
|
||||
plugin_path = os.path.join(plugins_dir, f"{plugin_name}.py")
|
||||
spec = importlib.util.spec_from_file_location(
|
||||
f"modules.launcher.plugins.{plugin_name}", plugin_path
|
||||
)
|
||||
|
||||
if spec and spec.loader:
|
||||
module = importlib.util.module_from_spec(spec)
|
||||
|
||||
# Add the module to sys.modules to support relative imports
|
||||
import sys
|
||||
|
||||
sys.modules[f"modules.launcher.plugins.{plugin_name}"] = module
|
||||
|
||||
spec.loader.exec_module(module)
|
||||
|
||||
# Look for plugin class
|
||||
for attr_name in dir(module):
|
||||
attr = getattr(module, attr_name)
|
||||
if (
|
||||
isinstance(attr, type)
|
||||
and issubclass(attr, PluginBase)
|
||||
and attr != PluginBase
|
||||
):
|
||||
self.plugin_classes[plugin_name] = attr
|
||||
break
|
||||
|
||||
except Exception as e:
|
||||
print(f"Failed to load plugin {plugin_name}: {e}")
|
||||
|
||||
def _activate_default_plugins(self):
|
||||
"""Activate default plugins."""
|
||||
default_plugins = [
|
||||
"applications",
|
||||
"calculator",
|
||||
"system",
|
||||
"clipboard",
|
||||
"power",
|
||||
"caffeine",
|
||||
"screencapture",
|
||||
"emoji",
|
||||
"wallpaper",
|
||||
"websearch",
|
||||
"reminders",
|
||||
"otp",
|
||||
"password",
|
||||
"bookmarks",
|
||||
"bash_scripts",
|
||||
"tmux",
|
||||
]
|
||||
|
||||
for plugin_name in default_plugins:
|
||||
self.activate_plugin(plugin_name)
|
||||
|
||||
def activate_plugin(self, plugin_name: str) -> bool:
|
||||
"""Activate a plugin by name."""
|
||||
if plugin_name in self.plugins:
|
||||
# Already activated
|
||||
return True
|
||||
|
||||
if plugin_name not in self.plugin_classes:
|
||||
return False
|
||||
|
||||
try:
|
||||
# Instantiate plugin
|
||||
plugin_class = self.plugin_classes[plugin_name]
|
||||
plugin_instance = plugin_class()
|
||||
|
||||
# Initialize plugin
|
||||
plugin_instance.initialize()
|
||||
|
||||
# Store plugin
|
||||
self.plugins[plugin_name] = plugin_instance
|
||||
self.active_plugins.append(plugin_name)
|
||||
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
print(f"Failed to activate plugin {plugin_name}: {e}")
|
||||
return False
|
||||
|
||||
def deactivate_plugin(self, plugin_name: str) -> bool:
|
||||
"""Deactivate a plugin by name."""
|
||||
if plugin_name not in self.plugins:
|
||||
return False
|
||||
|
||||
try:
|
||||
# Cleanup plugin
|
||||
plugin = self.plugins[plugin_name]
|
||||
plugin.cleanup()
|
||||
|
||||
# Remove from active plugins
|
||||
del self.plugins[plugin_name]
|
||||
if plugin_name in self.active_plugins:
|
||||
self.active_plugins.remove(plugin_name)
|
||||
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
print(f"Failed to deactivate plugin {plugin_name}: {e}")
|
||||
return False
|
||||
|
||||
def get_active_plugins(self) -> List[PluginBase]:
|
||||
"""Get list of active plugin instances."""
|
||||
return [
|
||||
self.plugins[name] for name in self.active_plugins if name in self.plugins
|
||||
]
|
||||
|
||||
def get_plugin_names(self) -> List[str]:
|
||||
"""Get list of available plugin names."""
|
||||
return list(self.plugin_classes.keys())
|
||||
|
||||
def get_active_plugin_names(self) -> List[str]:
|
||||
"""Get list of active plugin names."""
|
||||
return self.active_plugins.copy()
|
||||
|
||||
def reload_plugin(self, plugin_name: str) -> bool:
|
||||
"""Reload a plugin."""
|
||||
was_active = plugin_name in self.active_plugins
|
||||
|
||||
# Deactivate if active
|
||||
if was_active:
|
||||
self.deactivate_plugin(plugin_name)
|
||||
|
||||
# Remove from classes
|
||||
if plugin_name in self.plugin_classes:
|
||||
del self.plugin_classes[plugin_name]
|
||||
|
||||
# Reload from file
|
||||
plugins_dir = os.path.join(os.path.dirname(__file__), "plugins")
|
||||
self._load_plugin_from_file(plugins_dir, plugin_name)
|
||||
|
||||
# Reactivate if it was active
|
||||
if was_active and plugin_name in self.plugin_classes:
|
||||
return self.activate_plugin(plugin_name)
|
||||
|
||||
return plugin_name in self.plugin_classes
|
||||
Reference in new issue
Block a user