feat: find brew package installation dir

packages/setup-brew/package.json

@@ -1,6 +1,6 @@
 {
   "name": "setup-brew",
-  "version": "1.0.2",
+  "version": "1.1.0",
   "description": "Setup brew and brew packages",
   "repository": "https://github.com/aminya/setup-cpp",
   "homepage": "https://github.com/aminya/setup-cpp/tree/master/packages/setup-brew",

packages/setup-brew/src/install-pack.ts

@@ -1,10 +1,11 @@
 import { join } from "path"
-import { info } from "ci-log"
+import { info, warning } from "ci-log"
 import { execaSync } from "execa"
 import which from "which"
 import type { InstallationInfo } from "./InstallationInfo.js"
 import type { BrewPackOptions } from "./install-pack-options.js"
-import { getBrewBinDir, setupBrew } from "./install.js"
+import { getBrewBinDir, getBrewBinPath, setupBrew } from "./install.js"
+import { brewPackInstallDir, brewPackNameAndVersion } from "./pack-install-dir.js"
 
 /* eslint-disable require-atomic-updates */
 let hasBrew = false
@@ -36,14 +37,13 @@ export async function installBrewPack(
     hasBrew = true
   }
 
-  const binDir = getBrewBinDir()
-  const brewPath = join(binDir, "brew")
+  const brewPath = getBrewBinPath()
 
-  // Args
   const args = [
     "install",
-    (version !== undefined && version !== "") ? `${name}@${version}` : name,
+    brewPackNameAndVersion(name, version),
   ]
+
   // Add options to args
   for (const [key, value] of Object.entries(options)) {
     if (typeof value === "boolean" && value) {
@@ -56,5 +56,12 @@ export async function installBrewPack(
   // brew is not thread-safe
   execaSync(brewPath, args, { stdio: "inherit" })
 
-  return { binDir }
+  const installDir = await brewPackInstallDir(name, version)
+
+  if (installDir === undefined) {
+    warning(`Failed to find installation directory for ${name} ${version}`)
+    return { binDir: getBrewBinDir(), installDir: undefined }
+  }
+
+  return { installDir, binDir: join(installDir, "bin") }
 }

packages/setup-brew/src/install.ts

@@ -74,6 +74,10 @@ export function getBrewBinDir() {
   return join(getBrewDir(), "bin")
 }
 
+export function getBrewBinPath() {
+  return join(getBrewBinDir(), "brew")
+}
+
 /**
  * Get the path where brew is installed
  * @returns {string} The path where brew is installed

packages/setup-brew/src/pack-install-dir.ts

@@ -0,0 +1,84 @@
+import { existsSync } from "fs"
+import { join } from "path"
+import { execa } from "execa"
+import { getBrewBinPath, getBrewDir } from "./install.js"
+
+export function brewPackNameAndVersion(name: string, version: string | undefined) {
+  return (version !== undefined && version !== "") ? `${name}@${version}` : name
+}
+
+/**
+ * Get the installation directory of a package
+ * @param name The name of the package
+ * @param nameAndVersion The name and version of the package
+ * @returns The installation directory of the package
+ */
+
+export async function brewPackInstallDir(name: string, version: string | undefined) {
+  const nameAndVersion = brewPackNameAndVersion(name, version)
+
+  // first try with --prefix
+  const nameAndVersionPrefix = await getBrewPackPrefix(nameAndVersion)
+  if (nameAndVersionPrefix !== undefined) {
+    return nameAndVersionPrefix
+  }
+
+  // try with --prefix name
+  const namePrefix = await getBrewPackPrefix(name)
+  if (namePrefix !== undefined) {
+    return namePrefix
+  }
+
+  // if that fails, try with searchInstallDir
+  return searchInstallDir(name, nameAndVersion)
+}
+
+async function getBrewPackPrefix(packArg: string) {
+  try {
+    const brewPath = getBrewBinPath()
+    return (await execa(brewPath, ["--prefix", packArg], { stdio: ["pipe", "inherit", "inherit"] })).stdout
+  } catch {
+    return undefined
+  }
+}
+/**
+ * Searches for the installation directory of a package
+ * @param name The name of the package
+ * @param nameAndVersion The name and version of the package
+ * @returns The installation directory of the package
+ */
+function searchInstallDir(name: string, nameAndVersion: string) {
+  const brewDir = getBrewDir()
+
+  // Check in opt directory (most common location)
+  const nameAndVersionOptDir = join(brewDir, "opt", nameAndVersion)
+  if (existsSync(nameAndVersionOptDir)) {
+    return nameAndVersionOptDir
+  }
+  const nameOptDir = join(brewDir, "opt", name)
+  if (existsSync(nameOptDir)) {
+    return nameOptDir
+  }
+
+  // Check in Cellar (where casks and some formulae are installed)
+  const nameAndVersionCellarDir = join(brewDir, "Cellar", nameAndVersion)
+  if (existsSync(nameAndVersionCellarDir)) {
+    return nameAndVersionCellarDir
+  }
+  const nameCellarDir = join(brewDir, "Cellar", name)
+  if (existsSync(nameCellarDir)) {
+    return nameCellarDir
+  }
+
+  // Check in lib directory
+  const nameAndVersionLibDir = join(brewDir, "lib", nameAndVersion)
+  if (existsSync(nameAndVersionLibDir)) {
+    return nameAndVersionLibDir
+  }
+  const nameLibDir = join(brewDir, "lib", name)
+  if (existsSync(nameLibDir)) {
+    return nameLibDir
+  }
+
+  return undefined
+}