seito: Add --vim-config

Author: sol

.gitattributes

@@ -1,3 +1,4 @@
 *.cabal linguist-generated=true
 /ghc-diagnostics/test/fixtures/**/*.yaml linguist-generated=true
 /ghc-diagnostics/test/fixtures/**/Foo.hs linguist-generated=true
+/vim/test/assets/*.errors linguist-generated=true

.github/workflows/build.yml

@@ -70,6 +70,32 @@ jobs:
           compression-level: 0
         if: matrix.ghc == 'latest'
 
+      - shell: bash
+        run: |
+          for GHC in 9.4
+          do
+            ghcup install ghc $GHC --no-set
+            SENSEI_GHC=ghc-$GHC $(cabal list-bin spec)
+          done
+        if: matrix.ghc == 'latest'
+
+      - run: vim/test/run.vim
+        if: runner.os == 'Linux'
+
+      - run: vim/test/assets/generate.sh
+        if: ${{ runner.os == 'Linux' && matrix.ghc != '9.4' }}
+
+      - run: git diff --exit-code
+
+      - shell: bash
+        run: |
+          untracked=$(git ls-files --others --exclude-standard)
+          if [ -n "$untracked" ]; then
+            echo "Untracked files:"
+            echo "$untracked"
+            exit 1
+          fi
+
   success:
     needs: build
     runs-on: ubuntu-latest
@@ -81,7 +107,7 @@ jobs:
 
       - uses: actions/checkout@v4
       - name: Check for trailing whitespace
-        run: '! git grep -I "\s\+$"'
+        run: "! git grep -nI '[[:blank:]]$' -- . ':!vim/test/assets'"
 
       - run: curl -sSL https://raw.githubusercontent.com/sol/hpack/main/get-hpack.sh | bash
       - run: hpack && git diff --exit-code

README.md

@@ -55,13 +55,14 @@ instead:
 
 ### Vim integration
 
-You can use `sensei` to load the result of the last test run into your quickfix
+You can use `seito` to load the results of the last test run into your quickfix
 list by executing `:make` in Vim.
 
-For this to work, you can either create a `Makefile` or set `makeprg` to a
-custom value.
+For this to work, you can choose one out of three options:
 
-(In both cases, `sed` is used to strip ANSI color sequences.)
+1. Create a `Makefile`
+2. Set `makeprg` to a custom value
+3. Use [`sensei.vim`](vim/sensei.vim)
 
 #### Option 1: Create a `Makefile`
 
@@ -82,6 +83,15 @@ Add the following to your Vim configuration (e.g.
 :set makeprg=seito
 ```
 
+#### Option 3: Use `sensei.vim`:
+
+Add the following to your Vim configuration (e.g.
+`~/.vim/after/ftplugin/haskell.vim`):
+
+```vim
+execute 'source ' . system('seito --vim-config')
+```
+
 ### Emacs integration
 
 Similarly, you can use `sensei` to load the result of the last test run into an

driver/seito.hs

@@ -7,10 +7,12 @@ import           System.Environment
 import           Control.Monad
 import qualified Data.ByteString.Lazy as L
 
+import           Paths_sensei (getDataFileName)
+
 import           Client
 
 main :: IO ()
 main = do
-  (success, output) <- getArgs >>= client ""
+  (success, output) <- getArgs >>= client getDataFileName ""
   L.putStr output
   unless success exitFailure

package.yaml

@@ -22,6 +22,8 @@ default-extensions:
   - PackageImports
   - DerivingVia
 
+data-files: vim/sensei.vim
+
 dependencies:
   - ghc-diagnostics
 
@@ -74,6 +76,7 @@ executables:
 
   seito:
     source-dirs: driver
+    generated-other-modules: Paths_sensei
     main: seito.hs
 
 tests:

sensei.cabal

@@ -8,15 +8,17 @@ import Network.HTTP.Client
 import HTTP.Util (makeRequest)
 import Sensei.API qualified as API
 
-client :: FilePath -> [String] -> IO (Bool, LazyByteString)
-client dir args = case args of
+client :: (FilePath -> IO FilePath) -> FilePath -> [String] -> IO (Bool, LazyByteString)
+client getDataFileName dir args = case args of
   [] -> hIsTerminalDevice stdout >>= run
   ["--no-color"] -> run False
   ["--color"] -> run True
   ["trigger"] -> API.trigger dir
+  ["--vim-config"] -> (,) True . fromString <$> getDataFileName "vim/sensei.vim"
   _ -> do
     hPutStrLn stderr $ "Usage: seito [ --color | --no-color ]"
     hPutStrLn stderr $ "       seito trigger"
+    hPutStrLn stderr $ "       seito --vim-config"
     return (False, "")
   where
     run :: Bool -> IO (Bool, LazyByteString)

src/Client.hs

@@ -24,17 +24,21 @@ spec = do
   describe "client" $ do
     it "accepts --color" $ do
       withSuccess $ \ dir -> do
-        client dir ["--color"] `shouldReturn` (True, fromString $ withColor Green "success")
+        client return dir ["--color"] `shouldReturn` (True, fromString $ withColor Green "success")
 
     it "accepts --no-color" $ do
       withSuccess $ \ dir -> do
-        client dir ["--no-color"] `shouldReturn` (True, "success")
+        client return dir ["--no-color"] `shouldReturn` (True, "success")
 
     it "indicates failure" $ do
       withFailure $ \ dir -> do
-        client dir ["--color"] `shouldReturn` (False, fromString $ withColor Red "failure")
+        client return dir ["--color"] `shouldReturn` (False, fromString $ withColor Red "failure")
 
     context "when server socket is missing" $ do
       it "reports error" $ do
         withTempDirectory $ \ dir -> do
-          client dir [] `shouldReturn` (False, "could not connect to " <> fromString (socketName dir) <> "\n")
+          client return dir [] `shouldReturn` (False, "could not connect to " <> fromString (socketName dir) <> "\n")
+
+    context "with --vim-config" $ do
+      it "returns a path to a Vim support file" $ do
+        client return undefined ["--vim-config"] `shouldReturn` (True, "vim/sensei.vim")

test/ClientSpec.hs

@@ -0,0 +1,165 @@
+source vim/sensei.vim
+
+function Require(actual, required)
+  for i in range(len(a:required))
+    if a:actual[i] > a:required[i]
+      return 1
+    elseif a:actual[i] < a:required[i]
+      return 0
+    endif
+  endfor
+  return 1
+endfunction
+
+function GhcVersion(name)
+  return matchlist(a:name, '\vghc-(\d+)\.(\d+)\.(\d+)\.errors')[1:3]
+endfunction
+
+function PopulateQuickFixList(name)
+  SUCCESS a:name
+  execute "cgetfile " . a:name
+  return filter(getqflist(), 'v:val.valid')
+endfunction
+
+function GhcErrorsFor(name)
+  let errors = glob("vim/test/assets/" . a:name . ".*.errors", v:true, v:true)
+  call ShouldBe(len(errors), 5)
+  return errors
+endfunction
+
+for name in GhcErrorsFor("lexical-error.hs")
+  let errors = PopulateQuickFixList(name)
+  call ShouldBe(len(errors), 1)
+
+  let err = errors[0]
+  call ShouldBe(bufname(err.bufnr), "lexical-error.hs")
+  call ShouldBe(err.lnum, 1)
+  call ShouldBe(err.col, 11)
+  call ShouldBe(err.end_lnum, 0)
+  call ShouldBe(err.end_col, 0)
+  call ShouldBe(err.type, 'e')
+  if Require(GhcVersion(name), [9,6])
+    call ShouldBe(err.nr, 21231)
+  else
+    call ShouldBe(err.nr, -1)
+  endif
+  call ShouldBe(err.text, "\n    lexical error in string/character literal at character '\\n'")
+endfor
+
+for name in GhcErrorsFor("parse-error.hs")
+  let errors = PopulateQuickFixList(name)
+  call ShouldBe(len(errors), 1)
+
+  let err = errors[0]
+  call ShouldBe(bufname(err.bufnr), "parse-error.hs")
+  call ShouldBe(err.lnum, 1)
+  call ShouldBe(err.col, 1)
+  call ShouldBe(err.end_lnum, 0)
+  call ShouldBe(err.end_col, 0)
+  call ShouldBe(err.type, 'e')
+  if Require(GhcVersion(name), [9,8])
+    call ShouldBe(err.nr, 25277)
+  else
+    call ShouldBe(err.nr, -1)
+  endif
+  call ShouldBe(err.text, "\n    Parse error: module header, import declaration\n    or top-level declaration expected.")
+endfor
+
+for name in GhcErrorsFor("type-error.hs")
+  let errors = PopulateQuickFixList(name)
+  call ShouldBe(len(errors), 1)
+
+  let err = errors[0]
+  call ShouldBe(bufname(err.bufnr), "type-error.hs")
+  call ShouldBe(err.lnum, 2)
+  call ShouldBe(err.col, 7)
+  call ShouldBe(err.end_lnum, 0)
+  call ShouldBe(err.end_col, 0)
+  call ShouldBe(err.type, 'e')
+  if Require(GhcVersion(name), [9,6])
+    call ShouldBe(err.nr, 83865)
+    call ShouldBe(err.text, "\n    Couldn't match type ‘Int’ with ‘[Char]’\n    Expected: String\n      Actual: Int")
+  else
+    call ShouldBe(err.nr, -1)
+    call ShouldBe(err.text, "\n    • Couldn't match type ‘Int’ with ‘[Char]’\n      Expected: String\n        Actual: Int\n    • In the expression: 23 :: Int\n      In an equation for ‘foo’: foo = 23 :: Int")
+  endif
+endfor
+
+for name in GhcErrorsFor("suggested-fix.hs")
+  let errors = PopulateQuickFixList(name)
+  call ShouldBe(len(errors), 2)
+
+  let err = errors[0]
+  call ShouldBe(bufname(err.bufnr), "suggested-fix.hs")
+  call ShouldBe(err.lnum, 1)
+  call ShouldBe(err.col, 1)
+  call ShouldBe(err.end_lnum, 0)
+  call ShouldBe(err.end_col, 0)
+  call ShouldBe(err.type, 'e')
+  if Require(GhcVersion(name), [9,6])
+    call ShouldBe(err.nr, 44432)
+  else
+    call ShouldBe(err.nr, -1)
+  endif
+  call ShouldBe(err.text, "\n    The type signature for ‘foo’ lacks an accompanying binding")
+
+  let err = errors[1]
+  call ShouldBe(bufname(err.bufnr), "suggested-fix.hs")
+  call ShouldBe(err.lnum, 4)
+  call ShouldBe(err.col, 1)
+  call ShouldBe(err.end_lnum, 0)
+  call ShouldBe(err.end_col, 0)
+  call ShouldBe(err.type, 'e')
+  if Require(GhcVersion(name), [9,6])
+    call ShouldBe(err.nr, 44432)
+  else
+    call ShouldBe(err.nr, -1)
+  endif
+  call ShouldBe(err.text, "\n    The type signature for ‘bar’ lacks an accompanying binding")
+endfor
+
+for name in GhcErrorsFor("suggested-fix-multiline.hs")
+  let errors = PopulateQuickFixList(name)
+  call ShouldBe(len(errors), 2)
+
+  let err = errors[0]
+  call ShouldBe(bufname(err.bufnr), "suggested-fix-multiline.hs")
+  call ShouldBe(err.lnum, 1)
+  call ShouldBe(err.col, 1)
+  call ShouldBe(err.end_lnum, 0)
+  call ShouldBe(err.end_col, 0)
+  call ShouldBe(err.type, 'e')
+  if Require(GhcVersion(name), [9,6])
+    call ShouldBe(err.nr, 44432)
+  else
+    call ShouldBe(err.nr, -1)
+  endif
+  call ShouldBe(err.text, "\n    The type signature for ‘foo’ lacks an accompanying binding")
+
+  let err = errors[1]
+  call ShouldBe(bufname(err.bufnr), "suggested-fix-multiline.hs")
+  call ShouldBe(err.lnum, 4)
+  call ShouldBe(err.col, 1)
+  call ShouldBe(err.end_lnum, 0)
+  call ShouldBe(err.end_col, 0)
+  call ShouldBe(err.type, 'e')
+  if Require(GhcVersion(name), [9,6])
+    call ShouldBe(err.nr, 44432)
+  else
+    call ShouldBe(err.nr, -1)
+  endif
+  call ShouldBe(err.text, "\n    The type signature for ‘bar’ lacks an accompanying binding")
+endfor
+
+let errors = PopulateQuickFixList("vim/test/assets/hspec.hs.errors")
+call ShouldBe(len(errors), 1)
+
+let err = errors[0]
+call ShouldBe(bufname(err.bufnr), "vim/test/assets/hspec.hs")
+call ShouldBe(err.lnum, 6)
+call ShouldBe(err.col, 11)
+call ShouldBe(err.end_lnum, 0)
+call ShouldBe(err.end_col, 0)
+call ShouldBe(err.type, '')
+call ShouldBe(err.nr, -1)
+call ShouldBe(err.text, "")

vim/sensei.test.vim

@@ -0,0 +1,18 @@
+setlocal makeprg=seito
+
+" GHC
+setlocal errorformat=%A%f:%l:%c:\ %t%*[^:]:\ [GHC-%n]
+setlocal errorformat^=%A%f:%l:%c:\ %t%*[^:]: " GHC 9.6
+
+" lines that start with a space continue the previous message
+setlocal errorformat^=%+C\ %.%#
+
+" empty lines terminate a message
+setlocal errorformat^=%Z
+
+" ignore this part of the message
+setlocal errorformat^=%-G\ \ \ \ Suggested\ fix:%.%#
+setlocal errorformat^=%-G\ \ \ \ \ \ Perhaps\ you\ meant\ %.%# " GHC 9.2
+
+" Hspec
+setlocal errorformat^=\ \ %f:%l:%c:\ .%#