#!/bin/bash
#
# .---.                  .              .
# |                      |              |
# |--- .--. .-.  .-.  .-.|  .-. .--.--. |.-.  .-. .--.  .-.
# |    |   (.-' (.-' (   | (   )|  |  | |   )(   )|  | (.-'
# '    '     --'  --'  -' -  -' '  '   -' -'   -' '   -  --'
#
#                    Freedom in the Cloud
#
# This command is run on initial install in order to set up a mesh router
#
# License
# =======
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

PROJECT_NAME='freedombone'

export TEXTDOMAIN=${PROJECT_NAME}-image-mesh
export TEXTDOMAINDIR="/usr/share/locale"

MY_USERNAME='fbone'
PEER_ID=
INSTALL_DIR=/root/build
INSTALL_LOG=/var/log/${PROJECT_NAME}.log

DEFAULT_USERNAME=fbone

ZERONET_URL='http://127.0.0.1:43110'
ZERONET_PORT=15441

TOX_NODES=
#TOX_NODES=(
#  '192.254.75.102,2607:5600:284::2,33445,951C88B7E75C867418ACDB5D273821372BB5BD652740BCDF623A4FA293E75D2F,Tox RELENG,US'
#  '144.76.60.215,2a01:4f8:191:64d6::1,33445,04119E835DF3E78BACF0F84235B300546AF8B936F035185E2A8E9E0A67C8924F,sonOfRa,DE'
#)

MESH_INSTALL_DIR=/var/lib
MESH_INSTALL_COMPLETED=/root/.mesh_setup_completed
MESH_INSTALL_SETUP=/root/.initial_mesh_setup
MESH_AMNESIC=/root/.amnesic

function create_ram_disk {
	ramdisk_size_mb=$1
	if [ ! -d /mnt/ramdisk ]; then
		mkdir -p /mnt/ramdisk
	fi
	if ! grep -q "ramdisk" /etc/fstab; then
		mount -t tmpfs -o size=${ramdisk_size_mb}m tmpfs /mnt/ramdisk
		echo "tmpfs       /mnt/ramdisk tmpfs   nodev,nosuid,noexec,nodiratime,size=${ramdisk_size_mb}M   0 0" >> /etc/fstab
	fi
}

function set_hostname {
	DEFAULT_DOMAIN_NAME="$1"

	echo "$DEFAULT_DOMAIN_NAME" > /etc/hostname
	hostname $DEFAULT_DOMAIN_NAME

	if grep -q "127.0.1.1" /etc/hosts; then
		sed -i "s/127.0.1.1.*/127.0.1.1       $DEFAULT_DOMAIN_NAME/g" /etc/hosts
	else
		echo "127.0.1.1  $DEFAULT_DOMAIN_NAME" >> /etc/hosts
	fi
}

function change_avahi_name {
	decarray=( 1 2 3 4 5 6 7 8 9 0 )
	PEER_ID=${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}${decarray[$RANDOM%10]}
	sed -i "s|#host-name=.*|host-name=P$PEER_ID|g" /etc/avahi/avahi-daemon.conf
	sed -i "s|host-name=.*|host-name=P$PEER_ID|g" /etc/avahi/avahi-daemon.conf
	set_hostname P$PEER_ID

	echo "New avahi name for this peer is P$PEER_ID"
	toilet "mesh-$PEER_ID"
	echo $"avahi name changed to P${PEER_ID}.local" >> $INSTALL_LOG
}

function regenerate_ssh_host_keys {
	echo $"Regenerating ssh host keys" >> $INSTALL_LOG
	rm -f /etc/ssh/ssh_host_*
	dpkg-reconfigure openssh-server
	echo $'ssh host keys regenerated' >> $INSTALL_LOG
	# remove small moduli
	awk '$5 > 2000' /etc/ssh/moduli > ~/moduli
	mv ~/moduli /etc/ssh/moduli
	echo $'ssh small moduli removed' >> $INSTALL_LOG
}

function configure_zeronet {
	sed -i "s|ExecStart=.*|ExecStart=/usr/bin/python zeronet.py --ip_external P${PEER_ID}.local --trackers_file $MESH_INSTALL_DIR/zeronet/bootstrap|g" /etc/systemd/system/zeronet.service
	echo $"zeronet daemon updated to P${PEER_ID}.local" >> $INSTALL_LOG
}

