[haskell indexer facebookincubator#5] Optionally store file content

Author: simonmar

glean/lang/haskell/HieIndexer/Index.hs

@@ -398,6 +398,10 @@ indexHieFile writer HieIndexerOptions{..} path hie = do
     let fileLines = mkFileLines filefact offs
     Glean.makeFact_ @Src.FileLines fileLines
 
+    when storeSrc $
+      Glean.makeFactV_ @Src.FileContent filefact $
+        Text.decodeUtf8Lenient (hie_hs_src hie)
+
     Glean.makeFact_ @Hs.ModuleSource $
       Hs.ModuleSource_key modfact filefact
 

glean/lang/haskell/HieIndexer/Options.hs

@@ -36,6 +36,8 @@ data HieIndexerOptions = HieIndexerOptions
       -- ^ Prefix to add to source paths
   , unitName :: UnitName
       -- ^ How to handle unit names
+  , storeSrc :: Bool
+      -- ^ Whether to store the indexed source in src.FileContent
   , verbosity :: Int
   }
 
@@ -82,6 +84,10 @@ options = info (helper <*> parser) fullDesc
         metavar "PATH" <>
         help "Prefix to add to source paths in the DB"))
 
+      storeSrc <- switch (
+        long "store-src" <>
+        help "Store the full source file in src.FileContent")
+
       unitName <- option (maybeReader parseUnit) (
         long "unit" <>
         metavar "key|id" <>

glean/schema/source/src.angle

@@ -8,6 +8,9 @@ predicate File : string
 # Source code digests produced by unspecified hash function
 predicate FileDigest : File -> string
 
+# Source code content (if the indexer stored it)
+predicate FileContent : File -> string
+
 # Common source code location type
 type Loc = {
   file : File,