feat: use html parsing for github stars badge

Author: amio

libs/create-badgen-handler-next.ts

@@ -116,9 +116,13 @@ function simpleDecode (str: string): string {
 }
 
 export class BadgenError {
-  public status: string   // error badge param: status (required)
-  public color: string    // error badge param: color
-  public code: number     // status code for response
+  /** The "status" param for error badge (required) */
+  public status: string
+  /** The "color" param for error badge, red by default */
+  public color: string
+  /** The status code for error badge response, 500 by default */
+  public code: number
+  /** The error message */
   public message: string
 
   constructor ({ status, color = 'grey', code = 500, message = '' }) {

libs/query-html.ts

@@ -0,0 +1,8 @@
+import got from 'got'
+import { parse } from 'node-html-parser'
+
+export default async function queryHTML(url: string, cssSelector: string) {
+  const htmlString = await got(url).text()
+  const root = parse(htmlString)
+  return root.querySelector(cssSelector)
+}

package-lock.json

@@ -34,6 +34,7 @@
     "measurement-protocol": "^0.1.1",
     "millify": "^6.1.0",
     "my-way": "^2.0.0",
+    "node-html-parser": "^6.1.5",
     "original-url": "^1.2.3",
     "semver": "^7.3.5",
     "serve-handler": "^6.1.3",

package.json

@@ -4,6 +4,7 @@ import got from 'libs/got'
 import { restGithub, queryGithub } from 'libs/github'
 import { createBadgenHandler, PathArgs, BadgenError } from 'libs/create-badgen-handler-next'
 import { coverageColor, millify, version } from 'libs/utils'
+import queryHTML from 'libs/query-html'
 
 type DependentsType = 'REPOSITORY' | 'PACKAGE'
 
@@ -297,7 +298,7 @@ const makeRepoQuery = (topic, owner, repo, restArgs) => {
     case 'last-commit':
       queryBody = `
         ${
-          restArgs.ref ? 
+          restArgs.ref ?
             `branch: ref(qualifiedName: "${restArgs.ref}")` :
             'defaultBranchRef'
         } {
@@ -364,10 +365,20 @@ async function repoStats ({topic, owner, repo, ...restArgs}: PathArgs) {
         color: 'blue'
       }
     case 'stars':
-      const { stargazers_count } = await meta({ owner, repo })
+      // const { stargazers_count } = await meta({ owner, repo })
+      const starsElem = await queryHTML(`https://github.com/${owner}/${repo}`, '#repo-stars-counter-star')
+      const stargazers_count = starsElem?.innerText
+
+      if (stargazers_count === undefined) {
+        throw new BadgenError({
+          status: 'not found',
+          message: `Could not find stargazers_count for ${owner}/${repo}`
+        })
+      }
+
       return {
         subject: topic,
-        status: millify(stargazers_count),
+        status: stargazers_count,
         color: 'blue'
       }
     case 'license':
@@ -451,7 +462,7 @@ async function repoStats ({topic, owner, repo, ...restArgs}: PathArgs) {
       return {
         subject: topic,
         status: millify(
-          result.branch ? 
+          result.branch ?
             result.branch.target.history.totalCount :
             result.defaultBranchRef.target.history.totalCount
         ),