Language.Haskell.Tools.AST.FromGHC

The FromGHC module provides a way to transform the GHC AST into our AST. This transformation is done in the Ghc monad. The conversion can be performed from the Parsed and the Renamed GHC AST. If the renamed AST is given, additional semantic information is looked up while traversing the AST.

Documentation

addTypeInfos :: LHsBinds Id -> Ann Module (Dom Name) RangeStage -> Ghc (Ann Module IdDom RangeStage)

extractExprIds :: LHsBinds Id -> [Located Id]

createModuleInfo :: Module -> Trf (ModuleInfo Name)

trfModule :: Module -> Located (HsModule RdrName) -> Trf (Ann Module (Dom RdrName) RangeStage)

trfModuleRename :: Module -> Ann Module (Dom RdrName) RangeStage -> (HsGroup Name, [LImportDecl Name], Maybe [LIE Name], Maybe LHsDocString) -> Located (HsModule RdrName) -> Trf (Ann Module (Dom Name) RangeStage)

trfModuleHead :: TransformName n r => Maybe (Located ModuleName) -> Maybe (Located [LIE n]) -> Maybe (Located WarningTxt) -> Trf (AnnMaybe ModuleHead (Dom r) RangeStage)

trfFilePragmas :: Trf (AnnList FilePragma (Dom r) RangeStage)

trfLanguagePragma :: Located String -> Trf (Ann FilePragma (Dom r) RangeStage)

trfOptionsPragma :: Located String -> Trf (Ann FilePragma (Dom r) RangeStage)

trfModulePragma :: Maybe (Located WarningTxt) -> Trf (AnnMaybe ModulePragma (Dom r) RangeStage)

trfText' :: StringLiteral -> Trf (StringNode (Dom r) RangeStage)

trfExportList :: TransformName n r => SrcLoc -> Maybe (Located [LIE n]) -> Trf (AnnMaybe ExportSpecList (Dom r) RangeStage)

trfExportList' :: TransformName n r => [LIE n] -> Trf (ExportSpecList (Dom r) RangeStage)

trfExport :: TransformName n r => LIE n -> Trf (Maybe (Ann ExportSpec (Dom r) RangeStage))

trfImports :: TransformName n r => [LImportDecl n] -> Trf (AnnList ImportDecl (Dom r) RangeStage)

trfImport :: TransformName n r => LImportDecl n -> Trf (Ann ImportDecl (Dom r) RangeStage)

trfImportSpecs :: TransformName n r => Maybe (Bool, Located [LIE n]) -> Trf (AnnMaybe ImportSpec (Dom r) RangeStage)

trfIESpec :: TransformName n r => LIE n -> Trf (Maybe (Ann IESpec (Dom r) RangeStage))

trfIESpec' :: TransformName n r => IE n -> Trf (Maybe (IESpec (Dom r) RangeStage))

type Trf = ReaderT TrfInput Ghc

The transformation monad type

data TrfInput

The (immutable) data for the transformation

Constructors

TrfInput

Fields

• srcMap :: SourceMap

  The lexical tokens of the source file

• pragmaComms :: Map String [Located String]

  Pragma comments

• contRange :: SrcSpan

  The focus of the transformation

• localsInScope :: [[Name]]

  Local names visible

• defining :: Bool

  True, if names are defined in the transformed AST element.

• definingTypeVars :: Bool

  True, if type variable names are defined in the transformed AST element.

• originalNames :: Map SrcSpan RdrName

  Stores the original format of names.

trfInit :: Map ApiAnnKey [SrcSpan] -> Map String [Located String] -> TrfInput

define :: Trf a -> Trf a

Perform the transformation taking names as defined.

defineTypeVars :: Trf a -> Trf a

Perform the transformation taking type variable names as defined.

typeVarTransform :: Trf a -> Trf a

Transform as type variables

transformingPossibleVar :: HsHasName n => n -> Trf a -> Trf a

Transform a name as a type variable if it is one.

addToScope :: HsHasName e => e -> Trf a -> Trf a

Perform the transformation putting the given definition in a new local scope.

addToCurrentScope :: HsHasName e => e -> Trf a -> Trf a

Perform the transformation putting the given definitions in the current scope.

runTrf :: Map ApiAnnKey [SrcSpan] -> Map String [Located String] -> Trf a -> Ghc a

Performs the transformation given the tokens of the source file

setOriginalNames :: Map SrcSpan RdrName -> Trf a -> Trf a

