Add java support

Signed-off-by: Nikolaos Karaolidis <nick@karaolidis.com>

hosts/common/shells/python/default.nix

@@ -13,9 +13,24 @@ pkgs.mkShell {
   ];
 
   shellHook = ''
-    export P10K_EXTRA_RIGHT_PROMPT_ELEMENTS=(
-      virtualenv
-      "''${P10K_EXTRA_RIGHT_PROMPT_ELEMENTS[@]}"
-    )
+    if git rev-parse --is-inside-work-tree &> /dev/null; then
+      TOP="$(git rev-parse --show-toplevel)"
+      if ! grep -q "^\.venv$" "$TOP/.gitignore" "$TOP/.git/info/exclude"; then echo ".venv" >> "$TOP/.git/info/exclude"; fi
+    else
+      TOP="$(pwd)"
+    fi
+
+    if [ -d "$TOP/.venv" ]; then
+      source "$TOP/.venv/bin/activate"
+    else
+      echo "No virtual environment found. Do you want to create one? (y/N)"
+      read -r answer
+      if [[ "$answer" =~ ^[Yy]$ ]]; then
+        python -m venv "$TOP/.venv"
+        source "$TOP/.venv/bin/activate"
+        pip install --upgrade pip
+        if [ -f "$TOP/requirements.txt" ]; then pip install -r "$TOP/requirements.txt"; fi
+      fi
+    fi
   '';
 }