补全安全检查

main.js

@@ -8,14 +8,30 @@ const 作品 = new Map()
 const 游戏 = new Map()
 const 截图 = new Map()
 const 格式 = new Intl.DateTimeFormat('zh-CN', { year: 'numeric', month: '2-digit', day: '2-digit' })
+const articleRegex = /\/web\/v1\/article\/get\?id=(\d+)&userId=(\d+)/
+const imagesRegex = /\/api\/images\?similar=(\d+)/
 
 
-var 日期 = 格式.format(new Date()).replace(/\//g, '-')
-var 存档 = new Date()
+var 当前日期 = 格式.format(new Date()).replace(/\//g, '-')
+var 正在存档 = false
 
-if (fs.existsSync(`./data/${日期}.json`)) {
-    console.log(日期, '启动时读入数据..')
-    const file = fs.readFileSync(`./data/${日期}.json`, 'utf8')
+
+const 存档 = async (message = '存档中..') => {
+    if (正在存档) return
+    正在存档 = true
+    console.log(message)
+    await fs.promises.writeFile(`./data/${当前日期}.json`, JSON.stringify({
+        作品: Object.fromEntries(作品),
+        游戏: Object.fromEntries(游戏),
+        截图: Object.fromEntries(截图),
+    }, null, 2))
+    正在存档 = false
+}
+
+
+if (fs.existsSync(`./data/${当前日期}.json`)) {
+    console.log(当前日期, '启动时读入数据..')
+    const file = fs.readFileSync(`./data/${当前日期}.json`, 'utf8')
     if (file) {
         const data = JSON.parse(file)
         Object.entries(data.作品).forEach(([key, value]) => 作品.set(key, value))
@@ -25,45 +41,39 @@ if (fs.existsSync(`./data/${日期}.json`)) {
 }
 
 
-console.log(日期, '开始收集日志..')
+console.log(当前日期, '开始收集日志..')
+setInterval(() => 存档('10 分钟存档一次..'), 600000)
 
-
-tail.on('line', (line) => {
+tail.on('line', async (line) => {
     const item = JSON.parse(line)
     if (item.level !== 'debug') return
     if (item.msg !== 'upstream roundtrip') return
 
-    if ((new Date()).getTime() - 存档.getTime() > 600000) {
-        console.log('10 分钟存档一次..')
-        fs.promises.writeFile(`./data/${日期}.json`, JSON.stringify({
-            作品: Object.fromEntries(作品),
-            游戏: Object.fromEntries(游戏),
-            截图: Object.fromEntries(截图),
-        }, null, 2))
-        存档 = new Date()
-    }
-
     const 日志日期 = 格式.format(new Date(item.ts * 1000)).replace(/\//g, '-')
-    if (日期 !== 日志日期) {
-        fs.promises.writeFile(`./data/${日期} 作品.json`, JSON.stringify(Object.fromEntries(作品), null, 2))
-        fs.promises.writeFile(`./data/${日期} 游戏.json`, JSON.stringify(Object.fromEntries(游戏), null, 2))
-        fs.promises.writeFile(`./data/${日期} 截图.json`, JSON.stringify(Object.fromEntries(截图), null, 2))
-        日期 = 日志日期
+    if (当前日期 !== 日志日期) {
+        await 存档('跨日期存档..')
+        作品.clear()
+        游戏.clear()
+        截图.clear()
+        当前日期 = 日志日期
     }
 
     if (item.request.uri.startsWith('/web/v1/article/get')) {
-        const reg = /\/web\/v1\/article\/get\?id=(\d+)&userId=(\d+)/
-        const [uri, id, user_id] = item.request.uri.match(reg) ?? []
+        const [uri, id, user_id] = item.request.uri.match(articleRegex) ?? []
         if (uri && id && user_id && item.request.headers.Referer) {
-            if (item.request.headers.Referer[0].includes('/inspirationInfo/')) {
-                游戏.set(id, 游戏.has(id) ? 游戏.get(id) + 1 : 1)
-                return
-            }
             if (item.request.headers.Referer[0].includes('/articleDetails/')) {
                 作品.set(id, 作品.has(id) ? 作品.get(id) + 1 : 1)
                 return
             }
-            console.log(item.request.headers.Referer)
+            if (item.request.headers.Referer[0].includes('/inspirationInfo/')) {
+                游戏.set(id, 游戏.has(id) ? 游戏.get(id) + 1 : 1)
+                return
+            }
+            if (item.request.headers.Referer[0].includes('/game/')) {
+                游戏.set(id, 游戏.has(id) ? 游戏.get(id) + 1 : 1)
+                return
+            }
+            console.log('未处理 Referer:', item.request.headers.Referer)
         }
         return
     }
@@ -71,8 +81,7 @@ tail.on('line', (line) => {
         if (item.request.uri.includes('page=')) {
             return
         }
-        const reg = /\/api\/images\?similar=(\d+)/
-        const [uri, id] = item.request.uri.match(reg) ?? []
+        const [uri, id] = item.request.uri.match(imagesRegex) ?? []
         if (uri && id) {
             截图.set(id, 截图.has(id) ? 截图.get(id) + 1 : 1)
         }
@@ -80,26 +89,11 @@ tail.on('line', (line) => {
     }
 })
 
-tail.on('end', () => {
-    console.log('没有更多内容，结束读取')
-    const sorted = new Map([...游戏.entries()].sort((a, b) => b[1] - a[1]))
-    const top10 = Array.from(sorted).slice(0, 10)
-    console.log(top10)
-})
-
-tail.on('error', (error) => {
-    console.error('错误:', error)
-})
+tail.on('end', () => console.log('没有更多内容，结束读取'))
+tail.on('error', (error) => console.error('错误:', error))
 
 process.on('SIGINT', async () => {
     tail.unwatch()
-
-    console.log('退出前储存数据..')
-    await fs.promises.writeFile(`./data/${日期}.json`, JSON.stringify({
-        作品: Object.fromEntries(作品),
-        游戏: Object.fromEntries(游戏),
-        截图: Object.fromEntries(截图),
-    }, null, 2))
-
+    await 存档('退出前储存数据..')
     process.exit()
 })