#!/usr/bin/bash

### Help 
Help() {
    echo
    echo "Use fzf to search and install with Pacman and Yay"
    echo "Defaults to Pacman if no options passed"
    echo
    echo "Syntax: fpf -[a|l|la|R|Ra|h]"
    echo
    echo "options:"
    echo
    echo "a     Search/List and install from AUR with Yay"
    echo
    echo "l		Search/List installed packages from official repo"
    echo
    echo "la	Search/List installed packages from AUR repo"
    echo
    echo "R		Search/List installed packages from official repos for removal"
    echo
    echo "Ra	Search/List installed packages from AUR repo for removal"
    echo
    echo "h     Print this help screen."
    echo
}

### Manjaro Repo

#Get pkg infos and files lists, bypassing 'sudo pacman -Fy' because entering a password to view pkg info is silly

GetDB() {
	wget -P /tmp/core/ https://mirror.datacenter.by/pub/mirrors/manjaro/stable/core/"$type"/core.files.tar.gz >/dev/null 2>&1 && tar xzf /tmp/core/core.files.tar.gz -C /tmp/core 2> /dev/null
}

#Double check things are up to date
UpdateInfos() {
	[ -f /tmp/core/core.files.tar.gz ] || GetDB
	d1=$(stat -c %y /tmp/core/core.files.tar.gz)
	d2=$(stat -c %y /var/lib/pacman/sync/core.db)
	d1="${d1:0:10}"
	d2="${d2:0:10}"
	[[ "${d2///-/}" > "${d1//-/}" ]] && GetDB
}

#Get Manjaro package list, sort, mark installed, preview infos and finally hand off to pacman for install
Mrepo() {
	echo "Setting things up..."
	sort <(comm -23 <(pacman -Sl | sort) <(pacman -Q | sort)) <(comm -12 <(pacman -Sl | sort) <(pacman -Q | sort) | awk '{$1=""; print $0" \033[32m*\033[0m"}') >  /tmp/packages
	cat /tmp/packages |fzf -q "$1" -e -m --preview 'cat <(cat /tmp/core/{1}/desc) <(cat /tmp/core/{1}/files)' --preview-window=wrap --layout=reverse --marker='>>' --header="Select packages to install (use TAB to toggle selection)" --info=hidden --ansi | xargs -ro sudo pacman -S
}

# List installed pkgs
Installed() {
	sort <(pacman -Qqs) |fzf -q "$1" -e -m --preview 'cat <(pacman -Qi {1}) <(pacman -Fl {1} | awk "{print \$2}")' --preview-window=wrap --layout=reverse --marker='>>' --header="Select packages to install (use TAB to toggle selection)" --info=hidden --ansi | xargs -ro pacman -Qik
}

# Remove installed pkgs
Remove() {
	sort <(pacman -Qqs) |fzf -q "$1" -e -m --preview 'cat <(pacman -Si {1}) <(pacman -Fl {1} | awk "{print \$2}")' --preview-window=wrap --layout=reverse --marker='>>' --header="Select packages to install (use TAB to toggle selection)" --info=hidden --ansi | xargs -ro sudo pacman -Rsn
}

### AUR Repo
#Get AUR package database, remove unwanted lines, sort, mark installed, preview infos and finally hand off to yay for install
Arepo() {
	wget -P /tmp/aur/ https://aur.archlinux.org/packages.gz >/dev/null 2>&1 && gunzip -f /tmp/aur/packages.gz
	echo "$(tail -n +2 /tmp/aur/packages)" > /tmp/aur/packages
	sort <(comm -23 <(sort /tmp/aur/packages) <(pacman -Qq | sort)) <(comm -12 <(sort /tmp/aur/packages) <(pacman -Qq | sort) | awk '{print $0" \033[32m*\033[0m"}') > /tmp/aur_packages
	cat /tmp/aur_packages | fzf -q "$1" -e -m --preview  'cat <(yay -Si {1}) <(pacman -Ql {1} 2>/dev/null | awk "{print \$2}")' --preview-window=wrap --layout=reverse --marker='>>' --header="Select packages to install (use TAB to toggle selection)" --info=hidden --ansi | xargs -ro yay -S
}

# List installed pkgs from AUR
AurInstalled() {
	sort <(pacman -Qqm) |fzf -q "$1" -e -m --preview 'cat <(yay -Qi {1}) <(pacman -Ql {1} | awk "{print \$2}")' --preview-window=wrap --layout=reverse --marker='>>' --header="Select packages to install (use TAB to toggle selection)" --info=hidden --ansi | xargs -ro pacman -Qik
}

# Remove installed pkgs from AUR
Remove() {
	sort <(pacman -Qqm) |fzf -q "$1" -e -m --preview 'cat <(yay -Qi {1}) <(pacman -Ql {1} | awk "{print \$2}")' --preview-window=wrap --layout=reverse --marker='>>' --header="Select packages to install (use TAB to toggle selection)" --info=hidden --ansi | xargs -ro sudo pacman -Rsn
}

### ORPHANS
#Orphans() {
	
#}

### MAIN

#   Test for AUR option, if not run with pacman
if [[ ! "$1" =~ "-" ]]; then
	UpdateInfos
	Mrepo "$1"
else
	for opt in "$@"; do
		case $opt in
    		-a|--aur)
				echo -e "Syncing AUR package database..." 
				Arepo "$2"
				;;
       		-l|--list-installed)
       			Installed "$2"
       			;;
       		-la|--list-aur-installed)
       			AurInstalled "$2"
       			;;
       		-o|--orphans)
       			Orphans "$2"
       			;;
       		-R|--remove)
       			Remove "$2"
       			;;
       		-Ra|--remove-aur)
       			AurRemove "$2"
       			;;
    		-h|--help) #Usage // Help
       			Help
       			;;       			
    		-*) #Error catching
       			echo "Invalid Usage"
       			Help
       			;;
    	esac
	done
fi