distro-container-setup

#!/data/data/com.termux/files/usr/bin/bash

function setup_distro() {
    if [[ "$distro_answer" == "1" ]]; then
    selected_distro="debian"
    elif [[ "$distro_answer" == "2" ]]; then
    selected_distro="ubuntu"
    elif [[ "$distro_answer" == "3" ]]; then
    selected_distro="archlinux"
    else
    selected_distro="debian"
    fi
    banner
    echo "${R}[${W}-${R}]${G}${BOLD}Installing ${C}proot-distro"${W}
    echo
    update_sys
    package_install_and_check "proot-distro"
    banner
    echo "${R}[${W}-${R}]${G}${BOLD}Setup Selected Distro: ${C}${selected_distro}"${W}
    echo
    pd install $selected_distro

    distro_path="$PREFIX/var/lib/proot-distro/installed-rootfs/$selected_distro"
}

#########################################################################
############################# Fix Sound Issue ###########################
#########################################################################

function distro_fix_sound_issue() {
    banner
    echo "${R}[${W}-${R}]${G}${BOLD}Fixing Proot Distro Sound Problem..."${W}
    echo
    echo "export PULSE_SERVER=127.0.0.1" >> "$distro_path/etc/profile"
cat <<EOF > "$HOME/.${selected_distro}-sound-access"
pulseaudio --start --exit-idle-time=-1
pacmd load-module module-native-protocol-tcp auth-ip-acl=127.0.0.1 auth-anonymous=1
EOF
}

#########################################################################
######### Update Distro | Install Required Packages | Add User ##########
#########################################################################

function pd_package_install_and_check() {
	packs_list=($@)
#get distro id
    if [ -f /etc/os-release ]; then
    . /etc/os-release
    fi
#instll package
for package_name in "${packs_list[@]}"; do
    echo "${R}[${W}-${R}]${G}${BOLD} Installing package: ${C}$package_name ${W}"
    if [[ "$ID" == "debian" ]] || [[ "$ID" == "ubuntu" ]]; then
        sudo apt install "$package_name" -y
    elif [[ "$ID" == "archlinux" ]]; then
        sudo pacman -S --noconfirm "$package_name"
    fi
#handel error in installation
    if [ $? -ne 0 ]; then
        echo "${R}[${W}-${R}]${G}${BOLD} Error detected during installation of: ${C}$package_name ${W}"
        if [[ "$ID" == "debian" ]] || [[ "$ID" == "ubuntu" ]]; then
            sudo apt --fix-broken install -y
            sudo dpkg --configure -a
            sudo apt install "$package_name" -y
        elif [[ "$ID" == "archlinux" ]]; then
            sudo pacman -Syu --noconfirm
            sudo pacman -S --noconfirm "$package_name"
        fi
    fi
#check installation
if [[ "$ID" == "debian" ]] || [[ "$ID" == "ubuntu" ]]; then
    if dpkg -s "$package_name" >/dev/null 2>&1; then
        echo "${R}[${W}-${R}]${G} $package_name installed successfully ${W}"
    else
        if type -p "$package_name" &>/dev/null || [ -e "$PREFIX/bin/$package_name"* ] || [ -e "$PREFIX/bin/"*"$package_name" ]; then
            echo "${R}[${W}-${R}]${C} $package_name ${G}installed successfully ${W}"
        fi
    fi
elif [[ "$ID" == arch* ]]; then
    if pacman -Qi "$package_name" >/dev/null 2>&1; then
        echo "${R}[${W}-${R}]${G} $package_name installed successfully ${W}"
    else
        if type -p "$package_name" &>/dev/null || [ -e "$PREFIX/bin/$package_name"* ] || [ -e "$PREFIX/bin/"*"$package_name" ]; then
        echo "${R}[${W}-${R}]${C} $package_name ${G}installed successfully ${W}"
        fi
    fi
fi
done
echo ""
}

