feat(haskell): Look for .hs files

spaceship-prompt · May 24, 2022 · b2eb8a3 · b2eb8a3
1 parent 2d53862
commit b2eb8a3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sections/haskell.zsh b/sections/haskell.zsh
@@ -24,7 +24,8 @@ spaceship_haskell() {
   [[ $SPACESHIP_HASKELL_SHOW == false ]] && return
 
   # If there are stack files in the context
-  spaceship::upsearch stack.yaml &>/dev/null || return
+  local is_haskell_project=$(spaceship::upsearch stack.yaml)
+  [[ -n "$is_haskell_project" || -n *.hs(#qN^/) ]] || return
 
   # The command is stack, so do not change this to haskell.
   spaceship::exists stack || return