d0904ca405
# What this PR does Improves OpenAPI schema coverage for internal API: - Fixes/Improves `alert group` and `feature` endpoints - Adds `integration` and `user` endpoints ## Which issue(s) this PR fixes https://github.com/grafana/oncall/issues/3444 ## Checklist - [x] Unit, integration, and e2e (if applicable) tests updated - [x] Documentation added (or `pr:no public docs` PR label added if not required) - [x] `CHANGELOG.md` updated (or `pr:no changelog` PR label added if not required)
64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
import enum
|
|
|
|
from django.conf import settings
|
|
from drf_spectacular.plumbing import resolve_type_hint
|
|
from drf_spectacular.utils import extend_schema
|
|
from rest_framework import status
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
|
|
from apps.auth_token.auth import PluginAuthentication
|
|
from apps.base.utils import live_settings
|
|
from apps.labels.utils import is_labels_feature_enabled
|
|
|
|
|
|
class Feature(enum.StrEnum):
|
|
MSTEAMS = "msteams"
|
|
SLACK = "slack"
|
|
TELEGRAM = "telegram"
|
|
LIVE_SETTINGS = "live_settings"
|
|
GRAFANA_CLOUD_NOTIFICATIONS = "grafana_cloud_notifications"
|
|
GRAFANA_CLOUD_CONNECTION = "grafana_cloud_connection"
|
|
GRAFANA_ALERTING_V2 = "grafana_alerting_v2"
|
|
LABELS = "labels"
|
|
|
|
|
|
class FeaturesAPIView(APIView):
|
|
"""
|
|
Return whitelist of enabled features.
|
|
It is needed to disable features for On-prem installations.
|
|
"""
|
|
|
|
authentication_classes = (PluginAuthentication,)
|
|
|
|
@extend_schema(responses={status.HTTP_200_OK: resolve_type_hint(list[Feature])})
|
|
def get(self, request):
|
|
data = self._get_enabled_features(request)
|
|
return Response(data)
|
|
|
|
def _get_enabled_features(self, request):
|
|
enabled_features = []
|
|
|
|
if settings.FEATURE_SLACK_INTEGRATION_ENABLED:
|
|
enabled_features.append(Feature.SLACK)
|
|
|
|
if settings.FEATURE_TELEGRAM_INTEGRATION_ENABLED:
|
|
enabled_features.append(Feature.TELEGRAM)
|
|
|
|
if settings.IS_OPEN_SOURCE:
|
|
# Features below should be enabled only in OSS
|
|
enabled_features.append(Feature.GRAFANA_CLOUD_CONNECTION)
|
|
if settings.FEATURE_LIVE_SETTINGS_ENABLED:
|
|
enabled_features.append(Feature.LIVE_SETTINGS)
|
|
if live_settings.GRAFANA_CLOUD_NOTIFICATIONS_ENABLED:
|
|
enabled_features.append(Feature.GRAFANA_CLOUD_NOTIFICATIONS)
|
|
else:
|
|
enabled_features.append(Feature.MSTEAMS)
|
|
|
|
if settings.FEATURE_GRAFANA_ALERTING_V2_ENABLED:
|
|
enabled_features.append(Feature.GRAFANA_ALERTING_V2)
|
|
|
|
if is_labels_feature_enabled(self.request.auth.organization):
|
|
enabled_features.append(Feature.LABELS)
|
|
|
|
return enabled_features
|