function pd_root_package_install_and_check() {
	packs_list=($@)
#get distro id
    if [ -f /etc/os-release ]; then
    . /etc/os-release
    fi
#instll package
for package_name in "${packs_list[@]}"; do
    echo "${R}[${W}-${R}]${G}${BOLD} Installing package: ${C}$package_name ${W}"
    if [[ "$ID" == "debian" ]] || [[ "$ID" == "ubuntu" ]]; then
        apt install "$package_name" -y
    elif [[ "$ID" == "archlinux" ]]; then
        pacman -S --noconfirm "$package_name"
    fi
#handel error in installation
if [ $? -ne 0 ]; then
    echo "${R}[${W}-${R}]${G}${BOLD} Error detected during installation of: ${C}$package_name ${W}"
    if [[ "$ID" == "debian" ]] || [[ "$ID" == "ubuntu" ]]; then
        apt --fix-broken install -y
        dpkg --configure -a
        apt install "$package_name" -y
    elif [[ "$ID" == "archlinux" ]]; then
        pacman -Syu --noconfirm
        pacman -S --noconfirm "$package_name"
    fi
fi
#check installation
if [[ "$ID" == "debian" ]] || [[ "$ID" == "ubuntu" ]]; then
    if dpkg -s "$package_name" >/dev/null 2>&1; then
        echo "${R}[${W}-${R}]${G} $package_name installed successfully ${W}"
    else
        if type -p "$package_name" &>/dev/null || [ -e "$PREFIX/bin/$package_name"* ] || [ -e "$PREFIX/bin/"*"$package_name" ]; then
            echo "${R}[${W}-${R}]${C} $package_name ${G}installed successfully ${W}"
        fi
    fi
elif [[ "$ID" == arch* ]]; then
    if pacman -Qi "$package_name" >/dev/null 2>&1; then
        echo "${R}[${W}-${R}]${G} $package_name installed successfully ${W}"
    else
        if type -p "$package_name" &>/dev/null || [ -e "$PREFIX/bin/$package_name"* ] || [ -e "$PREFIX/bin/"*"$package_name" ]; then
        echo "${R}[${W}-${R}]${C} $package_name ${G}installed successfully ${W}"
        fi
    fi
fi
done
echo ""
}

function distro_basic_task() {
    banner
    echo "${R}[${W}-${R}]${G}${BOLD}Updating ${C}$selected_distro ${G}And Install Required Packages"${W}
    echo
    if [[ "$selected_distro" == "debian" ]] || [[ "$selected_distro" == "debian" ]]; then
    proot-distro login $selected_distro --shared-tmp -- env DISPLAY=:1.0 apt update
    elif [[ "$selected_distro" == "archlinux" ]]; then
    proot-distro login $selected_distro --shared-tmp -- env DISPLAY=:1.0 pacman -Syu --noconfirm
    fi
    if [[ "$pd_audio_config_answer" == "y" ]]; then
    distro_packs="sudo pulseaudio"
    distro_fix_sound_issue
    elif [[ "$pd_audio_config_answer" == "n" ]]; then
    distro_packs="sudo"
    fi
    echo "pd_audio_config_answer=\"${pd_audio_config_answer}\"" >> $config_file
cat <<'EOF' > $distro_path/root/pd_basic_setup.sh
#!/bin/bash
R="$(printf '\033[1;31m')"                           
G="$(printf '\033[1;32m')"
Y="$(printf '\033[1;33m')"
B="$(printf '\033[1;34m')"
C="$(printf '\033[1;36m')"                                       
W="$(printf '\033[0m')"
BOLD="$(printf '\033[1m')"

EOF
declare -f banner pd_root_package_install_and_check >> $distro_path/root/pd_basic_setup.sh
cat <<EOF >> $distro_path/root/pd_basic_setup.sh
banner
pd_root_package_install_and_check "${distro_packs}"
rm pd_basic_setup.sh
EOF
proot-distro login $selected_distro -- /bin/bash -c 'bash pd_basic_setup.sh'
    if [[ "$pd_useradd_answer" == "n" ]]; then
    final_user_name="root"
    elif [[ "$pd_useradd_answer" == "y" ]]; then
    final_user_name="${user_name}"
    fi
if [[ "$pd_pass_type" == "1" ]];then
cat <<EOF > $distro_path/root/useradd.sh
#!/bin/bash
groupadd storage
groupadd wheel
useradd -m -g users -s \$(which bash) ${final_user_name}
usermod -aG wheel,polkitd,audio,video,storage ${final_user_name}
echo "${final_user_name}:${pass}" | chpasswd
chmod u+rw /etc/sudoers
echo "$final_user_name ALL=(ALL:ALL) ALL" | tee -a /etc/sudoers
chmod u-w /etc/sudoers
rm useradd.sh
EOF
elif [[ "$pd_pass_type" == "2" ]];then
cat <<EOF > $distro_path/root/useradd.sh
#!/bin/bash
groupadd storage
groupadd wheel
useradd -m -g users -s \$(which bash) ${final_user_name}
usermod -aG wheel,polkitd,audio,video,storage ${final_user_name}
chmod u+rw /etc/sudoers
echo "$final_user_name ALL=(ALL) NOPASSWD:ALL" | tee -a /etc/sudoers
chmod u-w /etc/sudoers
rm useradd.sh
EOF
fi
proot-distro login $selected_distro -- /bin/sh -c 'bash useradd.sh'
}

