relic/packages/core/src/helpers/setup.js

import Logger from "../logger"

const Log = Logger.child({ service: "SETUP" })

import path from "node:path"
import fs from "node:fs"
import os from "node:os"
import admzip from "adm-zip"
import resolveOs from "../utils/resolveOs"
import chmodRecursive from "../utils/chmodRecursive"

import downloadFile from "../helpers/downloadHttpFile"

import Vars from "../vars"
import Prerequisites from "../prerequisites"

export default async () => {
    if (!fs.existsSync(Vars.binaries_path)) {
        Log.info(`Creating binaries directory: ${Vars.binaries_path}...`)
        await fs.promises.mkdir(Vars.binaries_path, { recursive: true })
    }

    for await (let prerequisite of Prerequisites) {
        try {
            Log.info(`Checking prerequisite: ${prerequisite.id}...`)

            if (Array.isArray(prerequisite.requireOs) && !prerequisite.requireOs.includes(os.platform())) {
                Log.info(`Prerequisite: ${prerequisite.id} is not required for this os.`)
                continue
            }

            if (!fs.existsSync(prerequisite.finalBin)) {
                Log.info(`Missing prerequisite: ${prerequisite.id}, installing...`)

                global._relic_eventBus.emit("app:setup", {
                    installed: false,
                    message: `Installing ${prerequisite.id}`,
                })

                if (fs.existsSync(prerequisite.destination)) {
                    Log.info(`Deleting temporal file [${prerequisite.destination}]`)

                    global._relic_eventBus.emit("app:setup", {
                        installed: false,
                        message: `Deleting temporal file [${prerequisite.destination}]`,
                    })

                    await fs.promises.rm(prerequisite.destination)
                }

                if (fs.existsSync(prerequisite.extract)) {
                    Log.info(`Deleting temporal directory [${prerequisite.extract}]`)

                    global._relic_eventBus.emit("app:setup", {
                        installed: false,
                        message: `Deleting temporal directory [${prerequisite.extract}]`,
                    })

                    await fs.promises.rm(prerequisite.extract, { recursive: true })
                }

                Log.info(`Creating base directory: ${Vars.binaries_path}/${prerequisite.id}...`)

                global._relic_eventBus.emit("app:setup", {
                    installed: false,
                    message: `Creating base directory: ${Vars.binaries_path}/${prerequisite.id}`,
                })

                await fs.promises.mkdir(path.resolve(Vars.binaries_path, prerequisite.id), { recursive: true })

                if (typeof prerequisite.url === "function") {
                    prerequisite.url = await prerequisite.url(resolveOs(), os.arch())
                    Log.info(`Resolved url: ${prerequisite.url}`)
                }

                Log.info(`Downloading ${prerequisite.id} from [${prerequisite.url}] to destination [${prerequisite.destination}]...`)

                global._relic_eventBus.emit("app:setup", {
                    installed: false,
                    message: `Starting download ${prerequisite.id} from [${prerequisite.url}] to destination [${prerequisite.destination}]`,
                })

                try {
                    await downloadFile(
                        prerequisite.url,
                        prerequisite.destination,
                        (progress) => {
                            global._relic_eventBus.emit("app:setup", {
                                installed: false,
                                message: `Downloaded ${progress.transferredString} / ${progress.totalString} | ${progress.speedString}/s`,
                            })
                        }
                    )
                } catch (error) {
                    if (fs.existsSync(prerequisite.destination)) {
                        await fs.promises.rm(prerequisite.destination)
                    }

                    throw error
                }

                if (typeof prerequisite.extract === "string") {
                    Log.info(`Extracting ${prerequisite.id} to destination [${prerequisite.extract}]...`)

                    global._relic_eventBus.emit("app:setup", {
                        installed: false,
                        message: `Extracting ${prerequisite.id} to destination [${prerequisite.extract}]`,
                    })

                    const zip = new admzip(prerequisite.destination)

                    await zip.extractAllTo(prerequisite.extract, true)

                    Log.info(`Extraction ok...`)
                }

                if (prerequisite.extractTargetFromName === true) {
                    let name = path.basename(prerequisite.url)
                    const ext = path.extname(name)

                    name = name.replace(ext, "")

                    if (fs.existsSync(path.resolve(prerequisite.extract, name))) {
                        await fs.promises.rename(path.resolve(prerequisite.extract, name), `${prerequisite.extract}_old`)
                        await fs.promises.rm(prerequisite.extract, { recursive: true })
                        await fs.promises.rename(`${prerequisite.extract}_old`, prerequisite.extract)
                    }
                }

                if (prerequisite.deleteBeforeExtract === true) {
                    Log.info(`Deleting temporal file [${prerequisite.destination}]`)

                    global._relic_eventBus.emit("app:setup", {
                        installed: false,
                        message: `Deleting temporal file [${prerequisite.destination}]`,
                    })

                    await fs.promises.unlink(prerequisite.destination)
                }

                if (typeof prerequisite.rewriteExecutionPermission !== "undefined") {
                    const to = typeof prerequisite.rewriteExecutionPermission === "string" ?
                        prerequisite.rewriteExecutionPermission :
                        prerequisite.finalBin

                    Log.info(`Rewriting permissions to ${to}...`)

                    global._relic_eventBus.emit("app:setup", {
                        installed: false,
                        message: `Rewriting permissions to ${to}`,
                    })

                    await chmodRecursive(to, 0o755)
                }

                if (Array.isArray(prerequisite.moveDirs)) {
                    for (const dir of prerequisite.moveDirs) {
                        if (Array.isArray(dir.requireOs)) {
                            if (!dir.requireOs.includes(resolveOs())) {
                                continue
                            }
                        }

                        Log.info(`Moving ${dir.from} to ${dir.to}...`)

                        global._relic_eventBus.emit("app:setup", {
                            installed: false,
                            message: `Moving ${dir.from} to ${dir.to}`,
                        })

                        await fs.promises.rename(dir.from, dir.to)

                        if (dir.deleteParentBefore === true) {
                            await fs.promises.rm(path.dirname(dir.from), { recursive: true })
                        }
                    }
                }
            }

            global._relic_eventBus.emit("app:setup", {
                installed: true,
                message: null,
            })

            Log.info(`Prerequisite: ${prerequisite.id} is ready!`)
        } catch (error) {
            global._relic_eventBus.emit("app:setup", {
                installed: false,
                error: error,
                message: error.message,
            })

            Log.error("Aborting setup due to an error...")
            Log.error(error)

            throw error
        }

        Log.info(`All prerequisites are ready!`)
    }
}