getOriginalName :: RdrName -> Trf String

Get the original format of a name (before scoping).

trfOperator :: TransformName n r => Located n -> Trf (Ann Operator (Dom r) RangeStage)

trfOperator' :: TransformName n r => n -> Trf (Operator (Dom r) RangeStage)

trfName :: TransformName n r => Located n -> Trf (Ann Name (Dom r) RangeStage)

trfName' :: TransformName n r => n -> Trf (Name (Dom r) RangeStage)

trfAmbiguousFieldName :: TransformName n r => Located (AmbiguousFieldOcc n) -> Trf (Ann Name (Dom r) RangeStage)

trfAmbiguousFieldName' :: forall n r. TransformName n r => SrcSpan -> AmbiguousFieldOcc n -> Trf (Ann Name (Dom r) RangeStage)

class (DataId n, Eq n, GHCName n) => TransformableName n where

Minimal complete definition

correctNameString, fromGHCName

Methods

correctNameString :: n -> Trf String

fromGHCName :: Name -> n

Instances

TransformableName RdrName

Methods

correctNameString :: RdrName -> Trf String

fromGHCName :: Name -> RdrName

TransformableName Name

Methods

correctNameString :: Name -> Trf String

fromGHCName :: Name -> Name

class (TransformableName name, HsHasName name, TransformableName res, HsHasName res, GHCName res) => TransformName name res where

This class allows us to use the same transformation code for multiple variants of the GHC AST. GHC Name annotated with name can be transformed to our representation with semantic annotations of res.

Minimal complete definition

transformName

Methods

transformName :: name -> res

Demote a given name

Instances

((~) * n r, TransformableName n, HsHasName n) => TransformName n r

Methods

transformName :: n -> r

(TransformableName res, GHCName res, HsHasName res) => TransformName Name res

Methods

transformName :: Name -> res

trfSimpleName :: TransformName n r => Located n -> Trf (Ann SimpleName (Dom r) RangeStage)

trfSimpleName' :: TransformName n r => n -> Trf (SimpleName (Dom r) RangeStage)

trfNameStr :: String -> Trf (AnnList UnqualName (Dom r) RangeStage)

Creates a qualified name from a name string

trfNameStr' :: String -> SrcLoc -> [Ann UnqualName (Dom r) RangeStage]

trfModuleName :: Located ModuleName -> Trf (Ann ModuleName (Dom r) RangeStage)

trfModuleName' :: ModuleName -> Trf (ModuleName (Dom r) RangeStage)

trfFastString :: Located FastString -> Trf (Ann StringNode (Dom r) RangeStage)

trfDataKeyword :: NewOrData -> Trf (Ann DataOrNewtypeKeyword (Dom r) RangeStage)

trfCallConv :: Located CCallConv -> Trf (Ann CallConv (Dom r) RangeStage)

trfCallConv' :: CCallConv -> Trf (CallConv (Dom r) RangeStage)

trfSafety :: SrcSpan -> Located Safety -> Trf (AnnMaybe Safety (Dom r) RangeStage)

trfOverlap :: Located OverlapMode -> Trf (Ann OverlapPragma (Dom r) RangeStage)

trfRole :: Located (Maybe Role) -> Trf (Ann Role (Dom r) RangeStage)

trfPhase :: Trf SrcLoc -> Activation -> Trf (AnnMaybe PhaseControl (Dom r) RangeStage)

trfPhaseNum :: PhaseNum -> Trf (Ann PhaseNumber (Dom r) RangeStage)

createNameInfo :: n -> Trf (NameInfo n)

Creates a semantic information for a name

createAmbigousNameInfo :: RdrName -> SrcSpan -> Trf (NameInfo n)

Creates a semantic information for an ambiguous name (caused by field disambiguation for example)

createImplicitNameInfo :: String -> Trf (NameInfo n)

Creates a semantic information for an implicit name

createImportData :: (HsHasName n, GHCName n) => ImportDecl (Dom n) stage -> Trf (ImportInfo n)

Adds semantic information to an impord declaration. See ImportInfo.

getImportedNames :: String -> Maybe String -> Trf (Module, [Name])

Get names that are imported from a given import

checkImportVisible :: (HsHasName n, GhcMonad m) => ImportDecl (Dom n) stage -> Name -> m Bool

Check is a given name is imported from an import with given import specification.

ieSpecMatches :: (HsHasName n, GhcMonad m) => IESpec (Dom n) stage -> Name -> m Bool