#########################################################################
################## Setup Zsh And Terminal Utility #######################
#########################################################################

function distro_zsh_answer() {
    banner
    if [[ "$pd_useradd_answer" == "y" ]]; then
    save_path="$distro_path/home/$final_user_name"
    elif [[ "$pd_useradd_answer" == "n" ]]; then
    save_path="$distro_path/root"
    fi
    if [[ "$distro_zsh_answer" == "n" ]]; then
	echo "${R}[${W}-${R}]${C}Canceling Distro Zsh Setup..."${W}
    sleep 1.5
	pd_shell_name="bash"
    else
    pd_shell_name="zsh"
	echo "${R}[${W}-${R}]${G}${BOLD} Configuring Zsh for ${selected_distro}${G}.."${W}
	echo
cat <<'EOF' > $save_path/pd_zsh_setup.sh
#!/bin/bash
R="$(printf '\033[1;31m')"                           
G="$(printf '\033[1;32m')"
Y="$(printf '\033[1;33m')"
B="$(printf '\033[1;34m')"
C="$(printf '\033[1;36m')"                                       
W="$(printf '\033[0m')"
BOLD="$(printf '\033[1m')"

EOF
declare -f banner get_latest_release pd_package_install_and_check >> $save_path/pd_zsh_setup.sh
cat <<'EOF' >> $save_path/pd_zsh_setup.sh
banner
pd_package_install_and_check "git zsh wget"
wget https://raw.githubusercontent.com/sabamdarif/short-linux-scripts/main/install-zsh.sh && bash install-zsh.sh
rm pd_zsh_setup.sh
EOF
    proot-distro login --user $final_user_name $selected_distro -- /bin/bash -c 'bash pd_zsh_setup.sh'
fi
}

