# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: MIT-0
#!/bin/bash

yum -y -q install jq mysql amazon-efs-utils
# Get OOD Stack data
OOD_STACK_NAME=$1
REGION=$(curl http://169.254.169.254/latest/meta-data/placement/region)

OOD_STACK=$(aws cloudformation describe-stacks --stack-name $OOD_STACK_NAME --region $REGION )


S3_CONFIG_BUCKET=$(echo $OOD_STACK | jq -r '.Stacks[].Outputs[] | select(.OutputKey=="ClusterConfigBucket") | .OutputValue')
EFS_ID=$(echo $OOD_STACK | jq -r '.Stacks[].Outputs[] | select(.OutputKey=="EFSMountId") | .OutputValue')

# Copy Common Munge Key
aws s3 cp s3://$S3_CONFIG_BUCKET/munge.key /etc/munge/munge.key
chown munge: /etc/munge/munge.key
chmod 400 /etc/munge/munge.key
systemctl restart munge

# Add entry for fstab so mounts on restart
mkdir /shared
echo "$(curl -s http://169.254.169.254/latest/meta-data/placement/availability-zone).${EFS_ID}.efs.$REGION.amazonaws.com:/ /shared efs _netdev,noresvport,tls,iam 0 0" >> /etc/fstab
mount -a

# Add spack-users group
groupadd spack-users -g 4000

# change to using ad user login, this is no longer needed 
#/shared/copy_users.sh

#fix the sssd script so getent passws command can find the domain user
#This line allows the users to login without the domain name
sed -i 's/use_fully_qualified_names = True/use_fully_qualified_names = False/g' /etc/sssd/sssd.conf
#This line configure sssd to create the home directories in the shared folder
sed -i 's/fallback_homedir = \/home\/%u/fallback_homedir = \/shared\/home\/%u/' -i /etc/sssd/sssd.conf
sleep 1
systemctl restart sssd

## install remote desktop packages
## uncomment the following if you want to run interacctive remote desktop session in OOD
##
yum install nmap-ncat -y

cat > /etc/yum.repos.d/TurboVNC.repo <<  'EOF'
[TurboVNC]
name=TurboVNC official RPMs
baseurl=https://sourceforge.net/projects/turbovnc/files
gpgcheck=1
gpgkey=https://sourceforge.net/projects/turbovnc/files/VGL-GPG-KEY
       https://sourceforge.net/projects/turbovnc/files/VGL-GPG-KEY-1024
enabled=1
EOF

yum install turbovnc -y

amazon-linux-extras install python3.8
ln -sf /usr/bin/python3.8 /usr/bin/python3

pip3 install --no-input websockify
pip3 install --no-input jupyter

amazon-linux-extras install mate-desktop1.x -y

#
cat >> /etc/bashrc << 'EOF'
PATH=$PATH:/opt/TurboVNC/bin
#this is to fix the dconf permission error
export XDG_RUNTIME_DIR="$HOME/.cache/dconf"
EOF