import {
Arch,
asArray,
AsyncTaskManager,
exec,
executeAppBuilder,
exists,
getArchSuffix,
getPath7za,
getPlatformIconFileName,
InvalidConfigurationError,
log,
spawnAndWrite,
statOrNull,
use,
walk,
} from "builder-util"
import { CURRENT_APP_INSTALLER_FILE_NAME, CURRENT_APP_PACKAGE_FILE_NAME, PackageFileInfo, UUID } from "builder-util-runtime"
import _debug from "debug"
import * as fs from "fs"
import { readFile, stat, unlink } from "fs-extra"
import * as path from "path"
import { Target } from "../../core"
import { DesktopShortcutCreationPolicy, getEffectiveOptions } from "../../options/CommonWindowsInstallerConfiguration"
import { chooseNotNull, computeSafeArtifactNameIfNeeded, normalizeExt } from "../../platformPackager"
import { hashFile } from "../../util/hash"
import { isMacOsCatalina } from "../../util/macosVersion"
import { time } from "../../util/timer"
import { execWine } from "../../wine"
import { WinPackager } from "../../winPackager"
import { archive, ArchiveOptions } from "../archive"
import { appendBlockmap, configureDifferentialAwareArchiveOptions, createBlockmap, createNsisWebDifferentialUpdateInfo } from "../differentialUpdateInfoBuilder"
import { getWindowsInstallationAppPackageName, getWindowsInstallationDirName } from "../targetUtil"
import { Commands } from "./Commands"
import { Defines } from "./Defines"
import { addCustomMessageFileInclude, createAddLangsMacro, LangConfigurator } from "./nsisLang"
import { computeLicensePage } from "./nsisLicense"
import { NsisOptions, PortableOptions } from "./nsisOptions"
import { NsisScriptGenerator } from "./nsisScriptGenerator"
import { AppPackageHelper, NSIS_PATH, NSIS_RESOURCES_PATH, NsisTargetOptions, nsisTemplatesDir, UninstallerReader } from "./nsisUtil"
const debug = _debug("electron-builder:nsis")
const ELECTRON_BUILDER_NS_UUID = UUID.parse("50e065bc-3134-11e6-9bab-38c9862bdaf3")
const USE_NSIS_BUILT_IN_COMPRESSOR = false
export class NsisTarget extends Target {
readonly options: NsisOptions
readonly archs: Map<Arch, string> = new Map()
readonly isAsyncSupported = false
constructor(
readonly packager: WinPackager,
readonly outDir: string,
targetName: string,
protected readonly packageHelper: AppPackageHelper
) {
super(targetName)
this.packageHelper.refCount++
this.options =
targetName === "portable"
? Object.create(null)
: {
preCompressedFileExtensions: [".avi", ".mov", ".m4v", ".mp4", ".m4p", ".qt", ".mkv", ".webm", ".vmdk"],
...this.packager.config.nsis,
}
if (targetName !== "nsis") {
Object.assign(this.options, (this.packager.config as any)[targetName === "nsis-web" ? "nsisWeb" : targetName])
}
const deps = packager.info.metadata.dependencies
if (deps != null && deps["electron-squirrel-startup"] != null) {
log.warn('"electron-squirrel-startup" dependency is not required for NSIS')
}
NsisTargetOptions.resolve(this.options)
}
get shouldBuildUniversalInstaller() {
const buildSeparateInstallers = this.options.buildUniversalInstaller === false
return !buildSeparateInstallers
}
build(appOutDir: string, arch: Arch) {
this.archs.set(arch, appOutDir)
if (!this.shouldBuildUniversalInstaller) {
return this.buildInstaller(new Map<Arch, string>().set(arch, appOutDir))
}
return Promise.resolve()
}
get isBuildDifferentialAware(): boolean {
return !this.isPortable && this.options.differentialPackage !== false
}
private getPreCompressedFileExtensions(): Array<string> | null {
const result = this.isWebInstaller ? null : this.options.preCompressedFileExtensions
return result == null ? null : asArray(result).map(it => (it.startsWith(".") ? it : `.${it}`))
}
async buildAppPackage(appOutDir: string, arch: Arch): Promise<PackageFileInfo> {
const options = this.options
const packager = this.packager
const isBuildDifferentialAware = this.isBuildDifferentialAware
const format = !isBuildDifferentialAware && options.useZip ? "zip" : "7z"
const archiveFile = path.join(this.outDir, `${packager.appInfo.sanitizedName}-${packager.appInfo.version}-${Arch[arch]}.nsis.${format}`)
const preCompressedFileExtensions = this.getPreCompressedFileExtensions()
const archiveOptions: ArchiveOptions = {
withoutDir: true,
compression: packager.compression,
excluded: preCompressedFileExtensions == null ? null : preCompressedFileExtensions.map(it => `*${it}`),
}
const timer = time(`nsis package, ${Arch[arch]}`)
await archive(format, archiveFile, appOutDir, isBuildDifferentialAware ? configureDifferentialAwareArchiveOptions(archiveOptions) : archiveOptions)
timer.end()
if (isBuildDifferentialAware && this.isWebInstaller) {
const data = await appendBlockmap(archiveFile)
return {
...data,
path: archiveFile,
}
} else {
return await createPackageFileInfo(archiveFile)
}
}
protected installerFilenamePattern(primaryArch?: Arch | null, defaultArch?: string): string {
const setupText = this.isPortable ? "" : "Setup "
const archSuffix = !this.shouldBuildUniversalInstaller && primaryArch != null ? getArchSuffix(primaryArch, defaultArch) : ""
return "${productName} " + setupText + "${version}" + archSuffix + ".${ext}"
}
private get isPortable(): boolean {
return this.name === "portable"
}
async finishBuild(): Promise<any> {
if (!this.shouldBuildUniversalInstaller) {
await super.finishBuild()
return this.packageHelper.finishBuild()
}
try {
const { pattern } = this.packager.artifactPatternConfig(this.options, this.installerFilenamePattern())
const builds = new Set([this.archs])
if (pattern.includes("${arch}") && this.archs.size > 1) {
;[...this.archs].forEach(([arch, appOutDir]) => builds.add(new Map().set(arch, appOutDir)))
}
const doBuildArchs = builds.values()
for (const archs of doBuildArchs) {
await this.buildInstaller(archs)
}
} finally {
await super.finishBuild()
await this.packageHelper.finishBuild()
}
}
private async buildInstaller(archs: Map<Arch, string>): Promise<any> {
const primaryArch: Arch | null = archs.size === 1 ? (archs.keys().next().value ?? null) : null
const packager = this.packager
const appInfo = packager.appInfo
const options = this.options
const defaultArch = chooseNotNull(this.packager.platformSpecificBuildOptions.defaultArch, this.packager.config.defaultArch) ?? undefined
const installerFilename = packager.expandArtifactNamePattern(options, "exe", primaryArch, this.installerFilenamePattern(primaryArch, defaultArch), false, defaultArch)
const oneClick = options.oneClick !== false
const installerPath = path.join(this.outDir, installerFilename)
const logFields: any = {
target: this.name,
file: log.filePath(installerPath),
archs: Array.from(archs.keys())
.map(it => Arch[it])
.join(", "),
}
const isPerMachine = options.perMachine === true
if (!this.isPortable) {
logFields.oneClick = oneClick
logFields.perMachine = isPerMachine
}
await packager.info.emitArtifactBuildStarted(
{
targetPresentableName: this.name,
file: installerPath,
arch: primaryArch,
},
logFields
)
const guid = options.guid || UUID.v5(appInfo.id, ELECTRON_BUILDER_NS_UUID)
const uninstallAppKey = guid.replace(/\\/g, " - ")
const defines: Defines = {
APP_ID: appInfo.id,
APP_GUID: guid,
UNINSTALL_APP_KEY: uninstallAppKey,
PRODUCT_NAME: appInfo.productName,
PRODUCT_FILENAME: appInfo.productFilename,
APP_FILENAME: getWindowsInstallationDirName(appInfo, !oneClick || isPerMachine),
APP_DESCRIPTION: appInfo.description,
VERSION: appInfo.version,
PROJECT_DIR: packager.projectDir,
BUILD_RESOURCES_DIR: packager.info.buildResourcesDir,
APP_PACKAGE_NAME: getWindowsInstallationAppPackageName(appInfo.name),
}
if (options.customNsisBinary?.debugLogging) {
defines.ENABLE_LOGGING_ELECTRON_BUILDER = null
}
if (uninstallAppKey !== guid) {
defines.UNINSTALL_REGISTRY_KEY_2 = `Software\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\${guid}`
}
const commands: Commands = {
OutFile: `"${installerPath}"`,
VIProductVersion: appInfo.getVersionInWeirdWindowsForm(),
VIAddVersionKey: this.computeVersionKey(),
Unicode: this.isUnicodeEnabled,
}
const isPortable = this.isPortable
const iconPath = (isPortable ? null : await packager.getResource(options.installerIcon, "installerIcon.ico")) || (await packager.getIconPath())
if (iconPath != null) {
if (isPortable) {
commands.Icon = `"${iconPath}"`
} else {
defines.MUI_ICON = iconPath
defines.MUI_UNICON = iconPath
}
}
const packageFiles: { [arch: string]: PackageFileInfo } = {}
let estimatedSize = 0
if (this.isPortable && options.useZip) {
for (const [arch, dir] of archs.entries()) {
defines[arch === Arch.x64 ? "APP_DIR_64" : arch === Arch.arm64 ? "APP_DIR_ARM64" : "APP_DIR_32"] = dir
}
} else if (USE_NSIS_BUILT_IN_COMPRESSOR && archs.size === 1) {
const value: Arch | undefined = archs.keys().next().value
use(value, v => (defines.APP_BUILD_DIR = archs.get(v)))
} else {
await Promise.all(
Array.from(archs.keys()).map(async arch => {
const { fileInfo, unpackedSize } = await this.packageHelper.packArch(arch, this)
const file = fileInfo.path
const defineKey = arch === Arch.x64 ? "APP_64" : arch === Arch.arm64 ? "APP_ARM64" : "APP_32"
defines[defineKey] = file
const defineNameKey = `${defineKey}_NAME` as "APP_64_NAME" | "APP_ARM64_NAME" | "APP_32_NAME"
defines[defineNameKey] = path.basename(file)
const defineHashKey = `${defineKey}_HASH` as "APP_64_HASH" | "APP_ARM64_HASH" | "APP_32_HASH"
defines[defineHashKey] = Buffer.from(fileInfo.sha512, "base64").toString("hex").toUpperCase()
const defineUnpackedSizeKey = `${defineKey}_UNPACKED_SIZE` as "APP_64_UNPACKED_SIZE" | "APP_ARM64_UNPACKED_SIZE" | "APP_32_UNPACKED_SIZE"
defines[defineUnpackedSizeKey] = Math.ceil(unpackedSize / 1024).toString()
if (this.isWebInstaller) {
await packager.info.emitArtifactBuildCompleted({
file,
target: this,
arch,
packager,
})
packageFiles[Arch[arch]] = fileInfo
}
const path7za = await getPath7za()
const archiveInfo = (await exec(path7za, ["l", file])).trim()
const match = /(\d+)\s+\d+\s+\d+\s+files/.exec(archiveInfo)
if (match == null) {
log.warn({ output: archiveInfo }, "cannot compute size of app package")
} else {
estimatedSize += parseInt(match[1], 10)
}
})
)
}
this.configureDefinesForAllTypeOfInstaller(defines)
if (isPortable) {
const { unpackDirName, requestExecutionLevel, splashImage } = options as PortableOptions
defines.REQUEST_EXECUTION_LEVEL = requestExecutionLevel || "user"
if (typeof unpackDirName === "string" || !unpackDirName) {
defines.UNPACK_DIR_NAME = unpackDirName || (await executeAppBuilder(["ksuid"]))
}
if (splashImage != null) {
defines.SPLASH_IMAGE = path.resolve(packager.projectDir, splashImage)
}
} else {
await this.configureDefines(oneClick, defines)
}
if (estimatedSize !== 0) {
defines.ESTIMATED_SIZE = Math.round(estimatedSize / 1024)
}
if (packager.compression === "store") {
commands.SetCompress = "off"
} else {
commands.SetCompressor = "zlib"
if (!this.isWebInstaller) {
defines.COMPRESS = "auto"
}
}
debug(defines)
debug(commands)
if (packager.packagerOptions.effectiveOptionComputed != null && (await packager.packagerOptions.effectiveOptionComputed([defines, commands]))) {
return
}
const definesUninstaller = { ...defines }
const commandsUninstaller = { ...commands }
if (appInfo.shortVersion != null) {
definesUninstaller.VERSION = appInfo.shortVersion
commandsUninstaller.VIProductVersion = appInfo.shortVersionWindows
commandsUninstaller.VIAddVersionKey = this.computeVersionKey(true)
}
this.buildQueueManager.add(async () => {
const sharedHeader = await this.computeCommonInstallerScriptHeader()
const script = isPortable
? await readFile(path.join(nsisTemplatesDir, "portable.nsi"), "utf8")
: await this.computeScriptAndSignUninstaller(definesUninstaller, commandsUninstaller, installerPath, sharedHeader, archs)
defines.UNINSTALLER_OUT_FILE = definesUninstaller.UNINSTALLER_OUT_FILE
await this.executeMakensis(defines, commands, sharedHeader + (await this.computeFinalScript(script, true, archs)))
await Promise.all<any>([packager.sign(installerPath), defines.UNINSTALLER_OUT_FILE == null ? Promise.resolve() : unlink(defines.UNINSTALLER_OUT_FILE)])
const safeArtifactName = computeSafeArtifactNameIfNeeded(installerFilename, () => this.generateGitHubInstallerName(primaryArch, defaultArch))
let updateInfo: any
if (this.isWebInstaller) {
updateInfo = createNsisWebDifferentialUpdateInfo(installerPath, packageFiles)
} else if (this.isBuildDifferentialAware) {
updateInfo = await createBlockmap(installerPath, this, packager, safeArtifactName)
}
if (updateInfo != null && isPerMachine && (oneClick || options.packElevateHelper)) {
updateInfo.isAdminRightsRequired = true
}
await packager.info.emitArtifactBuildCompleted({
file: installerPath,
updateInfo,
target: this,
packager,
arch: primaryArch,
safeArtifactName,
isWriteUpdateInfo: !this.isPortable,
})
})
}
protected generateGitHubInstallerName(primaryArch: Arch | null, defaultArch: string | undefined): string {
const appInfo = this.packager.appInfo
const classifier = appInfo.name.toLowerCase() === appInfo.name ? "setup-" : "Setup-"
const archSuffix = !this.shouldBuildUniversalInstaller && primaryArch != null ? getArchSuffix(primaryArch, defaultArch) : ""
return `${appInfo.name}-${this.isPortable ? "" : classifier}${appInfo.version}${archSuffix}.exe`
}
private get isUnicodeEnabled(): boolean {
return this.options.unicode !== false
}
get isWebInstaller(): boolean {
return false
}
private async computeScriptAndSignUninstaller(defines: Defines, commands: Commands, installerPath: string, sharedHeader: string, archs: Map<Arch, string>): Promise<string> {
const packager = this.packager
const customScriptPath = await packager.getResource(this.options.script, "installer.nsi")
const script = await readFile(customScriptPath || path.join(nsisTemplatesDir, "installer.nsi"), "utf8")
if (customScriptPath != null) {
log.info({ reason: "custom NSIS script is used" }, "uninstaller is not signed by electron-builder")
return script
}
const uninstallerPath = path.join(this.outDir, `${path.basename(installerPath, "exe")}__uninstaller.exe`)
const isWin = process.platform === "win32"
defines.BUILD_UNINSTALLER = null
defines.UNINSTALLER_OUT_FILE = isWin ? uninstallerPath : path.win32.join("Z:", uninstallerPath)
await this.executeMakensis(defines, commands, sharedHeader + (await this.computeFinalScript(script, false, archs)))
if (isMacOsCatalina()) {
try {
await UninstallerReader.exec(installerPath, uninstallerPath)
} catch (error: any) {
log.warn(`packager.vm is used: ${error.message}`)
const vm = await packager.vm.value
await vm.exec(installerPath, [])
let i = 0
while (!(await exists(uninstallerPath)) && i++ < 100) {
await new Promise((resolve, _reject) => setTimeout(resolve, 300))
}
}
} else {
await execWine(installerPath, null, [], { env: { __COMPAT_LAYER: "RunAsInvoker" } })
}
await packager.sign(uninstallerPath)
delete defines.BUILD_UNINSTALLER
defines.UNINSTALLER_OUT_FILE = uninstallerPath
return script
}
private computeVersionKey(short = false) {
const localeId = this.options.language || "1033"
const appInfo = this.packager.appInfo
const versionKey = [
`/LANG=${localeId} ProductName "${appInfo.productName}"`,
`/LANG=${localeId} ProductVersion "${appInfo.version}"`,
`/LANG=${localeId} LegalCopyright "${appInfo.copyright}"`,
`/LANG=${localeId} FileDescription "${appInfo.description}"`,
`/LANG=${localeId} FileVersion "${appInfo.buildVersion}"`,
]
if (short) {
versionKey[1] = `/LANG=${localeId} ProductVersion "${appInfo.shortVersion}"`
versionKey[4] = `/LANG=${localeId} FileVersion "${appInfo.shortVersion}"`
}
use(this.packager.platformSpecificBuildOptions.legalTrademarks, it => versionKey.push(`/LANG=${localeId} LegalTrademarks "${it}"`))
use(appInfo.companyName, it => versionKey.push(`/LANG=${localeId} CompanyName "${it}"`))
return versionKey
}
protected configureDefines(oneClick: boolean, defines: Defines): Promise<any> {
const packager = this.packager
const options = this.options
const asyncTaskManager = new AsyncTaskManager(packager.info.cancellationToken)
if (oneClick) {
defines.ONE_CLICK = null
if (options.runAfterFinish !== false) {
defines.RUN_AFTER_FINISH = null
}
asyncTaskManager.add(async () => {
const installerHeaderIcon = await packager.getResource(options.installerHeaderIcon, "installerHeaderIcon.ico")
if (installerHeaderIcon != null) {
defines.HEADER_ICO = installerHeaderIcon
}
})
} else {
if (options.runAfterFinish === false) {
defines.HIDE_RUN_AFTER_FINISH = null
}
asyncTaskManager.add(async () => {
const installerHeader = await packager.getResource(options.installerHeader, "installerHeader.bmp")
if (installerHeader != null) {
defines.MUI_HEADERIMAGE = null
defines.MUI_HEADERIMAGE_RIGHT = null
defines.MUI_HEADERIMAGE_BITMAP = installerHeader
}
})
asyncTaskManager.add(async () => {
const bitmap = (await packager.getResource(options.installerSidebar, "installerSidebar.bmp")) || "${NSISDIR}\\Contrib\\Graphics\\Wizard\\nsis3-metro.bmp"
defines.MUI_WELCOMEFINISHPAGE_BITMAP = bitmap
defines.MUI_UNWELCOMEFINISHPAGE_BITMAP = (await packager.getResource(options.uninstallerSidebar, "uninstallerSidebar.bmp")) || bitmap
})
if (options.allowElevation !== false) {
defines.MULTIUSER_INSTALLMODE_ALLOW_ELEVATION = null
}
}
if (options.perMachine === true) {
defines.INSTALL_MODE_PER_ALL_USERS = null
}
if (options.selectPerMachineByDefault === true) {
defines.INSTALL_MODE_PER_ALL_USERS_DEFAULT = null
}
if (!oneClick || options.perMachine === true) {
defines.INSTALL_MODE_PER_ALL_USERS_REQUIRED = null
}
if (options.allowToChangeInstallationDirectory) {
if (oneClick) {
throw new InvalidConfigurationError("allowToChangeInstallationDirectory makes sense only for assisted installer (please set oneClick to false)")
}
defines.allowToChangeInstallationDirectory = null
}
if (options.removeDefaultUninstallWelcomePage) {
defines.removeDefaultUninstallWelcomePage = null
}
const commonOptions = getEffectiveOptions(options, packager)
if (commonOptions.menuCategory != null) {
defines.MENU_FILENAME = commonOptions.menuCategory
}
defines.SHORTCUT_NAME = commonOptions.shortcutName
if (options.deleteAppDataOnUninstall) {
defines.DELETE_APP_DATA_ON_UNINSTALL = null
}
asyncTaskManager.add(async () => {
const uninstallerIcon = await packager.getResource(options.uninstallerIcon, "uninstallerIcon.ico")
if (uninstallerIcon != null) {
defines.UNINSTALLER_ICON = uninstallerIcon
defines.MUI_UNICON = uninstallerIcon
}
})
defines.UNINSTALL_DISPLAY_NAME = packager.expandMacro(options.uninstallDisplayName || "${productName} ${version}", null, {}, false)
if (commonOptions.isCreateDesktopShortcut === DesktopShortcutCreationPolicy.NEVER) {
defines.DO_NOT_CREATE_DESKTOP_SHORTCUT = null
}
if (commonOptions.isCreateDesktopShortcut === DesktopShortcutCreationPolicy.ALWAYS) {
defines.RECREATE_DESKTOP_SHORTCUT = null
}
if (!commonOptions.isCreateStartMenuShortcut) {
defines.DO_NOT_CREATE_START_MENU_SHORTCUT = null
}
if (options.displayLanguageSelector === true) {
defines.DISPLAY_LANG_SELECTOR = null
}
return asyncTaskManager.awaitTasks()
}
private configureDefinesForAllTypeOfInstaller(defines: Defines): void {
const appInfo = this.packager.appInfo
const companyName = appInfo.companyName
if (companyName != null) {
defines.COMPANY_NAME = companyName
}
if (defines.APP_FILENAME !== appInfo.productFilename) {
defines.APP_PRODUCT_FILENAME = appInfo.productFilename
}
if (this.isWebInstaller) {
defines.APP_PACKAGE_STORE_FILE = `${appInfo.updaterCacheDirName}\\${CURRENT_APP_PACKAGE_FILE_NAME}`
} else {
defines.APP_INSTALLER_STORE_FILE = `${appInfo.updaterCacheDirName}\\${CURRENT_APP_INSTALLER_FILE_NAME}`
}
if (!this.isWebInstaller && defines.APP_BUILD_DIR == null) {
const options = this.options
if (options.useZip) {
defines.ZIP_COMPRESSION = null
}
defines.COMPRESSION_METHOD = options.useZip ? "zip" : "7z"
}
}
private async executeMakensis(defines: Defines, commands: Commands, script: string): Promise<void> {
const args: Array<string> = this.options.warningsAsErrors === false ? [] : ["-WX"]
args.push("-INPUTCHARSET", "UTF8")
for (const name of Object.keys(defines)) {
const value: any = defines[name as keyof Defines]
if (value == null) {
args.push(`-D${name}`)
} else {
args.push(`-D${name}=${value}`)
}
}
for (const name of Object.keys(commands)) {
const value = commands[name as keyof Commands]
if (Array.isArray(value)) {
for (const c of value) {
args.push(`-X${name} ${c}`)
}
} else {
args.push(`-X${name} ${value}`)
}
}
args.push("-")
if (this.packager.debugLogger.isEnabled) {
this.packager.debugLogger.add("nsis.script", script)
}
const nsisPath = await NSIS_PATH()
const command = path.join(
nsisPath,
process.platform === "darwin" ? "mac" : process.platform === "win32" ? "Bin" : "linux",
process.platform === "win32" ? "makensis.exe" : "makensis"
)
await ensureNotBusy(commands["OutFile"].replace(/"/g, ""))
await spawnAndWrite(command, args, script, {
env: { ...process.env, NSISDIR: nsisPath },
cwd: nsisTemplatesDir,
})
}
private async computeCommonInstallerScriptHeader(): Promise<string> {
const packager = this.packager
const options = this.options
const scriptGenerator = new NsisScriptGenerator()
const langConfigurator = new LangConfigurator(options)
scriptGenerator.include(path.join(nsisTemplatesDir, "include", "StdUtils.nsh"))
const includeDir = path.join(nsisTemplatesDir, "include")
scriptGenerator.addIncludeDir(includeDir)
scriptGenerator.flags(["updated", "force-run", "keep-shortcuts", "no-desktop-shortcut", "delete-app-data", "allusers", "currentuser"])
createAddLangsMacro(scriptGenerator, langConfigurator)
const taskManager = new AsyncTaskManager(packager.info.cancellationToken)
const pluginArch = this.isUnicodeEnabled ? "x86-unicode" : "x86-ansi"
taskManager.add(async () => {
scriptGenerator.addPluginDir(pluginArch, path.join(await NSIS_RESOURCES_PATH(), "plugins", pluginArch))
})
taskManager.add(async () => {
const userPluginDir = path.join(packager.info.buildResourcesDir, pluginArch)
const stat = await statOrNull(userPluginDir)
if (stat != null && stat.isDirectory()) {
scriptGenerator.addPluginDir(pluginArch, userPluginDir)
}
})
taskManager.addTask(addCustomMessageFileInclude("messages.yml", packager, scriptGenerator, langConfigurator))
if (!this.isPortable) {
if (options.oneClick === false) {
taskManager.addTask(addCustomMessageFileInclude("assistedMessages.yml", packager, scriptGenerator, langConfigurator))
}
taskManager.add(async () => {
const customInclude = await packager.getResource(this.options.include, "installer.nsh")
if (customInclude != null) {
scriptGenerator.addIncludeDir(packager.info.buildResourcesDir)
scriptGenerator.include(customInclude)
}
})
}
await taskManager.awaitTasks()
return scriptGenerator.build()
}
private async computeFinalScript(originalScript: string, isInstaller: boolean, archs: Map<Arch, string>): Promise<string> {
const packager = this.packager
const options = this.options
const langConfigurator = new LangConfigurator(options)
const scriptGenerator = new NsisScriptGenerator()
const taskManager = new AsyncTaskManager(packager.info.cancellationToken)
if (isInstaller) {
taskManager.add(() => computeLicensePage(packager, options, scriptGenerator, langConfigurator.langs))
}
await taskManager.awaitTasks()
if (this.isPortable) {
return scriptGenerator.build() + originalScript
}
const preCompressedFileExtensions = this.getPreCompressedFileExtensions()
if (preCompressedFileExtensions != null && preCompressedFileExtensions.length !== 0) {
for (const [arch, dir] of archs.entries()) {
await generateForPreCompressed(preCompressedFileExtensions, dir, arch, scriptGenerator)
}
}
const fileAssociations = packager.fileAssociations
if (fileAssociations.length !== 0) {
scriptGenerator.include(path.join(path.join(nsisTemplatesDir, "include"), "FileAssociation.nsh"))
if (isInstaller) {
const registerFileAssociationsScript = new NsisScriptGenerator()
for (const item of fileAssociations) {
const extensions = asArray(item.ext).map(normalizeExt)
for (const ext of extensions) {
const customIcon = await packager.getResource(getPlatformIconFileName(item.icon, false), `${extensions[0]}.ico`)
let installedIconPath = "$appExe,0"
if (customIcon != null) {
installedIconPath = `$INSTDIR\\resources\\${path.basename(customIcon)}`
registerFileAssociationsScript.file(installedIconPath, customIcon)
}
const icon = `"${installedIconPath}"`
const commandText = `"Open with ${packager.appInfo.productName}"`
const command = '"$appExe $\\"%1$\\""'
registerFileAssociationsScript.insertMacro("APP_ASSOCIATE", `"${ext}" "${item.name || ext}" "${item.description || ""}" ${icon} ${commandText} ${command}`)
}
}
scriptGenerator.macro("registerFileAssociations", registerFileAssociationsScript)
} else {
const unregisterFileAssociationsScript = new NsisScriptGenerator()
for (const item of fileAssociations) {
for (const ext of asArray(item.ext)) {
unregisterFileAssociationsScript.insertMacro("APP_UNASSOCIATE", `"${normalizeExt(ext)}" "${item.name || ext}"`)
}
}
scriptGenerator.macro("unregisterFileAssociations", unregisterFileAssociationsScript)
}
}
return scriptGenerator.build() + originalScript
}
}
async function generateForPreCompressed(preCompressedFileExtensions: Array<string>, dir: string, arch: Arch, scriptGenerator: NsisScriptGenerator): Promise<void> {
const resourcesDir = path.join(dir, "resources")
const dirInfo = await statOrNull(resourcesDir)
if (dirInfo == null || !dirInfo.isDirectory()) {
return
}
const nodeModules = `${path.sep}node_modules`
const preCompressedAssets = await walk(resourcesDir, (file, stat) => {
if (stat.isDirectory()) {
return !file.endsWith(nodeModules)
} else {
return preCompressedFileExtensions.some(it => file.endsWith(it))
}
})
if (preCompressedAssets.length !== 0) {
const macro = new NsisScriptGenerator()
for (const file of preCompressedAssets) {
macro.file(`$INSTDIR\\${path.relative(dir, file).replace(/\//g, "\\")}`, file)
}
scriptGenerator.macro(`customFiles_${Arch[arch]}`, macro)
}
}
async function ensureNotBusy(outFile: string): Promise<void> {
function isBusy(wasBusyBefore: boolean): Promise<boolean> {
return new Promise((resolve, reject) => {
fs.open(outFile, "r+", (error, fd) => {
try {
if (error != null && error.code === "EBUSY") {
if (!wasBusyBefore) {
log.info({}, "output file is locked for writing (maybe by virus scanner) => waiting for unlock...")
}
resolve(false)
} else if (fd == null) {
resolve(true)
} else {
fs.close(fd, () => resolve(true))
}
} catch (error: any) {
reject(error)
}
})
}).then(result => {
if (result) {
return true
} else {
return new Promise(resolve => setTimeout(resolve, 2000)).then(() => isBusy(true))
}
})
}
await isBusy(false)
}
async function createPackageFileInfo(file: string): Promise<PackageFileInfo> {
return {
path: file,
size: (await stat(file)).size,
sha512: await hashFile(file),
}
}