Files
metrics/source/plugins/starlists/index.mjs
2022-03-15 01:01:23 +01:00

106 lines
4.8 KiB
JavaScript

//Setup
export default async function({login, q, imports, data, account}, {enabled = false} = {}) {
//Plugin execution
try {
//Check if plugin is enabled and requirements are met
if ((!enabled) || (!q.starlists))
return null
//Load inputs
let {limit, ignored, only, "limit.repositories":_limit, languages, "limit.languages":_limit_languages, "shuffle.repositories":_shuffle} = imports.metadata.plugins.starlists.inputs({data, account, q})
ignored = ignored.map(imports.stripemojis)
only = only.map(imports.stripemojis)
//Start puppeteer and navigate to star lists
console.debug(`metrics/compute/${login}/plugins > starlists > starting browser`)
const browser = await imports.puppeteer.launch()
console.debug(`metrics/compute/${login}/plugins > starlists > started ${await browser.version()}`)
const page = await browser.newPage()
//Fetch star lists
console.debug(`metrics/compute/${login}/plugins > starlists > fetching lists`)
await page.goto(`https://github.com/${login}?tab=stars`)
let lists = (await page.evaluate(login => [...document.querySelectorAll(`[href^='/stars/${login}/lists']`)].map(element => ({
link:element.href,
name:element.querySelector("h3")?.innerText ?? "",
description:element.querySelector("span")?.innerText ?? "",
count:Number(element.querySelector("div")?.innerText.match(/(?<count>\d+)/)?.groups.count),
repositories:[],
})), login
))
const count = lists.length
console.debug(`metrics/compute/${login}/plugins > starlists > found [${lists.map(({name}) => name)}]`)
lists = lists
.filter(({name}) => {
name = imports.stripemojis(name ?? "").trim().toLocaleLowerCase()
if (only.length)
return only.includes(name)
return !ignored.includes(name)
})
.slice(0, limit)
console.debug(`metrics/compute/${login}/plugins > starlists > extracted ${lists.length} lists`)
//Compute lists content
const colors = {}
for (const list of lists) {
//Fetch star list content
console.debug(`metrics/compute/${login}/plugins > starlists > fetching ${list.name}`)
const repositories = []
for (let i = 1; i <= (languages ? 100 : 1); i++) {
console.debug(`metrics/compute/${login}/plugins > starlists > fetching page ${i}`)
await page.goto(`${list.link}?page=${i}`)
repositories.push(
...await page.evaluate(() => [...document.querySelectorAll("#user-list-repositories > div:not(.paginate-container)")].map(element => ({
name:element.querySelector("div:first-child")?.innerText.replace(" / ", "/") ?? "",
description:element.querySelector(".py-1")?.innerText ?? "",
language:{
name:element.querySelector("[itemprop='programmingLanguage']")?.innerText ?? "",
color:element.querySelector(".repo-language-color")?.style?.backgroundColor?.match(/\d+/g)?.map(x => Number(x).toString(16).padStart(2, "0")).join("") ?? null,
},
stargazers:Number(element.querySelector("[href$='/stargazers']")?.innerText.trim().replace(/[^\d]/g, "") ?? NaN),
forks:Number(element.querySelector("[href$='/network/members']")?.innerText.trim().replace(/[^\d]/g, "") ?? NaN),
}))
),
)
if (await page.evaluate(() => document.querySelector(".next_page.disabled"))) {
console.debug(`metrics/compute/${login}/plugins > starlists > reached last page`)
break
}
}
list.repositories.push(...repositories)
if (_shuffle)
list.repositories = imports.shuffle(list.repositories)
//Compute languages statistics
if (languages) {
list.languages = {}
for (const {language:{name, color}} of repositories) {
if (name)
list.languages[name] = (list.languages[name] ?? 0) + 1
if (color)
colors[name] = color
}
list.languages = Object.entries(list.languages).sort((a, b) => b[1] - a[1]).slice(0, _limit_languages || Infinity)
const visible = list.languages.map(([_, value]) => value).reduce((a, b) => a + b, 0)
list.languages = list.languages.map(([name, value]) => ({name, value, color:name in colors ? `#${colors[name]}` : null, x:0, p:value / visible}))
for (let i = 1; i < list.languages.length; i++)
list.languages[i].x = (list.languages[i - 1]?.x ?? 0) + (list.languages[i - 1]?.value ?? 0) / visible
}
//Limit repositories
list.repositories = list.repositories.slice(0, _limit)
}
//Close browser
console.debug(`metrics/compute/${login}/plugins > starlists > closing browser`)
await browser.close()
//Results
return {lists, count}
}
//Handle errors
catch (error) {
throw {error:{message:"An error occured", instance:error}}
}
}