Championship stats & predictions

Hamilton Academical vs. Dundee United

<|repo_name|>konojunya/ghc<|file_sep|>/compiler/GHC/Plugins/Parsing/ParserUtils.hs {-# LANGUAGE CPP #-} {-# LANGUAGE OverloadedStrings #-} {-# LANGUAGE ScopedTypeVariables #-} module GHC.Plugins.Parsing.ParserUtils where import Control.Monad import Control.Monad.IO.Class (MonadIO(..)) import Data.Foldable (toList) import Data.List.NonEmpty (NonEmpty(..), nonEmpty) import Data.Maybe (fromMaybe) import Data.String import GHC import GHC.Core import GHC.Core.Opt.CSE import GHC.Core.TyCo.Reify import GHC.Driver.Env import GHC.Driver.Ppr import GHC.Hs import GHC.Plugins.Types import GHC.Plugins.TypeUtils import qualified Data.ByteString.Char8 as B8 #if MIN_VERSION_ghc(9,4,0) import GHC.Plugins.Internal.GhcPluginsCompat (fromRdrName) #endif -- | @parseModule s@ parses 'RdrName's from @s@ into a list of top-level bindings, -- along with any errors encountered during parsing. parseModule :: MonadIO m => String -> m ([(String, Located RdrName)], [ParseError]) parseModule s = do #if MIN_VERSION_ghc(9,4,0) parseModule' s >>= case #else parseModule' s >>= case #endif ParseOk m _ -> pure ([], getTopBindings m) ParseFailed errs -> pure ([], errs) parseModule' :: MonadIO m => String -> m ParseResult GhcPsModule GhcPsBinds GhcPsDecs GhcPsType GhcPsExpr GhcPsStmt GhcPsLHsDecl GhcPsLHsExpr GhcPsLHsPat GhcPsLHsType GhcPsLHsQSpec GhcPsLHsBinds GhcPsLHsBndr GhcPsLHsBndrs GhcPsLHsLocId -> m ParseResult GhcPsModule GhcPsBinds GhcPsDecs GhcPsType GhcPsExpr GhcPsStmt GhcPsLHsDecl GhcPsLHsExpr GhcPsLHsPat GhcPsLHsType GhcPsLHsQSpec GhcPsLHsBinds GhcPsLHsBndr GhcPsLHsBndrs GhcPsLHsLocId parseModule' s = do #if MIN_VERSION_ghc(9,4,0) _ <- liftIO $ runGhc (Just libdir) $ do #else _ <- liftIO $ runGhc (Just libdir) $ do #endif -- TODO: maybe allow setting other options? setSessionDynFlags defaultDynFlags { hscTarget = HscInterpreted } -- TODO: this is not ideal; it should not depend on compiler settings like this... -- TODO: is there any way around this? setPluginHooks $ defaultPluginHooks { pprPanic = const (pure ()) } parseModuleWithComments s getTopBindings :: MonadIO m => HscEnv -> m [(String, Located RdrName)] getTopBindings hsc_env = do #if MIN_VERSION_ghc(9,4,0) -- TODO: we should make sure these are exactly equivalent; -- it would be nice if we didn't have different APIs here... let top_bindings = fmap ((_, x) -> x) $ getOrphans $ hscEnvOrphanDecls hsc_env -- If there are no top-level bindings returned by 'getOrphans', -- then try using 'hsGetParsedSource'. if null top_bindings then do -- See https://gitlab.haskell.org/ghc/ghc/-/issues/16874; it seems that sometimes -- getOrphans returns nothing even though there are top-level declarations. hs_mod <- hsGetParsedSource Nothing hsc_env let top_bindings = case hs_mod_source hs_mod of HsValD ls | null ls -> [] | otherwise -> fmap ((_, x) -> x) ls HsModD ls | null ls -> [] | otherwise -> fmap ((_, x) -> x) ls ++ getTopBindingsFromDataDecls hs_mod_source hs_mod HsGroupD ls | null ls -> [] | otherwise -> fmap ((_, x) -> x) ls ++ getTopBindingsFromDataDecls hs_mod_source hs_mod _ -> [] pure top_bindings else pure top_bindings #else let top_bindings = fmap ((_, x) -> x) $ getOrphans $ hscEnvOrphanDecls hsc_env let hs_mod = hscGetParsedSource Nothing hsc_env let top_bindings_from_hsmod = case hs_mod_source hs_mod of HsValD ls | null ls -> [] | otherwise -> fmap ((_, x) -> x) ls HsModD ls | null ls -> [] | otherwise -> fmap ((_, x) -> x) ls ++ getTopBindingsFromDataDecls hs_mod_source hs_mod HsGroupD ls | null ls -> [] | otherwise -> fmap ((_, x) -> x) ls ++ getTopBindingsFromDataDecls hs_mod_source hs_mod _ -> [] pure (top_bindings ++ top_bindings_from_hsmod) #endif -- | @getTopBindingsFromDataDecls decl_src mod@ extracts all top-level bindings from data declarations @decl_src@ found in @mod@. getTopBindingsFromDataDecls :: MonadIO m => HsDecl SrcSpanInfo -> HscSource -> m [(String, Located RdrName)] getTopBindingsFromDataDecls decl_src mod = #if MIN_VERSION_ghc(9,4,0) case decl_src of #else case decl_src of {- HsDataDefn lqname ty_vars cons _deriving -> fmap (x@(Located _ n@(Unqual _)) -> (lqnameFS lqname, case n `elem` ty_vars of {- True -> Located lqnameFS lqname $ Unqual $ mkRdrQual lqnameFS $ mkRdrVar n {-TODO: I don't think this should depend on 'lqnameFS'...-} False -> Located lqnameFS lqname n -} ) <$> nonEmpty cons -} HsNewtypeDefn lqname ty_vars con _deriving -> fmap (x@(Located _ n@(Unqual _)) -> (lqnameFS lqname, case n `elem` ty_vars of {- True -> Located lqnameFS lqname $ Unqual $ mkRdrQual lqnameFS $ mkRdrVar n {-TODO: I don't think this should depend on 'lqnameFS'...-} False -> Located lqnameFS lqname n -} ) <$> nonEmpty [con] {-TODO: I don't think there should only ever be one constructor-} -}_ #endif HsDataDefn{} -> pure . concatMap extractDataDecl mod_decls_data_decls where mod_decls_data_decls = getModDeclsDataDecls mod_decls mod_body -- TODO: make sure these two functions are equivalent! HsNewtypeDefn{} -> pure . concatMap extractNewtypeDecl mod_decls_data_decls where mod_decls_data_decls = getModDeclsDataDecls mod_decls mod_body -- TODO: make sure these two functions are equivalent! _ -> pure [] -- | @extractDataDecl decl@ extracts all top-level bindings from data declaration @decl@. extractDataDecl :: MonadIO m => HsDataDefn SrcSpanInfo -> m [(String,LitPlaceholder)] extractDataDecl (HsDataDefn lqname ty_vars cons _deriving) #if MIN_VERSION_ghc(9,4,0) = concat <$> traverse extractConstructor cons <* traverse extractTyVar ty_vars where {-# SCC extractConstructor #-} #else = concat <$> traverse extractConstructor cons <* traverse extractTyVar ty_vars where {-# SCC extractConstructor #-} #endif -- | @extractNewtypeDecl decl@ extracts all top-level bindings from newtype declaration @decl@. extractNewtypeDecl :: MonadIO m => HsNewtypeDefn SrcSpanInfo -> m [(String,LitPlaceholder)] extractNewtypeDecl (HsNewtypeDefn lqname ty_vars con _deriving) #if MIN_VERSION_ghc(9,4,0) = concat <$> traverse extractConstructor [con] <* traverse extractTyVar ty_vars where {-# SCC extractConstructor #-} #else = concat <$> traverse extractConstructor [con] <* traverse extractTyVar ty_vars where {-# SCC extractConstructor #-} #endif -- | @extractTyVar tv@ extracts any type variable bindings from @tv@. extractTyVar :: MonadIO m => TyVarBndr SrcSpanInfo -> m [(String,LitPlaceholder)] extractTyVar tv@(PlainTV name) #if MIN_VERSION_ghc(9,4,0) = pure [(litPlaceholder name,toRdrName name)] extractTyVar tv@(KindedTV name _) = pure [] #elif MIN_VERSION_ghc(9,2,0) = pure [(litPlaceholder name,toRdrName name)] extractTyVar tv@(KindedTV name _) = pure [] #else = pure [(litPlaceholder name,toRdrName name)] extractTyVar tv@(KindedTV name _) = pure [] #endif -- | @extractConstructor c@ extracts any constructor bindings from @cons@. extractConstructor :: MonadIO m => ConLike SrcSpanInfo () () () () () () () () () () () () () () () () () () () () () () -> m [(String,LitPlaceholder)] #if MIN_VERSION_ghc(9,4,0) -- TODO: it seems like there are constructors which have no fields; -- does that mean that they cannot have type variable bindings? -- Or does it just mean that they have no fields? -- It seems like there shouldn't be any constructors without type variables though... extractConstructor c@(RecC name fields) = extractRecordConFields c *> pure [(litPlaceholder name,toRdrName name)] extractConstructor c@(NormalC name fields) = concat <$> traverse extractConField fields <* pure [(litPlaceholder name,toRdrName name)] extractConstructor c@(InfixC field1 con field2) = concat <$> traverse extractConField [field1] <* traverse extractConField [field2] -- TODO: figure out what kind of data constructor 'ForallC' actually is... -- We assume it doesn't have any type variable bindings... -- -- Also note that if you're using record syntax then you may need some sort of record syntax extension enabled; -- see https://gitlab.haskell.org/ghc/ghc/-/issues/16874. #if !MIN_VERSION_ghc(9,6) -- This case doesn't seem necessary because it looks like 'ForallC' -- always gets desugared into 'RecC'. -- -- I'm not sure why it still exists though... #elif !MIN_VERSION_ghc(9,8) #else #elif !MIN_VERSION_ghc(9,10) #else #endif extractConstructor c@(ForallC ty_var_bndrs ctx con) #if MIN_VERSION_ghc(9,6) -- NOTE: I don't think there should be any record syntax