function distro_terminal_utility_setup() {
    if [[ "$distro_terminal_utility_setup_answer" == "n" ]];then
    banner
	echo "${R}[${W}-${R}]${C}Skipping Terminal Utility Setup For ${selected_distro}..."${W}
	echo
	elif [[ "$distro_terminal_utility_setup_answer" == "y" ]];then
	banner
    echo "${R}[${W}-${R}]${C}${BOLD}Configuring Terminal Utility..."${W}
    echo
    if [[ "$selected_distro" == "debian" ]] || [[ "$selected_distro" == "debian" ]]; then
cat <<'EOF' > $distro_path/root/pd_setup.sh
#!/bin/bash

R="$(printf '\033[1;31m')"                           
G="$(printf '\033[1;32m')"
Y="$(printf '\033[1;33m')"
B="$(printf '\033[1;34m')"
C="$(printf '\033[1;36m')"                                       
W="$(printf '\033[0m')"
BOLD="$(printf '\033[1m')"
EOF
declare -f banner get_latest_release pd_package_install_and_check >> $distro_path/root/pd_setup.sh
cat <<'EOF' >> $distro_path/root/pd_setup.sh
banner
pd_package_install_and_check "nala zoxide bat wget gpg"
device_arch=$(uname -m)
    case "$device_arch" in
        aarch64)
            archtype="arm64"
            ;;
        arm)
            archtype="armhf"
            ;;
        amd64|x86_64)
            archtype="amd64"
            ;;
        i*86|x86)
            archtype="i386"
            ;;
        *)
            echo "${R} unknown architecture"${W}
            ;;
    esac
    version=$(get_latest_release "fastfetch-cli" "fastfetch")
    wget -O $distro_path/root/fastfetch.deb https://github.com/fastfetch-cli/fastfetch/releases/download/${version}/fastfetch-linux-${archtype}.deb
    apt install ./fastfetch.deb -y
    rm fastfetch.deb
mkdir -p /etc/apt/keyrings
wget -qO- https://raw.githubusercontent.com/eza-community/eza/main/deb.asc | sudo gpg --dearmor -o /etc/apt/keyrings/gierens.gpg
echo "deb [signed-by=/etc/apt/keyrings/gierens.gpg] http://deb.gierens.de stable main" | sudo tee /etc/apt/sources.list.d/gierens.list
chmod 644 /etc/apt/keyrings/gierens.gpg /etc/apt/sources.list.d/gierens.list
apt update
apt install -y eza
rm pd_setup.sh
EOF
elif [[ "$selected_distro" == "archlinux" ]]; then
cat <<'EOF' > $distro_path/root/pd_setup.sh
#!/bin/bash

R="$(printf '\033[1;31m')"                           
G="$(printf '\033[1;32m')"
Y="$(printf '\033[1;33m')"
B="$(printf '\033[1;34m')"
C="$(printf '\033[1;36m')"                                       
W="$(printf '\033[0m')"
BOLD="$(printf '\033[1m')"
EOF
declare -f banner pd_package_install_and_check >> $distro_path/root/pd_setup.sh
cat <<'EOF' >> $distro_path/root/pd_setup.sh
pd_package_install_and_check "eza zoxide bat fastfetch"
EOF
fi
proot-distro login $selected_distro -- /bin/bash -c 'bash pd_setup.sh'

if [[ "$selected_distro" == "debian" ]] || [[ "$selected_distro" == "ubuntu" ]]; then
cat <<'EOF' >> "$save_path/.${pd_shell_name}rc"
alias apt='sudo nala $@'
alias apt-get='sudo nala $@'
EOF
fi

