Use /usr/bin/env shebang in tests

src/Nixfmt/Pretty.hs

@@ -17,7 +17,7 @@ import qualified Data.Text as Text (null, takeWhile)
 
 import Nixfmt.Predoc
  (Doc, GroupAnn(..), Pretty, emptyline, group, group', hardline, hardspace, hcat, line, line',
- nest, offset, newline, pretty, sepBy, surroundWith, softline, text, comment, trailingComment, trailing, textWidth,
+ nest, offset, newline, pretty, sepBy, surroundWith, softline, softline', text, comment, trailingComment, trailing, textWidth,
  unexpandSpacing')
 import Nixfmt.Types
  (Ann(..), Binder(..), Expression(..), Item(..), Items(..), Leaf,
@@ -139,9 +139,18 @@ prettyTerm (Token t) = pretty t
 prettyTerm (SimpleString (Ann leading s trailing')) = pretty leading <> prettySimpleString s <> pretty trailing'
 prettyTerm (IndentedString (Ann leading s trailing')) = pretty leading <> prettyIndentedString s <> pretty trailing'
 prettyTerm (Path p) = pretty p
-prettyTerm (Selection term selectors Nothing) = pretty term <> hcat selectors
-prettyTerm (Selection term selectors (Just (kw, def))) =
- pretty term <> hcat selectors <> hardspace <> pretty kw <> hardspace <> pretty def
+prettyTerm (Selection term selectors rest) =
+ pretty term <> sep <> hcat selectors
+ <> pretty ((\(kw, def) -> softline <> nest (pretty kw <> hardspace <> pretty def)) <$> rest)
+ where
+ -- Selection (`foo.bar.baz`) case distinction on the first element (`foo`):
+ sep = case term of
+ -- If it is an ident, keep it all together
+ (Token _) -> mempty
+ -- If it is a parenthesized expression, maybe add a line break
+ (Parenthesized _ _ _) -> softline'
+ -- Otherwise, very likely add a line break
+ _ -> line'
 
 -- Empty list
 prettyTerm (List (Ann leading paropen Nothing) (Items []) (Ann [] parclose trailing'))

test/test.sh

@@ -1,4 +1,4 @@
-#!/bin/env bash
+#!/usr/bin/env bash
 set -euo pipefail
 
 # Simple test runner for nixfmt.