#!/bin/bash

# Define constants
# Define Gitea repository paths
GITEA_HOST="code.060070.xyz"
GITUSER="sky"
REPONAME="simple-admin"
GITTREE="beta-curl"
GITMAINTREE="beta-curl"
GITDEVTREE="dev"

# Construct Gitea URLs
# Gitea 的 raw 文件 URL 格式为: https://[host]/[user]/[repo]/raw/branch/[branch]/[file]
GITROOT="https://$GITEA_HOST/$GITUSER/$REPONAME/raw/branch/$GITTREE"
GITROOTMAIN="https://$GITEA_HOST/$GITUSER/$REPONAME/raw/branch/$GITMAINTREE"
GITROOTDEV="https://$GITEA_HOST/$GITUSER/$REPONAME/raw/branch/$GITDEVTREE"

# Define filesystem path
DIR_NAME="simpleadmin"
SERVICE_FILE="/lib/systemd/system/install_simpleadmin.service"
SERVICE_NAME="install_simpleadmin"
TMP_SCRIPT="/tmp/install_simpleadmin.sh"
LOG_FILE="/tmp/install_simpleadmin.log"
export PATH=/bin:/sbin:/usr/bin:/usr/sbin:/opt/bin:/opt/sbin:/usrdata/root/bin

# Tmp Script dependent constants 
SIMPLE_ADMIN_DIR="/usrdata/simpleadmin"
# Function to remount file system as read-write
remount_rw() {
    mount -o remount,rw /
}

# Function to remount file system as read-only
remount_ro() {
    mount -o remount,ro /
}

# Installation Prep
remount_rw
systemctl daemon-reload
rm $SERVICE_FILE > /dev/null 2>&1
rm $SERVICE_NAME > /dev/null 2>&1

# Create the systemd service file
cat <<EOF > "$SERVICE_FILE"
[Unit]
Description=Update $DIR_NAME temporary service

[Service]
Type=oneshot
ExecStart=/bin/bash $TMP_SCRIPT > $LOG_FILE 2>&1

[Install]
WantedBy=multi-user.target
EOF

# Create and populate the temporary shell script for installation
cat <<EOF > "$TMP_SCRIPT"
#!/bin/bash

# Define Gitea repository paths
GITEA_HOST="code.060070.xyz"
GITUSER="sky"
REPONAME="simple-admin"
GITTREE="beta-curl"
GITMAINTREE="beta-curl"
GITDEVTREE="dev"

# Construct Gitea URLs
# Gitea 的 raw 文件 URL 格式为: https://[host]/[user]/[repo]/raw/branch/[branch]/[file]
GITROOT="https://$GITEA_HOST/$GITUSER/$REPONAME/raw/branch/$GITTREE"
GITROOTMAIN="https://$GITEA_HOST/$GITUSER/$REPONAME/raw/branch/$GITMAINTREE"
GITROOTDEV="https://$GITEA_HOST/$GITUSER/$REPONAME/raw/branch/$GITDEVTREE"

# Define filesystem path
SIMPLE_ADMIN_DIR="/usrdata/simpleadmin"
export HOME=/usrdata/root
export PATH=/bin:/sbin:/usr/bin:/usr/sbin:/opt/bin:/opt/sbin:/usrdata/root/bin

# Function to remount file system as read-write
remount_rw() {
    mount -o remount,rw /
}

