init

Author: DriveFLY

.env.development

@@ -0,0 +1,10 @@
+NODE_ENV=development
+
+FEISHU_APP_ID=cli_a6f4f052acb1500d
+FEISHU_APP_SECRET=Nm3ejTKsq6ZXSc3M7qW78eksCA28gfYA
+
+# 多维表格记录
+FEISHU_BITABLE_APP_TOKEN=Rmu3b3iUAaqTaUs65GEcfWGMnGe
+FEISHU_BITABLE_USER_TABLE_ID=tblxfv5W6nzGQcHL
+FEISHU_BITABLE_RECORD_TABLE_ID=tbl59pIqnkfJJaUz
+FEISHU_BITABLE_STATUS_TABLE_ID=tblL9nLlquhdU8wR

.env.example

@@ -0,0 +1,10 @@
+NODE_ENV=
+
+FEISHU_APP_ID=
+FEISHU_APP_SECRET=
+
+# 多维表格记录
+FEISHU_BITABLE_APP_TOKEN=
+FEISHU_BITABLE_USER_TABLE_ID=
+FEISHU_BITABLE_RECORD_TABLE_ID=
+FEISHU_BITABLE_STATUS_TABLE_ID=

.prettierrc

@@ -0,0 +1,7 @@
+{
+    "printWidth": 80,
+    "trailingComma": "es5",
+    "tabWidth": 2,
+    "semi": false,
+    "singleQuote": true
+}

package.json

@@ -0,0 +1,23 @@
+{
+  "name": "leetcode-learning-bot",
+  "version": "1.0.0",
+  "type": "module",
+  "description": "",
+  "main": "index.js",
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1",
+    "dev": "dotenvx run -f .env.development -- node src/main.js",
+    "prod": "dotenvx run -f .env.production -- node src/main.js"
+  },
+  "keywords": [],
+  "author": "",
+  "license": "ISC",
+  "dependencies": {
+    "@larksuiteoapi/node-sdk": "^1.46.0",
+    "axios": "^1.8.4"
+  },
+  "devDependencies": {
+    "@dotenvx/dotenvx": "^1.39.1",
+    "prettier": "3.5.3"
+  }
+}

src/LeetCodeHistory.js

@@ -0,0 +1,107 @@
+import larkClient, { lark } from './larkClient.js'
+
+// 创建实时解题个数记录
+const setLeetCodeHistory = async (open_id, easy, medium, hard) => {
+  const data = new Map([
+    ['Easy', easy],
+    ['Medium', medium],
+    ['Hard', hard],
+    ['日期', new Date().getTime()],
+    ['人员', [{ id: open_id }]],
+  ])
+
+  await larkClient.bitable.v1.appTableRecord
+    .create({
+      path: {
+        app_token: process.env.FEISHU_BITABLE_APP_TOKEN,
+        table_id: process.env.FEISHU_BITABLE_RECORD_TABLE_ID,
+      },
+      data: {
+        fields: Object.fromEntries(data),
+      },
+    })
+    .then((res) => {
+      //   console.log(res)
+    })
+    .catch((e) => {
+      console.error(JSON.stringify(e.response.data, null, 4))
+    })
+}
+
+// 获取历史解题个数记录
+const getLeetCodeHistoryCount = async (open_id) => {
+  const res = await larkClient.bitable.v1.appTableRecord
+    .search({
+      path: {
+        app_token: process.env.FEISHU_BITABLE_APP_TOKEN,
+        table_id: process.env.FEISHU_BITABLE_RECORD_TABLE_ID,
+      },
+      data: {
+        sort: [
+          {
+            field_name: '日期',
+            desc: true,
+          },
+        ],
+        filter: {
+          conjunction: 'and',
+          conditions: [
+            {
+              field_name: '人员',
+              operator: 'is',
+              value: [open_id],
+            },
+            {
+              field_name: '日期',
+              operator: 'is',
+              value: ['Yesterday'],
+            },
+          ],
+        },
+      },
+    })
+    .then((res) => {
+      // console.log(res)
+      return res
+    })
+    .catch((e) => {
+      console.error(JSON.stringify(e.response.data, null, 4))
+    })
+  if (res.data.items.length === 0) {
+    return -1
+  }
+  return res.data.items[0].fields.Total.value[0]
+}
+
+// 创建打卡状态
+const setLeetCodeStatus = async (open_id, status) => {
+  if (status === true) {
+    status = '打卡成功'
+  } else {
+    status = '打卡失败'
+  }
+  const data = new Map([
+    ['日期', new Date().getTime()],
+    ['人员', [{ id: open_id }]],
+    ['状态', status],
+  ])
+
+  await larkClient.bitable.v1.appTableRecord
+    .create({
+      path: {
+        app_token: process.env.FEISHU_BITABLE_APP_TOKEN,
+        table_id: process.env.FEISHU_BITABLE_STATUS_TABLE_ID,
+      },
+      data: {
+        fields: Object.fromEntries(data),
+      },
+    })
+    .then((res) => {
+        // console.log(res)
+    })
+    .catch((e) => {
+      console.error(JSON.stringify(e.response.data, null, 4))
+    })
+}
+
+export { setLeetCodeHistory, getLeetCodeHistoryCount, setLeetCodeStatus }