cat <<'EOF' >> "$save_path/.${pd_shell_name}rc"
alias cat='bat $@'
alias ls='eza --icons $@'
alias mkdir='mkdir -p'
alias neofetch='fastfetch'
alias tx11start='echo please run it from termux, not inside proot distro.'
alias tx11stop='echo please run it from termux, not inside proot distro.'
#######################################################
# SPECIAL FUNCTIONS
#######################################################
# Extracts any archive(s) (if unp isn't installed)
extract() {
	for archive in "$@"; do
		if [ -f "$archive" ]; then
			case $archive in
			*.tar.bz2) tar xvjf $archive ;;
                        *.tar.xz) tar -xvf $archive ;;
                        *.tar.gz) tar -xzvf $archive ;;
			*.tar.gz) tar xvzf $archive ;;
			*.bz2) bunzip2 $archive ;;
			*.rar) rar x $archive ;;
			*.gz) gunzip $archive ;;
			*.tar) tar xvf $archive ;;
			*.tbz2) tar xvjf $archive ;;
			*.tgz) tar xvzf $archive ;;
			*.zip) unzip $archive ;;
			*.Z) uncompress $archive ;;
			*.7z) 7z x $archive ;;
			*) echo "don't know how to extract '$archive'..." ;;
			esac
		else
			echo "'$archive' is not a valid file!"
		fi
	done
}
# Searches for text in all files in the current folder
ftext() {
	# -i case-insensitive
	# -I ignore binary files
	# -H causes filename to be printed
	# -r recursive search
	# -n causes line number to be printed
	# optional: -F treat search term as a literal, not a regular expression
	# optional: -l only print filenames and not the matching lines ex. grep -irl "$1" *
	grep -iIHrn --color=always "$1" . | less -r
}
# Copy file with a progress bar
cpp() {
	set -e
	strace -q -ewrite cp -- "${1}" "${2}" 2>&1 |
		awk '{
	count += $NF
	if (count % 10 == 0) {
		percent = count / total_size * 100
		printf "%3d%% [", percent
		for (i=0;i<=percent;i++)
			printf "="
			printf ">"
			for (i=percent;i<100;i++)
				printf " "
				printf "]\r"
			}
		}
	END { print "" }' total_size="$(stat -c '%s' "${1}")" count=0
}
# Copy and go to the directory
cpg() {
	if [ -d "$2" ]; then
		cp "$1" "$2" && cd "$2"
	else
		cp "$1" "$2"
	fi
}
# Move and go to the directory
mvg() {
	if [ -d "$2" ]; then
		mv "$1" "$2" && cd "$2"
	else
		mv "$1" "$2"
	fi
}
# Create and go to the directory
mkdirg() {
	mkdir -p "$1"
	cd "$1"
}
EOF
cat <<EOF >> "$save_path/.${pd_shell_name}rc"
#set zoxide as cd
eval "\$(zoxide init --cmd cd ${pd_shell_name})"
EOF
	fi
}

#########################################################################
#################### Create App Installer Shortcut ######################
#########################################################################

function handel_debian_based_app_installer() {
cat <<TOP_EOF > "$PREFIX/bin/$selected_distro"
#!/data/data/com.termux/files/usr/bin/bash

if [[ "\$#" -eq 0 ]]; then
    proot-distro login --user $final_user_name $selected_distro --shared-tmp
elif [[ "\$1" = "install" ]]; then
if [[ ! -d "/data/data/com.termux/files/usr/share/applications/pd_added" ]]; then
        mkdir -p "/data/data/com.termux/files/usr/share/applications/pd_added"
        fi
proot-distro login --user $final_user_name $selected_distro --shared-tmp -- env DISPLAY=:1.0 sudo apt install \${@:2}
cat <<EOF > "$save_root_path/packinstall.sh"
packages=(\${@:2})
EOF
cat <<'EOF' >> "$save_root_path/packinstall.sh"
for package_name in "\${packages[@]}"; do
desktop_files=\$(dpkg-query -W -f='\${binary:Package}\n' | grep "^\$package_name\(-.*\)\?\$" | xargs dpkg-query -L | grep "^/usr/share/applications/.*\.desktop\$")
if [ -z "\$desktop_files" ]; then
    echo "${R}[${W}-${R}]${G} No .desktop files found for package ${C}'\$package_name' ${G}in/usr/share/applications."
else
    for desktop_files_name in \$desktop_files; do
        desktop_files_with_ext=\$(basename "\$desktop_files_name")
        desktop_files_without_ext="\${desktop_files_with_ext%.desktop}"
        echo "${R}[${W}-${R}]${G} Adding ${C}\${desktop_files_without_ext} ${G}To Termux Menu"
        cp "/usr/share/applications/\${desktop_files_with_ext}" "/data/data/com.termux/files/usr/share/applications/pd_added/"
        sed -i 's/Exec=/Exec=pdrun /g' "/data/data/com.termux/files/usr/share/applications/pd_added/\${desktop_files_with_ext}"
    done
fi
done
rm packinstall.sh
EOF
proot-distro login $selected_distro -- /bin/bash -c 'bash packinstall.sh'

elif [[ "\$1" = "remove" ]] || [[ "\$1" = "remove" ]]; then
proot-distro login --user $final_user_name $selected_distro --shared-tmp -- env DISPLAY=:1.0 sudo apt \$1 \${@:2}
cat <<EOF > "$save_root_path/packremove.sh"
packages=(\${@:2})
EOF
cat <<'EOF' >> "$save_root_path/packremove.sh"
for package_name in "\${packages[@]}"; do
desktop_files=\$(dpkg-query -W -f='\${binary:Package}\n' | grep "^\$package_name\(-.*\)\?\$" | xargs dpkg-query -L | grep "^/usr/share/applications/.*\.desktop\$")
if [ -z "\$desktop_files" ]; then
    echo "${R}[${W}-${R}]${G} No .desktop files found for package ${C}'\$package_name' ${G}in/usr/share/applications."
else
    for desktop_files_name in \$desktop_files; do
        desktop_files_with_ext=\$(basename "\$desktop_files_name")
        desktop_files_without_ext="\${desktop_files_with_ext%.desktop}"
        echo "${R}[${W}-${R}]${G} Removing ${C}\${desktop_files_without_ext} ${G}From Termux Menu" 
        rm /data/data/com.termux/files/usr/share/applications/pd_added/\${desktop_files_with_ext}
    done
fi
done
rm packremove.sh
EOF
proot-distro login $selected_distro -- /bin/bash -c 'bash packremove.sh'
else
echo "sudo apt \$@
rm else.sh
" > $save_root_path/else.sh
proot-distro login $selected_distro -- /bin/sh -c 'bash else.sh'
fi

if [[ \$1 = "install" ]] || [[ \$1 = "remove" ]] || [[ \$1 = "autoremove" ]]; then
gtk-update-icon-cache
xfce4-panel -r >/dev/null 2>&1
fi

TOP_EOF
}

