Skip to content

Commit

Permalink
✨ Migrate action helper functions
Browse files Browse the repository at this point in the history
  • Loading branch information
AnandChowdhary committed Oct 13, 2020
1 parent d11b709 commit 27d693d
Show file tree
Hide file tree
Showing 4 changed files with 595 additions and 0 deletions.
135 changes: 135 additions & 0 deletions src/graph.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,135 @@
import { Octokit } from "@octokit/rest";
import slugify from "@sindresorhus/slugify";
import { readFile, writeFile, ensureDir, writeJson, readJson } from "fs-extra";
import { safeLoad } from "js-yaml";
import { join } from "path";
import { CanvasRenderService } from "chartjs-node-canvas";

const canvasRenderService = new CanvasRenderService(600, 400);

export const generateGraphs = async () => {
const config = safeLoad(await readFile(join(".", ".upptimerc.yml"), "utf8")) as {
sites: { name: string; url: string }[];
owner: string;
repo: string;
userAgent?: string;
PAT?: string;
assignees?: string[];
};
const owner = config.owner;
const repo = config.repo;

const octokit = new Octokit({
auth: config.PAT || process.env.GH_PAT || process.env.GITHUB_TOKEN,
userAgent: config.userAgent || process.env.USER_AGENT || "KojBot",
});

await ensureDir(join(".", "graphs"));

for await (const site of config.sites) {
const slug = slugify(site.name);

let uptime = 0;
let responseTime = 0;
try {
const api: {
slug: string;
uptime: string;
time: number;
}[] = await readJson(join(".", "history", "summary.json"));
const item = api.find((site) => site.slug === slug);
if (item) {
uptime = parseFloat(item.uptime);
responseTime = item.time;
}
} catch (error) {}
await ensureDir(join(".", "api", slug));
await writeJson(join(".", "api", slug, "uptime.json"), {
schemaVersion: 1,
label: "uptime",
message: `${uptime}%`,
color:
uptime > 95
? "brightgreen"
: uptime > 90
? "green"
: uptime > 85
? "yellowgreen"
: uptime > 80
? "yellow"
: uptime > 75
? "orange"
: "red",
});
await writeJson(join(".", "api", slug, "response-time.json"), {
schemaVersion: 1,
label: "response time",
message: `${responseTime} ms`,
color:
responseTime < 200
? "brightgreen"
: responseTime < 400
? "green"
: responseTime < 600
? "yellowgreen"
: responseTime < 800
? "yellow"
: responseTime < 1000
? "orange"
: "red",
});

const history = await octokit.repos.listCommits({
owner,
repo,
path: `history/${slug}.yml`,
per_page: 10,
});
if (!history.data.length) continue;
const data: [number, string][] = history.data
.filter(
(item) =>
item.commit.message.includes(" in ") &&
Number(item.commit.message.split(" in ")[1].split("ms")[0]) !== 0
)
.map((item) => [
Number(item.commit.message.split(" in ")[1].split("ms")[0]),
String(item.commit.author.date),
]);
const image = await canvasRenderService.renderToBuffer({
type: "line",
data: {
labels: [1, ...data.map((item) => item[1])],
datasets: [
{
backgroundColor: "#89e0cf",
borderColor: "#1abc9c",
data: [1, ...data.map((item) => item[0])],
},
],
},
options: {
legend: { display: false },
scales: {
xAxes: [
{
display: false,
gridLines: {
display: false,
},
},
],
yAxes: [
{
display: false,
gridLines: {
display: false,
},
},
],
},
},
});
await writeFile(join(".", "graphs", `${slug}.png`), image);
}
};
3 changes: 3 additions & 0 deletions src/readme.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
import { generateSummary } from "./summary";

generateSummary();
176 changes: 176 additions & 0 deletions src/summary.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,176 @@
import { Octokit } from "@octokit/rest";
import slugify from "@sindresorhus/slugify";
import { readFile } from "fs-extra";
import { safeLoad } from "js-yaml";
import { join } from "path";

