fix issues

create_shortcut.sh

@@ -30,7 +30,7 @@ Exec="${FULL_PATH}/Lotion"
 Terminal=false
 Categories=Office;TextEditor;Utility
 Type=Application
-Icon=${WORKING_DIR}/icon.png
+Icon=${WORKING_DIR}/Lotion/resources/app/icon.png
 StartupWMClass=Notion
 EOS
 chmod +x Notion_native.desktop

install.sh

@@ -26,7 +26,7 @@ if [[ $command == 'web' ]]; then
 	wget https://github.com/puneetsl/lotion/releases/download/V-0.05/Lotion-web.tar.xz
 	tar xvf $PD/Lotion-web.tar.xz -C $INSTALL_DIR --strip 1
 	rm Lotion-web.tar.xz
-	/bin/bash $PD/create_shortcut.sh Lotion
+	/bin/bash $PD/create_shortcut.sh $command
 	echo done
 	exit 0
 fi

uninstall.sh

@@ -2,7 +2,19 @@
 
 export check="\xE2\x9C\x94"
 export cross="\xE2\x9D\x8C"
-
+command=$1
+if [[  ! $command =~ native|web ]]; then
+	printf "\nSelect which version you would want to uninstall:\n\n"
+	select command in web native
+	do
+		if [[ $command =~ native|web ]]; then
+			echo $command
+			break
+		else
+			echo "Please input 1 or 2"
+		fi
+	done
+fi
 uninstall_prep() {
   export install_dir
   export shortcut_dir
@@ -12,11 +24,11 @@ uninstall_prep() {
   share_dir="/usr/share"
 
   if [[ -d ${share_dir}/lotion ]]; then
-    install_dir="${share_dir}/lotion"
+    install_dir="${share_dir}/lotion-$command"
     shortcut_dir="${share_dir}/applications"
     exec_dir="/usr/bin"
   else
-    install_dir="$(pwd)/Lotion"
+    install_dir="$(pwd)/Lotion-$command"
     shortcut_dir="$HOME/.local/share/applications"
     exec_dir="$HOME/.local/bin"
   fi
@@ -30,10 +42,15 @@ delete_directory() {
     echo -e "${cross} ${install_dir} directory was not found. Skipping."
   fi
 }
+if [[ $command == 'web' ]]; then
+  shortcut_file=Lotion.desktop
+else
+  shortcut_file=Notion_native.desktop
+fi
 
 delete_files() {
   declare -a files
-  files=("${shortcut_dir}/Lotion.desktop" "${exec_dir}/lotion" "${exec_dir}/lotion_uninstall")
+  files=("${shortcut_dir}/${shortcut_file}" "${exec_dir}/lotion-$command" "${exec_dir}/lotion_uninstall")
 
   for file in "${files[@]}"; do
     if [[ -f "${file}" ]]; then