Cleaned up WORKSPACE/HOME usage in setup

- Removed HOME from SAVE_ENV since we don't set it anywhere
- Added WORKSPACE to SAVE_ENV as we do set that one
- Removed chown for /opt as that could break certain usages of the image and is not needed

Signed-off-by: Craig Magina <cmagina@redhat.com>

Author: cmagina

scripts/setup.sh

@@ -23,7 +23,6 @@ declare -a SAVE_VARS=(
 	"CUDA_VERSION"
 	"CUSTOM_LLVM"
 	"DISPLAY"
-	"HOME"
 	"INSTALL_JUPYTER"
 	"INSTALL_TOOLS"
 	"INSTALL_TRITON"
@@ -34,6 +33,7 @@ declare -a SAVE_VARS=(
 	"TRITON_CPU_BACKEND"
 	"USE_CCACHE"
 	"WAYLAND_DISPLAY"
+	"WORKSPACE"
 	"XDG_RUNTIME_DIR"
 )
 

scripts/setup_user.sh

@@ -110,10 +110,10 @@ create_user() {
 fix_permissions() {
 	echo "Fixing permissions for user $USERNAME ..."
 	chown "$USERNAME:$GROUP_ID" -R "$HOME"
-	chown "$USERNAME:$GROUP_ID" -R /opt
-	chown "$USERNAME:$GROUP_ID" -R "$WORKSPACE"
-	mkdir -p "/run/user/$USER_ID"
-	chown "$USERNAME:$GROUP_ID" "/run/user/$USER_ID"
+	[[ -n "$WORKSPACE" && -d $WORKSPACE ]] && chown "$USERNAME:$GROUP_ID" -R "$WORKSPACE"
+
+	mkdir -p "/run/user/$USER_UID"
+	chown "$USERNAME:$USER_GID" "/run/user/$USER_UID"
 }
 
 ##