export const generateSummary = async () => {
const config = safeLoad(await readFile(join(".", ".upptimerc.yml"), "utf8")) as {
sites: { name: string; url: string }[];
owner: string;
repo: string;
userAgent?: string;
PAT?: string;
assignees?: string[];
};
const owner = config.owner;
const repo = config.repo;

const octokit = new Octokit({
auth: config.PAT || process.env.GH_PAT || process.env.GITHUB_TOKEN,
userAgent: config.userAgent || process.env.USER_AGENT || "KojBot",
});

let readmeContent = await readFile(join(".", "README.md"), "utf8");

const startText = readmeContent.split("<!--start: status pages-->")[0];
const endText = readmeContent.split("<!--end: status pages-->")[1];

const pageStatuses: Array<{
url: string;
status: string;
slug: string;
time: number;
uptime: string;
name: string;
}> = [];

let numberOfDown = 0;
for await (const site of config.sites) {
const slug = slugify(site.name);
let startTime = new Date().toISOString();
try {
startTime =
(await readFile(join(".", "history", `${slug}.yml`), "utf8"))
.split("\n")
.find((line) => line.toLocaleLowerCase().includes("- starttime"))
?.split("startTime:")[1]
.trim() || new Date().toISOString();
} catch (error) {}
let secondsDown = 0;
const history = await octokit.repos.listCommits({
owner,
repo,
path: `history/${slug}.yml`,
per_page: 100,
});
const issues = await octokit.issues.listForRepo({
owner,
repo,
labels: slug,
filter: "all",
per_page: 100,
});
issues.data.forEach((issue) => {
if (issue.closed_at)
secondsDown += Math.floor(
(new Date(issue.closed_at).getTime() - new Date(issue.created_at).getTime()) / 1000
);
else
secondsDown += Math.floor(
(new Date().getTime() - new Date(issue.created_at).getTime()) / 1000
);
});
const uptime = (
100 -
100 * (secondsDown / ((new Date().getTime() - new Date(startTime).getTime()) / 1000))
).toFixed(2);
if (!history.data.length) continue;
const averageTime =
history.data
.filter(
(item) =>
item.commit.message.includes(" in ") &&
Number(item.commit.message.split(" in ")[1].split("ms")[0]) !== 0 &&
!isNaN(Number(item.commit.message.split(" in ")[1].split("ms")[0]))
)
.map((item) => Number(item.commit.message.split(" in ")[1].split("ms")[0]))
.reduce((p, c) => p + c, 0) / history.data.length;
const status = history.data[0].commit.message.split(" ")[0].includes("🟩") ? "up" : "down";
pageStatuses.push({
name: site.name,
url: site.url,
slug,
status,
uptime,
time: Math.floor(averageTime),
});
if (status === "down") {
numberOfDown++;
}
}

if (readmeContent.includes("<!--start: status pages-->")) {
readmeContent = `${startText}<!--start: status pages-->
| URL | Status | History | Response Time | Uptime |
| --- | ------ | ------- | ------------- | ------ |
${pageStatuses
.map(
(page) =>
`| ${page.url.startsWith("$") ? page.name : `[${page.name}](${page.url})`} | ${
page.status === "up" ? "🟩 Up" : "🟥 Down"
} | [${page.slug}.yml](https://github.com/${owner}/${repo}/commits/master/history/${
page.slug
}.yml) | <img alt="Response time graph" src="./graphs/${page.slug}.png" height="20"> ${
page.time
}ms | ![Uptime ${
page.uptime
}%](https://img.shields.io/endpoint?url=https%3A%2F%2Fraw.githubusercontent.com%2F${owner}%2F${repo}%2Fmaster%2Fapi%2F${
page.slug
}%2Fuptime.json)`
)
.join("\n")}
<!--end: status pages-->${endText}`;
}

// Add live status line
readmeContent = readmeContent
.split("\n")
.map((line) => {
if (line.includes("<!--live status-->")) {
line = `${line.split("<!--live status-->")[0]}<!--live status--> **${
numberOfDown === 0
? "🟩 All systems operational"
: numberOfDown === config.sites.length
? "🟥 Complete outage"
: "🟨 Partial outage"
}**`;
}
return line;
})
.join("\n");

const sha = (
await octokit.repos.getContent({
owner,
repo,
path: "README.md",
})
).data.sha;
await octokit.repos.createOrUpdateFileContents({
owner,
repo,
path: "README.md",
message: ":pencil: Update summary in README [skip ci]",
content: Buffer.from(readmeContent).toString("base64"),
sha,
});
let summarySha: string | undefined = undefined;
try {
summarySha = (
await octokit.repos.getContent({
owner,
repo,
path: "history/summary.json",
})
).data.sha;
} catch (error) {}
await octokit.repos.createOrUpdateFileContents({
owner,
repo,
path: "history/summary.json",
message: ":card_file_box: Update status summary [skip ci]",
content: Buffer.from(JSON.stringify(pageStatuses, null, 2)).toString("base64"),
sha: summarySha,
});
};
Loading

0 comments on commit 27d693d

Please sign in to comment.