From 9b7448753d46791b897ed91d7851c800ee11690d Mon Sep 17 00:00:00 2001
From: Jeff Thompson <jeff@thefirst.org>
Date: Fri, 25 Oct 2024 15:36:05 +0200
Subject: [PATCH] .github/workflows: copy utils folder from berty

Signed-off-by: Jeff Thompson <jeff@thefirst.org>
---
 .../utils/get-self-hosted-runner-status.js    | 115 ++++++++++++++++++
 .../utils/mac-runner-matrix-builder.js        |  47 +++++++
 .github/workflows/utils/variables.json        |  12 ++
 3 files changed, 174 insertions(+)
 create mode 100755 .github/workflows/utils/get-self-hosted-runner-status.js
 create mode 100755 .github/workflows/utils/mac-runner-matrix-builder.js
 create mode 100644 .github/workflows/utils/variables.json

diff --git a/.github/workflows/utils/get-self-hosted-runner-status.js b/.github/workflows/utils/get-self-hosted-runner-status.js
new file mode 100755
index 0000000..269163f
--- /dev/null
+++ b/.github/workflows/utils/get-self-hosted-runner-status.js
@@ -0,0 +1,115 @@
+#!/usr/bin/env node
+const https = require("https");
+
+// Function that request Github API to get self-hosted runners status
+async function getSelfHostedRunnersStatus() {
+	const options = {
+		hostname: "ghrunners-lambda.berty.io",
+		port: 443,
+		method: "GET",
+		timeout: 30000,
+	};
+
+	return new Promise((resolve, reject) => {
+		const req = https.request(options, (res) => {
+			const body = [];
+			res.on("data", (chunk) => body.push(chunk));
+			res.on("end", () => {
+				const resString = Buffer.concat(body).toString();
+
+				if (res.statusCode < 200 || res.statusCode > 299) {
+					return reject(
+						new Error(`HTTP status code ${res.statusCode}: ${resString}`)
+					);
+				}
+				resolve(resString);
+			});
+		});
+
+		req.on("error", (err) => {
+			reject(err);
+		});
+
+		req.on("timeout", () => {
+			req.destroy();
+			reject(new Error("Request timed out"));
+		});
+
+		req.end();
+	});
+}
+
+// Return hom many self-hosted runners are online
+async function getSelfHostedRunnersOnlineCount() {
+	return new Promise((resolve, reject) => {
+		getSelfHostedRunnersStatus()
+			.then((json) => {
+				const stat = JSON.parse(json);
+				const total = stat.runners.length;
+				var count = 0;
+				var desc = "None";
+
+				for (const runner of stat.runners) {
+					if (runner.status === "online") count++;
+				}
+
+				if (count === total && total > 0) desc = "All";
+				else if (count > 0) desc = "Partially";
+
+				resolve(`${desc} (${count}/${total})`);
+			})
+			.catch((err) => {
+				reject(err);
+			});
+	});
+}
+
+// Returns true if at least one self-hosted runner is available
+async function isSelfHostedRunnerAvailable() {
+	try {
+		const json = await getSelfHostedRunnersStatus();
+		const stat = JSON.parse(json);
+
+		for (const runner of stat.runners) {
+			if (runner.status === "online" && runner.busy === false) return true;
+		}
+	} catch (err) {
+		console.error("Github API request self-hosted status error:", err);
+	}
+
+	return false;
+}
+
+// If executed as a script
+if (require.main === module) {
+	const onlineCheck = process.argv[2];
+
+	if (process.argv.length > 3 || (onlineCheck && onlineCheck !== "online")) {
+		console.error(`Usage: ${process.argv[1]} [online]`);
+		process.exit(1);
+	}
+
+	if (onlineCheck) {
+		getSelfHostedRunnersOnlineCount()
+			.then((count) => console.log(count))
+			.catch((err) => {
+				console.error(err);
+				process.exit(1);
+			});
+	} else {
+		getSelfHostedRunnersStatus()
+			.then((json) => {
+				console.log(JSON.stringify(JSON.parse(json), undefined, 2));
+			})
+			.catch((err) => {
+				console.error(err);
+				process.exit(1);
+			});
+	}
+}
+
+// Export module functions
+module.exports = {
+	getSelfHostedRunnersStatus: getSelfHostedRunnersStatus,
+	isSelfHostedRunnerAvailable: isSelfHostedRunnerAvailable,
+};
diff --git a/.github/workflows/utils/mac-runner-matrix-builder.js b/.github/workflows/utils/mac-runner-matrix-builder.js
new file mode 100755
index 0000000..b31e52c
--- /dev/null
+++ b/.github/workflows/utils/mac-runner-matrix-builder.js
@@ -0,0 +1,47 @@
+#!/usr/bin/env node
+const statusGetter = require("./get-self-hosted-runner-status");
+
+// Check parameters
+const mode = process.argv[2] || "self-hosted";
+
+if (
+	!["self-hosted", "github", "optimized"].includes(mode) ||
+	process.argv.length > 3
+) {
+	console.error(`Usage: ${process.argv[1]} [self-hosted | github | optimized]`);
+	process.exit(1);
+}
+
+// Returns matrix according to parameters and self-hosted runner availability
+async function getMatrix() {
+	const runners = {
+		selfhosted: {
+			name: "self-hosted",
+			runner: ["self-hosted", "macOS", "public"],
+			selfhosted: true,
+		},
+		github: {
+			name: "github",
+			runner: ["macos-12"],
+			selfhosted: false,
+		},
+	};
+
+	switch (mode) {
+		case "self-hosted":
+			return [runners.selfhosted];
+		case "github":
+			return [runners.github];
+		case "optimized":
+			return (await statusGetter.isSelfHostedRunnerAvailable())
+				? [runners.selfhosted]
+				: [runners.github];
+	}
+}
+
+getMatrix().then((matrix) => {
+	console.debug("matrix: " + JSON.stringify(matrix, undefined, 2));
+	console.log(
+		"::set-output name=matrix::" + JSON.stringify({ include: matrix })
+	);
+});
diff --git a/.github/workflows/utils/variables.json b/.github/workflows/utils/variables.json
new file mode 100644
index 0000000..b7f0936
--- /dev/null
+++ b/.github/workflows/utils/variables.json
@@ -0,0 +1,12 @@
+{
+  "cache-versions": {
+    "node": "3",
+    "nodeweb": "1",
+    "go": "1",
+    "ruby": "1",
+    "bridgeframework": "1",
+    "pushframework": "1",
+    "xcodeproject": "1",
+    "xcodeworkspace": "1"
+  }
+}