From 944244ef98ec2e5aac184915d1f1a3af34de6525 Mon Sep 17 00:00:00 2001
From: Vladlen Chekunov <chekunov@national.shitposting.agency>
Date: Thu, 27 Feb 2025 10:55:23 +0300
Subject: [PATCH] Add snap support.

---
 script/install.sh   | 10 ++++++++++
 script/uninstall.sh | 10 ++++++++++
 2 files changed, 20 insertions(+)

diff --git a/script/install.sh b/script/install.sh
index 40df055..0083d9a 100755
--- a/script/install.sh
+++ b/script/install.sh
@@ -4,6 +4,7 @@ set -e
 
 OS_NAME=$(uname -s)
 JB_PRODUCTS="idea clion phpstorm goland pycharm webstorm webide rider datagrip rubymine appcode dataspell gateway jetbrains_client jetbrainsclient studio devecostudio rustrover"
+JB_PRODUCTS_SNAPS="pycharm-community pycharm-professional rubymine clion rider webstorm intellij-idea-community kotlin pycharm-educational rustrover datagrip goland space intellij-idea-ultimate phpstorm dataspell"
 
 BASE_PATH=$(dirname $(
   cd $(dirname "$0")
@@ -79,6 +80,15 @@ else
   sed -i '/___MY_VMOPTIONS_SHELL_FILE="${HOME}\/\.jetbrains\.vmoptions\.sh"; if /d' "${ZSH_PROFILE_PATH}" >/dev/null 2>&1
 fi
 
+if command -v snap >/dev/null 2>&1; then
+  echo 'snap found. put environment variables.'
+
+  for SNAP in $JB_PRODUCTS_SNAPS; do
+    echo "Processing: ${SNAP}"
+    snap set "${SNAP}" env.___MY_VMOPTIONS_SHELL_FILE="${HOME}/.jetbrains.vmoptions.sh" >/dev/null 2>&1;
+  done
+fi
+
 echo "${EXEC_LINE}" >>"${PROFILE_PATH}"
 echo "${EXEC_LINE}" >>"${BASH_PROFILE_PATH}"
 echo "${EXEC_LINE}" >>"${ZSH_PROFILE_PATH}"
diff --git a/script/uninstall.sh b/script/uninstall.sh
index f53c8aa..f72885b 100755
--- a/script/uninstall.sh
+++ b/script/uninstall.sh
@@ -4,6 +4,7 @@ set -e
 
 OS_NAME=$(uname -s)
 JB_PRODUCTS="idea clion phpstorm goland pycharm webstorm webide rider datagrip rubymine appcode dataspell gateway jetbrains_client jetbrainsclient studio devecostudio rustrover"
+JB_PRODUCTS_SNAPS="pycharm-community pycharm-professional rubymine clion rider webstorm intellij-idea-community kotlin pycharm-educational rustrover datagrip goland space intellij-idea-ultimate phpstorm dataspell"
 
 KDE_ENV_DIR="${HOME}/.config/plasma-workspace/env"
 
@@ -26,6 +27,15 @@ MY_VMOPTIONS_SHELL_FILE="${HOME}/.${MY_VMOPTIONS_SHELL_NAME}"
 
 rm -rf "${MY_VMOPTIONS_SHELL_FILE}"
 
+if command -v snap >/dev/null 2>&1; then
+  echo 'snap found. put environment variables.'
+
+  for SNAP in $JB_PRODUCTS_SNAPS; do
+    echo "Processing: ${SNAP}"
+    snap unset "${SNAP}" env.___MY_VMOPTIONS_SHELL_FILE >/dev/null 2>&1;
+  done
+fi
+
 if [ $OS_NAME = "Darwin" ]; then
   for PRD in $JB_PRODUCTS; do
     ENV_NAME=$(echo $PRD | tr '[a-z]' '[A-Z]')"_VM_OPTIONS"