setup-system.sh

#!/usr/bin/env bash

# version modified by JLC for RPi4 2020/02/13
#
# 2020/02/03 JLC install_custom_raspiconfig() seems to be obsolete for RPi4
# 2020/04/07 JLC use $(hrpi-version) to do conditional process for rpi-3, rpi-4 and buster

username=$1
password=$2

install_custom_raspiconfig()
{
    wget https://raw.githubusercontent.com/poppy-project/raspi-config/master/raspi-config
    chmod +x raspi-config
    sudo chown root:root raspi-config
    sudo mv raspi-config /usr/bin/
}

setup_user()
{
    username=$1
    password=$2

    echo -e "\e[33mCreate a user \e[4m$username\e[0m."

    pass=$(perl -e 'print crypt($ARGV[0], "password")' "$password")
    sudo useradd -m -p "$pass" -s /bin/bash "$username"

    # copy groups from pi user
    user_groups=$(id -Gn pi)
    user_groups=${user_groups// /,}
    sudo usermod -a -G "$user_groups" "$username"

    echo "$username ALL=(ALL) NOPASSWD: ALL" | sudo tee /etc/sudoers.d/011_poppy-nopasswd
}

system_setup()
{
    # Change password for pi user
    sudo usermod -p "$(mkpasswd --method=sha-512 raspoppy)" pi

    # Add more langs (GB, US, FR)
    sudo sed -i 's/^#\s*\(en_GB.UTF-8 UTF-8\)/\1/g' /etc/locale.gen
    sudo sed -i 's/^#\s*\(en_US.UTF-8 UTF-8\)/\1/g' /etc/locale.gen
    sudo sed -i 's/^#\s*\(fr_FR.UTF-8 UTF-8\)/\1/g' /etc/locale.gen
    sudo locale-gen

    #JLC: don't know if the stuff bellow must be done with RPi4 under RaspBian buster ?
    #JLC: => anyway it can always be done by typing "sudo raspi-config" in aterminal !

    if [ "$(hrpi-version)" = "rpi-3" ]; then
	echo -e "\e[33mEnable camera.\e[0m"
    	echo "start_x=1" | sudo tee --append /boot/config.txt
    	echo "bcm2835-v4l2" | sudo tee /etc/modules-load.d/bcm2835-v4l2.conf

    	echo -e "\e[33mSetup serial communication.\e[0m"
    	sudo raspi-config --disable-serial-log
    	sudo tee --append /boot/config.txt > /dev/null <<EOF
init_uart_clock=16000000
dtoverlay=pi3-miniuart-bt
EOF
     fi
}

install_additional_packages()
{
    sudo apt-get update && sudo apt upgrade -y && sudo apt autoremove -y

    #JLC: added python3-venev & libatalas-base-dev for RaspBian buster:
    #JLC-2020/04/08: removed samba* & dhcpcd, added libhdf5-dev libhdf5-serial-dev libjasper-dev for opencv
    sudo apt-get install -y \
        build-essential unzip whois \
        network-manager \
        git \
        avahi-autoipd avahi-utils \
        libxslt-dev \
	python3-venv \
	libatlas-base-dev \
	libhdf5-dev libhdf5-serial-dev libjasper-dev libqtgui4 libqt4-test

    # board version utility
    #JLC: hrpi-version comptaible rpi-3 & rpi-4 is replaced by the new version included in the zip file
    sudo cp hrpi-version-2.sh /usr/bin/hrpi-version
    sudo chmod +x /usr/bin/hrpi-version
}

setup_network_tools()
{
    # samba
    sudo sed -i 's/map to guest = .*/map to guest = never/g' /etc/samba/smb.conf
    (echo "poppy"; echo "poppy") | sudo smbpasswd -s -a poppy

    # avahi services
    sudo tee /etc/avahi/services/ssh.service <<EOF
<?xml version="1.0" standalone='no'?>
<!DOCTYPE service-group SYSTEM "avahi-service.dtd">
<service-group>
  <name replace-wildcards="yes">SSH on %h</name>
  <service>
    <type>_ssh._tcp</type>
    <port>22</port>
  </service>
  <service>
    <type>_sftp-ssh._tcp</type>
    <port>22</port>
  </service>
</service-group>
EOF

    sudo tee /etc/avahi/services/poppy.service <<EOF
<?xml version="1.0" standalone='no'?>
<!DOCTYPE service-group SYSTEM "avahi-service.dtd">
<service-group>
  <name replace-wildcards="yes">poppy on %h</name>
  <service>
    <type>_poppy-robot._tcp</type>
    <port>9</port>
  </service>
  <service>
    <type>_device-info._tcp</type>
    <port>0</port>
  </service>
  <service>
    <type>_smb._tcp</type>
    <port>445</port>
  </service>
</service-group>
EOF
}

#JLC: install_additional_packages is run first to make hrpi-version available:
install_additional_packages
if [ "$(hrpi-version)" = "rpi-3" ]; then
    install_custom_raspiconfig
fi
setup_user "$username" "$password"
system_setup
setup_network_tools