Update BinaryManager JSDocs, move validVersions to required binary objects

This commit is contained in:
advplyr 2024-02-17 17:40:33 -06:00
parent a119b05d85
commit d9e7f5d133
2 changed files with 63 additions and 19 deletions

View File

@ -11,16 +11,15 @@ const fileUtils = require('../utils/fileUtils')
class BinaryManager { class BinaryManager {
defaultRequiredBinaries = [ defaultRequiredBinaries = [
{ name: 'ffmpeg', envVariable: 'FFMPEG_PATH' }, { name: 'ffmpeg', envVariable: 'FFMPEG_PATH', validVersions: ['5.1', '6'] },
{ name: 'ffprobe', envVariable: 'FFPROBE_PATH' } { name: 'ffprobe', envVariable: 'FFPROBE_PATH', validVersions: ['5.1', '6'] }
] ]
goodVersions = [ '5.1', '6' ]
constructor(requiredBinaries = this.defaultRequiredBinaries) { constructor(requiredBinaries = this.defaultRequiredBinaries) {
this.requiredBinaries = requiredBinaries this.requiredBinaries = requiredBinaries
this.mainInstallPath = process.pkg ? path.dirname(process.execPath) : global.appRoot this.mainInstallPath = process.pkg ? path.dirname(process.execPath) : global.appRoot
this.altInstallPath = global.ConfigPath this.altInstallPath = global.ConfigPath
this.initialized = false
this.exec = exec this.exec = exec
} }
@ -31,30 +30,45 @@ class BinaryManager {
await this.removeOldBinaries(missingBinaries) await this.removeOldBinaries(missingBinaries)
await this.install(missingBinaries) await this.install(missingBinaries)
const missingBinariesAfterInstall = await this.findRequiredBinaries() const missingBinariesAfterInstall = await this.findRequiredBinaries()
if (missingBinariesAfterInstall.length != 0) { if (missingBinariesAfterInstall.length) {
Logger.error(`[BinaryManager] Failed to find or install required binaries: ${missingBinariesAfterInstall.join(', ')}`) Logger.error(`[BinaryManager] Failed to find or install required binaries: ${missingBinariesAfterInstall.join(', ')}`)
process.exit(1) process.exit(1)
} }
this.initialized = true this.initialized = true
} }
/**
* Remove old/invalid binaries in main or alt install path
*
* @param {string[]} binaryNames
*/
async removeOldBinaries(binaryNames) { async removeOldBinaries(binaryNames) {
for (const binaryName of binaryNames) { for (const binaryName of binaryNames) {
const executable = this.getExecutableFileName(binaryName) const executable = this.getExecutableFileName(binaryName)
const mainInstallPath = path.join(this.mainInstallPath, executable) const mainInstallPath = path.join(this.mainInstallPath, executable)
const altInstallPath = path.join(this.altInstallPath, executable) if (await fs.pathExists(mainInstallPath)) {
Logger.debug(`[BinaryManager] Removing old binaries: ${mainInstallPath}, ${altInstallPath}`) Logger.debug(`[BinaryManager] Removing old binary: ${mainInstallPath}`)
await fs.remove(mainInstallPath) await fs.remove(mainInstallPath)
}
const altInstallPath = path.join(this.altInstallPath, executable)
if (await fs.pathExists(altInstallPath)) {
Logger.debug(`[BinaryManager] Removing old binary: ${altInstallPath}`)
await fs.remove(altInstallPath) await fs.remove(altInstallPath)
} }
} }
}
/**
* Find required binaries and return array of binary names that are missing
*
* @returns {Promise<string[]>}
*/
async findRequiredBinaries() { async findRequiredBinaries() {
const missingBinaries = [] const missingBinaries = []
for (const binary of this.requiredBinaries) { for (const binary of this.requiredBinaries) {
const binaryPath = await this.findBinary(binary.name, binary.envVariable) const binaryPath = await this.findBinary(binary.name, binary.envVariable, binary.validVersions)
if (binaryPath) { if (binaryPath) {
Logger.info(`[BinaryManager] Found good ${binary.name} at ${binaryPath}`) Logger.info(`[BinaryManager] Found valid binary ${binary.name} at ${binaryPath}`)
if (process.env[binary.envVariable] !== binaryPath) { if (process.env[binary.envVariable] !== binaryPath) {
Logger.info(`[BinaryManager] Updating process.env.${binary.envVariable}`) Logger.info(`[BinaryManager] Updating process.env.${binary.envVariable}`)
process.env[binary.envVariable] = binaryPath process.env[binary.envVariable] = binaryPath
@ -67,40 +81,70 @@ class BinaryManager {
return missingBinaries return missingBinaries
} }
async findBinary(name, envVariable) { /**
* Find absolute path for binary
*
* @param {string} name
* @param {string} envVariable
* @param {string[]} [validVersions]
* @returns {Promise<string>} Path to binary
*/
async findBinary(name, envVariable, validVersions = []) {
const executable = this.getExecutableFileName(name) const executable = this.getExecutableFileName(name)
// 1. check path specified in environment variable
const defaultPath = process.env[envVariable] const defaultPath = process.env[envVariable]
if (await this.isBinaryGood(defaultPath)) return defaultPath if (await this.isBinaryGood(defaultPath, validVersions)) return defaultPath
// 2. find the first instance of the binary in the PATH environment variable
const whichPath = which.sync(executable, { nothrow: true }) const whichPath = which.sync(executable, { nothrow: true })
if (await this.isBinaryGood(whichPath)) return whichPath if (await this.isBinaryGood(whichPath, validVersions)) return whichPath
// 3. check main install path (binary root dir)
const mainInstallPath = path.join(this.mainInstallPath, executable) const mainInstallPath = path.join(this.mainInstallPath, executable)
if (await this.isBinaryGood(mainInstallPath)) return mainInstallPath if (await this.isBinaryGood(mainInstallPath, validVersions)) return mainInstallPath
// 4. check alt install path (/config)
const altInstallPath = path.join(this.altInstallPath, executable) const altInstallPath = path.join(this.altInstallPath, executable)
if (await this.isBinaryGood(altInstallPath)) return altInstallPath if (await this.isBinaryGood(altInstallPath, validVersions)) return altInstallPath
return null return null
} }
async isBinaryGood(binaryPath) { /**
* Check binary path exists and optionally check version is valid
*
* @param {string} binaryPath
* @param {string[]} [validVersions]
* @returns {Promise<boolean>}
*/
async isBinaryGood(binaryPath, validVersions = []) {
if (!binaryPath || !await fs.pathExists(binaryPath)) return false if (!binaryPath || !await fs.pathExists(binaryPath)) return false
if (!validVersions.length) return true
try { try {
const { stdout } = await this.exec('"' + binaryPath + '"' + ' -version') const { stdout } = await this.exec('"' + binaryPath + '"' + ' -version')
const version = stdout.match(/version\s([\d\.]+)/)?.[1] const version = stdout.match(/version\s([\d\.]+)/)?.[1]
if (!version) return false if (!version) return false
return this.goodVersions.some(goodVersion => version.startsWith(goodVersion)) return validVersions.some(validVersion => version.startsWith(validVersion))
} catch (err) { } catch (err) {
Logger.error(`[BinaryManager] Failed to check version of ${binaryPath}`) Logger.error(`[BinaryManager] Failed to check version of ${binaryPath}`)
return false return false
} }
} }
/**
*
* @param {string[]} binaries
*/
async install(binaries) { async install(binaries) {
if (binaries.length == 0) return if (!binaries.length) return
Logger.info(`[BinaryManager] Installing binaries: ${binaries.join(', ')}`) Logger.info(`[BinaryManager] Installing binaries: ${binaries.join(', ')}`)
let destination = await fileUtils.isWritable(this.mainInstallPath) ? this.mainInstallPath : this.altInstallPath let destination = await fileUtils.isWritable(this.mainInstallPath) ? this.mainInstallPath : this.altInstallPath
await ffbinaries.downloadBinaries(binaries, { destination, version: '6.1', force: true }) await ffbinaries.downloadBinaries(binaries, { destination, version: '6.1', force: true })
Logger.info(`[BinaryManager] Binaries installed to ${destination}`) Logger.info(`[BinaryManager] Binaries installed to ${destination}`)
} }
/**
* Append .exe to binary name for Windows
*
* @param {string} name
* @returns {string}
*/
getExecutableFileName(name) { getExecutableFileName(name) {
return name + (process.platform == 'win32' ? '.exe' : '') return name + (process.platform == 'win32' ? '.exe' : '')
} }

View File

@ -366,7 +366,7 @@ module.exports.encodeUriPath = (path) => {
* This method is necessary because fs.access(directory, fs.constants.W_OK) does not work on Windows * This method is necessary because fs.access(directory, fs.constants.W_OK) does not work on Windows
* *
* @param {string} directory * @param {string} directory
* @returns {boolean} * @returns {Promise<boolean>}
*/ */
module.exports.isWritable = async (directory) => { module.exports.isWritable = async (directory) => {
try { try {