function handel_arch_based_app_installer() {
cat <<TOP_EOF > "$PREFIX/bin/$selected_distro"
#!/data/data/com.termux/files/usr/bin/bash

if [[ "\$#" -eq 0 ]]; then
    proot-distro login --user $final_user_name $selected_distro --shared-tmp
elif [[ "\$1" = "install" ]]; then
if [[ ! -d "/data/data/com.termux/files/usr/share/applications/pd_added" ]]; then
        mkdir -p "/data/data/com.termux/files/usr/share/applications/pd_added"
        fi
proot-distro login --user $final_user_name $selected_distro --shared-tmp -- env DISPLAY=:1.0 sudo pacman -S --noconfirm "\${@:2}"
cat <<EOF > "$save_root_path/packinstall.sh"
packages=("\${@:2}")
EOF
cat <<'EOF' >> "$save_root_path/packinstall.sh"
for package_name in "\${packages[@]}"; do
desktop_files=\$(pacman -Ql \$package_name | grep "/usr/share/applications/.*\.desktop$")
if [ -z "\$desktop_files" ]; then
    echo "${R}[${W}-${R}]${G} No .desktop files found for package ${C}'\$package_name' ${G}in /usr/share/applications."
else
    for desktop_files_name in \$desktop_files; do
        desktop_files_with_ext=\$(basename "\$desktop_files_name")
        desktop_files_without_ext="\${desktop_files_with_ext%.desktop}"
        echo "${R}[${W}-${R}]${G} Adding ${C}\${desktop_files_without_ext} ${G}To Termux Menu"
        cp "/usr/share/applications/\${desktop_files_with_ext}" "/data/data/com.termux/files/usr/share/applications/pd_added/"
        sed -i 's/Exec=/Exec=pdrun /g' "/data/data/com.termux/files/usr/share/applications/pd_added/\${desktop_files_with_ext}"
    done
fi
done
rm packinstall.sh
EOF
proot-distro login $selected_distro -- /bin/bash -c 'bash packinstall.sh'

elif [[ "\$1" = "remove" ]]; then
proot-distro login --user $final_user_name $selected_distro --shared-tmp -- env DISPLAY=:1.0 sudo pacman -R --noconfirm "\${@:2}"
cat <<EOF > "$save_root_path/packremove.sh"
packages=("\${@:2}")
EOF
cat <<'EOF' >> "$save_root_path/packremove.sh"
for package_name in "\${packages[@]}"; do
desktop_files=\$(pacman -Ql \$package_name | grep "/usr/share/applications/.*\.desktop$")
if [ -z "\$desktop_files" ]; then
    echo "${R}[${W}-${R}]${G} No .desktop files found for package ${C}'\$package_name' ${G}in /usr/share/applications."
else
    for desktop_files_name in \$desktop_files; do
        desktop_files_with_ext=\$(basename "\$desktop_files_name")
        desktop_files_without_ext="\${desktop_files_with_ext%.desktop}"
        echo "${R}[${W}-${R}]${G} Removing ${C}\${desktop_files_without_ext} ${G}From Termux Menu"
        rm /data/data/com.termux/files/usr/share/applications/pd_added/\${desktop_files_with_ext}
    done
fi
done
rm packremove.sh
EOF
proot-distro login $selected_distro -- /bin/bash -c 'bash packremove.sh'
else
echo "sudo pacman -S --noconfirm \$@
rm else.sh
" > $save_root_path/else.sh
proot-distro login $selected_distro -- /bin/sh -c 'bash else.sh'
fi

TOP_EOF
}