noSemaInfo :: src -> NodeInfo NoSemanticInfo src

nothing :: String -> String -> Trf SrcLoc -> Trf (AnnMaybe e (Dom n) RangeStage)

Creates a place for a missing node with a default location

emptyList :: String -> Trf SrcLoc -> Trf (AnnList e (Dom n) RangeStage)

makeList :: String -> Trf SrcLoc -> Trf [Ann e (Dom n) RangeStage] -> Trf (AnnList e (Dom n) RangeStage)

Creates a place for a list of nodes with a default place if the list is empty.

makeListBefore :: String -> String -> Trf SrcLoc -> Trf [Ann e (Dom n) RangeStage] -> Trf (AnnList e (Dom n) RangeStage)

makeListAfter :: String -> String -> Trf SrcLoc -> Trf [Ann e (Dom n) RangeStage] -> Trf (AnnList e (Dom n) RangeStage)

makeNonemptyList :: String -> Trf [Ann e (Dom n) RangeStage] -> Trf (AnnList e (Dom n) RangeStage)

makeIndentedList :: Trf SrcLoc -> Trf [Ann e (Dom n) RangeStage] -> Trf (AnnList e (Dom n) RangeStage)

Creates a place for an indented list of nodes with a default place if the list is empty.

makeIndentedListNewlineBefore :: Trf SrcLoc -> Trf [Ann e (Dom n) RangeStage] -> Trf (AnnList e (Dom n) RangeStage)

makeIndentedListBefore :: String -> Trf SrcLoc -> Trf [Ann e (Dom n) RangeStage] -> Trf (AnnList e (Dom n) RangeStage)

makeNonemptyIndentedList :: Trf [Ann e (Dom n) RangeStage] -> Trf (AnnList e (Dom n) RangeStage)

trfLoc :: (a -> Trf (b (Dom n) RangeStage)) -> Trf (SemanticInfo (Dom n) b) -> Located a -> Trf (Ann b (Dom n) RangeStage)

Transform a located part of the AST by automatically transforming the location. Sets the source range for transforming children.

trfLocNoSema :: SemanticInfo (Dom n) b ~ NoSemanticInfo => (a -> Trf (b (Dom n) RangeStage)) -> Located a -> Trf (Ann b (Dom n) RangeStage)

trfMaybe :: String -> String -> (Located a -> Trf (Ann e (Dom n) RangeStage)) -> Maybe (Located a) -> Trf (AnnMaybe e (Dom n) RangeStage)

Transforms a possibly-missing node with the default location of the end of the focus.

trfMaybeDefault :: String -> String -> (Located a -> Trf (Ann e (Dom n) RangeStage)) -> Trf SrcLoc -> Maybe (Located a) -> Trf (AnnMaybe e (Dom n) RangeStage)

Transforms a possibly-missing node with a default location

trfLocCorrect :: Trf (SemanticInfo (Dom n) b) -> (SrcSpan -> Trf SrcSpan) -> (a -> Trf (b (Dom n) RangeStage)) -> Located a -> Trf (Ann b (Dom n) RangeStage)

Transform a located part of the AST by automatically transforming the location with correction by applying the given function. Sets the source range for transforming children.

trfMaybeLoc :: (a -> Trf (Maybe (b (Dom n) RangeStage))) -> SemanticInfo (Dom n) b -> Located a -> Trf (Maybe (Ann b (Dom n) RangeStage))

Transform a located part of the AST by automatically transforming the location. Sets the source range for transforming children.

trfMaybeLocNoSema :: SemanticInfo (Dom n) b ~ NoSemanticInfo => (a -> Trf (Maybe (b (Dom n) RangeStage))) -> Located a -> Trf (Maybe (Ann b (Dom n) RangeStage))

trfAnnList :: SemanticInfo (Dom n) b ~ NoSemanticInfo => String -> (a -> Trf (b (Dom n) RangeStage)) -> [Located a] -> Trf (AnnList b (Dom n) RangeStage)

Creates a place for a list of nodes with the default place at the end of the focus if the list is empty.

trfAnnList' :: String -> (Located a -> Trf (Ann b (Dom n) RangeStage)) -> [Located a] -> Trf (AnnList b (Dom n) RangeStage)

nonemptyAnnList :: [Ann e (Dom n) RangeStage] -> AnnList e (Dom n) RangeStage

Creates a place for a list of nodes that cannot be empty.

