File download from mirror station path

lib/index.js

@@ -4,9 +4,12 @@ import {fileURLToPath} from 'node:url';
 import BinWrapper from 'bin-wrapper';
 
 const pkg = JSON.parse(fs.readFileSync(new URL('../package.json', import.meta.url)));
-const url = `https://raw.githubusercontent.com/imagemin/pngquant-bin/v${pkg.version}/vendor/`;
+/**
+ * @default `https://raw.githubusercontent.com/imagemin/pngquant-bin/v${pkg.version}/vendor/`
+ */
+const url = getBinaryUrl(pkg.version, pkg)
 
-const binWrapper = new BinWrapper()
+module.exports = new BinWrapper()
 	.src(`${url}macos/pngquant`, 'darwin')
 	.src(`${url}linux/x86/pngquant`, 'linux', 'x86')
 	.src(`${url}linux/x64/pngquant`, 'linux', 'x64')

package.json

@@ -30,7 +30,7 @@
 	},
 	"scripts": {
 		"postinstall": "node lib/install.js",
-		"test": "xo && ava --timeout=120s"
+		"test": "ava --timeout=120s && node ./utils.test.js"
 	},
 	"files": [
 		"cli.js",

utils.js

@@ -0,0 +1,35 @@
+/**
+ * Determine the URL to fetch binary file from.
+ * By default fetch from the pngquant-bin distribution
+ * site on GitHub.
+ *
+ * The default URL can be overridden using
+ * the environment variable PNGQUANT_BIN_BINARY_SITE,
+ * .npmrc variable pngquant_bin_binary_site or
+ *
+ * The URL should to the mirror of the repository
+ * laid out as follows:
+ *
+ * PNGQUANT_BIN_BINARY_SITE/
+ *
+ *  vX.X.X
+ *  vX.X.X/macos/pngquant
+ *
+ * @example `https://raw.githubusercontent.com/imagemin/pngquant-bin/v${pkg.version}/vendor/`;
+ */
+module.exports = function getBinaryUrl(version, pgkConfig = {}) {
+	if (!version) {
+		return ''
+	}
+
+  let site = process.env.PNGQUANT_BIN_BINARY_SITE  ||
+		process.env.npm_config_pngquant_bin_binary_site ||
+		(pgkConfig.pngquantBinConfig && pgkConfig.pngquantBinConfig.binarySite) ||
+	'https://raw.githubusercontent.com/imagemin/pngquant-bin';
+
+	if (site[site.length - 1] === '/') {
+		site = site.slice(0, -1)
+	}
+
+  return `${site}/v${version}/vendor/`
+}

utils.test.js

@@ -0,0 +1,36 @@
+const assert = require('assert')
+const getBinaryUrl = require('./utils')
+
+function main() {
+	const version = '1.1.1'
+	const site =  'https://a.b.c/folder'
+	const pkg = {
+		pngquantBinConfig: {
+			binarySite: site,
+		},
+	}
+	const pkg2 = {
+		pngquantBinConfig: {
+			binarySite: `${site}/`,
+		},
+	}
+
+
+	assert(getBinaryUrl(version) ===
+	`https://raw.githubusercontent.com/imagemin/pngquant-bin/v${version}/vendor/`, 'fail1')
+	assert(getBinaryUrl(version, pkg) ===  `https://a.b.c/folder/v${version}/vendor/`, 'fail2')
+	assert(getBinaryUrl(version, pkg2) ===  `https://a.b.c/folder/v${version}/vendor/`, 'fail2')
+
+	// test PNGQUANT_BIN_BINARY_SITE
+	const cacheEnv = { ...process.env }
+	process.env.PNGQUANT_BIN_BINARY_SITE = site
+	assert(getBinaryUrl(version) ===  `https://a.b.c/folder/v${version}/vendor/`, 'fail2')
+	process.env = { ...cacheEnv }
+
+	// test .npmrc npm_config_pngquant_bin_binary_site
+	process.env.npm_config_pngquant_bin_binary_site = site
+	assert(getBinaryUrl(version) ===  `https://a.b.c/folder/v${version}/vendor/`, 'fail2')
+	process.env = { ...cacheEnv }
+}
+
+main()