src/getActivityUser.js

@@ -0,0 +1,32 @@
+import larkClient, { lark } from './larkClient.js'
+
+const getActivityUser = async () => {
+  const userData = []
+  await larkClient.bitable.v1.appTableRecord
+    .search({
+      path: {
+        app_token: process.env.FEISHU_BITABLE_APP_TOKEN,
+        table_id: process.env.FEISHU_BITABLE_USER_TABLE_ID,
+      },
+      params: {
+        user_id_type: 'open_id',
+        page_size: 20,
+      },
+    })
+    .then((res) => {
+      res.data.items.forEach((item) => {
+        userData.push({
+          name: item.fields['人员'][0].name,
+          open_id: item.fields['人员'][0].id,
+          leetcode_id: item.fields['LeetCodeName'][0].text,
+        })
+      })
+      //   console.log(userData)
+    })
+    .catch((e) => {
+      console.error(JSON.stringify(e.response.data, null, 4))
+    })
+  return userData
+}
+
+export default getActivityUser

src/getFeishuUser.js

@@ -0,0 +1,37 @@
+import larkClient, { lark } from './larkClient.js'
+import fs from 'fs'
+
+const getFeishuUser = async () => {
+  const userData = []
+  await larkClient.contact.v3.user
+    .findByDepartment(
+      {
+        params: {
+          user_id_type: 'open_id',
+          department_id_type: 'open_department_id',
+          department_id: 'od-09661b4dbbeb079b90a5a93cc348e8d3',
+          page_size: 50,
+        },
+      }
+    )
+    .then((res) => {
+      // console.log(res.data.items)
+      userData.push(
+        ...res.data.items.map((item) => ({
+          name: item.name,
+          mobile: item.mobile,
+          open_id: item.open_id,
+        }))
+      )
+      // console.log(userData)
+    })
+    .then(() => {
+      fs.writeFileSync('userData.json', JSON.stringify(userData, null, 2))
+      console.log('File userData.json has been written')
+    })
+    .catch((e) => {
+      console.error(e)
+    })
+}
+
+export default getFeishuUser

src/getLeetCode.js

@@ -0,0 +1,12 @@
+// http://8.137.12.168:3000/userProfile/chi-la-cmeng-6
+
+import axios from 'axios'
+
+const getLeetcode = async (leetcode_id) => {
+  const response = await axios.get(
+    `http://8.137.12.168:3000/userProfile/${leetcode_id}`
+  )
+  return response.data
+}
+
+export default getLeetcode

src/larkClient.js

@@ -0,0 +1,18 @@
+import * as lark from '@larksuiteoapi/node-sdk'
+
+const FEISHU_APP_ID = process.env.FEISHU_APP_ID
+const FEISHU_APP_SECRET = process.env.FEISHU_APP_SECRET
+
+if (!FEISHU_APP_ID || !FEISHU_APP_SECRET) {
+  console.log('飞书应用配置为空')
+  process.exit(1)
+}
+
+const larkClient = new lark.Client({
+  appId: FEISHU_APP_ID,
+  appSecret: FEISHU_APP_SECRET,
+  disableTokenCache: false,
+})
+
+export default larkClient
+export { lark }

src/main.js

@@ -0,0 +1,46 @@
+import getActivityUser from './getActivityUser.js'
+import getLeetcode from './getLeetcode.js'
+import {
+  setLeetCodeHistory,
+  getLeetCodeHistoryCount,
+  setLeetCodeStatus,
+} from './LeetCodeHistory.js'
+import sendMessage from './sendMessage.js'
+
+const userList = await getActivityUser()
+for (const user of userList) {
+  console.log(`开始获取Leetcode数据：${JSON.stringify(user)}`)
+  const leetcode_data = await getLeetcode(user.leetcode_id)
+  if (Object.keys(leetcode_data).length === 0) {
+    console.log(`获取Leetcode数据失败：${user.name},${user.leetcode_id}`)
+    continue
+  }
+  console.log(`获取Leetcode数据成功：${JSON.stringify(leetcode_data)}`)
+
+  await setLeetCodeHistory(
+    user.open_id,
+    leetcode_data.easySolved,
+    leetcode_data.mediumSolved,
+    leetcode_data.hardSolved
+  )
+
+  let totalSolved =
+    leetcode_data.easySolved +
+    leetcode_data.mediumSolved +
+    leetcode_data.hardSolved
+  let message = `实时 Leetcode 数据：Easy ${leetcode_data.easySolved} 题、Medium ${leetcode_data.mediumSolved} 题、Hard ${leetcode_data.hardSolved} 题，共 ${totalSolved} 题。`
+
+  const historyCount = await getLeetCodeHistoryCount(user.open_id)
+  if (historyCount === -1) {
+    message += `暂未获取到昨日数据。`
+    await setLeetCodeStatus(user.open_id, true)
+  } else if (historyCount === totalSolved) {
+    message += `今日暂未完成打卡！`
+    await setLeetCodeStatus(user.open_id, false)
+  } else {
+    message += `较昨日增加 ${totalSolved - historyCount} 题，打卡成功！`
+    await setLeetCodeStatus(user.open_id, true)
+  }
+
+  await sendMessage(user.open_id, message)
+}