makeJust :: Ann e (Dom n) RangeStage -> AnnMaybe e (Dom n) RangeStage

Creates an optional node from an existing element

annLoc :: Trf (SemanticInfo (Dom n) b) -> Trf SrcSpan -> Trf (b (Dom n) RangeStage) -> Trf (Ann b (Dom n) RangeStage)

Annotates a node with the given location and focuses on the given source span.

annLocNoSema :: SemanticInfo (Dom n) b ~ NoSemanticInfo => Trf SrcSpan -> Trf (b (Dom n) RangeStage) -> Trf (Ann b (Dom n) RangeStage)

Focus manipulation

focusOn :: SrcSpan -> Trf a -> Trf a

between :: AnnKeywordId -> AnnKeywordId -> Trf a -> Trf a

Focuses the transformation to go between tokens. The tokens must be found inside the current range.

betweenIfPresent :: AnnKeywordId -> AnnKeywordId -> Trf a -> Trf a

Focuses the transformation to go between tokens if they are present

focusAfter :: AnnKeywordId -> Trf a -> Trf a

Focuses the transformation to be performed after the given token. The token must be found inside the current range.

focusAfterIfPresent :: AnnKeywordId -> Trf a -> Trf a

focusBefore :: AnnKeywordId -> Trf a -> Trf a

Focuses the transformation to be performed after the given token. The token must be found inside the current range.

focusBeforeIfPresent :: AnnKeywordId -> Trf a -> Trf a

before :: AnnKeywordId -> Trf SrcLoc

Gets the position before the given token

after :: AnnKeywordId -> Trf SrcLoc

Gets the position after the given token

annFrom :: AnnKeywordId -> Trf (SemanticInfo (Dom n) e) -> Trf (e (Dom n) RangeStage) -> Trf (Ann e (Dom n) RangeStage)

The element should span from the given token to the end of focus

atTheStart :: Trf SrcLoc

Gets the position at the beginning of the focus

atTheEnd :: Trf SrcLoc

Gets the position at the end of the focus

tokenLoc :: AnnKeywordId -> Trf SrcSpan

Searches for a token inside the focus and retrieves its location

allTokenLoc :: AnnKeywordId -> Trf [SrcSpan]

tokenLocBack :: AnnKeywordId -> Trf SrcSpan

Searches for a token backward inside the focus and retrieves its location

tokenBefore :: SrcLoc -> AnnKeywordId -> Trf SrcSpan

tokensLoc :: [AnnKeywordId] -> Trf SrcSpan

Searches for tokens in the given order inside the parent element and returns their combined location

uniqueTokenAnywhere :: AnnKeywordId -> Trf SrcSpan

Searches for a token and retrieves its location anywhere

annCont :: Trf (SemanticInfo (Dom n) e) -> Trf (e (Dom n) RangeStage) -> Trf (Ann e (Dom n) RangeStage)

Annotates the given element with the current focus as a location.

annContNoSema :: SemanticInfo (Dom n) e ~ NoSemanticInfo => Trf (e (Dom n) RangeStage) -> Trf (Ann e (Dom n) RangeStage)

copyAnnot :: SemanticInfo (Dom n) a ~ SemanticInfo (Dom n) b => (Ann a (Dom n) RangeStage -> b (Dom n) RangeStage) -> Trf (Ann a (Dom n) RangeStage) -> Trf (Ann b (Dom n) RangeStage)

Annotates the element with the same annotation that is on the other element

foldLocs :: [SrcSpan] -> SrcSpan

Combine source spans into one that contains them all

advanceStr :: String -> SrcLoc -> SrcLoc

The location after the given string

updateCol :: (Int -> Int) -> SrcLoc -> SrcLoc

Update column information in a source location

collectLocs :: [Located e] -> SrcSpan

Combine source spans of elements into one that contains them all

orderDefs :: [Ann e (Dom n) RangeStage] -> [Ann e (Dom n) RangeStage]

Rearrange definitions to appear in the order they are defined in the source file.

orderAnnList :: AnnList e (Dom n) RangeStage -> AnnList e (Dom n) RangeStage

Orders a list of elements to the order they are defined in the source file.

trfScopedSequence :: HsHasName d => (d -> Trf e) -> [d] -> Trf [e]

Transform a list of definitions where the defined names are in scope for subsequent definitions

splitLocated :: Located String -> [Located String]

Splits a given string at whitespaces while calculating the source location of the fragments