From 8f10d90f2ce6f174f14002e366760d1fb55aba66 Mon Sep 17 00:00:00 2001
From: Alexander Gehrke <github@qwertyuiop.de>
Date: Fri, 6 Aug 2021 17:27:14 +0200
Subject: [PATCH] Add flatpak dependency handling

---
 flatpaks/.repos/flathub |  1 +
 flatpaks/discord        |  1 +
 flatpaks/schildichat    |  4 ++++
 install.zsh             | 41 ++++++++++++++++++++++++++++++++++-------
 lib/common.zsh          |  2 ++
 5 files changed, 42 insertions(+), 7 deletions(-)
 create mode 100644 flatpaks/.repos/flathub
 create mode 100755 flatpaks/schildichat

diff --git a/flatpaks/.repos/flathub b/flatpaks/.repos/flathub
new file mode 100644
index 0000000..4c2fafa
--- /dev/null
+++ b/flatpaks/.repos/flathub
@@ -0,0 +1 @@
+https://flathub.org/repo/flathub.flatpakrepo
diff --git a/flatpaks/discord b/flatpaks/discord
index 78aa6ce..887f773 100755
--- a/flatpaks/discord
+++ b/flatpaks/discord
@@ -1,3 +1,4 @@
 #!/bin/zsh
+#flatpak: flathub:com.discordapp.Discord
 export $(dbus-launch)
 exec flatpak run com.discordapp.Discord
diff --git a/flatpaks/schildichat b/flatpaks/schildichat
new file mode 100755
index 0000000..e8dc904
--- /dev/null
+++ b/flatpaks/schildichat
@@ -0,0 +1,4 @@
+#!/bin/zsh
+#flatpak: flathub:chat.schildi.desktop 
+export $(dbus-launch)
+exec flatpak run chat.schildi.desktop
diff --git a/install.zsh b/install.zsh
index f6634d1..1ea3302 100755
--- a/install.zsh
+++ b/install.zsh
@@ -24,7 +24,24 @@ uses_interpreter() {
 }
 
 get_dependencies() {
-	sed -n '2{/^#dep:/{s/^#dep://;p;q}}' $1
+	sed -n '2,4{/^#dep:/{s/^#dep://;p;q}}' $1
+}
+get_flatpak_dependencies() {
+	sed -n '2,4{/^#flatpak:/{s/^#flatpak://;p;q}}' $1
+}
+
+have_flatpak() {
+	check "Checking for flatpak: $1… "
+	flatpak info "$@" &> /dev/null
+}
+
+install_flatpak() {
+	if [[ ! -e flatpaks/.repos/$1 ]]; then
+		fail "Unknown flatpak repo: $1. Add its URL in flatpaks/.repos/$1"
+	else
+		flatpak --user remote-add --if-not-exists $1 $(<flatpaks/.repos/$1)
+		flatpak --user install $1 $2
+	fi
 }
 
 if [[ -z $1 ]]; then
@@ -35,12 +52,13 @@ if [[ -z $1 ]]; then
 	  -f, --force       overwrite existing files in the same dir
 	  -s, --skip        when installing several scripts, skip scripts with unmet
 	                    dependencies instead of aborting.
+	  --no-flatpak      do not automatically install flatpak deps (still checks for them)
 	HELP
 	exit 1
 fi
 
 zparseopts -D -E p:=install_path -path:=install_path f=force -force=force \
-	s=skip -skip=skip
+	s=skip -skip=skip -no-flatpak=noflatpakinstall
 
 if [[ ! $install_path ]]; then
 	install_path=$HOME/.local/bin
@@ -56,11 +74,20 @@ for prog in $@; do
 	done
 	for dep in $(get_dependencies $prog); do
 		if ! have_dependency $dep; then
-			if [[ $skip ]]; then
-				continue 2
-			else
-				exit 1
-			fi
+			[[ $skip ]] && continue 2 || exit 1
+		fi
+	done
+	for dep in $(get_flatpak_dependencies $prog); do
+		local repo=${dep%:*}
+		local pak=${dep#*:}
+		if have_flatpak $pak; then
+			succeed
+		elif [[ $noflatpakinstall ]]; then
+			fail "not installed and --no-flatpak given"
+			[[ $skip ]] && continue 2 || exit 1
+		else
+			info "missing. Installing now..."
+			install_flatpak $repo $pak
 		fi
 	done
 	if [[ -e $install_path/${prog:t} && ! $force ]]; then
diff --git a/lib/common.zsh b/lib/common.zsh
index 726aba6..f68bc5f 100644
--- a/lib/common.zsh
+++ b/lib/common.zsh
@@ -3,6 +3,7 @@ autoload -Uz colors && colors
 local c_fail="${fg_bold[red]}✘$reset_color"
 local c_warn="${fg_bold[yellow]}⚠$reset_color"
 local c_success="${fg_bold[green]}✔$reset_color"
+local c_info="${fg_bold[green]}💡$reset_color"
 
 check() {
 	echo -n " ? $*"
@@ -18,6 +19,7 @@ result() {
 
 fail() { result $c_fail $@ }
 warn() { result $c_warn $@ }
+info() { result $c_info $@ }
 succeed() { result $c_success $@ }
 error() { echo -n "  "; fail $@ }
 warning() { echo -n "  "; warn $@ }