From 855ce255ddee3fb63d4e2b584fe52583a8692531 Mon Sep 17 00:00:00 2001
From: romain lespinasse <romain.lespinasse@gmail.com>
Date: Wed, 6 Nov 2019 01:27:30 +0100
Subject: [PATCH] feat: add GITHUB_HEAD_REF_SLUG

---
 README.md     |  9 ++++++---
 entrypoint.sh | 10 +++++++++-
 2 files changed, 15 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 84b5908..df036dd 100644
--- a/README.md
+++ b/README.md
@@ -10,9 +10,10 @@ This action slug and expose some github variables.
 
 ## Environement Variables
 
-### `GITHUB_REF_SLUG`
-
-Slug from `GITHUB_REF` env variables
+| GitHub environment variable | Slug variable |
+| - | - |
+| GITHUB_REF | GITHUB_REF_SLUG |
+| GITHUB_HEAD_REF | GITHUB_HEAD_REF_SLUG |
 
 ## Example usage
 
@@ -21,4 +22,6 @@ Slug from `GITHUB_REF` env variables
 - name: Print slug variables
   run: |
     echo ${{ env.GITHUB_REF_SLUG }}
+    echo ${{ env.GITHUB_HEAD_REF_SLUG }}
+    echo ${{ env.GITHUB_BASE_REF_SLUG }}
 ```
diff --git a/entrypoint.sh b/entrypoint.sh
index a533ecb..d279890 100755
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -1,3 +1,11 @@
 #!/bin/sh -l
 
-echo ::set-env name=GITHUB_REF_SLUG::$(echo $GITHUB_REF | tr A-Z a-z | sed -r 's#refs/.*/##;s/[~\^]+//g;s/[^a-zA-Z0-9]+/-/g;s/^-+\|-+$//g' | cut -c1-63)
+slug_ref() {
+    echo $1 \
+         | tr A-Z a-z \
+         | sed -r 's#refs/.*/##;s/[~\^]+//g;s/[^a-zA-Z0-9]+/-/g;s/^-+\|-+$//g' \
+         | cut -c1-63
+}
+
+echo ::set-env name=GITHUB_REF_SLUG::$(slug_ref $GITHUB_REF)
+echo ::set-env name=GITHUB_HEAD_REF_SLUG::$(slug_ref $GITHUB_HEAD_REF)