add feishu bot

config.yaml

@@ -29,6 +29,12 @@ notify:
     access_token:
     secret:
 
+  #https://open.feishu.cn/document/client-docs/bot-v3/add-custom-bot#9fe10f9b
+  feishu:
+    enable: false
+    webhook:
+    secret:
+
   #邮箱配置，在每天6点会推送前一天的漏洞汇总，注意password为授权码
   email:
     smtp_server:

notifications/feishu.py

@@ -0,0 +1,30 @@
+import hashlib
+import base64
+import hmac
+import requests
+import time
+
+
+def gen_sign(timestamp, secret):
+    string_to_sign = '{}\n{}'.format(timestamp, secret)
+    hmac_code = hmac.new(string_to_sign.encode("utf-8"), digestmod=hashlib.sha256).digest()
+    sign = base64.b64encode(hmac_code).decode('utf-8')
+    return sign
+
+
+def feishu_notification(webhook, secret, content):
+    webhook_url = 'https://open.feishu.cn/open-apis/bot/v2/hook/' + webhook
+    timestamp = str(int(time.time()))
+    sign = gen_sign(timestamp, secret)
+    headers = {
+        "Content-Type": "application/json"
+    }
+    data = {
+        "timestamp": timestamp,
+        "sign": sign,
+        "msg_type": "text",
+        "content": {
+            "text": content
+        }
+    }
+    requests.post(webhook_url, json=data, headers=headers)

notifications/notifier.py

@@ -3,6 +3,7 @@
 from notifications.mail import email_notification
 from notifications.dingtalk import dingtalk_notification
 from notifications.wxwork import wxwork_notification
+from notifications.feishu import feishu_notification
 
 
 def send_realtime_notifications(content):
@@ -11,6 +12,8 @@ def send_realtime_notifications(content):
         wxwork_notification(notify['wxwork']['key'], content)
     if notify['dingtalk']['enable']:
         dingtalk_notification(notify['dingtalk']['access_token'], notify['dingtalk']['secret'], content)
+    if notify['feishu']['enable']:
+        feishu_notification(notify['feishu']['webhook'], notify['feishu']['secret'], content)
 
 
 def send_daily_notifications(date):