Merge pull request #71 from haskell-works/new-create-index-command

New create-index command

Author: newhoggy

app/App/Commands/CreateIndex.hs

@@ -0,0 +1,60 @@
+{-# LANGUAGE DataKinds            #-}
+{-# LANGUAGE DeriveGeneric        #-}
+{-# LANGUAGE FlexibleInstances    #-}
+{-# LANGUAGE OverloadedStrings    #-}
+{-# LANGUAGE ScopedTypeVariables  #-}
+{-# LANGUAGE TypeApplications     #-}
+{-# LANGUAGE TypeSynonymInstances #-}
+
+module App.Commands.CreateIndex
+  ( cmdCreateIndex
+  ) where
+
+import Control.Lens
+import Control.Monad
+import Data.Generics.Product.Any
+import Data.Semigroup                             ((<>))
+import HaskellWorks.Data.Xml.Succinct.Cursor.MMap
+import Options.Applicative                        hiding (columns)
+
+import qualified App.Commands.Types                as Z
+import qualified Data.ByteString.Lazy              as LBS
+import qualified HaskellWorks.Data.ByteString.Lazy as LBS
+
+runCreateIndex :: Z.CreateIndexOptions -> IO ()
+runCreateIndex opt = do
+  let input         = opt ^. the @"input"
+  let maybeIbOutput = opt ^. the @"ibOutput"
+  let maybeBpOutput = opt ^. the @"bpOutput"
+
+  cursor <- mmapSlowCursor input
+
+  forM_ maybeIbOutput $ flip LBS.writeFile (LBS.toLazyByteString (cursor ^. the @"interests"      . the @1))
+  forM_ maybeBpOutput $ flip LBS.writeFile (LBS.toLazyByteString (cursor ^. the @"balancedParens" . the @1))
+
+  return ()
+
+optsCreateIndex :: Parser Z.CreateIndexOptions
+optsCreateIndex = Z.CreateIndexOptions
+  <$> strOption
+      (   long "input"
+      <>  help "Input file"
+      <>  metavar "FILE"
+      )
+  <*> optional
+      ( strOption
+        (   long "ib-output"
+        <>  help "Interest Bits output"
+        <>  metavar "FILE"
+        )
+      )
+  <*> optional
+      ( strOption
+        (   long "bp-output"
+        <>  help "Balanced Parens output"
+        <>  metavar "FILE"
+        )
+      )
+
+cmdCreateIndex :: Mod CommandFields (IO ())
+cmdCreateIndex = command "create-index"  $ flip info idm $ runCreateIndex <$> optsCreateIndex

app/App/Commands/Types.hs

@@ -3,6 +3,7 @@
 
 module App.Commands.Types
   ( CountOptions(..)
+  , CreateIndexOptions(..)
   , DemoOptions(..)
   ) where
 
@@ -17,3 +18,9 @@ data CountOptions = CountOptions
   , xpath  :: XPath
   , method :: Text
   } deriving (Eq, Show, Generic)
+
+data CreateIndexOptions = CreateIndexOptions
+  { input    :: FilePath
+  , ibOutput :: Maybe FilePath
+  , bpOutput :: Maybe FilePath
+  } deriving (Eq, Show, Generic)

hw-xml.cabal

@@ -37,8 +37,8 @@ common generic-lens               { build-depends: generic-lens               >=
 common ghc-prim                   { build-depends: ghc-prim                   >= 0.5        && < 0.6    }
 common hedgehog                   { build-depends: hedgehog                   >= 1.0        && < 1.1    }
 common hspec                      { build-depends: hspec                      >= 2.5        && < 3.0    }
-common hw-balancedparens          { build-depends: hw-balancedparens          >= 0.3.0.0    && < 0.4    }
-common hw-bits                    { build-depends: hw-bits                    >= 0.7.0.6    && < 0.8    }
+common hw-balancedparens          { build-depends: hw-balancedparens          >= 0.3.0.1    && < 0.4    }
+common hw-bits                    { build-depends: hw-bits                    >= 0.7.0.7    && < 0.8    }
 common hw-hspec-hedgehog          { build-depends: hw-hspec-hedgehog          >= 0.1        && < 0.2    }
 common hw-parser                  { build-depends: hw-parser                  >= 0.1.0.1    && < 0.2    }
 common hw-prim                    { build-depends: hw-prim                    >= 0.6.2.33   && < 0.7    }
@@ -139,6 +139,7 @@ executable hw-xml
   other-modules:        Paths_hw_xml
                         App.Commands
                         App.Commands.Count
+                        App.Commands.CreateIndex
                         App.Commands.Demo
                         App.Commands.Types
                         App.Options