[terraform] use oncreate.sh

Author: joshuanianji

src/terraform-cli-persistence/devcontainer-feature.json

@@ -17,5 +17,7 @@
         "ghcr.io/devcontainers/features/common-utils",
         "ghcr.io/meaningful-ooo/devcontainer-features/fish"
     ],
-    "postCreateCommand": "/usr/local/share/terraform-cli-persistence-post-create.sh"
+    "onCreateCommand": {
+        "terraform-cli-persistence-setup": "/usr/local/share/terraform-cli-persistence/scripts/oncreate.sh"
+    }
 }

src/terraform-cli-persistence/install.sh

@@ -1,48 +1,52 @@
 #!/bin/sh
-set -e
 
+USERNAME=${USERNAME:-${_REMOTE_USER}}
 FEATURE_ID="terraform-cli-persistence"
-
-echo "Activating feature '$FEATURE_ID'"
-echo "User: ${_REMOTE_USER}     User home: ${_REMOTE_USER_HOME}"
-
-if [ -z "$_REMOTE_USER" ] || [ -z "$_REMOTE_USER_HOME" ]; then
-    echo "***********************************************************************************"
-    echo "*** Require _REMOTE_USER and _REMOTE_USER_HOME to be set (by dev container CLI) ***"
-    echo "***********************************************************************************"
-    exit 1
-fi
-
-# make /dc/terraform-cli folder if doesn't exist
-mkdir -p "/dc/terraform-cli"
-
-# as to why we move around the folder, check `github-cli-persistence/install.sh`
-if [ -e "$_REMOTE_USER_HOME/.terraform.d" ]; then
-    echo "Moving existing .terraform.d folder to .terraform.d-old"
-    mv "$_REMOTE_USER_HOME/.terraform.d" "$_REMOTE_USER_HOME/.terraform.d-old"
-fi
-
-ln -s /dc/terraform-cli "$_REMOTE_USER_HOME/.terraform.d"
-# chown the entire `.config` folder because devcontainers creates
-# a `~/.config/vscode-dev-containers` folder later on
-chown -R "${_REMOTE_USER}:${_REMOTE_USER}" "$_REMOTE_USER_HOME/.terraform.d"
-
-# --- Generate a '$FEATURE_ID-post-create.sh' script to be executed by the 'postCreateCommand' lifecycle hook
-# Looks like this is the best way to run a script in lifecycle hooks
-# Source: https://github.com/devcontainers/features/blob/562305d37b97d47331d96306ffc2a0a3cce55e64/src/git-lfs/install.sh#L190C1-L190C109
-POST_CREATE_SCRIPT_PATH="/usr/local/share/$FEATURE_ID-post-create.sh"
-
-tee "$POST_CREATE_SCRIPT_PATH" >/dev/null \
-    <<'EOF'
-#!/bin/sh
+LIFECYCLE_SCRIPTS_DIR="/usr/local/share/${FEATURE_ID}/scripts"
 
 set -e
 
-# if the user is not root, chown /dc/aws-cli to the user
-if [ "$(id -u)" != "0" ]; then
-    echo "Running post-start.sh for user $USER"
-    sudo chown -R "$USER:$USER" /dc/terraform-cli
+create_cache_dir() {
+    if [ -d "$1" ]; then
+        echo "Cache directory $1 already exists. Skip creation..."
+    else
+        echo "Create cache directory $1..."
+        mkdir -p "$1"
+    fi
+
+    if [ -z "$2" ]; then
+        echo "No username provided. Skip chown..."
+    else
+        echo "Change owner of $1 to $2..."
+        chown -R "$2:$2" "$1"
+    fi
+}
+
+create_symlink_dir() {
+    local local_dir=$1
+    local cache_dir=$2
+    local username=$3
+
+    runuser -u "$username" -- mkdir -p "$(dirname "$local_dir")"
+    runuser -u "$username" -- mkdir -p "$cache_dir"
+
+    # if the folder we want to symlink already exists, the ln -s command will create a folder inside the existing folder
+    if [ -e "$local_dir" ]; then
+        echo "Moving existing $local_dir folder to $local_dir-old"
+        mv "$local_dir" "$local_dir-old"
+    fi
+
+    echo "Symlink $local_dir to $cache_dir for $username..."
+    runuser -u "$username" -- ln -s "$cache_dir" "$local_dir"
+}
+
+create_cache_dir "/dc/terraform-cli" "${USERNAME}"
+create_symlink_dir "$_REMOTE_USER_HOME/.terraform.d" "/dc/terraform-cli" "${USERNAME}"
+
+# Set Lifecycle scripts
+if [ -f oncreate.sh ]; then
+    mkdir -p "${LIFECYCLE_SCRIPTS_DIR}"
+    cp oncreate.sh "${LIFECYCLE_SCRIPTS_DIR}/oncreate.sh"
 fi
-EOF
 
-chmod 755 "$POST_CREATE_SCRIPT_PATH"
+echo "Finished installing $FEATURE_ID"

src/terraform-cli-persistence/oncreate.sh

@@ -0,0 +1,9 @@
+#!/bin/sh
+
+set -e
+
+# if the user is not root, chown /dc/aws-cli to the user
+if [ "$(id -u)" != "0" ]; then
+    echo "Running post-start.sh for user $USER"
+    sudo chown -R "$USER:$USER" /dc/terraform-cli
+fi