feat: 添加 飞书 (#6602)

* feat: 添加 飞书

Co-authored-by: xinwen <coderWen@126.com>
Co-authored-by: wenyann <64353056+wenyann@users.noreply.github.com>

apps/settings/api/__init__.py

@@ -2,3 +2,4 @@ from .common import *
 from .ldap import *
 from .wecom import *
 from .dingtalk import *
+from .feishu import *

apps/settings/api/common.py

@@ -130,6 +130,7 @@ class PublicSettingApi(generics.RetrieveAPIView):
                 },
                 "AUTH_WECOM": settings.AUTH_WECOM,
                 "AUTH_DINGTALK": settings.AUTH_DINGTALK,
+                "AUTH_FEISHU": settings.AUTH_FEISHU,
                 'SECURITY_WATERMARK_ENABLED': settings.SECURITY_WATERMARK_ENABLED
             }
         }
@@ -148,6 +149,7 @@ class SettingsApi(generics.RetrieveUpdateAPIView):
         'email_content': serializers.EmailContentSettingSerializer,
         'wecom': serializers.WeComSettingSerializer,
         'dingtalk': serializers.DingTalkSettingSerializer,
+        'feishu': serializers.FeiShuSettingSerializer,
     }
 
     def get_serializer_class(self):

apps/settings/api/feishu.py

@@ -0,0 +1,41 @@
+from rest_framework.views import Response
+from rest_framework.generics import GenericAPIView
+from rest_framework.exceptions import APIException
+from rest_framework import status
+from django.utils.translation import gettext_lazy as _
+
+from settings.models import Setting
+from common.permissions import IsSuperUser
+from common.message.backends.feishu import FeiShu
+
+from .. import serializers
+
+
+class FeiShuTestingAPI(GenericAPIView):
+    permission_classes = (IsSuperUser,)
+    serializer_class = serializers.FeiShuSettingSerializer
+
+    def post(self, request):
+        serializer = self.serializer_class(data=request.data)
+        serializer.is_valid(raise_exception=True)
+
+        app_id = serializer.validated_data['FEISHU_APP_ID']
+        app_secret = serializer.validated_data.get('FEISHU_APP_SECRET')
+
+        if not app_secret:
+            secret = Setting.objects.filter(name='FEISHU_APP_SECRET').first()
+            if secret:
+                app_secret = secret.cleaned_value
+
+        app_secret = app_secret or ''
+
+        try:
+            feishu = FeiShu(app_id=app_id, app_secret=app_secret)
+            feishu.send_text(['test'], 'test')
+            return Response(status=status.HTTP_200_OK, data={'msg': _('Test success')})
+        except APIException as e:
+            try:
+                error = e.detail['errmsg']
+            except:
+                error = e.detail
+            return Response(status=status.HTTP_400_BAD_REQUEST, data={'error': error})

apps/settings/serializers/settings.py

@@ -7,6 +7,7 @@ __all__ = [
     'BasicSettingSerializer', 'EmailSettingSerializer', 'EmailContentSettingSerializer',
     'LDAPSettingSerializer', 'TerminalSettingSerializer', 'SecuritySettingSerializer',
     'SettingsSerializer', 'WeComSettingSerializer', 'DingTalkSettingSerializer',
+    'FeiShuSettingSerializer',
 ]
 
 
@@ -218,6 +219,12 @@ class DingTalkSettingSerializer(serializers.Serializer):
     AUTH_DINGTALK = serializers.BooleanField(default=False, label=_('Enable DingTalk Auth'))
 
 
+class FeiShuSettingSerializer(serializers.Serializer):
+    FEISHU_APP_ID = serializers.CharField(max_length=256, required=True, label='App ID')
+    FEISHU_APP_SECRET = serializers.CharField(max_length=256, required=False, label='App Secret', write_only=True)
+    AUTH_FEISHU = serializers.BooleanField(default=False, label=_('Enable FeiShu Auth'))
+
+
 class SettingsSerializer(
     BasicSettingSerializer,
     EmailSettingSerializer,
@@ -227,6 +234,7 @@ class SettingsSerializer(
     SecuritySettingSerializer,
     WeComSettingSerializer,
     DingTalkSettingSerializer,
+    FeiShuSettingSerializer,
 ):
 
     # encrypt_fields 现在使用 write_only 来判断了

apps/settings/urls/api_urls.py

@@ -15,6 +15,7 @@ urlpatterns = [
     path('ldap/cache/refresh/', api.LDAPCacheRefreshAPI.as_view(), name='ldap-cache-refresh'),
     path('wecom/testing/', api.WeComTestingAPI.as_view(), name='wecom-testing'),
     path('dingtalk/testing/', api.DingTalkTestingAPI.as_view(), name='dingtalk-testing'),
+    path('feishu/testing/', api.FeiShuTestingAPI.as_view(), name='feishu-testing'),
 
     path('setting/', api.SettingsApi.as_view(), name='settings-setting'),
     path('public/', api.PublicSettingApi.as_view(), name='public-setting'),