134 lines
3.4 KiB
TypeScript
134 lines
3.4 KiB
TypeScript
import fetch from "node-fetch";
|
|
import cheerio from "cheerio";
|
|
import { formatParams, Options } from "./formatParams";
|
|
|
|
/**
|
|
* get the contribution grid from a github user page
|
|
*
|
|
* @param userName github user name
|
|
* @param options set the time range: from / to or year
|
|
*/
|
|
export const getGithubUserContribution = async (
|
|
userName: string,
|
|
options: Options = {}
|
|
) => {
|
|
// either use github.com/users/xxxx/contributions for previous years
|
|
// or github.com/xxxx ( which gives the latest update to today result )
|
|
const url =
|
|
"year" in options || "from" in options || "to" in options
|
|
? `https://github.com/users/${userName}/contributions?` +
|
|
formatParams(options)
|
|
: `https://github.com/${userName}`;
|
|
|
|
const res = await fetch(url);
|
|
|
|
if (!res.ok) throw new Error(res.statusText);
|
|
|
|
const resText = await res.text();
|
|
|
|
return parseUserPage(resText);
|
|
};
|
|
|
|
const defaultColorScheme = [
|
|
"#ebedf0",
|
|
"#9be9a8",
|
|
"#40c463",
|
|
"#30a14e",
|
|
"#216e39",
|
|
];
|
|
|
|
const parseUserPage = (content: string) => {
|
|
const $ = cheerio.load(content);
|
|
|
|
//
|
|
// parse colorScheme
|
|
const colorScheme = [...defaultColorScheme];
|
|
const colorSchemeMap: Record<string, number> = Object.fromEntries(
|
|
defaultColorScheme.map((color, i) => [color, i])
|
|
);
|
|
$("ul.legend > li")
|
|
.toArray()
|
|
.forEach((x, i) => {
|
|
const bgColor = x.attribs.style.match(/background\-color: +(.+)/)![1]!;
|
|
if (bgColor) {
|
|
const color = bgColor.replace(/\s/g, "");
|
|
colorSchemeMap[color] = i;
|
|
|
|
if (!color.startsWith("var(--")) colorScheme[i] = color;
|
|
}
|
|
});
|
|
|
|
//
|
|
// parse cells
|
|
const rawCells = $(".js-calendar-graph rect[data-count]")
|
|
.toArray()
|
|
.map((x) => {
|
|
const color = x.attribs.fill.trim();
|
|
const count = +x.attribs["data-count"];
|
|
const date = x.attribs["data-date"];
|
|
|
|
const colorIndex = colorSchemeMap[color];
|
|
|
|
if (colorIndex === -1) throw new Error("could not map the cell color");
|
|
|
|
return {
|
|
svgPosition: getSvgPosition(x),
|
|
color: colorScheme[colorIndex],
|
|
count,
|
|
date,
|
|
};
|
|
});
|
|
|
|
const xMap: Record<number, true> = {};
|
|
const yMap: Record<number, true> = {};
|
|
rawCells.forEach(({ svgPosition: { x, y } }) => {
|
|
xMap[x] = true;
|
|
yMap[y] = true;
|
|
});
|
|
|
|
const xRange = Object.keys(xMap)
|
|
.map((x) => +x)
|
|
.sort((a, b) => +a - +b);
|
|
const yRange = Object.keys(yMap)
|
|
.map((x) => +x)
|
|
.sort((a, b) => +a - +b);
|
|
|
|
const cells = rawCells.map(({ svgPosition, ...c }) => ({
|
|
...c,
|
|
x: xRange.indexOf(svgPosition.x),
|
|
y: yRange.indexOf(svgPosition.y),
|
|
}));
|
|
|
|
return { cells, colorScheme };
|
|
};
|
|
|
|
// returns the position of the svg elements, accounting for it's transform and it's parent transform
|
|
// ( only accounts for translate transform )
|
|
const getSvgPosition = (e: cheerio.Element): { x: number; y: number } => {
|
|
if (!e || e.tagName === "svg") return { x: 0, y: 0 };
|
|
|
|
const p = getSvgPosition(e.parent);
|
|
|
|
if (e.attribs.x) p.x += +e.attribs.x;
|
|
if (e.attribs.y) p.y += +e.attribs.y;
|
|
|
|
if (e.attribs.transform) {
|
|
const m = e.attribs.transform.match(
|
|
/translate\( *([\.\d]+) *, *([\.\d]+) *\)/
|
|
);
|
|
|
|
if (m) {
|
|
p.x += +m[1];
|
|
p.y += +m[2];
|
|
}
|
|
}
|
|
|
|
return p;
|
|
};
|
|
|
|
type ThenArg<T> = T extends PromiseLike<infer U> ? U : T;
|
|
|
|
export type Res = ThenArg<ReturnType<typeof getGithubUserContribution>>;
|
|
|
|
export type Cell = Res["cells"][number];
|