\begin{code}
module HsSyn (
-
-- NB: don't reexport HsCore
-- this module tells about "real Haskell"
- module HsSyn,
module HsBinds,
module HsDecls,
module HsExpr,
module HsTypes,
Fixity, NewOrData,
+ HsModule(..),
+ collectStmtsBinders, collectStmtBinders,
collectHsBinders, collectLocatedHsBinders,
collectMonoBinders, collectLocatedMonoBinders,
- collectSigTysFromMonoBinds,
- hsModule, hsImports
+ collectSigTysFromHsBinds, collectSigTysFromMonoBinds
) where
#include "HsVersions.h"
\begin{code}
data HsModule name
= HsModule
- Module
- (Maybe Version) -- source interface version number
- (Maybe [IE name]) -- export list; Nothing => export everything
- -- Just [] => export *nothing* (???)
+ (Maybe Module) -- Nothing => "module X where" is omitted
+ -- (in which case the next field is Nothing too)
+ (Maybe [IE name]) -- Export list; Nothing => export list omitted, so export everything
+ -- Just [] => export *nothing*
-- Just [...] => as you would expect...
[ImportDecl name] -- We snaffle interesting stuff out of the
-- imported interfaces early on, adding that
instance (NamedThing name, OutputableBndr name)
=> Outputable (HsModule name) where
- ppr (HsModule name iface_version exports imports
- decls deprec src_loc)
+ ppr (HsModule Nothing _ imports decls _ src_loc)
+ = pp_nonnull imports $$ pp_nonnull decls
+
+ ppr (HsModule (Just name) exports imports decls deprec src_loc)
= vcat [
case exports of
Nothing -> pp_header (ptext SLIT("where"))
pp_modname = ptext SLIT("module") <+> ppr name
- pp_nonnull [] = empty
- pp_nonnull xs = vcat (map ppr xs)
-
-hsModule (HsModule mod _ _ _ _ _ _) = mod
-hsImports (HsModule mod vers exports imports decls deprec src_loc) = imports
+pp_nonnull [] = empty
+pp_nonnull xs = vcat (map ppr xs)
\end{code}
\begin{code}
collectLocatedHsBinders :: HsBinds name -> [(name,SrcLoc)]
+-- Used at top level only; so no need for an IPBinds case
collectLocatedHsBinders EmptyBinds = []
collectLocatedHsBinders (MonoBind b _ _)
= collectLocatedMonoBinders b
= collectLocatedHsBinders b1 ++ collectLocatedHsBinders b2
collectHsBinders :: HsBinds name -> [name]
-collectHsBinders EmptyBinds = []
-collectHsBinders (MonoBind b _ _)
- = collectMonoBinders b
-collectHsBinders (ThenBinds b1 b2)
- = collectHsBinders b1 ++ collectHsBinders b2
+collectHsBinders EmptyBinds = []
+collectHsBinders (IPBinds _ _) = [] -- Implicit parameters don't create
+ -- ordinary bindings
+collectHsBinders (MonoBind b _ _) = collectMonoBinders b
+collectHsBinders (ThenBinds b1 b2) = collectHsBinders b1 ++ collectHsBinders b2
collectLocatedMonoBinders :: MonoBinds name -> [(name,SrcLoc)]
collectLocatedMonoBinders binds
go (PatMonoBind pat _ loc) acc = collectPatBinders pat ++ acc
go (FunMonoBind f _ _ loc) acc = f : acc
go (AndMonoBinds bs1 bs2) acc = go bs1 (go bs2 acc)
+ go (VarMonoBind v _) acc = v : acc
+ go (AbsBinds _ _ dbinds _ binds) acc
+ = [dp | (_,dp,_) <- dbinds] ++ go binds acc
\end{code}
+
%************************************************************************
%* *
\subsection{Getting patterns out of bindings}
Get all the pattern type signatures out of a bunch of bindings
\begin{code}
+collectSigTysFromHsBinds :: HsBinds name -> [HsType name]
+collectSigTysFromHsBinds EmptyBinds = []
+collectSigTysFromHsBinds (IPBinds _ _) = []
+collectSigTysFromHsBinds (MonoBind b _ _) = collectSigTysFromMonoBinds b
+collectSigTysFromHsBinds (ThenBinds b1 b2) = collectSigTysFromHsBinds b1 ++
+ collectSigTysFromHsBinds b2
+
+
collectSigTysFromMonoBinds :: MonoBinds name -> [HsType name]
collectSigTysFromMonoBinds bind
= go bind []
go_matches (match : matches) acc = go_matches matches acc
\end{code}
+\begin{code}
+collectStmtsBinders :: [Stmt id] -> [id]
+collectStmtsBinders = concatMap collectStmtBinders
+
+collectStmtBinders :: Stmt id -> [id]
+ -- Id Binders for a Stmt... [but what about pattern-sig type vars]?
+collectStmtBinders (BindStmt pat _ _) = collectPatBinders pat
+collectStmtBinders (LetStmt binds) = collectHsBinders binds
+collectStmtBinders (ExprStmt _ _ _) = []
+collectStmtBinders (ResultStmt _ _) = []
+collectStmtBinders (RecStmt ss _ _ _) = collectStmtsBinders ss
+collectStmtBinders other = panic "collectStmtBinders"
+\end{code}
+