setup-cpp/packages/sudo-tools/src/index.ts

49 lines
1.5 KiB
TypeScript
Raw Normal View History

2022-08-08 08:18:49 +08:00
import which from "which"
import execa from "execa"
let isSudoCache: boolean | undefined = undefined
/**
* Detect if sudo is available and the user has root privileges
*
* @note it caches the result for the subsequent calls to this function.
*/
export function isSudo(): boolean {
2022-08-08 08:18:49 +08:00
if (isSudoCache !== undefined) {
return isSudoCache
}
// eslint-disable-next-line @typescript-eslint/strict-boolean-expressions, @typescript-eslint/no-unnecessary-condition
isSudoCache = (Boolean(process.env.CI) || isRoot()) && which.sync("sudo", { nothrow: true }) !== null
2022-08-08 08:18:49 +08:00
return isSudoCache
}
/** Detect if the process has root privileges */
export function isRoot(): boolean {
return process.getuid?.() === 0
}
2022-08-08 08:18:49 +08:00
/** Prepend `sudo` to the command if sudo is available */
export function prependSudo(command: string) {
if (isSudo()) {
2022-08-08 08:18:49 +08:00
return `sudo ${command}`
}
return command
}
/**
* Execute a command as root if sudo is available. Otherwise executes the command normally without sudo.
2022-08-08 08:18:49 +08:00
*
2022-08-08 08:24:44 +08:00
* @param program The program to spawn
2022-08-08 08:18:49 +08:00
* @param args The command arguments
* @param execOptions The options passed to `execa`.
2022-08-08 08:24:44 +08:00
*
* Defaults to `{ stdio: "inherit" }`
2022-08-08 08:18:49 +08:00
*/
export function execRoot(program: string, args: string[] = [], execOptions: execa.SyncOptions = { stdio: "inherit" }) {
if (isSudo()) {
2022-08-08 08:24:44 +08:00
return execa.commandSync(`sudo ${[program, ...args].map((arg) => `'${arg}'`).join(" ")}`, execOptions)
2022-08-08 08:18:49 +08:00
} else {
2022-08-08 08:24:44 +08:00
return execa.sync(program, args, execOptions)
2022-08-08 08:18:49 +08:00
}
}