Skip to content
Back to top

# Source code for mindroot.coreplugins.index.handlers.plugin_ops

    import json
    import logging
    from datetime import datetime
    from pathlib import Path
    from fastapi.responses import JSONResponse
    from fastapi import HTTPException
    from ..models import PluginEntry, PluginManifest
    logging.basicConfig(level=logging.DEBUG)
    logger = logging.getLogger(__name__)

    async def get_installed_plugin_metadata(plugin_name: str) -> dict:
        """Get metadata from main plugin manifest"""
        try:
            manifest_path = Path('data/plugin_manifest.json')
            logger.debug(f'Reading main plugin manifest from: {manifest_path}')
            if not manifest_path.exists():
                raise ValueError(f'Main plugin manifest not found at {manifest_path}')
            with open(manifest_path) as f:
                manifest_data = json.load(f)
            installed_plugins = manifest_data.get('plugins', {}).get('installed', {})
            plugin_data = installed_plugins.get(plugin_name)
            if not plugin_data:
                raise ValueError(f'Plugin {plugin_name} not found in manifest')
            logger.debug(f'Found plugin data in manifest: {plugin_data}')
            return {'commands': plugin_data.get('commands', []), 'services': plugin_data.get('services', []), 'dependencies': plugin_data.get('dependencies', [])}
        except Exception as e:
            logger.error(f'Error reading plugin manifest: {str(e)}')
            raise

    async def create_distributable_entry(plugin: PluginEntry) -> dict:
        """Transform plugin entry to distributable format"""
        try:
            metadata = await get_installed_plugin_metadata(plugin.name)
            if not plugin.remote_source and plugin.source_path:
                plugin_id = Path(plugin.source_path).parent.name
                remote_source = f'runvnc/{plugin_id}'
            else:
                remote_source = plugin.remote_source or ''
            return {'name': plugin.name, 'version': plugin.version, 'description': plugin.description, 'source': 'github', 'github_url': f'https://github.com/{remote_source}', 'commands': metadata['commands'], 'services': metadata['services'], 'dependencies': metadata['dependencies'], 'added_at': datetime.now().isoformat()}
        except Exception as e:
            logger.error(f'Error creating distributable entry: {str(e)}')
            raise

    async def add_plugin(INDEX_DIR: Path, index_name: str, plugin: PluginEntry):
        """Add a plugin to an index"""
        try:
            logger.debug(f'Adding plugin: {plugin.name} to index: {index_name}')
            logger.debug(f'Plugin data: {plugin.dict()}')
            index_dir = INDEX_DIR / index_name
            index_file = index_dir / 'index.json'
            if not index_file.exists():
                return JSONResponse({'success': False, 'message': 'Index not found'})
            has_github_info = plugin.remote_source or plugin.github_url or getattr(plugin, 'metadata', {}).get('github_url')
            if not has_github_info:
                return JSONResponse({'success': False, 'message': 'Plugin missing GitHub repository information. Cannot add to index.'})
            if plugin.source == 'local':
                return JSONResponse({'success': False, 'message': 'Local plugins cannot be added to an index'})
            with open(index_file, 'r') as f:
                index_data = json.load(f)
            if any((p['name'] == plugin.name for p in index_data['plugins'])):
                return JSONResponse({'success': False, 'message': 'Plugin already in index'})
            try:
                plugin_data = await create_distributable_entry(plugin)
            except Exception as e:
                return JSONResponse({'success': False, 'message': f'Failed to create distributable plugin entry: {str(e)}'})
            index_data['plugins'].append(plugin_data)
            with open(index_file, 'w') as f:
                json.dump(index_data, f, indent=2)
            return JSONResponse({'success': True, 'data': index_data})
        except Exception as e:
            logger.error(f'Error adding plugin: {str(e)}', exc_info=True)
            raise HTTPException(status_code=500, detail=str(e))

    async def remove_plugin(INDEX_DIR: Path, index_name: str, plugin_name: str):
        """Remove a plugin from an index"""
        try:
            index_dir = INDEX_DIR / index_name
            index_file = index_dir / 'index.json'
            if not index_file.exists():
                return JSONResponse({'success': False, 'message': 'Index not found'})
            with open(index_file, 'r') as f:
                index_data = json.load(f)
            index_data['plugins'] = [p for p in index_data['plugins'] if p['name'] != plugin_name]
            with open(index_file, 'w') as f:
                json.dump(index_data, f, indent=2)
            return JSONResponse({'success': True, 'data': index_data})
        except Exception as e:
            logger.error(f'Error removing plugin: {str(e)}', exc_info=True)
            raise HTTPException(status_code=500, detail=str(e))