mirror of
https://github.com/UNITRONIX/BetterDesk.git
synced 2026-10-26 06:19:53 +00:00
UNITRONIX
107 lines
4.3 KiB
JavaScript
107 lines
4.3 KiB
JavaScript
'use strict';
|
|
|
|
const fs = require('fs');
|
|
const fsp = fs.promises;
|
|
const path = require('path');
|
|
|
|
const INSTALLER_SCRIPT = String.raw`[CmdletBinding()]
|
|
param(
|
|
[switch]$UseScheduledTask
|
|
)
|
|
|
|
$ErrorActionPreference = 'Stop'
|
|
$profileInstallsService = __INSTALL_SERVICE__
|
|
$profileAutostarts = __AUTOSTART__
|
|
$app = @(
|
|
(Join-Path $PSScriptRoot 'betterdesk.exe'),
|
|
(Join-Path $PSScriptRoot 'rustdesk.exe')
|
|
) | Where-Object { Test-Path -LiteralPath $_ } | Select-Object -First 1
|
|
|
|
if (-not $app) {
|
|
throw 'BetterDesk executable was not found next to this installer.'
|
|
}
|
|
|
|
$appName = [IO.Path]::GetFileNameWithoutExtension($app)
|
|
$serviceName = $appName
|
|
$taskName = "$appName Tray"
|
|
$startupShortcut = Join-Path $env:ProgramData "Microsoft\Windows\Start Menu\Programs\Startup\$appName Tray.lnk"
|
|
$isUninstall = [IO.Path]::GetFileNameWithoutExtension($MyInvocation.MyCommand.Name) -match 'Uninstall'
|
|
$identity = [Security.Principal.WindowsIdentity]::GetCurrent()
|
|
$principal = [Security.Principal.WindowsPrincipal]::new($identity)
|
|
$isAdministrator = $principal.IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)
|
|
|
|
if (-not $isAdministrator) {
|
|
$arguments = @(
|
|
'-NoProfile', '-ExecutionPolicy', 'Bypass', '-File',
|
|
$MyInvocation.MyCommand.Path
|
|
)
|
|
if ($UseScheduledTask) {
|
|
$arguments += '-UseScheduledTask'
|
|
}
|
|
$elevated = Start-Process -FilePath 'powershell.exe' -Verb RunAs -ArgumentList $arguments -Wait -PassThru
|
|
exit $elevated.ExitCode
|
|
}
|
|
|
|
if ($isUninstall) {
|
|
Unregister-ScheduledTask -TaskName $taskName -Confirm:$false -ErrorAction SilentlyContinue
|
|
if ($profileInstallsService -and (Test-Path -LiteralPath $app)) {
|
|
$process = Start-Process -FilePath $app -ArgumentList '--uninstall-service' -Wait -PassThru -WindowStyle Hidden
|
|
if ($process.ExitCode -ne 0) {
|
|
throw "BetterDesk service uninstall failed with exit code $($process.ExitCode)."
|
|
}
|
|
}
|
|
exit 0
|
|
}
|
|
|
|
if ($profileInstallsService) {
|
|
$process = Start-Process -FilePath $app -ArgumentList '--install-service' -Wait -PassThru -WindowStyle Hidden
|
|
if ($process.ExitCode -ne 0) {
|
|
throw "BetterDesk service installation failed with exit code $($process.ExitCode)."
|
|
}
|
|
$service = Get-Service -Name $serviceName -ErrorAction Stop
|
|
if ($service.Status -notin @('Running', 'StartPending')) {
|
|
throw "BetterDesk service is not running (status: $($service.Status))."
|
|
}
|
|
}
|
|
|
|
if ($profileAutostarts -and (-not $profileInstallsService -or $UseScheduledTask)) {
|
|
Remove-Item -LiteralPath $startupShortcut -Force -ErrorAction SilentlyContinue
|
|
$taskAction = New-ScheduledTaskAction -Execute $app -Argument '--tray' -WorkingDirectory (Split-Path -Parent $app)
|
|
$taskTrigger = New-ScheduledTaskTrigger -AtLogOn -User $identity.Name
|
|
$taskPrincipal = New-ScheduledTaskPrincipal -UserId $identity.Name -LogonType InteractiveToken -RunLevel Highest
|
|
$taskSettings = New-ScheduledTaskSettingsSet -RestartCount 3 -RestartInterval (New-TimeSpan -Minutes 1) -AllowStartIfOnBatteries -DontStopIfGoingOnBatteries
|
|
Register-ScheduledTask -TaskName $taskName -Action $taskAction -Trigger $taskTrigger -Principal $taskPrincipal -Settings $taskSettings -Force | Out-Null
|
|
}
|
|
|
|
if (-not $profileInstallsService -and -not $profileAutostarts) {
|
|
Write-Host "BetterDesk installed without service or autostart."
|
|
} else {
|
|
Write-Host "BetterDesk Support startup configuration installed successfully."
|
|
}
|
|
`;
|
|
|
|
function buildWindowsSupportInstallerScript({ installService = false, autostart = false } = {}) {
|
|
return INSTALLER_SCRIPT
|
|
.replace('__INSTALL_SERVICE__', installService ? '$true' : '$false')
|
|
.replace('__AUTOSTART__', autostart ? '$true' : '$false');
|
|
}
|
|
|
|
async function writeWindowsSupportInstallers(stageDir, options = {}) {
|
|
const script = buildWindowsSupportInstallerScript(options);
|
|
await fsp.writeFile(path.join(stageDir, 'Install-BetterDesk.ps1'), script, 'utf8');
|
|
await fsp.writeFile(path.join(stageDir, 'Uninstall-BetterDesk.ps1'), script, 'utf8');
|
|
}
|
|
|
|
function hasWindowsSupportInstallers(stageDir) {
|
|
return [
|
|
'Install-BetterDesk.ps1',
|
|
'Uninstall-BetterDesk.ps1',
|
|
].every((name) => fs.existsSync(path.join(stageDir, name)));
|
|
}
|
|
|
|
module.exports = {
|
|
buildWindowsSupportInstallerScript,
|
|
writeWindowsSupportInstallers,
|
|
hasWindowsSupportInstallers,
|
|
};
|