dcae98b02a
Related to https://github.com/grafana/oncall/issues/96 --------- Co-authored-by: Ravishankar <ravishankar.gnanaprakasam@gmail.com>
82 lines
3 KiB
Python
82 lines
3 KiB
Python
from rest_framework import mixins, status, viewsets
|
|
from rest_framework.decorators import action
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
|
|
from apps.api.permissions import RBACPermission
|
|
from apps.auth_token.auth import PluginAuthentication
|
|
from apps.mattermost.auth import MattermostEventAuthentication
|
|
from apps.mattermost.events.event_manager import EventManager
|
|
from apps.mattermost.models import MattermostChannel
|
|
from apps.mattermost.serializers import MattermostChannelSerializer
|
|
from common.api_helpers.mixins import PublicPrimaryKeyMixin
|
|
from common.insight_log.chatops_insight_logs import ChatOpsEvent, ChatOpsTypePlug, write_chatops_insight_log
|
|
|
|
|
|
class MattermostChannelViewSet(
|
|
PublicPrimaryKeyMixin[MattermostChannel],
|
|
mixins.CreateModelMixin,
|
|
mixins.RetrieveModelMixin,
|
|
mixins.ListModelMixin,
|
|
mixins.DestroyModelMixin,
|
|
viewsets.GenericViewSet,
|
|
):
|
|
authentication_classes = (PluginAuthentication,)
|
|
permission_classes = (IsAuthenticated, RBACPermission)
|
|
|
|
rbac_permissions = {
|
|
"list": [RBACPermission.Permissions.CHATOPS_READ],
|
|
"retrieve": [RBACPermission.Permissions.CHATOPS_READ],
|
|
"create": [RBACPermission.Permissions.CHATOPS_UPDATE_SETTINGS],
|
|
"destroy": [RBACPermission.Permissions.CHATOPS_UPDATE_SETTINGS],
|
|
"set_default": [RBACPermission.Permissions.CHATOPS_UPDATE_SETTINGS],
|
|
}
|
|
|
|
serializer_class = MattermostChannelSerializer
|
|
|
|
def get_queryset(self):
|
|
return MattermostChannel.objects.filter(organization=self.request.user.organization)
|
|
|
|
@action(detail=True, methods=["post"])
|
|
def set_default(self, request, pk):
|
|
mattermost_channel = self.get_object()
|
|
mattermost_channel.make_channel_default(request.user)
|
|
|
|
return Response(status=status.HTTP_200_OK)
|
|
|
|
def perform_create(self, serializer):
|
|
serializer.save()
|
|
instance = serializer.instance
|
|
write_chatops_insight_log(
|
|
author=self.request.user,
|
|
event_name=ChatOpsEvent.CHANNEL_CONNECTED,
|
|
chatops_type=ChatOpsTypePlug.MATTERMOST.value,
|
|
channel_name=instance.channel_name,
|
|
)
|
|
|
|
def perform_destroy(self, instance):
|
|
write_chatops_insight_log(
|
|
author=self.request.user,
|
|
event_name=ChatOpsEvent.CHANNEL_DISCONNECTED,
|
|
chatops_type=ChatOpsTypePlug.MATTERMOST.value,
|
|
channel_name=instance.channel_name,
|
|
channel_id=instance.channel_id,
|
|
)
|
|
instance.delete()
|
|
|
|
|
|
class MattermostEventView(APIView):
|
|
authentication_classes = (MattermostEventAuthentication,)
|
|
permission_classes = (IsAuthenticated, RBACPermission)
|
|
|
|
rbac_permissions = {
|
|
"post": [RBACPermission.Permissions.ALERT_GROUPS_WRITE],
|
|
}
|
|
|
|
def get(self, request, format=None):
|
|
return Response("hello")
|
|
|
|
def post(self, request):
|
|
EventManager.process_request(request=request)
|
|
return Response(status=200)
|