@@ -5,7 +5,6 @@ module StackTrace.Plugin (plugin) where
55
66import Control.Arrow (first)
77import Data.Monoid (Any(Any, getAny))
8−import GHC.Types.SrcLoc
98#if __GLASGOW_HASKELL__ >= 900
109import GHC.Plugins
1110#else
@@ -50,14 +49,9 @@ importDeclQualified = QualifiedPre
5049
5150ghcStackImport :: Located (ImportDecl (GhcPass p))
5251ghcStackImport =
53− L srcSpan $
52+ noLoc $
5453 (simpleImportDecl $ mkModuleName "GHC.Stack")
55− { ideclQualified = importDeclQualified, ideclAs = Just $ noLoc ghcStackModuleName }
56− where
57− -- This is for GHC-9 related problems. @noLoc@ causes GHC to throw warnings
58− -- about unused imports. Even if the import is used
59− -- See: https://github.com/waddlaw/haskell-stack-trace-plugin/issues/16
60− srcSpan = RealSrcSpan (realSrcLocSpan $ mkRealSrcLoc "haskell-stack-trace-plugin:very-unique-file-name-to-avoid-collision" 1 1) Nothing
54+ {ideclQualified = importDeclQualified, ideclAs = Just $ noLoc ghcStackModuleName}
6155
6256#if __GLASGOW_HASKELL__ >= 900
6357updateHsModule :: HsModule -> HsModule
@@ -166,17 +160,15 @@ updateLHsType = traverse
166160
167161-- Main process
168162updateHsType :: HsType GhcPs -> (Any, HsType GhcPs)
169−updateHsType ty@(HsQualTy xty ctxt body) =
170− if hasHasCallStack (unLoc ctxt)
171− then pure ty
172− else flagASTModified $ HsQualTy xty (fmap appendHSC ctxt) body
163+updateHsType (HsQualTy xty ctxt body) =
164+ flagASTModified $ HsQualTy xty (fmap appendHSC ctxt) body
173165updateHsType ty@HsTyVar {} =
174166 flagASTModified $ HsQualTy xQualTy (noLoc $ appendHSC []) (noLoc ty)
175167updateHsType ty@HsAppTy {} =
176168 flagASTModified $ HsQualTy xQualTy (noLoc $ appendHSC []) (noLoc ty)
177169updateHsType ty@HsFunTy {} =
178170 flagASTModified $ HsQualTy xQualTy (noLoc $ appendHSC []) (noLoc ty)
179−updateHsType ty@HsListTy {} =
171+updateHsType ty@HsListTy {} =
180172 flagASTModified $ HsQualTy xQualTy (noLoc $ appendHSC []) (noLoc ty)
181173updateHsType ty@HsTupleTy {} =
182174 flagASTModified $ HsQualTy xQualTy (noLoc $ appendHSC []) (noLoc ty)
@@ -196,13 +188,6 @@ flagASTModified a = (Any True, a)
196188appendHSC :: HsContext GhcPs -> HsContext GhcPs
197189appendHSC cs = mkHSC : cs
198190
199−hasHasCallStack :: HsContext GhcPs -> Bool
200−hasHasCallStack = any (checkHsType . unLoc)
201− where
202− checkHsType :: HsType GhcPs -> Bool
203− checkHsType (HsTyVar _ _ lid) = unLoc lid == (mkRdrUnqual $ mkClsOcc "HasCallStack")
204− checkHsType _ = False
205−
206191-- make HasCallStack => constraint
207192mkHSC :: LHsType GhcPs
208193mkHSC = noLoc $ HsTyVar xQualTy NotPromoted lId