CallStack.hs

{-# LANGUAGE CPP #-}
{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE OverloadedStrings #-}

module Skeletest.TestUtils.CallStack (
  sanitizeTraceback,
) where

#if __GLASGOW_HASKELL__ == 910
import Data.Text qualified as Text
#endif

-- GHC 9.10 specifically added a backtrace to SomeException, which was reverted in 9.12
-- https://github.com/haskell/core-libraries-committee/issues/285
sanitizeTraceback :: String -> String
#if __GLASGOW_HASKELL__ == 910
sanitizeTraceback s =
  let (pre, post) = break ("HasCallStack backtrace:" `Text.isInfixOf`) $ Text.lines $ Text.pack s
      (_, post2) = span isCallStackLine $ drop 1 post
   in Text.unpack . Text.unlines $ pre ++ post2
 where
  isCallStackLine line =
    or
      [ ", called at" `Text.isInfixOf` line
      , -- True if this line is the wrapped continuation of the previous line
        (fmap fst . Text.uncons) line `notElem` map Just ['│', '╰']
      ]
#else
sanitizeTraceback = id
#endif