38 lines
1.3 KiB
Haskell
38 lines
1.3 KiB
Haskell
module Language.Haskell.GhcMod.List (listModules, modules) where
|
|
|
|
import Control.Applicative ((<$>))
|
|
import Control.Exception (SomeException(..))
|
|
import Control.Monad (void)
|
|
import Data.List (nub, sort)
|
|
import GHC (Ghc)
|
|
import qualified GHC as G
|
|
import Language.Haskell.GhcMod.GHCApi
|
|
import Language.Haskell.GhcMod.Types
|
|
import Packages (pkgIdMap, exposedModules, sourcePackageId, display)
|
|
import UniqFM (eltsUFM)
|
|
|
|
----------------------------------------------------------------
|
|
|
|
-- | Listing installed modules.
|
|
listModules :: Options -> Cradle -> IO String
|
|
listModules opt cradle = withGHC' $ do
|
|
void $ initializeFlagsWithCradle opt cradle [] False
|
|
modules opt
|
|
|
|
-- | Listing installed modules.
|
|
modules :: Options -> Ghc String
|
|
modules opt = convert opt . arrange <$> (getModules `G.gcatch` handler)
|
|
where
|
|
getModules = getExposedModules <$> G.getSessionDynFlags
|
|
getExposedModules = concatMap exposedModules'
|
|
. eltsUFM . pkgIdMap . G.pkgState
|
|
exposedModules' p =
|
|
map G.moduleNameString (exposedModules p)
|
|
`zip`
|
|
repeat (display $ sourcePackageId p)
|
|
arrange = nub . sort . map dropPkgs
|
|
dropPkgs (name, pkg)
|
|
| detailed opt = name ++ " " ++ pkg
|
|
| otherwise = name
|
|
handler (SomeException _) = return []
|