update

Author: awxiaoxian2020

src/utilities/fetch-package-readmes.mjs

@@ -2,9 +2,9 @@ import _ from 'lodash';
 import fs from 'fs';
 import path from 'path';
 import { promisify } from 'util';
-import mkdirp from 'mkdirp';
-import fetch from 'node-fetch';
+import { mkdirp } from 'mkdirp';
 import { fileURLToPath } from 'url';
+import api from './githubAPI.mjs';
 
 import yamlHeadmatter from './yaml-headmatter.mjs';
 import processReadme from './process-readme.mjs';
@@ -13,7 +13,6 @@ const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
 
 const writeFile = promisify(fs.writeFile);
-const rename = promisify(fs.rename);
 const readFile = promisify(fs.readFile);
 const cwd = process.cwd();
 
@@ -32,28 +31,35 @@ const loaderGroup = {
   'style-loader': 'CSS',
   'stylus-loader': 'CSS',
 };
+const communityPackages = [{
+  name: 'svg-chunk-webpack-plugin',
+  contributors: ['yoriiis', 'alexander-akait']
+}];
 
 async function main() {
   for (const type of types) {
     const outputDir = pathMap[type];
 
     await mkdirp(outputDir);
 
+    /** @type string[] */
     const repos = JSON.parse(
       await readFile(path.resolve(__dirname, `../../repositories/${type}.json`))
     );
 
     for (const repo of repos) {
-      const [, packageName] = repo.split('/');
-      const url = `https://raw.githubusercontent.com/${repo}/master/README.md`;
+      const [owner, packageName] = repo.split('/');
+
+      const response = await api.repos.get({
+        owner,
+        repo: packageName,
+      });
+
+      const defaultBranch =  response.data.default_branch;
+      const url = `https://raw.githubusercontent.com/${repo}/${defaultBranch}/README.md`;
       const htmlUrl = `https://github.com/${repo}`;
-      const editUrl = `${htmlUrl}/edit/master/README.md`;
-<<<<<<< HEAD
+      const editUrl = `${htmlUrl}/edit/${defaultBranch}/README.md`;
       const fileName = path.resolve(outputDir, `_${packageName}.mdx`);
-=======
-      const fileName = path.resolve(outputDir, `${packageName}.mdx`);
-      const mdxFileName = path.resolve(outputDir, `${packageName}.mdx`);
->>>>>>> d054d4656cbfb5228fc96f5de26d4f0ce349f742
 
       let title = packageName;
 
@@ -67,27 +73,29 @@ async function main() {
       let headmatter;
 
       if (type === 'plugins') {
+        let group = 'webpack contrib';
+        let contributors = [];
+        const packageFromCommunity = communityPackages.find((item) => item.name === packageName);
+        if (packageFromCommunity) {
+            group = 'Community';
+            contributors = packageFromCommunity.contributors;
+        }
         headmatter = yamlHeadmatter({
           title: title,
-          group: 'webpack contrib',
+          group,
+          contributors,
           source: url,
           edit: editUrl,
           repo: htmlUrl,
-<<<<<<< HEAD
-          thirdParty: true
-=======
->>>>>>> d054d4656cbfb5228fc96f5de26d4f0ce349f742
+          thirdParty: true,
         });
       } else {
         let basic = {
           title: title,
           source: url,
           edit: editUrl,
           repo: htmlUrl,
-<<<<<<< HEAD
-          thirdParty: true
-=======
->>>>>>> d054d4656cbfb5228fc96f5de26d4f0ce349f742
+          thirdParty: true,
         };
 
         if (loaderGroup[packageName]) {
@@ -96,12 +104,16 @@ async function main() {
         headmatter = yamlHeadmatter(basic);
       }
 
-      const response = await fetch(url);
-      const content = await response.text();
+      const { data: content } = await api.repos.getReadme({
+        owner,
+        repo: packageName,
+        mediaType: {
+          format: 'raw',
+        },
+      });
       const body = processReadme(content, { source: url });
       await writeFile(fileName, headmatter + body);
-      await rename(fileName, mdxFileName);
-      console.log('Generated:', path.relative(cwd, mdxFileName));
+      console.log('Generated:', path.relative(cwd, fileName));
     }
   }
 }