function distro_create_app_installer() {
    banner
    echo "${R}[${W}-${R}]${G}${BOLD}Creating App Launcher"${W}

if [[ "$pd_useradd_answer" == "y" ]]; then
save_user_path="$distro_path/home/$final_user_name"
fi
save_root_path="$distro_path/root"

if [[ "$selected_distro" == "debian" ]] || [[ "$selected_distro" == "ubuntu" ]]; then
handel_debian_based_app_installer
elif [[ "$selected_distro" == "archlinux" ]]; then
handel_arch_based_app_installer
fi

chmod +x "$PREFIX/bin/$selected_distro"


#########################################################################
########################### Add To Menu Setup ###########################
#########################################################################
banner
package_install_and_check "zenity"

check_and_create_directory "$PREFIX/share/applications/pd_added"

echo "#!/data/data/com.termux/files/usr/bin/bash
distro_path="$distro_path"
" > $PREFIX/bin/add2menu
cat <<'EOF' >> $PREFIX/bin/add2menu

list_desktop_files_with_name() {
    find "$1" -name "*.desktop" -type f -exec awk -F '=' '/^Name=/ {gsub(/.desktop$/,"",$2); print $2}' {} \;
}

action=$(zenity --list --title="Choose Action" --text="Select an action:" --checklist --column="" --column="Action" FALSE "Add Application To Termux Menu" FALSE "Remove Application From Termux Menu")

if [[ -z $action ]]; then
  zenity --info --text="No action selected. Quitting..." --title="Operation Cancelled"
  exit 0
fi

if [[ $action == *"Add Application To Termux Menu"* ]]; then
  # Get the list of .desktop files with Name parameter
  files=$(list_desktop_files_with_name "$distro_path/usr/share/applications/")

  # Display the list of files and ask user to select
  selected_names=$(zenity --list --title="Add To Termux Menu" --text="Select The App Name To Add:" --checklist --column="Include" --column="Name" --separator="," $files)

  if [[ -z $selected_names ]]; then
    zenity --info --text="No file selected. Quitting..." --title="Operation Cancelled"
    exit 0
  fi

  IFS=',' read -ra selected_names_array <<< "$selected_names"

  for selected_name in "${selected_names_array[@]}"; do
    selected_file=$(find "$distro_path/usr/share/applications/" -name "*.desktop" -type f -exec awk -F '=' -v name="$selected_name" '$0 ~ "Name=" name {print FILENAME}' {} \;)

    cp "$selected_file" "$PREFIX/share/applications/pd_added/"
    sed -i 's/Exec=/Exec=pdrun /g' "$PREFIX/share/applications/pd_added/${selected_file##*/}"
    gtk-update-icon-cache
    xfce4-panel -r >/dev/null 2>&1

    zenity --info --text="File '$selected_name' added successfully!" --title="Success"
  done
fi

if [[ $action == *"Remove Application From Termux Menu"* ]]; then
  # Get the list of .desktop files with Name parameter
  files=$(list_desktop_files_with_name "$PREFIX/share/applications/pd_added/")

  # Display the list of files and ask user to select
  selected_names=$(zenity --list --title="Remove From Termux Menu" --text="Choose .desktop files to remove:" --checklist --column="Include" --column="Name" --separator="," $files)

  if [[ -z $selected_names ]]; then
    zenity --info --text="No file selected for removal. Quitting..." --title="Operation Cancelled"
    exit 0
  fi

  IFS=',' read -ra selected_names_array <<< "$selected_names"

  for selected_name in "${selected_names_array[@]}"; do
    selected_file=$(find "$PREFIX/share/applications/pd_added/" -name "*.desktop" -type f -exec awk -F '=' -v name="$selected_name" '$0 ~ "Name=" name {print FILENAME}' {} \;)

    rm "$selected_file"
    gtk-update-icon-cache
    xfce4-panel -r >/dev/null 2>&1

    zenity --info --text="File '$selected_name' has been removed successfully!" --title="Success"
  done
fi
gtk-update-icon-cache
EOF
chmod +x "$PREFIX/bin/add2menu"

check_and_create_directory "$PREFIX/share/applications/"
cat <<EOF > "$PREFIX/share/applications/add2menu.desktop"
[Desktop Entry]
Version=1.0
Type=Application
Name=Add To Menu
Comment=
Exec=add2menu
Icon=edit-move
Categories=System;
Path=
Terminal=false
StartupNotify=false
EOF
chmod +x $PREFIX/share/applications/add2menu.desktop
cp $PREFIX/share/applications/add2menu.desktop $HOME/Desktop/
}

