#!/bin/bash
set -e
source /etc/profile

export VAGRANT_KEY="ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAQEA6NF8iallvQVp22WDkTkyrtvp9eWW6A8YVr+kz4TjGYe7gHzIw+niNltGEFHzD8+v1I2YJ6oXevct1YeS0o9HZyN1Q9qgCgzUFtdOKLv6IedplqoPkcmF0aYet2PkEDo3MlTBckFXPITAMzF8dJSIFo9D8HfdOV0IAdx4O7PtixWKn5y2hMNG0zQPyUecp4pzC6kivAIhyfHilFR61RGL+GPXQ2MWZWFYbAGjyiYJnAmCP3NOTd0jMZEnDkbUvxhMmBYSdETk1rRgm+R4LOzFUGaHqHDLKLX+FIPKcF96hrucXzcWyLbIbEgE98OHlnVYCzRdK8jlqm8tehUc9c9WhQ== vagrant insecure public key"

# Create vagrant user
if $(grep -q 'vagrant' /etc/shadow); then
  echo 'Skipping vagrant user creation'
elif $(grep -q 'ubuntu' /etc/shadow); then
  echo 'vagrant user does not exist, renaming ubuntu user...'
  mv /home/{ubuntu,vagrant}
  usermod -l vagrant -d /home/vagrant ubuntu
  groupmod -n vagrant ubuntu
  echo -n 'vagrant:vagrant' | chpasswd
  echo 'Renamed ubuntu user to vagrant and changed password.'
elif [ ${DISTRIBUTION} = 'centos' -o ${DISTRIBUTION} = 'fedora' ]; then
  echo 'Creating vagrant user...'
  useradd --create-home -s /bin/bash -u 1000 vagrant
  echo -n 'vagrant:vagrant' | chpasswd
  sed -i 's/^Defaults\s\+requiretty/# Defaults requiretty/' /etc/sudoers
else
  echo 'Creating vagrant user...'
  useradd --create-home -s /bin/bash vagrant
  echo -n 'vagrant:vagrant' | chpasswd
fi

# Configure SSH access
mkdir -p /home/vagrant/.ssh
echo $VAGRANT_KEY > /home/vagrant/.ssh/authorized_keys
chown -R vagrant /home/vagrant/.ssh
chmod +x /home/vagrant/.ssh
echo 'SSH credentials configured for the vagrant user.'

# Enable passwordless sudo for the vagrant user
echo "vagrant ALL=(ALL) NOPASSWD:ALL" > /etc/sudoers.d/vagrant
chmod 0440 /etc/sudoers.d/vagrant
echo 'Sudoers file created.'