From d48150562f4bb618a8442fdb03214df19444df4c Mon Sep 17 00:00:00 2001
From: fengkx <liangkx8237@gmail.com>
Date: Wed, 29 Apr 2020 14:08:32 +0800
Subject: [PATCH] feat: RSS of cve.mitre.org (#4599)

---
 docs/en/programming.md   |  6 ++++++
 docs/programming.md      |  6 ++++++
 lib/router.js            |  3 +++
 lib/routes/cve/search.js | 26 ++++++++++++++++++++++++++
 4 files changed, 41 insertions(+)
 create mode 100644 lib/routes/cve/search.js

diff --git a/docs/en/programming.md b/docs/en/programming.md
index 4b3d273fa15f5b..941848fe4b4c00 100644
--- a/docs/en/programming.md
+++ b/docs/en/programming.md
@@ -4,6 +4,12 @@ pageClass: routes
 
 # Programming
 
+## cve.mitre.org
+
+### Search Result
+
+<Route author="fengkx" example="/cve/search/PostgreSQL" path="/cve/search/:keyword" :paramsDesc="['keyword'] >
+
 ## GitHub
 
 ::: tip
diff --git a/docs/programming.md b/docs/programming.md
index 13860cd266490a..ac30144fe10b6e 100644
--- a/docs/programming.md
+++ b/docs/programming.md
@@ -18,6 +18,12 @@ pageClass: routes
 
 > AlgoCasts 需要付费订阅, RSS 仅做更新提醒, 不含付费内容.
 
+## cve.mitre.org
+
+### 搜索结果
+
+<Route author="fengkx" example="/cve/search/PostgreSQL" path="/cve/search/:keyword" :paramsDesc="['关键词'] >
+
 ## Dockone
 
 ### 周报
diff --git a/lib/router.js b/lib/router.js
index bd75326a2bffa0..4f97f83a6666bd 100755
--- a/lib/router.js
+++ b/lib/router.js
@@ -2607,4 +2607,7 @@ router.get('/unit-image/films/:type?', require('./routes/unit-image/films'));
 // digic-picture
 router.get('/digic-pictures/:menu/:tags?', require('./routes/digic-pictures/index'));
 
+// cve.mitre.org
+router.get('/cve/search/:keyword', require('./routes/cve/search'));
+
 module.exports = router;
diff --git a/lib/routes/cve/search.js b/lib/routes/cve/search.js
new file mode 100644
index 00000000000000..4f486e572f4e12
--- /dev/null
+++ b/lib/routes/cve/search.js
@@ -0,0 +1,26 @@
+const got = require('@/utils/got');
+const cheerio = require('cheerio');
+
+module.exports = async (ctx, next) => {
+    const { keyword } = ctx.params;
+    const link = `https://cve.mitre.org/cgi-bin/cvekey.cgi?keyword=${keyword}`;
+    const body = await got(link).text();
+    const $ = cheerio.load(body);
+    const rows = $('#TableWithRules').find('tbody').first().find('tr').get().slice(0, 10);
+    const item = rows.map((row) => {
+        const [cveName, description] = $(row).find('td').get().map($);
+        return {
+            title: cveName.text(),
+            link: 'https://cve.mitre.org' + cveName.find('a').first().attr('href'),
+            description: description.text(),
+            guid: this.link,
+        };
+    });
+    ctx.state.data = {
+        title: `CVE search result ${keyword}`,
+        link,
+        description: this.title,
+        item,
+    };
+    await next();
+};