module Language.PureScript.Ide.Filter
( Filter
, moduleFilter
, prefixFilter
, equalityFilter
, applyFilters
) where
import Protolude hiding (isPrefixOf)
import Data.Aeson
import Data.Text (isPrefixOf)
import qualified Language.PureScript as P
import Language.PureScript.Ide.Types
import Language.PureScript.Ide.Util
newtype Filter = Filter (Endo [Module]) deriving(Monoid)
type Module = (P.ModuleName, [IdeDeclarationAnn])
mkFilter :: ([Module] -> [Module]) -> Filter
mkFilter = Filter . Endo
moduleFilter :: [P.ModuleName] -> Filter
moduleFilter =
mkFilter . moduleFilter'
moduleFilter' :: [P.ModuleName] -> [Module] -> [Module]
moduleFilter' moduleIdents = filter (flip elem moduleIdents . fst)
prefixFilter :: Text -> Filter
prefixFilter "" = mkFilter identity
prefixFilter t = mkFilter $ identFilter prefix t
where
prefix :: IdeDeclaration -> Text -> Bool
prefix ed search = search `isPrefixOf` identifierFromIdeDeclaration ed
equalityFilter :: Text -> Filter
equalityFilter = mkFilter . identFilter equality
where
equality :: IdeDeclaration -> Text -> Bool
equality ed search = identifierFromIdeDeclaration ed == search
identFilter :: (IdeDeclaration -> Text -> Bool) -> Text -> [Module] -> [Module]
identFilter predicate search =
filter (not . null . snd) . fmap filterModuleDecls
where
filterModuleDecls :: Module -> Module
filterModuleDecls (moduleIdent, decls) =
(moduleIdent, filter (flip predicate search . discardAnn) decls)
runFilter :: Filter -> [Module] -> [Module]
runFilter (Filter f) = appEndo f
applyFilters :: [Filter] -> [Module] -> [Module]
applyFilters = runFilter . fold
instance FromJSON Filter where
parseJSON = withObject "filter" $ \o -> do
(filter' :: Text) <- o .: "filter"
case filter' of
"exact" -> do
params <- o .: "params"
search <- params .: "search"
return $ equalityFilter search
"prefix" -> do
params <- o.: "params"
search <- params .: "search"
return $ prefixFilter search
"modules" -> do
params <- o .: "params"
modules <- map P.moduleNameFromString <$> params .: "modules"
return $ moduleFilter modules
_ -> mzero