From 9b7c5ba9ea699745d2f393914ea9352296ffa951 Mon Sep 17 00:00:00 2001
From: Yuki Okushi <huyuumi.dev@gmail.com>
Date: Thu, 31 Dec 2020 01:58:13 +0900
Subject: [PATCH] Add `custom_domain` key for github-actions/static-websites

---
 github-actions/static-websites/action.yml    | 4 ++++
 github-actions/static-websites/entrypoint.sh | 7 +++++++
 2 files changed, 11 insertions(+)

diff --git a/github-actions/static-websites/action.yml b/github-actions/static-websites/action.yml
index 2349c2bdf..99c90648a 100644
--- a/github-actions/static-websites/action.yml
+++ b/github-actions/static-websites/action.yml
@@ -20,6 +20,10 @@ inputs:
     description: 'AWS IAM credentials used to invalidate Cloudfront'
     required: false
     default: ''
+  custom_domain:
+    description: 'The custom domain for GitHub Pages'
+    required: false
+    default: ''
 
 runs:
   using: docker
diff --git a/github-actions/static-websites/entrypoint.sh b/github-actions/static-websites/entrypoint.sh
index eb36367b5..3efe66a5f 100755
--- a/github-actions/static-websites/entrypoint.sh
+++ b/github-actions/static-websites/entrypoint.sh
@@ -5,6 +5,7 @@ IFS=$'\n\t'
 deploy_dir="${GITHUB_WORKSPACE}/${INPUT_DEPLOY_DIR}"
 github_token="${INPUT_GITHUB_TOKEN}"
 cloudfront_distribution="${INPUT_CLOUDFRONT_DISTRIBUTION-}"
+custom_domain="${CUSTOM_DOMAIN-}"
 
 export AWS_ACCESS_KEY_ID="${INPUT_AWS_ACCESS_KEY_ID-}"
 export AWS_SECRET_ACCESS_KEY="${INPUT_AWS_SECRET_ACCESS_KEY-}"
@@ -16,6 +17,12 @@ touch "${deploy_dir}/.nojekyll"
 
 # Push the website to GitHub pages
 cd "${deploy_dir}"
+
+# Set custom domain if present.
+if [[ -n "${custom_domain}" ]]; then
+    echo "${custom_domain}" > ./CNAME
+fi
+
 rm -rf .git
 git init
 git config user.name "Deploy from CI"