# Function to remount file system as read-only
remount_ro() {
    mount -o remount,ro /
}
remount_rw
uninstall_simpleadmin() {
	echo "Uninstalling Simpleadmin..."
		
	# Clean up Tailscale if present
	if [ -d "/usrdata/tailscale" ]; then
		echo "Cleaning up Tailscale..."
		# Stop and logout from Tailscale
		if [ -f "/usrdata/tailscale/tailscale" ]; then
			/usrdata/tailscale/tailscale logout
			/usrdata/tailscale/tailscale down
		fi
		
		systemctl stop tailscale 2>/dev/null
		systemctl stop tailscale-webui 2>/dev/null
		rm -f /lib/systemd/system/tailscale.service
		
		# Remove Tailscale files and directories
		rm -rf /usrdata/tailscale
		rm -f /bin/tailscale
		rm -f /bin/tailscaled
		
		# Reload systemd
		systemctl daemon-reload
	fi
		
	# Check if Lighttpd service is installed and remove it if present
	if [ -f "/lib/systemd/system/lighttpd.service" ]; then
		echo "Lighttpd detected, uninstalling Lighttpd webserver and its modules..."
		systemctl stop lighttpd
		rm -f /lib/systemd/system/lighttpd.service
		opkg --force-remove --force-removal-of-dependent-packages remove lighttpd-mod-authn_file lighttpd-mod-auth lighttpd-mod-cgi lighttpd-mod-openssl lighttpd-mod-proxy lighttpd
	fi
	echo -e "\e[1;34mUninstalling simpleadmin content...\e[0m"
	systemctl stop simpleadmin_generate_status
	systemctl stop simpleadmin_httpd
	rm -f /lib/systemd/system/simpleadmin_httpd.service
	rm -f /lib/systemd/system/simpleadmin_generate_status.service
	systemctl daemon-reload
	
	echo -e "\e[1;34mUninstalling ttyd...\e[0m"
    systemctl stop ttyd
    rm -rf /usrdata/ttyd
	rm -rf "$SIMPLE_ADMIN_DIR"
    rm -f /lib/systemd/system/ttyd.service
    rm -f /lib/systemd/system/multi-user.target.wants/ttyd.service
    rm -f /bin/ttyd
    echo -e "\e[1;32mttyd has been uninstalled.\e[0m"

    echo "Uninstallation process completed."
}

