aboutsummaryrefslogtreecommitdiff
path: root/src/GHCUtils.hs
blob: b5448dd43be3da02f4fb58ff550fcc1379d43e1d (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
module GHCUtils where

import HsSyn
import SrcLoc

getMainDeclBinder :: HsDecl name -> Maybe name
getMainDeclBinder (TyClD d) = Just (tcdName d)
getMainDeclBinder (ValD d)
   = case collectAcc d [] of
        []       -> Nothing   -- see rn003
        (name:_) -> Just (unLoc name)
getMainDeclBinder (SigD d) = sigNameNoLoc d
getMainDeclBinder (ForD (ForeignImport name _ _)) = Just (unLoc name)
getMainDeclBinder (ForD (ForeignExport name _ _)) = Just (unLoc name)
getMainDeclBinder _ = Nothing