first commit

src/main.ts

@@ -0,0 +1,99 @@
+import { program } from 'commander'
+import { fetchRepos, fetchContributorsInfo } from './fetch'
+import { checkContribsPersistence, saveContribsPersistence } from './persistence'
+import { saveSVG as saveSVG } from './save-svg'
+import { generateContributorsSVGFile } from './svg-codegen'
+import { getRepoName } from './utils'
+import type { CliOptions, RepoInfo, ContributorsInfo } from './types'
+
+
+async function main() {
+  const { Github_token: defaultToken, Github_owner: defaultOwner } = process.env
+  const defaultRepoName = await getRepoName()
+  const GITHUBReg = /https:\/\/github.com\/([\w\-_]+)\/([\w\-_]+)/
+  let urlInfo = null
+  program
+    .name('gh-contrib-svg')
+    .arguments('[url]')
+    .option('-t, --token <token>', 'Personal GitHub token', defaultToken)
+    .option('-o, --owner <owner>', 'Repo owner name', defaultOwner)
+    .option('-r, --repo <repo>', 'GitHub repo path', defaultRepoName)
+    .option('-s, --size <size>', 'Single avatar block size (pixel)', "120")
+    .option('-w, --width <width>', 'Output image width (pixel)', "1000")
+    .option('-c, --count <count>', 'Avatar count in one line', "8")
+    .action((url) => {
+      if (!url) return
+      const match = url.match(GITHUBReg)
+      if (!match)
+        throw new Error('Invalid GitHub Repo URL')
+      const [_, owner, repo] = match
+      urlInfo = {
+        owner,
+        repo
+      }
+    })
+    .parse(process.argv)
+  const options = Object.assign(program.opts(), urlInfo)
+  const { token, repo, owner, size: avatarBlockSize, width, count: lineCount } = options as CliOptions
+
+  if (token && owner) {
+    let repos: RepoInfo[] = []
+    let identifier = 'contributor_'
+    if (repo) {
+      // fetch <owner>/<repo> contributors info
+      identifier += repo;
+      repos.push({ owner, repo })
+    } else {
+      // fetch <owner> contributors info
+      identifier += owner;
+      const ownerRepos = await fetchRepos({ token, owner })
+      if (!ownerRepos || ownerRepos.length === 0) {
+        throw new Error('No repos found')
+      }
+      repos = [...repos, ...ownerRepos]
+    }
+    const startTime = performance.now()
+    const allContributorsInfos = new Map<string, ContributorsInfo>()
+    for (const { owner, repo } of repos) {
+      const contributorsInfos = await fetchContributorsInfo({ token, repo, owner });
+      contributorsInfos.forEach((info, username) => {
+        allContributorsInfos.set(username, info);
+      });
+    }
+
+    // sort contributors by commit count and pull request count
+    const sortedContributors = [...allContributorsInfos.entries()]
+      .sort(([, userInfoA], [, userInfoB]) => {
+        const countA = userInfoA.commitURLs.length
+        const countB = userInfoB.commitURLs.length
+        return countB - countA
+      })
+    const contribUserNames = sortedContributors.map(([userName,]) => userName);
+    checkContribsPersistence(
+      contribUserNames,
+      identifier
+    )
+
+    const svgString = await generateContributorsSVGFile({
+      imgWidth: Number(width),
+      blockSize: Number(avatarBlockSize),
+      lineCount: Number(lineCount),
+    }, new Map(sortedContributors))
+
+    saveSVG(svgString, identifier);
+    saveContribsPersistence(
+      contribUserNames,
+      identifier
+    )
+
+    const endTime = performance.now()
+    console.log(`Time cost: ${Math.round((endTime - startTime) / 1000)}s`)
+  } else {
+    if (!token)
+      throw new Error('Personal GitHub token is required')
+    if (!owner)
+      throw new Error('GitHub repo path is required')
+  }
+}
+
+main()