install_lighttpd() {
	# Check for simpleadmin_httpd service and remove if exists
    if [ -f "/lib/systemd/system/simpleadmin_httpd.service" ]; then
        systemctl stop simpleadmin_httpd
        rm /lib/systemd/system/simpleadmin_httpd.service
        rm /lib/systemd/system/multi-user.target.wants/simpleadmin_httpd.service
    fi

    /opt/bin/opkg install sudo lighttpd lighttpd-mod-auth lighttpd-mod-authn_file lighttpd-mod-cgi lighttpd-mod-openssl lighttpd-mod-proxy
    # Ensure rc.unslung doesn't try to start it
    # Dynamically find and remove any Lighttpd-related init script
    for script in /opt/etc/init.d/*lighttpd*; do
        if [ -f "$script" ]; then
            echo "Removing existing Lighttpd init script: $script"
            rm "$script" # Remove the script if it contains 'lighttpd' in its name
        fi
    done
    systemctl stop lighttpd
    echo -e "\033[0;32mInstalling/Updating Lighttpd...\033[0m"
    mkdir -p "$SIMPLE_ADMIN_DIR"
    curl -o "$SIMPLE_ADMIN_DIR/lighttpd.conf" "$GITROOT/simpleadmin/lighttpd.conf"
    curl -o "/lib/systemd/system/lighttpd.service" "$GITROOT/simpleadmin/systemd/lighttpd.service"
    ln -sf "/lib/systemd/system/lighttpd.service" "/lib/systemd/system/multi-user.target.wants/"
    echo "www-data ALL = (root) NOPASSWD: /usr/sbin/iptables, /usr/sbin/ip6tables, /usrdata/simplefirewall/ttl-override, /bin/echo, /bin/cat" > /opt/etc/sudoers.d/www-data

    openssl req -new -newkey rsa:2048 -days 3650 -nodes -x509 \
        -subj "/C=US/ST=MI/L=Romulus/O=RMIITools/CN=localhost" \
        -keyout $SIMPLE_ADMIN_DIR/server.key -out $SIMPLE_ADMIN_DIR/server.crt
    systemctl daemon-reload
    systemctl start lighttpd
    
    echo -e "\033[0;32mLighttpd installation/update complete.\033[0m"
}

install_simpleadmin() {
remount_rw
echo -e "\e[1;31m2) Installing simpleadmin from the $GITTREE branch\e[0m"
			mkdir $SIMPLE_ADMIN_DIR
			mkdir $SIMPLE_ADMIN_DIR/systemd
			mkdir $SIMPLE_ADMIN_DIR/script
    		mkdir $SIMPLE_ADMIN_DIR/console
			mkdir $SIMPLE_ADMIN_DIR/console/menu
			mkdir $SIMPLE_ADMIN_DIR/console/services
			mkdir $SIMPLE_ADMIN_DIR/console/services/systemd
      		mkdir $SIMPLE_ADMIN_DIR/www
			mkdir $SIMPLE_ADMIN_DIR/www/cgi-bin
			mkdir $SIMPLE_ADMIN_DIR/www/css
    		mkdir $SIMPLE_ADMIN_DIR/www/js
            cd $SIMPLE_ADMIN_DIR/systemd
            curl -o lighttpd.service "$GITROOT/simpleadmin/systemd/lighttpd.service"
			sleep 1
			cd $SIMPLE_ADMIN_DIR/script
			curl -o ttl_script.sh "$GITROOT/simpleadmin/script/ttl_script.sh"
			curl -o remove_watchcat.sh "$GITROOT/simpleadmin/script/remove_watchcat.sh"
			curl -o create_watchcat.sh "$GITROOT/simpleadmin/script/create_watchcat.sh"
			sleep 1
			cd $SIMPLE_ADMIN_DIR/console
			curl -o .profile "$GITROOT/simpleadmin/console/.profile"
			sleep 1
			cd $SIMPLE_ADMIN_DIR/console/menu
			curl -o start_menu.sh "$GITROOT/simpleadmin/console/menu/start_menu.sh"
			ln -f $SIMPLE_ADMIN_DIR/console/menu/start_menu.sh /usrdata/root/bin/menu
			curl -o sfirewall_settings.sh "$GITROOT/simpleadmin/console/menu/sfirewall_settings.sh"
			curl -o LAN_settings.sh "$GITROOT/simpleadmin/console/menu/LAN_settings.sh"
			curl -o start_menu.sh "$GITROOT/simpleadmin/console/menu/start_menu.sh"
			sleep 1
			cd $SIMPLE_ADMIN_DIR/www
			curl -o deviceinfo.html "$GITROOT/simpleadmin/www/deviceinfo.html"
   			curl -o favicon.ico "$GITROOT/simpleadmin/www/favicon.ico"
			curl -o index.html "$GITROOT/simpleadmin/www/index.html"
    		curl -o network.html "$GITROOT/simpleadmin/www/network.html"
			curl -o settings.html "$GITROOT/simpleadmin/www/settings.html"
			curl -o sms.html "$GITROOT/simpleadmin/www/sms.html"
			curl -o scanner.html "$GITROOT/simpleadmin/www/scanner.html"
			curl -o watchcat.html "$GITROOT/simpleadmin/www/watchcat.html"
			sleep 1
			cd $SIMPLE_ADMIN_DIR/www/js
			curl -o alpinejs.min.js "$GITROOT/simpleadmin/www/js/alpinejs.min.js"
			curl -o bootstrap.bundle.min.js "$GITROOT/simpleadmin/www/js/bootstrap.bundle.min.js"
			curl -o dark-mode.js "$GITROOT/simpleadmin/www/js/dark-mode.js"
			curl -o generate-freq-box.js "$GITROOT/simpleadmin/www/js/generate-freq-box.js"
			curl -o parse-settings.js "$GITROOT/simpleadmin/www/js/parse-settings.js"
			curl -o populate-checkbox.js "$GITROOT/simpleadmin/www/js/populate-checkbox.js"
    		sleep 1
    		cd $SIMPLE_ADMIN_DIR/www/css
    		curl -o bootstrap.min.css "$GITROOT/simpleadmin/www/css/bootstrap.min.css"
      		curl -o styles.css "$GITROOT/simpleadmin/www/css/styles.css"
			sleep 1
			cd $SIMPLE_ADMIN_DIR/www/cgi-bin
			curl -o get_atcommand "$GITROOT/simpleadmin/www/cgi-bin/get_atcommand"
			curl -o user_atcommand "$GITROOT/simpleadmin/www/cgi-bin/user_atcommand"
			curl -o get_ping "$GITROOT/simpleadmin/www/cgi-bin/get_ping"
			curl -o get_sms "$GITROOT/simpleadmin/www/cgi-bin/get_sms"
			curl -o get_system_stats "$GITROOT/simpleadmin/www/cgi-bin/get_system_stats"
    		curl -o get_ttl_status "$GITROOT/simpleadmin/www/cgi-bin/get_ttl_status"
      		curl -o set_ttl "$GITROOT/simpleadmin/www/cgi-bin/set_ttl"
			curl -o send_sms "$GITROOT/simpleadmin/www/cgi-bin/send_sms"
			curl -o get_uptime "$GITROOT/simpleadmin/www/cgi-bin/get_uptime"
			curl -o get_watchcat_status "$GITROOT/simpleadmin/www/cgi-bin/get_watchcat_status"
			curl -o set_watchcat "$GITROOT/simpleadmin/www/cgi-bin/set_watchcat"
			curl -o watchcat_maker "$GITROOT/simpleadmin/www/cgi-bin/watchcat_maker"
			sleep 1
			cd /
            chmod +x $SIMPLE_ADMIN_DIR/www/cgi-bin/*
			chmod +x $SIMPLE_ADMIN_DIR/script/*
			chmod +x $SIMPLE_ADMIN_DIR/console/menu/*
			chmod +x $SIMPLE_ADMIN_DIR/console/.profile
			cp -f $SIMPLE_ADMIN_DIR/console/.profile /usrdata/root/.profile
			chmod +x /usrdata/root/.profile
            cp -rf $SIMPLE_ADMIN_DIR/systemd/* /lib/systemd/system
			sleep 1
            systemctl daemon-reload
			sleep 1
}
install_ttyd() {
    echo -e "\e[1;34mStarting ttyd installation process...\e[0m"
    cd $SIMPLE_ADMIN_DIR/console
    curl -L -o ttyd https://code.060070.xyz/sky/simple-admin/raw/branch/main/tools/ttyd.armhf && chmod +x ttyd
    curl -o ttyd.bash "$GITROOT/simpleadmin/console/ttyd.bash" && chmod +x ttyd.bash
    cd $SIMPLE_ADMIN_DIR/systemd/
	curl -o ttyd.service "$GITROOT/simpleadmin/systemd/ttyd.service"
    cp -f $SIMPLE_ADMIN_DIR/systemd/ttyd.service /lib/systemd/system/
    ln -sf /usrdata/simpleadmin/ttyd /bin
    
    # Enabling and starting ttyd service
    systemctl daemon-reload
    ln -sf /lib/systemd/system/ttyd.service /lib/systemd/system/multi-user.target.wants/
    systemctl start ttyd
    if [ "$?" -ne 0 ]; then
        echo -e "\e[1;31mFailed to start ttyd service. Please check the systemd service file and ttyd binary.\e[0m"
        exit 1
    fi

    echo -e "\e[1;32mInstallation Complete! ttyd server is up.\e[0m"
}
uninstall_simpleadmin
install_lighttpd
install_simpleadmin
install_ttyd
remount_ro
exit 0
EOF

# Make the temporary script executable
chmod +x "$TMP_SCRIPT"

# Reload systemd to recognize the new service and start the update
systemctl daemon-reload
systemctl start $SERVICE_NAME