Allow to install stack hook

haskell · Dec 30, 2024 · db0dac8 · db0dac8
1 parent f144b0d
commit db0dac8
Show file tree

Hide file tree

Showing 5 changed files with 94 additions and 37 deletions.
diff --git a/.github/workflows/test-ghcup.yaml b/.github/workflows/test-ghcup.yaml
@@ -22,6 +22,7 @@ jobs:
 
       - uses: ./
         with:
+          stack_hook: true
           version: ${{ matrix.version }}
 
       - run: ghcup config
@@ -52,6 +53,12 @@ jobs:
       - if: runner.os == 'Windows'
         run: ghcup run -m sh -- -c 'pacman --version'
 
+      - name: Stack hook test
+        run: |
+          ghcup install stack latest
+          cat $(stack path --stack-root)/hooks/ghc-install.sh
+        shell: bash
+
   vanilla-channel:
     strategy:
       matrix:

diff --git a/action.yml b/action.yml
@@ -13,6 +13,9 @@ inputs:
     description: Set the release-channels
     default: |
       GHCupURL
+  stack_hook:
+    description: Install the GHCup stack hook
+    default: false
 
 outputs:
   path:

diff --git a/ghcup/dist/index.js b/ghcup/dist/index.js
diff --git a/ghcup/src/index.ts b/ghcup/src/index.ts
@@ -1,10 +1,11 @@
-import { main } from "./main.ts"
+import { main, parseYAMLBoolean } from "./main.ts"
 import core from "@actions/core";
 
 try {
   main({
     version: core.getInput("version"),
-    release_channels: core.getMultilineInput("release-channels")
+    release_channels: core.getMultilineInput("release-channels"),
+    stack_hook: parseYAMLBoolean("stack_hook", core.getInput("stack_hook")),
   })
 } catch (error) {
   core.setFailed((error as Error).message);

diff --git a/ghcup/src/main.ts b/ghcup/src/main.ts
@@ -1,4 +1,6 @@
-import * as path from 'path'
+import * as path from 'path';
+import * as fs from 'fs';
+import os from 'os';
 import { chmod } from 'fs/promises';
 
 import tc from '@actions/tool-cache';
@@ -27,11 +29,13 @@ const ghcup_os_map: Map<Platform, GHCupOS> = new Map([
   ['freebsd', 'portbld-freebsd']
 ]);
 
-function ghcup_url(version: string, arch: GHCupArch, os: GHCupOS): string {
+const hook_url: string = 'https://www.haskell.org/ghcup/sh/hooks/stack/ghc-install.sh';
+
+function ghcup_url(version: string, arch: GHCupArch, gos: GHCupOS): string {
   if (version == 'latest') {
-    return `https://downloads.haskell.org/ghcup/${arch}-${os}-ghcup${ext}`;
+    return `https://downloads.haskell.org/ghcup/${arch}-${gos}-ghcup${ext}`;
   } else {
-    return `https://downloads.haskell.org/ghcup/${version}/${arch}-${os}-ghcup-${version}${ext}`;
+    return `https://downloads.haskell.org/ghcup/${version}/${arch}-${gos}-ghcup-${version}${ext}`;
   }
 }
 
@@ -46,12 +50,12 @@ async function ghcup(version: string) {
       throw `GHCup does not support architecture ${platform.arch}`;
     }
 
-    const os = ghcup_os_map.get(platform.platform);
-    if (os == undefined) {
+    const gos = ghcup_os_map.get(platform.platform);
+    if (gos == undefined) {
       throw `GHCup does not support platform ${platform.platform}`;
     }
 
-    const url = ghcup_url(version, arch, os);
+    const url = ghcup_url(version, arch, gos);
 
     const tempDirectory = process.env['RUNNER_TEMP'] || '';
     const ghcupExeName = `ghcup${ext}`;
@@ -69,9 +73,45 @@ async function ghcup(version: string) {
   }
 }
 
+function getStackRoot() {
+  if (platform.isWindows) {
+	const appdata = process.env['APPDATA'] || '';
+    return process.env['STACK_ROOT'] ?? path.join(appdata, 'stack');
+  } else {
+	const hdir = os.homedir();
+    return process.env['STACK_ROOT'] ?? path.join(hdir, '.stack');
+  }
+}
+
+async function installStackHook() {
+  const stack_root = getStackRoot();
+  const hook_dest = path.join(stack_root, 'hooks', 'ghc-install.sh')
+  fs.rmSync(hook_dest, {
+    force: true,
+  });
+  // we do not cache, it isn't versioned
+  const hookPath = await tc.downloadTool(hook_url, hook_dest);
+  if (!(platform.isWindows)) {
+    await chmod(hook_dest, "0765");
+  }
+  core.debug(`stack ghcup hook is at ${hookPath}`);
+}
+
+export function parseYAMLBoolean(name: string, val: string): boolean {
+  const trueValue = ['true', 'True', 'TRUE'];
+  const falseValue = ['false', 'False', 'FALSE'];
+  if (trueValue.includes(val)) return true;
+  if (falseValue.includes(val)) return false;
+  throw new TypeError(
+    `Action input "${name}" does not meet YAML 1.2 "Core Schema" specification: \n` +
+      `Supported boolean values: \`true | True | TRUE | false | False | FALSE\``
+  );
+}
+
 export type Opts = {
   version: string,
-  release_channels: string[]
+  release_channels: string[],
+  stack_hook: boolean
 }
 
 export async function main(opts: Opts) {
@@ -94,7 +134,12 @@ export async function main(opts: Opts) {
       core.debug(`GHCUP_MSYS2 is ${ghcup_msys2}`);
   }
 
+  if (opts.stack_hook) {
+	  installStackHook()
+  }
+
   await exec.exec(ghcupPath, [
     'config', 'set', 'url-source', JSON.stringify(opts.release_channels)
   ]);
 }
+