module Language.Haskell.Tools.Refactor.Predefined.GenerateExports (generateExports, DomGenerateExports) where
import Control.Reference ((^?), (.=), (&))
import Language.Haskell.Tools.Refactor
import qualified GHC (NamedThing(..), Name(..))
import Control.Applicative ((<|>))
import Data.Maybe (Maybe(..), catMaybes)
type DomGenerateExports dom = (Domain dom, HasNameInfo dom)
generateExports :: DomGenerateExports dom => LocalRefactoring dom
generateExports mod = return (modHead & annJust & mhExports & annMaybe
.= Just (createExports (getTopLevels mod)) $ mod)
getTopLevels :: DomGenerateExports dom => Module dom -> [(GHC.Name, Bool)]
getTopLevels mod = catMaybes $ map (\d -> fmap (,exportContainOthers d)
(foldl (<|>) Nothing $ map semanticsName $ d ^? elementName))
(mod ^? modDecl & annList)
where exportContainOthers :: Decl dom -> Bool
exportContainOthers (DataDecl {}) = True
exportContainOthers (ClassDecl {}) = True
exportContainOthers _ = False
createExports :: [(GHC.Name, Bool)] -> ExportSpecs dom
createExports elems = mkExportSpecs $ map (mkExportSpec . createExport) elems
where createExport (n, False) = mkIESpec (mkUnqualName' (GHC.getName n)) Nothing
createExport (n, True) = mkIESpec (mkUnqualName' (GHC.getName n)) (Just mkSubAll)