Use fetch

Author: stevenjoezhang

lib/comment.js

@@ -1,16 +1,12 @@
 const fs = require('fs');
 const path = require('path');
-const https = require('https');
 const yaml = require('js-yaml');
 const { spawnSync } = require('child_process');
 const ssri = require('ssri');
 const { getVendors } = require('hexo-theme-next/scripts/events/lib/utils');
 const yargs = require('yargs/yargs');
 const { hideBin } = require('yargs/helpers');
 
-// Process won't quit if keepAlive
-const agent = new https.Agent({ keepAlive: false });
-
 const argv = yargs(hideBin(process.argv))
   .option('diff', {
     describe: 'Enable diff mode',
@@ -26,41 +22,18 @@ const dependencies = yaml.load(vendorsFile);
 
 const newPlugins = require('../package.json').dependencies;
 
-async function request(url) {
-  return new Promise((resolve, reject) => {
-    const req = https.get(url, { agent }, response => {
-      resolve(response.statusCode);
-    });
-    req.on('error', error => {
-      reject(error.status);
-    });
-  });
-}
-
-async function checkIntegrity(url) {
-  return new Promise((resolve, reject) => {
-    const req = https.get(url, response => {
-      ssri.fromStream(response, { algorithms: ['sha256'] })
-        .then(integrity => integrity.toString())
-        .then(resolve);
-    });
-    req.on('error', error => {
-      reject(error.status);
-    });
-  });
-}
-
 async function formatTable(name, links, groundTruth) {
   let content = '';
   for (const [key, url] of Object.entries(links)) {
     if (!['jsdelivr', 'unpkg', 'cdnjs'].includes(key)) continue;
-    const statusCode = await request(url);
-    let integrityMatch;
+    const response = await fetch(url);
+    const statusCode = response.status;
+    let integrityMatch = '❌';
     if (statusCode === 200) {
-      const integrity = await checkIntegrity(url);
-      integrityMatch = integrity === groundTruth ? '✅' : '❌';
-    } else {
-      integrityMatch = '❌';
+      const text = await response.text();
+      if (ssri.checkData(text, groundTruth, { algorithms: ['sha256'] })) {
+        integrityMatch = '✅';
+      }
     }
     content += `| [${key}](${url}) | ${statusCode === 200 ? '✅' : '❌'} ${statusCode} | ${integrityMatch} |\n`;
   }