From 129e4d67c760e62f50b035845c949f6af3d4644c Mon Sep 17 00:00:00 2001
From: tangyanbiao <tangyanbiao@xeno.com>
Date: Tue, 21 Dec 2021 14:31:29 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=94=AF=E6=8C=81telegram=20bot?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .env.example                  |  5 ++++-
 api/{bot/feishu.ts => txc.ts} | 29 +++++++----------------------
 bot/feishu.ts                 | 20 ++++++++++++++++++++
 bot/telegram.ts               | 15 +++++++++++++++
 vercel.json                   |  2 +-
 5 files changed, 47 insertions(+), 24 deletions(-)
 rename api/{bot/feishu.ts => txc.ts} (72%)
 create mode 100644 bot/feishu.ts
 create mode 100644 bot/telegram.ts

diff --git a/.env.example b/.env.example
index 01dd99e..8ade2b0 100644
--- a/.env.example
+++ b/.env.example
@@ -1 +1,4 @@
-WEBHOOK_URL=飞书群机器人 webhook 地址
\ No newline at end of file
+BOTS='TELEGRAM'
+FEISHU_TOKEN=''
+TELEGRAM_TOKEN=''
+TELEGRAM_CHAT_ID=''
\ No newline at end of file
diff --git a/api/bot/feishu.ts b/api/txc.ts
similarity index 72%
rename from api/bot/feishu.ts
rename to api/txc.ts
index 32d51ad..0bf795c 100644
--- a/api/bot/feishu.ts
+++ b/api/txc.ts
@@ -1,25 +1,4 @@
 import {VercelRequest, VercelResponse} from '@vercel/node'
-import fetch from 'node-fetch'
-
-async function send(content: string) {
-    const body = JSON.stringify({
-        "msg_type": "text",
-        "content": {
-            "text": content
-        }
-    })
-    const res = await fetch(process.env.WEBHOOK_URL, {
-        method: 'POST',
-        headers: {
-            'Content-Type': 'application/json'
-        },
-        body,
-    })
-    const data = await res.json()
-    if (data.StatusCode !== 0) {
-        console.error(data.StatusMessage)
-    }
-}
 
 function handleMessage(name: string, data: any): string {
     let action = ''
@@ -56,10 +35,16 @@ async function main(req: VercelRequest, res: VercelResponse) {
         res.setHeader('Allow', 'POST')
         return res.status(405).end('Method Not Allowed')
     }
+    const bots = (process.env.BOTS || '').split(',').map(v => v.toUpperCase())
     try {
         const product = req.query.product.toString() || 'Unknown'
         const content = handleMessage(product, req.body)
-        await send(content)
+        if (bots.indexOf('TELEGRAM') >= 0) {
+          await sendToTelegram(content)
+        }
+        if (bots.indexOf('FEISHU') >= 0) {
+          await sendToFeishu(content)
+        }
         return res.status(200).end('ok')
     } catch (e) {
         console.error(e)
diff --git a/bot/feishu.ts b/bot/feishu.ts
new file mode 100644
index 0000000..61c6f59
--- /dev/null
+++ b/bot/feishu.ts
@@ -0,0 +1,20 @@
+async function sendToFeishu(content: string) {
+  const URL = `https://open.feishu.cn/open-apis/bot/v2/hook/${process.env.FEISHU_TOKEN}`
+  const body = JSON.stringify({
+      "msg_type": "text",
+      "content": {
+          "text": content
+      }
+  })
+  const res = await fetch(URL, {
+      method: 'POST',
+      headers: {
+          'Content-Type': 'application/json'
+      },
+      body,
+  })
+  const data = await res.json()
+  if (data.StatusCode !== 0) {
+      console.error(data.StatusMessage)
+  }
+}
\ No newline at end of file
diff --git a/bot/telegram.ts b/bot/telegram.ts
new file mode 100644
index 0000000..f8c4780
--- /dev/null
+++ b/bot/telegram.ts
@@ -0,0 +1,15 @@
+async function sendToTelegram(text: string) {
+  const URL = `https://api.telegram.org/bot${process.env.TELEGRAM_TOKEN}/sendMessage`
+  const body = `chat_id=${process.env.TELEGRAM_CHAT_ID}&text=${text}`
+  const res = await fetch(URL, {
+      method: 'POST',
+      headers: {
+          'Content-Type': 'application/x-www-form-urlencoded'
+      },
+      body,
+  })
+  const data = await res.json()
+  if (!data.ok) {
+      console.error(data.StatusMessage)
+  }
+}
\ No newline at end of file
diff --git a/vercel.json b/vercel.json
index 638784d..fcd203f 100644
--- a/vercel.json
+++ b/vercel.json
@@ -1,5 +1,5 @@
 {
     "rewrites": [
-        { "source": "/hooks/:product", "destination": "/api/bot/feishu" }
+        { "source": "/hooks/:product", "destination": "/api/txc" }
     ]
 }
\ No newline at end of file