#########################################################################
########################## App Launcher Setup ###########################
#########################################################################

function distro_app_launch_setup() {
cat <<EOF > "$PREFIX/bin/pdrun"
#!/data/data/com.termux/files/usr/bin/bash

bash ~/.${selected_distro}-sound-access
xhost + > /dev/null 2>&1
case \$1 in
--no-gpu)
proot-distro login --user $final_user_name $selected_distro --shared-tmp -- env DISPLAY=:1.0 \$@
;;
--help|-h)
echo "${G}Type ${C}pdrun COMMAND_TO_RUN${G}to start with gpu acceleration"${W}
echo "${G}Type ${C}pdrun ---no-gpu COMMAND_TO_RUN${G}to start without gpu acceleration"${W}
*)
;;
proot-distro login --user $final_user_name $selected_distro --shared-tmp -- env DISPLAY=:1.0 $selected_pd_hw_method \$@
;;
esac
EOF
chmod +x "$PREFIX/bin/pdrun"
}

#########################################################################
############### Distro Directory , Theme And Icon Link ##################
#########################################################################

function distro_folders_link() {
    echo "${R}[${W}-${R}]${G}${BOLD}Link Distro Directories With Internal Storage"${W}
cat <<EOF > "$save_path/setup-folders.sh"
#!/bin/bash
directories=(Desktop Music Downloads Pictures Videos)
for dir in "\${directories[@]}"; do
ln -s "/data/data/com.termux/files/home/\$dir" "\$HOME/"
done
rm setup-folders.sh
EOF
proot-distro login --user $final_user_name $selected_distro -- /bin/sh -c 'bash setup-folders.sh'
    echo "${R}[${W}-${R}]${G}${BOLD}Link Distro Termux Themes And Icons With $selected_distro "${W}
    sleep 0.5
    check_and_create_directory "$HOME/.icons"
    ln -s /data/data/com.termux/files/home/.icons $save_path
    check_and_create_directory "$HOME/.themes"
    ln -s /data/data/com.termux/files/home/.themes $save_path

}

setup_distro
distro_basic_task
distro_zsh_answer
distro_terminal_utility_setup
distro_create_app_installer
distro_app_launch_setup
distro_folders_link