gotesttools-gotestfmt-action/index.js
Andrew Ferrazzutti 8de621c2cf Download artifact for the correct architecture
This adds support for running the action for CPU architectures other
than amd64/x86_64.
2025-10-02 11:39:43 -04:00

124 lines
4.5 KiB
JavaScript

const core = require('@actions/core');
const { Octokit } = require("@octokit/rest");
const fs = require("fs")
const { execSync } = require("child_process")
const { arch: processArch } = require("node:process");
async function downloadRelease(octokit, os, arch, org, repo, release, token) {
const postfix = `_${os}_${arch}.${os === "windows" ? "zip" : "tar.gz"}`;
const tempdir = os === "windows" ? process.env.TEMP + "\\" : "/tmp/";
const extract = os === "windows" ? "tar -xvf" : "tar -xvzf";
const archive = `${tempdir}gotestfmt${postfix}`;
const releaseAssets = await octokit.rest.repos.listReleaseAssets({
owner: org,
repo: repo,
release_id: release.id,
})
for (let asset of releaseAssets.data) {
console.log("Examining release asset " + asset.name + " at " + asset.browser_download_url + " ...")
if (asset.name.endsWith(postfix)) {
console.log("Found binary named " + asset.name + " at " + asset.browser_download_url + " , attempting download...")
if (token) {
execSync(`curl -L -o ${archive} -H "Authorization: Bearer ${token}" ${asset.browser_download_url}`)
} else {
execSync(`curl -L -o ${archive} ${asset.browser_download_url}`)
}
console.log("Unpacking archive file...")
core.addPath(process.env.GITHUB_WORKSPACE)
process.chdir(process.env.GITHUB_WORKSPACE)
execSync(`${extract} "${archive}"`)
console.log("Removing asset archive...")
fs.unlinkSync(archive)
console.log("Successfully set up gotestfmt.")
return
}
}
throw `No release asset matched postfix '${postfix}'.`
}
async function downloadGofmt(octokit, version, versionPrefix, os, arch, org, repo, token) {
if (version !== "") {
if (!version.startsWith(versionPrefix)) {
throw "Specified version " + version + " does not start with required version prefix " + versionPrefix + "."
}
console.log("Downloading gotestfmt version " + version + " from " + org + "/" + repo + "...")
} else {
console.log("Downloading latest stable gotestfmt version starting with " + versionPrefix + " from " + org + "/" + repo + "...")
}
const releases = await octokit.rest.repos.listReleases({
owner: org,
repo: repo
// No pagination added, we are optimistic that there is a stable release within the first 100
// releases.
})
let tries = 0
for (let release of releases.data) {
if ((version !== "" && release.name === version) || (!release.prerelease && release.name.startsWith(versionPrefix))) {
console.log("Found release " + release.name + " matching criteria, attempting to download binary...")
try {
await downloadRelease(octokit, os, arch, org, repo, release, token)
return
} catch (e) {
tries++
if (tries > 3) {
console.log("Binary download failed, tried " + tries + " times, giving up. (" + e + ")")
throw e
}
console.log("Binary download failed, trying next release. (" + e + ")")
}
}
}
console.log("Failed to find a release matching the criteria.")
throw "Failed to find a release matching the criteria."
}
async function determineOS() {
const uname = execSync("uname")
let os = uname.toString().trim().toLowerCase()
if (os.indexOf("msys_nt") === 0)
{
os = "windows";
}
console.log(`Running on OS '${os}'`)
return os
}
function determineArch() {
switch (processArch) {
case "x64":
return "amd64";
default:
return processArch;
}
}
async function main() {
try {
// versionPrefix is the prefix of the version gotestfmt-action supports.
const versionPrefix = "v2."
const token = core.getInput('token');
const version = core.getInput('version');
const org = core.getInput("org")
const repo = core.getInput("repo")
const octokit = new Octokit({
auth: token,
})
const os = await determineOS()
const arch = determineArch()
await downloadGofmt(octokit, version, versionPrefix, os, arch, org, repo, token)
console.log("Setup complete.")
} catch (error) {
console.log("Setup failed.")
core.setFailed(error);
}
}
main()