diff options
author | Ben Gamari <ben@smart-cactus.org> | 2017-11-02 14:35:01 -0400 |
---|---|---|
committer | Ben Gamari <ben@smart-cactus.org> | 2017-11-02 17:03:54 -0400 |
commit | d9b6015d1942aa176e85bb71f34200bab54e1c9c (patch) | |
tree | 416d1dca0af01e30570b8aa6ce662e12579c8b59 /utils/check-api-annotations/Main.hs | |
parent | bd765f4b1332b3d2a7908de3f9ff1d50da0e0b1d (diff) | |
download | haskell-d9b6015d1942aa176e85bb71f34200bab54e1c9c.tar.gz |
Revert "Move check-ppr and check-api-annotations to testsuite/utils"
Unfortunately this (ironically) ended up breaking bindist testing since
we didn't have a package-data.mk. Unfortunately there is no easy way to
fix this.
This reverts commit 1e9f90af7311c33de0f7f5b7dba594725596d675.
Diffstat (limited to 'utils/check-api-annotations/Main.hs')
-rw-r--r-- | utils/check-api-annotations/Main.hs | 122 |
1 files changed, 122 insertions, 0 deletions
diff --git a/utils/check-api-annotations/Main.hs b/utils/check-api-annotations/Main.hs new file mode 100644 index 0000000000..6b973e12e8 --- /dev/null +++ b/utils/check-api-annotations/Main.hs @@ -0,0 +1,122 @@ +{-# LANGUAGE RankNTypes #-} + +import Data.Data +import Data.List +import GHC +import DynFlags +import Outputable +import ApiAnnotation +import System.Environment( getArgs ) +import System.Exit +import qualified Data.Map as Map +import qualified Data.Set as Set + +main::IO() +main = do + args <- getArgs + case args of + [libdir,fileName] -> testOneFile libdir fileName + _ -> putStrLn "invoke with the libdir and a file to parse." + +testOneFile :: FilePath -> String -> IO () +testOneFile libdir fileName = do + let modByFile m = + case ml_hs_file $ ms_location m of + Nothing -> False + Just fn -> fn == fileName + ((anns,_cs),p) <- runGhc (Just libdir) $ do + dflags <- getSessionDynFlags + _ <- setSessionDynFlags dflags + addTarget Target { targetId = TargetFile fileName Nothing + , targetAllowObjCode = True + , targetContents = Nothing } + _ <- load LoadAllTargets + graph <- getModuleGraph + let modSum = + case filter modByFile (mgModSummaries graph) of + [x] -> x + xs -> error $ "Can't find module, got:" + ++ show (map (ml_hs_file . ms_location) xs) + p <- parseModule modSum + return (pm_annotations p,p) + + let sspans = Set.fromList $ getAllSrcSpans (pm_parsed_source p) + + exploded = [((kw,ss),[anchor]) + | ((anchor,kw),sss) <- Map.toList anns,ss <- sss] + + exploded' = Map.toList $ Map.fromListWith (++) exploded + + problems' = filter (\(_,anchors) + -> not (any (\a -> Set.member a sspans) anchors)) + exploded' + + problems'' = filter (\((a,_),_) -> a /= AnnEofPos) problems' + + putStrLn "---Problems (should be empty list)---" + putStrLn (intercalate "\n" [pp $ Map.fromList $ map fst problems'']) + putStrLn "---Annotations-----------------------" + putStrLn "-- SrcSpan the annotation is attached to, AnnKeywordId," + putStrLn "-- list of locations the keyword item appears in" + -- putStrLn (intercalate "\n" [showAnns anns]) + putStrLn (showAnns anns) + if null problems'' + then exitSuccess + else exitFailure + + where + getAllSrcSpans :: (Data t) => t -> [SrcSpan] + getAllSrcSpans ast = everything (++) ([] `mkQ` getSrcSpan) ast + where + getSrcSpan :: SrcSpan -> [SrcSpan] + getSrcSpan ss = [ss] + + +showAnns :: Map.Map ApiAnnKey [SrcSpan] -> String +showAnns anns = "[\n" ++ (intercalate ",\n" + $ map (\((s,k),v) + -- -> ("(AK " ++ pp s ++ " " ++ show k ++" = " ++ pp v ++ ")\n")) + -> ("((" ++ pp s ++ "," ++ show k ++"), " ++ pp v ++ ")")) + $ Map.toList anns) + ++ "\n]\n" + +pp :: (Outputable a) => a -> String +pp a = showPpr unsafeGlobalDynFlags a + + +-- --------------------------------------------------------------------- + +-- Copied from syb for the test + + +-- | Generic queries of type \"r\", +-- i.e., take any \"a\" and return an \"r\" +-- +type GenericQ r = forall a. Data a => a -> r + + +-- | Make a generic query; +-- start from a type-specific case; +-- return a constant otherwise +-- +mkQ :: ( Typeable a + , Typeable b + ) + => r + -> (b -> r) + -> a + -> r +(r `mkQ` br) a = case cast a of + Just b -> br b + Nothing -> r + + + +-- | Summarise all nodes in top-down, left-to-right order +everything :: (r -> r -> r) -> GenericQ r -> GenericQ r + +-- Apply f to x to summarise top-level node; +-- use gmapQ to recurse into immediate subterms; +-- use ordinary foldl to reduce list of intermediate results + +everything k f x = foldl k (f x) (gmapQ (everything k f) x) |