function remove_zeronet_blog {
	blog_address=$(cat $MESH_INSTALL_DIR/zeronet/blog.txt | grep "Site address" | awk -F ':' '{print $2}')
	ZERONET_BLOG_ADDRESS=${blog_address//[[:blank:]]/}
	if [ -d $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS ]; then
		shred -zu $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS/*.json
		shred -zu $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS/*.html
		shred -zu $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS/data/*
		shred -zu $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS/img/post/*
		rm -rf $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS/img/post
		shred -zu $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS/img/*
		rm -rf $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS
	fi
}

function configure_zeronet_blog {
	if [ -f $MESH_INSTALL_COMPLETED ]; then
		remove_zeronet_blog
	fi

	echo $'Updating ZeroNet Blog' >> $INSTALL_LOG
	ZERONET_DEFAULT_BLOG_TITLE="${MY_USERNAME}'s Blog"

	cd $MESH_INSTALL_DIR/zeronet
	python zeronet.py --batch siteCreate 2> $MESH_INSTALL_DIR/zeronet/blog.txt
	if [ ! -f $MESH_INSTALL_DIR/zeronet/blog.txt ]; then
		echo $'Unable to create ZeroNet blog' >> $INSTALL_LOG
		exit 7386
	fi
	blog_address=$(cat $MESH_INSTALL_DIR/zeronet/blog.txt | grep "Site address" | awk -F ':' '{print $2}')
	blog_private_key=$(cat $MESH_INSTALL_DIR/zeronet/blog.txt | grep "Site private key" | awk -F ':' '{print $2}')
	ZERONET_BLOG_ADDRESS=${blog_address//[[:blank:]]/}
	ZERONET_BLOG_PRIVATE_KEY=${blog_private_key//[[:blank:]]/}

	if [ ${#ZERONET_BLOG_ADDRESS} -lt 20 ]; then
		echo $"Address:    $ZERONET_BLOG_ADDRESS" >> $INSTALL_LOG
		echo $"Public key: $ZERONET_BLOG_PRIVATE_KEY" >> $INSTALL_LOG
		echo $'Unable to create zeronet blog address' >> $INSTALL_LOG
		exit 7358
	fi

	if [ ${#ZERONET_BLOG_PRIVATE_KEY} -lt 20 ]; then
		echo $"Address:    $ZERONET_BLOG_ADDRESS" >> $INSTALL_LOG
		echo $"Public key: $ZERONET_BLOG_PRIVATE_KEY" >> $INSTALL_LOG
		echo $'Unable to create zeronet blog private key' >> $INSTALL_LOG
		exit 1639
	fi

	if [ ! -d "$MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS" ]; then
		echo $"Unable to find site directory: $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS" >> $INSTALL_LOG
		exit 7638
	fi

	echo $"ZeroNet Blog address:     $ZERONET_BLOG_ADDRESS" >> $INSTALL_LOG
	echo $"ZeroNet Blog private key: $ZERONET_BLOG_PRIVATE_KEY" >> $INSTALL_LOG
	cp -r $MESH_INSTALL_DIR/zeronet/ZeroBlog/* $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS

	if [ ! -d $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS/data ]; then
		mkdir $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS/data
	fi
	cp $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS/data-default/data.json $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS/data
	sed -i "s/MyZeroBlog/$ZERONET_DEFAULT_BLOG_TITLE/g" $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS/data/data.json
	sed -i "s/My ZeroBlog./$ZERONET_DEFAULT_BLOG_TAGLINE/g" $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS/data/data.json
	sed -i "s/ZeroBlog Demo/$ZERONET_DEFAULT_BLOG_TITLE/g" $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS/index.html
	sed -i "s|<h3 class=\"description\">.*|<h3 class=\"description\">$ZERONET_DEFAULT_BLOG_TAGLINE</h3>|g" $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS/index.html
	sed -i "s/Blogging platform Demo/Blogging platform/g" $MESH_INSTALL_DIR/zeronet/data/$ZERONET_BLOG_ADDRESS/content.json
	python zeronet.py siteSign $ZERONET_BLOG_ADDRESS $ZERONET_BLOG_PRIVATE_KEY

	# update the avahi service
	echo '<?xml version="1.0" standalone="no"?><!--*-nxml-*-->' > /tmp/zeronet-blog.service
	echo '<!DOCTYPE service-group SYSTEM "avahi-service.dtd">' >> /tmp/zeronet-blog.service
	echo '<service-group>' >> /tmp/zeronet-blog.service
	echo '  <name replace-wildcards="yes">%h ZeroNet Blog</name>' >> /tmp/zeronet-blog.service
	echo '  <service>' >> /tmp/zeronet-blog.service
	echo '    <type>_zeronet._udp</type>' >> /tmp/zeronet-blog.service
	echo "    <port>$ZERONET_PORT</port>" >> /tmp/zeronet-blog.service
	echo "    <txt-record>$ZERONET_URL/$ZERONET_BLOG_ADDRESS</txt-record>" >> /tmp/zeronet-blog.service
	echo '  </service>' >> /tmp/zeronet-blog.service
	echo '</service-group>' >> /tmp/zeronet-blog.service
	cp /tmp/zeronet-blog.service /etc/avahi/services/zeronet-blog.service

	if [ ! -d /home/${MY_USERNAME}/.config/zeronet ]; then
		mkdir -p /home/${MY_USERNAME}/.config/zeronet
		chown -R ${MY_USERNAME}:${MY_USERNAME} /home/${MY_USERNAME}/.config
	fi
	echo "$ZERONET_URL/$ZERONET_BLOG_ADDRESS" > /home/${MY_USERNAME}/.config/zeronet/myblog

	sed -i "s|ZeroNet Blog address.*|ZeroNet Blog address: $ZERONET_BLOG_ADDRESS|g" /home/${MY_USERNAME}/README
	sed -i "s|ZeroNet Blog private key.*|ZeroNet Blog private key: $ZERONET_BLOG_PRIVATE_KEY|g" /home/${MY_USERNAME}/README

	if [ -d ${MESH_INSTALL_DIR}/zeronet ]; then
		chown -R zeronet:zeronet ${MESH_INSTALL_DIR}/zeronet
	fi

	echo $'Update of ZeroNet Blog completed' >> $INSTALL_LOG
}

function remove_zeronet_mail {
	mail_address=$(cat $MESH_INSTALL_DIR/zeronet/mail.txt | grep "Site address" | awk -F ':' '{print $2}')
	ZERONET_MAIL_ADDRESS=${mail_address//[[:blank:]]/}
	if [ -d $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS ]; then
		shred -zu $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS/*.json
		shred -zu $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS/*.html
		shred -zu $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS/data/users/*
		rm -rf $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS/data/users
		shred -zu $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS/data/*
		rm -rf $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS
	fi
}

function configure_zeronet_mail {
	if [ -f $MESH_INSTALL_COMPLETED ]; then
		remove_zeronet_mail
	fi

	echo $'Updating ZeroNet Mail' >> $INSTALL_LOG
	ZERONET_DEFAULT_MAIL_TITLE="${MY_USERNAME}'s Mail"

	cd $MESH_INSTALL_DIR/zeronet
	python zeronet.py --batch siteCreate 2> $MESH_INSTALL_DIR/zeronet/mail.txt
	if [ ! -f $MESH_INSTALL_DIR/zeronet/mail.txt ]; then
		echo $'Unable to create ZeroNet mail' >> $INSTALL_LOG
		exit 72574
	fi
	mail_address=$(cat $MESH_INSTALL_DIR/zeronet/mail.txt | grep "Site address" | awk -F ':' '{print $2}')
	mail_private_key=$(cat $MESH_INSTALL_DIR/zeronet/mail.txt | grep "Site private key" | awk -F ':' '{print $2}')
	ZERONET_MAIL_ADDRESS=${mail_address//[[:blank:]]/}
	ZERONET_MAIL_PRIVATE_KEY=${mail_private_key//[[:blank:]]/}

	if [ ${#ZERONET_MAIL_ADDRESS} -lt 20 ]; then
		echo $"Address:    $ZERONET_MAIL_ADDRESS" >> $INSTALL_LOG
		echo $"Public key: $ZERONET_MAIL_PRIVATE_KEY" >> $INSTALL_LOG
		echo $'Unable to create zeronet mail address' >> $INSTALL_LOG
		exit 7358
	fi

	if [ ${#ZERONET_MAIL_PRIVATE_KEY} -lt 20 ]; then
		echo $"Address:    $ZERONET_MAIL_ADDRESS" >> $INSTALL_LOG
		echo $"Public key: $ZERONET_MAIL_PRIVATE_KEY" >> $INSTALL_LOG
		echo $'Unable to create zeronet mail private key' >> $INSTALL_LOG
		exit 1639
	fi

	if [ ! -d "$MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS" ]; then
		echo $"Unable to find site directory: $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS" >> $INSTALL_LOG
		exit 7638
	fi

	echo $"ZeroNet Mail address:     $ZERONET_MAIL_ADDRESS" >> $INSTALL_LOG
	echo $"ZeroNet Mail private key: $ZERONET_MAIL_PRIVATE_KEY" >> $INSTALL_LOG
	cp -r $MESH_INSTALL_DIR/zeronet/ZeroMail/* $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS

	if [ ! -d $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS/data ]; then
		mkdir $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS/data
	fi
	cp $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS/data-default/data.json $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS/data
	sed -i "s/MyZeroMail/$ZERONET_DEFAULT_MAIL_TITLE/g" $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS/data/data.json
	sed -i "s/My ZeroMail./$ZERONET_DEFAULT_MAIL_TAGLINE/g" $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS/data/data.json
	sed -i "s/ZeroMail Demo/$ZERONET_DEFAULT_MAIL_TITLE/g" $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS/index.html
	sed -i "s|<h3 class=\"description\">.*|<h3 class=\"description\">$ZERONET_DEFAULT_MAIL_TAGLINE</h3>|g" $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS/index.html
	sed -i "s/Mailging platform Demo/Mailging platform/g" $MESH_INSTALL_DIR/zeronet/data/$ZERONET_MAIL_ADDRESS/content.json
	python zeronet.py siteSign $ZERONET_MAIL_ADDRESS $ZERONET_MAIL_PRIVATE_KEY

	# Add an avahi service
	echo '<?xml version="1.0" standalone="no"?><!--*-nxml-*-->' > /tmp/zeronet-mail.service
	echo '<!DOCTYPE service-group SYSTEM "avahi-service.dtd">' >> /tmp/zeronet-mail.service
	echo '<service-group>' >> /tmp/zeronet-mail.service
	echo '  <name replace-wildcards="yes">%h ZeroNet Mail</name>' >> /tmp/zeronet-mail.service
	echo '  <service>' >> /tmp/zeronet-mail.service
	echo '    <type>_zeronet._udp</type>' >> /tmp/zeronet-mail.service
	echo "    <port>$ZERONET_PORT</port>" >> /tmp/zeronet-mail.service
	echo "    <txt-record>$ZERONET_URL/$ZERONET_MAIL_ADDRESS</txt-record>" >> /tmp/zeronet-mail.service
	echo '  </service>' >> /tmp/zeronet-mail.service
	echo '</service-group>' >> /tmp/zeronet-mail.service
	cp /tmp/zeronet-mail.service /etc/avahi/services/zeronet-mail.service

	if [ ! -d /home/${MY_USERNAME}/.config/zeronet ]; then
		mkdir -p /home/${MY_USERNAME}/.config/zeronet
		chown -R ${MY_USERNAME}:${MY_USERNAME} /home/${MY_USERNAME}/.config
	fi
	echo "$ZERONET_URL/$ZERONET_MAIL_ADDRESS" > /home/${MY_USERNAME}/.config/zeronet/mymail

	sed -i "s|ZeroNet Mail address.*|ZeroNet Mail address: $ZERONET_MAIL_ADDRESS|g" /home/${MY_USERNAME}/README
	sed -i "s|ZeroNet Mail private key.*|ZeroNet Mail private key: $ZERONET_MAIL_PRIVATE_KEY|g" /home/${MY_USERNAME}/README

	if [ -d ${MESH_INSTALL_DIR}/zeronet ]; then
		chown -R zeronet:zeronet ${MESH_INSTALL_DIR}/zeronet
	fi

	echo $'Update of ZeroNet Mail completed' >> $INSTALL_LOG
}

function remove_zeronet_forum {
	forum_address=$(cat $MESH_INSTALL_DIR/zeronet/forum.txt | grep "Site address" | awk -F ':' '{print $2}')
	ZERONET_FORUM_ADDRESS=${forum_address//[[:blank:]]/}
	if [ -d $MESH_INSTALL_DIR/zeronet/data/$ZERONET_FORUM_ADDRESS ]; then
		shred -zu $MESH_INSTALL_DIR/zeronet/data/$ZERONET_FORUM_ADDRESS/data/users/*
		rm -rf $MESH_INSTALL_DIR/zeronet/data/$ZERONET_FORUM_ADDRESS/data/users
		shred -zu $MESH_INSTALL_DIR/zeronet/data/$ZERONET_FORUM_ADDRESS/data/*
		shred -zu $MESH_INSTALL_DIR/zeronet/data/$ZERONET_FORUM_ADDRESS/*.html
		shred -zu $MESH_INSTALL_DIR/zeronet/data/$ZERONET_FORUM_ADDRESS/*.json
		rm -rf $MESH_INSTALL_DIR/zeronet/data/$ZERONET_FORUM_ADDRESS
	fi
}

function configure_zeronet_forum {
	if [ -f $MESH_INSTALL_COMPLETED ]; then
		remove_zeronet_forum
	fi
	echo $'Updating ZeroNet Forum' >> $INSTALL_LOG
	ZERONET_DEFAULT_FORUM_TITLE="${MY_USERNAME}'s Forum"

	cd $MESH_INSTALL_DIR/zeronet
	python zeronet.py --batch siteCreate 2> $MESH_INSTALL_DIR/zeronet/forum.txt
	if [ ! -f $MESH_INSTALL_DIR/zeronet/forum.txt ]; then
		echo $'Unable to create ZeroNet forum' >> $INSTALL_LOG
		exit 47962
	fi
	forum_address=$(cat $MESH_INSTALL_DIR/zeronet/forum.txt | grep "Site address" | awk -F ':' '{print $2}')
	forum_private_key=$(cat $MESH_INSTALL_DIR/zeronet/forum.txt | grep "Site private key" | awk -F ':' '{print $2}')
	ZERONET_FORUM_ADDRESS=${forum_address//[[:blank:]]/}
	ZERONET_FORUM_PRIVATE_KEY=${forum_private_key//[[:blank:]]/}

	if [ ${#ZERONET_FORUM_ADDRESS} -lt 20 ]; then
		echo $"Address:    $ZERONET_FORUM_ADDRESS" >> $INSTALL_LOG
		echo $"Public key: $ZERONET_FORUM_PRIVATE_KEY" >> $INSTALL_LOG
		echo $'Unable to create zeronet forum address' >> $INSTALL_LOG
		exit 76352
	fi

	if [ ${#ZERONET_FORUM_PRIVATE_KEY} -lt 20 ]; then
		echo $"Address:    $ZERONET_FORUM_ADDRESS" >> $INSTALL_LOG
		echo $"Public key: $ZERONET_FORUM_PRIVATE_KEY" >> $INSTALL_LOG
		echo $'Unable to create zeronet forum private key' >> $INSTALL_LOG
		exit 87356
	fi

	if [ ! -d "$MESH_INSTALL_DIR/zeronet/data/$ZERONET_FORUM_ADDRESS" ]; then
		echo $"Unable to find site directory: $MESH_INSTALL_DIR/zeronet/data/$ZERONET_FORUM_ADDRESS" >> $INSTALL_LOG
		exit 7638
	fi

	echo $"Forum address:     $ZERONET_FORUM_ADDRESS"
	echo $"Forum private key: $ZERONET_FORUM_PRIVATE_KEY"
	cp -r $MESH_INSTALL_DIR/zeronet/ZeroTalk/* $MESH_INSTALL_DIR/zeronet/data/$ZERONET_FORUM_ADDRESS
	sed -i "s/ZeroBoard/$ZERONET_DEFAULT_FORUM_TITLE/g" $MESH_INSTALL_DIR/zeronet/data/$ZERONET_FORUM_ADDRESS/index.html
	sed -i "s/ZeroTalk/$ZERONET_DEFAULT_FORUM_TITLE/g" $MESH_INSTALL_DIR/zeronet/data/$ZERONET_FORUM_ADDRESS/index.html
	sed -i "s|Demo for dynamic, decentralized content publishing.|$ZERONET_DEFAULT_FORUM_TAGLINE|g" $MESH_INSTALL_DIR/zeronet/data/$ZERONET_FORUM_ADDRESS/index.html
	sed -i 's/Messaging Board Demo/Messaging Board/g' $MESH_INSTALL_DIR/zeronet/data/$ZERONET_FORUM_ADDRESS/content.json
	sed -i "s/ZeroBoard/$ZERONET_DEFAULT_FORUM_TITLE/g" $MESH_INSTALL_DIR/zeronet/data/$ZERONET_FORUM_ADDRESS/content.json
	python zeronet.py siteSign $ZERONET_FORUM_ADDRESS $ZERONET_FORUM_PRIVATE_KEY --inner_path data/users/content.json

	# Add an avahi service
	echo '<?xml version="1.0" standalone="no"?><!--*-nxml-*-->' > /tmp/zeronet-forum.service
	echo '<!DOCTYPE service-group SYSTEM "avahi-service.dtd">' >> /tmp/zeronet-forum.service
	echo '<service-group>' >> /tmp/zeronet-forum.service
	echo '  <name replace-wildcards="yes">%h ZeroNet Forum</name>' >> /tmp/zeronet-forum.service
	echo '  <service>' >> /tmp/zeronet-forum.service
	echo '    <type>_zeronet._udp</type>' >> /tmp/zeronet-forum.service
	echo "    <port>$ZERONET_PORT</port>" >> /tmp/zeronet-forum.service
	echo "    <txt-record>$ZERONET_URL/$ZERONET_FORUM_ADDRESS</txt-record>" >> /tmp/zeronet-forum.service
	echo '  </service>' >> /tmp/zeronet-forum.service
	echo '</service-group>' >> /tmp/zeronet-forum.service
	sudo cp /tmp/zeronet-forum.service /etc/avahi/services/zeronet-forum.service

	if [ ! -d /home/${MY_USERNAME}/.config/zeronet ]; then
		mkdir -p /home/${MY_USERNAME}/.config/zeronet
		chown -R ${MY_USERNAME}:${MY_USERNAME} /home/${MY_USERNAME}/.config
	fi
	echo "$ZERONET_URL/$ZERONET_FORUM_ADDRESS" > /home/${MY_USERNAME}/.config/zeronet/myforum

	sed -i "s|ZeroNet Forum address.*|ZeroNet Forum address: $ZERONET_FORUM_ADDRESS|g" /home/${MY_USERNAME}/README
	sed -i "s|ZeroNet Forum private key.*|ZeroNet Forum private key: $ZERONET_FORUM_PRIVATE_KEY|g" /home/${MY_USERNAME}/README

	if [ -d ${MESH_INSTALL_DIR}/zeronet ]; then
		chown -R zeronet:zeronet ${MESH_INSTALL_DIR}/zeronet
	fi

	echo $'Update of ZeroNet Forum completed' >> $INSTALL_LOG
}

function remove_zeronet_id {
	zeroid_address=$(cat $MESH_INSTALL_DIR/zeronet/zeroid.txt | grep "Site address" | awk -F ':' '{print $2}')
	ZERONET_ID_ADDRESS=${zeroid_address//[[:blank:]]/}
	if [ -d $MESH_INSTALL_DIR/zeronet/data/$ZERONET_ID_ADDRESS ]; then
		shred -zu $MESH_INSTALL_DIR/zeronet/data/$ZERONET_ID_ADDRESS/*
		rm -rf $MESH_INSTALL_DIR/zeronet/data/$ZERONET_ID_ADDRESS
	fi
}

function configure_zeronet_id {
	if [ -f $MESH_INSTALL_COMPLETED ]; then
		remove_zeronet_id
	fi
	echo $'Updating ZeroID' >> $INSTALL_LOG

	cd $MESH_INSTALL_DIR/zeronet
	python zeronet.py --batch siteCreate 2> $MESH_INSTALL_DIR/zeronet/zeroid.txt
	if [ ! -f $MESH_INSTALL_DIR/zeronet/zeroid.txt ]; then
		echo $'Unable to create ZeroID' >> $INSTALL_LOG
		exit 47962
	fi
	zeroid_address=$(cat $MESH_INSTALL_DIR/zeronet/zeroid.txt | grep "Site address" | awk -F ':' '{print $2}')
	zeroid_private_key=$(cat $MESH_INSTALL_DIR/zeronet/zeroid.txt | grep "Site private key" | awk -F ':' '{print $2}')
	ZERONET_ID_ADDRESS=${zeroid_address//[[:blank:]]/}
	ZERONET_ID_PRIVATE_KEY=${zeroid_private_key//[[:blank:]]/}

	if [ ${#ZERONET_ID_ADDRESS} -lt 20 ]; then
		echo $"Address:    $ZERONET_ID_ADDRESS" >> $INSTALL_LOG
		echo $"Public key: $ZERONET_ID_PRIVATE_KEY" >> $INSTALL_LOG
		echo $'Unable to create ZeroID address' >> $INSTALL_LOG
		exit 76352
	fi

	if [ ${#ZERONET_ID_PRIVATE_KEY} -lt 20 ]; then
		echo $"Address:    $ZERONET_ID_ADDRESS" >> $INSTALL_LOG
		echo $"Public key: $ZERONET_ID_PRIVATE_KEY" >> $INSTALL_LOG
		echo $'Unable to create ZeroID private key' >> $INSTALL_LOG
		exit 87356
	fi

	if [ ! -d "$MESH_INSTALL_DIR/zeronet/data/$ZERONET_ID_ADDRESS" ]; then
		echo $"Unable to find site directory: $MESH_INSTALL_DIR/zeronet/data/$ZERONET_ID_ADDRESS" >> $INSTALL_LOG
		exit 378434
	fi

	echo $"ZeroID address:     $ZERONET_ID_ADDRESS"
	echo $"ZeroID private key: $ZERONET_ID_PRIVATE_KEY"
	cp -r $MESH_INSTALL_DIR/zeronet/ZeroID/* $MESH_INSTALL_DIR/zeronet/data/$ZERONET_ID_ADDRESS
	# TODO
	sed -i "s///g" $MESH_INSTALL_DIR/zeronet/data/$ZERONET_ID_ADDRESS/content.json
	python zeronet.py siteSign $ZERONET_ID_ADDRESS $ZERONET_ID_PRIVATE_KEY --inner_path data/users/content.json

	# Add an avahi service
	echo '<?xml version="1.0" standalone="no"?><!--*-nxml-*-->' > /tmp/zeronet-id.service
	echo '<!DOCTYPE service-group SYSTEM "avahi-service.dtd">' >> /tmp/zeronet-id.service
	echo '<service-group>' >> /tmp/zeronet-id.service
	echo '  <name replace-wildcards="yes">%h ZeroNet ID Service</name>' >> /tmp/zeronet-id.service
	echo '  <service>' >> /tmp/zeronet-id.service
	echo '    <type>_zeronet._udp</type>' >> /tmp/zeronet-id.service
	echo "    <port>$ZERONET_PORT</port>" >> /tmp/zeronet-id.service
	echo "    <txt-record>$ZERONET_URL/$ZERONET_ID_ADDRESS</txt-record>" >> /tmp/zeronet-id.service
	echo '  </service>' >> /tmp/zeronet-id.service
	echo '</service-group>' >> /tmp/zeronet-id.service
	sudo cp /tmp/zeronet-id.service /etc/avahi/services/zeronet-id.service

	if [ ! -d /home/${MY_USERNAME}/.config/zeronet ]; then
		mkdir -p /home/${MY_USERNAME}/.config/zeronet
		chown -R ${MY_USERNAME}:${MY_USERNAME} /home/${MY_USERNAME}/.config
	fi
	echo "$ZERONET_URL/$ZERONET_ID_ADDRESS" > /home/${MY_USERNAME}/.config/zeronet/myzeroid

	sed -i "s|ZeroID address.*|ZeroID address: $ZERONET_ID_ADDRESS|g" /home/${MY_USERNAME}/README
	sed -i "s|ZeroID private key.*|ZeroID private key: $ZERONET_ID_PRIVATE_KEY|g" /home/${MY_USERNAME}/README

	if [ -d ${MESH_INSTALL_DIR}/zeronet ]; then
		chown -R zeronet:zeronet ${MESH_INSTALL_DIR}/zeronet
	fi

	echo $'Update of ZeroID completed' >> $INSTALL_LOG
}

function amnesic_remove_tox {
	echo $'Clearing qtox client settings' >> $INSTALL_LOG
	if [ -d /home/${MY_USERNAME}/.config/tox ]; then
		shred -zu /home/${MY_USERNAME}/.config/tox/*.tox
		shred -zu /home/${MY_USERNAME}/.config/tox/*.db
		shred -zu /home/${MY_USERNAME}/.config/tox/*.ini
		rm -rf /home/${MY_USERNAME}/.config/tox
	fi
}

function configure_toxcore {
	echo $'Configuring toxcore' >> $INSTALL_LOG

	TOXIC_FILE=$(cat /usr/share/${PROJECT_NAME}/apps/${PROJECT_NAME}-app-tox | grep "TOXIC_FILE=" | head -n 1 | awk -F '=' '{print $2}')

	echo $'Enabling toxcore daemon' >> $INSTALL_LOG
	systemctl enable tox-bootstrapd.service
	echo $'Regenerating Tox bootstrap node keys' >> $INSTALL_LOG
	systemctl stop tox-bootstrapd.service
	if [ -f /var/lib/tox-bootstrapd/keys ]; then
		rm /var/lib/tox-bootstrapd/keys
	fi
	systemctl start tox-bootstrapd.service
	# sleep for a while so that the tox keys can be generated
	sleep 30
	TOX_BOOTSTRAP_ID_FILE=/var/lib/tox-bootstrapd/pubkey.txt
	TOX_PUBLIC_KEY=$(cat /var/log/syslog | grep tox | grep "Public Key" | awk -F ' ' '{print $8}' | tail -1)
	if [ ${#TOX_PUBLIC_KEY} -lt 30 ]; then
		echo $'WARNING: Could not obtain the tox node public key' >> $INSTALL_LOG
		exit 46362
	fi

	# save the public key for later reference
	echo "$TOX_PUBLIC_KEY" > $TOX_BOOTSTRAP_ID_FILE
	echo $'Configured toxcore' >> $INSTALL_LOG

	amnesic_remove_tox
	echo $'toxic client configured' >> $INSTALL_LOG
}

function disable_password_logins {
	if [ ! -d /home/$DEFAULT_USERNAME ]; then
		echo $"Home directory for $DEFAULT_USERNAME not found"
		exit 3682572
	fi

	# create an authorized_keys file
	if [ ! -d /home/$DEFAULT_USERNAME/.ssh ]; then
		mkdir -p /home/$DEFAULT_USERNAME/.ssh
	fi
	if [ ! -f /home/$DEFAULT_USERNAME/.ssh/authorized_keys ]; then
		touch /home/$DEFAULT_USERNAME/.ssh/authorized_keys
	fi
	chown -R $DEFAULT_USERNAME:$DEFAULT_USERNAME /home/$DEFAULT_USERNAME/.ssh

	# disable password logins
	sed -i 's|#PasswordAuthentication.*|PasswordAuthentication no|g' /etc/ssh/sshd_config
	sed -i 's|PasswordAuthentication.*|PasswordAuthentication no|g' /etc/ssh/sshd_config
}

function disable_screen_lock {
	dconf write /org/gnome/desktop/screensaver/lock-enabled false
}

function enable_batman_daemon {
	systemctl enable batman
	systemctl daemon-reload
}

function mesh_amnesic {
	if [ ! -f $MESH_AMNESIC ]; then
		return
	fi

	echo '#!/bin/bash' > /usr/bin/amnesic
	echo '' >> /usr/bin/amnesic
	echo "shred -zu /var/log/${PROJECT_NAME}*" >> /usr/bin/amnesic
	echo '' >> /usr/bin/amnesic
	echo 'MY_USERNAME=$1' >> /usr/bin/amnesic
	echo -n "${PROJECT_NAME}-image-mesh " >> /usr/bin/amnesic
	echo '$MY_USERNAME amnesic' >> /usr/bin/amnesic
	echo '' >> /usr/bin/amnesic
	echo 'exit 0' >> /usr/bin/amnesic
	chmod +x /usr/bin/amnesic

	if [ ! -f /etc/systemd/system/amnesic.service ]; then
		echo '[Unit]' > /etc/systemd/system/amnesic.service
		echo 'Description=Amnesic Mesh' >> /etc/systemd/system/amnesic.service
		echo '' >> /etc/systemd/system/amnesic.service
		echo '[Service]' >> /etc/systemd/system/amnesic.service
		echo 'User=root' >> /etc/systemd/system/amnesic.service
		echo 'Group=root' >> /etc/systemd/system/amnesic.service
		echo 'Type=oneshot' >> /etc/systemd/system/amnesic.service
		echo 'RemainAfterExit=true' >> /etc/systemd/system/amnesic.service
		echo 'ExecStart=/bin/true' >> /etc/systemd/system/amnesic.service
		echo "ExecStop=/usr/bin/amnesic $MY_USERNAME" >> /etc/systemd/system/amnesic.service
		echo '' >> /etc/systemd/system/amnesic.service
		echo '[Install]' >> /etc/systemd/system/amnesic.service
		echo 'WantedBy=multi-user.target' >> /etc/systemd/system/amnesic.service

		systemctl daemon-reload
	fi
	systemctl enable amnesic
	systemctl start amnesic
}

function mesh_restart_daemons {
	systemctl restart avahi-daemon
	systemctl restart ssh
	systemctl restart tox-bootstrapd
	systemctl restart zeronet
	echo $'Daemons restarted' >> $INSTALL_LOG
}

# whether to reset the identity
set_new_identity=
if [ $2 ]; then
	if [[ "$2" == $"new"* ]]; then
		if [ ! -f $MESH_INSTALL_SETUP ]; then
			touch $MESH_INSTALL_SETUP
		fi
		set_new_identity=1
	fi
	if [[ "$2" == $"amnesic"* ]]; then
		if [ ! -f $MESH_AMNESIC ]; then
			touch $MESH_AMNESIC
		fi
		if [ ! -f $MESH_INSTALL_SETUP ]; then
			touch $MESH_INSTALL_SETUP
		fi
		set_new_identity=1
	fi
fi

if [ -f $MESH_INSTALL_SETUP ]; then
	if [ $1 ]; then
		MY_USERNAME=$1
	fi

	if [ ! $set_new_identity ]; then
		# sleep in order to allow other daemons to start up
		sleep 15
	fi

	# clear the install log
	if [ -f $INSTALL_LOG ]; then
		rm $INSTALL_LOG
	fi

	echo $'Beginning mesh node setup' >> $INSTALL_LOG
	enable_batman_daemon
	create_ram_disk 1
	change_avahi_name
	regenerate_ssh_host_keys
	configure_zeronet_blog
	configure_zeronet_mail
	configure_zeronet_forum
	configure_zeronet_id
	configure_toxcore
	configure_zeronet
	disable_password_logins
	disable_screen_lock
	mesh_amnesic

	if [ ! -f $MESH_AMNESIC ]; then
		rm $MESH_INSTALL_SETUP
		systemctl disable mesh-setup.service
	fi

	mesh_restart_daemons
	if [ ! -f $MESH_INSTALL_COMPLETED ]; then
		echo $'Mesh node setup complete' >> $INSTALL_LOG
		touch $MESH_INSTALL_COMPLETED
	fi
fi

exit 0