airano-ir
2c9d4ed3d2
- Added update_webhook and delete_label tools (58 total Gitea tools) - 97 unit tests (test_gitea_plugin.py) - Gitea now public for all users (DEFAULT_PUBLIC_PLUGINS) - Fixed merge_pull_request optional field handling Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
"""Plugin visibility control for public vs admin users (Track F.1).
|
|
|
|
Controls which plugins are visible to public (OAuth) users vs admin users.
|
|
Admin users (MASTER_API_KEY) always see all plugins. Public users only see
|
|
plugins listed in the ENABLED_PLUGINS setting (DB > ENV > default).
|
|
|
|
Usage:
|
|
from core.plugin_visibility import get_public_plugin_types, is_plugin_public
|
|
|
|
public_types = get_public_plugin_types() # {"wordpress", "woocommerce", "supabase"}
|
|
if is_plugin_public("gitea"): # True
|
|
...
|
|
"""
|
|
|
|
import os
|
|
|
|
# Default plugins available to public (OAuth) users
|
|
DEFAULT_PUBLIC_PLUGINS = {"wordpress", "woocommerce", "supabase", "openpanel", "gitea"}
|
|
|
|
|
|
def _parse_plugins(val: str) -> set[str]:
|
|
"""Parse comma-separated plugin string into set."""
|
|
return {p.strip().lower() for p in val.split(",") if p.strip()}
|
|
|
|
|
|
def get_public_plugin_types() -> set[str]:
|
|
"""Return the set of plugin types visible to public users.
|
|
|
|
Checks DB settings first (sync-safe), then env var, then defaults.
|
|
|
|
Returns:
|
|
Set of lowercase plugin type strings.
|
|
"""
|
|
# Try DB setting (sync access via cached value)
|
|
try:
|
|
from core.settings import _cached_plugins
|
|
|
|
if _cached_plugins is not None:
|
|
return set(_cached_plugins)
|
|
except (ImportError, AttributeError):
|
|
pass
|
|
|
|
# Fallback to env var
|
|
env_val = os.getenv("ENABLED_PLUGINS", "").strip()
|
|
if not env_val:
|
|
return set(DEFAULT_PUBLIC_PLUGINS)
|
|
return _parse_plugins(env_val)
|
|
|
|
|
|
def is_plugin_public(plugin_type: str) -> bool:
|
|
"""Check if a plugin type is enabled for public users.
|
|
|
|
Args:
|
|
plugin_type: Plugin type string (e.g., "wordpress").
|
|
|
|
Returns:
|
|
True if the plugin is in the public set.
|
|
"""
|
|
return plugin_type.lower() in get_public_plugin_types()
|