{-# LANGUAGE CPP #-}
module BishBosh.ContextualNotation.PGNDatabase(
PGNDatabase,
Decompressor,
MaybeMaximumGames,
parseIO
) where
import Control.DeepSeq(($!!))
import qualified BishBosh.ContextualNotation.PGN as ContextualNotation.PGN
import qualified BishBosh.ContextualNotation.StandardAlgebraic as ContextualNotation.StandardAlgebraic
import qualified BishBosh.Data.Exception as Data.Exception
import qualified BishBosh.Type.Count as Type.Count
import qualified Control.Exception
import qualified Control.Monad
import qualified Data.Maybe
import qualified System.Exit
import qualified System.FilePath
import qualified System.IO
import qualified System.Process
#ifdef USE_POLYPARSE
import qualified BishBosh.Text.Poly as Text.Poly
# if USE_POLYPARSE == 'L'
import qualified Text.ParserCombinators.Poly.Lazy as Poly
# elif USE_POLYPARSE == 'P'
import qualified Control.Arrow
import qualified Text.ParserCombinators.Poly.Plain as Poly
# else
# error "USE_POLYPARSE invalid"
# endif
#else /* Parsec */
import qualified Control.Arrow
import qualified Text.ParserCombinators.Parsec as Parsec
import Text.ParserCombinators.Parsec((<?>))
#endif
type PGNDatabase = [ContextualNotation.PGN.PGN]
parser
:: ContextualNotation.PGN.IsStrictlySequential
-> ContextualNotation.StandardAlgebraic.ValidateMoves
-> [ContextualNotation.PGN.Tag]
#ifdef USE_POLYPARSE
-> Text.Poly.TextParser PGNDatabase
parser :: IsStrictlySequential
-> IsStrictlySequential -> [Tag] -> TextParser PGNDatabase
parser IsStrictlySequential
isStrictlySequential IsStrictlySequential
validateMoves [Tag]
identificationTags = Parser Char PGN -> Parser Char () -> TextParser PGNDatabase
forall (p :: * -> *) a z.
(PolyParse p, Show a) =>
p a -> p z -> p [a]
Poly.manyFinally' Parser Char PGN
parser' (Parser Char () -> TextParser PGNDatabase)
-> Parser Char () -> TextParser PGNDatabase
forall a b. (a -> b) -> a -> b
$ Parser Char ()
Text.Poly.spaces Parser Char () -> Parser Char () -> Parser Char ()
forall (m :: * -> *) a b. Monad m => m a -> m b -> m b
>> Parser Char ()
forall t. Parser t ()
Poly.eof
#else /* Parsec */
-> Parsec.Parser PGNDatabase
parser isStrictlySequential validateMoves identificationTags = Parsec.manyTill parser' (Parsec.try $ Parsec.spaces >> Parsec.try Parsec.eof) <?> "PGN-database"
#endif
where
parser' :: Parser Char PGN
parser' = IsStrictlySequential
-> IsStrictlySequential -> [Tag] -> Parser Char PGN
ContextualNotation.PGN.parser IsStrictlySequential
isStrictlySequential IsStrictlySequential
validateMoves [Tag]
identificationTags
type PGNPredicate = ContextualNotation.PGN.PGN -> Bool
type MaybeMaximumGames = Maybe Type.Count.NGames
parse
:: String
-> ContextualNotation.PGN.IsStrictlySequential
-> ContextualNotation.StandardAlgebraic.ValidateMoves
-> [ContextualNotation.PGN.Tag]
-> PGNPredicate
-> MaybeMaximumGames
-> String
-> Either String PGNDatabase
#ifdef USE_POLYPARSE
# if USE_POLYPARSE == 'L'
parse :: Tag
-> IsStrictlySequential
-> IsStrictlySequential
-> [Tag]
-> PGNPredicate
-> MaybeMaximumGames
-> Tag
-> Either Tag PGNDatabase
parse Tag
_ IsStrictlySequential
isStrictlySequential IsStrictlySequential
validateMoves [Tag]
identificationTags PGNPredicate
pgnPredicate MaybeMaximumGames
maybeMaximumGames = PGNDatabase -> Either Tag PGNDatabase
forall a b. b -> Either a b
Right
# elif USE_POLYPARSE == 'P'
parse name isStrictlySequential validateMoves identificationTags pgnPredicate maybeMaximumGames = Control.Arrow.left (showString "regarding " . shows name . showString ", ")
# else
# error "USE_POLYPARSE invalid"
# endif
(PGNDatabase -> Either Tag PGNDatabase)
-> (Tag -> PGNDatabase) -> Tag -> Either Tag PGNDatabase
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (PGNDatabase, Tag) -> PGNDatabase
forall a b. (a, b) -> a
fst ((PGNDatabase, Tag) -> PGNDatabase)
-> (Tag -> (PGNDatabase, Tag)) -> Tag -> PGNDatabase
forall b c a. (b -> c) -> (a -> b) -> a -> c
. TextParser PGNDatabase -> Tag -> (PGNDatabase, Tag)
forall t a. Parser t a -> [t] -> (a, [t])
Poly.runParser TextParser PGNDatabase
parser'
#else /* Parsec */
parse name isStrictlySequential validateMoves identificationTags pgnPredicate maybeMaximumGames = Control.Arrow.left (showString "failed to parse; " . show) . Parsec.parse parser' name
#endif
where
parser' :: TextParser PGNDatabase
parser' = (
(PGNDatabase -> PGNDatabase)
-> (NGames -> PGNDatabase -> PGNDatabase)
-> MaybeMaximumGames
-> PGNDatabase
-> PGNDatabase
forall b a. b -> (a -> b) -> Maybe a -> b
Data.Maybe.maybe PGNDatabase -> PGNDatabase
forall a. a -> a
id (NGames -> PGNDatabase -> PGNDatabase
forall a. NGames -> [a] -> [a]
take (NGames -> PGNDatabase -> PGNDatabase)
-> (NGames -> NGames) -> NGames -> PGNDatabase -> PGNDatabase
forall b c a. (b -> c) -> (a -> b) -> a -> c
. NGames -> NGames
forall a b. (Integral a, Num b) => a -> b
fromIntegral) MaybeMaximumGames
maybeMaximumGames (PGNDatabase -> PGNDatabase)
-> (PGNDatabase -> PGNDatabase) -> PGNDatabase -> PGNDatabase
forall b c a. (b -> c) -> (a -> b) -> a -> c
. PGNPredicate -> PGNDatabase -> PGNDatabase
forall a. (a -> IsStrictlySequential) -> [a] -> [a]
filter PGNPredicate
pgnPredicate
) (PGNDatabase -> PGNDatabase)
-> TextParser PGNDatabase -> TextParser PGNDatabase
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
`fmap` IsStrictlySequential
-> IsStrictlySequential -> [Tag] -> TextParser PGNDatabase
parser IsStrictlySequential
isStrictlySequential IsStrictlySequential
validateMoves [Tag]
identificationTags
type Decompressor = String
parseIO
:: System.FilePath.FilePath
-> Maybe Decompressor
-> ContextualNotation.PGN.IsStrictlySequential
-> ContextualNotation.StandardAlgebraic.ValidateMoves
-> System.IO.TextEncoding
-> [ContextualNotation.PGN.Tag]
-> PGNPredicate
-> MaybeMaximumGames
-> IO (Either String PGNDatabase)
parseIO :: Tag
-> Maybe Tag
-> IsStrictlySequential
-> IsStrictlySequential
-> TextEncoding
-> [Tag]
-> PGNPredicate
-> MaybeMaximumGames
-> IO (Either Tag PGNDatabase)
parseIO Tag
filePath Maybe Tag
maybeDecompressionCommand IsStrictlySequential
isStrictlySequential IsStrictlySequential
validateMoves TextEncoding
textEncoding [Tag]
identificationTags PGNPredicate
pgnPredicate MaybeMaximumGames
maybeMaximumGames = Tag
-> IsStrictlySequential
-> IsStrictlySequential
-> [Tag]
-> PGNPredicate
-> MaybeMaximumGames
-> Tag
-> Either Tag PGNDatabase
parse Tag
filePath IsStrictlySequential
isStrictlySequential IsStrictlySequential
validateMoves [Tag]
identificationTags PGNPredicate
pgnPredicate MaybeMaximumGames
maybeMaximumGames (Tag -> Either Tag PGNDatabase)
-> IO Tag -> IO (Either Tag PGNDatabase)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
`fmap` IO Tag -> (Tag -> IO Tag) -> Maybe Tag -> IO Tag
forall b a. b -> (a -> b) -> Maybe a -> b
Data.Maybe.maybe (
Tag -> IOMode -> (Handle -> IO Tag) -> IO Tag
forall r. Tag -> IOMode -> (Handle -> IO r) -> IO r
System.IO.withFile Tag
filePath IOMode
System.IO.ReadMode ((Handle -> IO Tag) -> IO Tag) -> (Handle -> IO Tag) -> IO Tag
forall a b. (a -> b) -> a -> b
$ \Handle
fileHandle -> do
Handle -> TextEncoding -> IO ()
System.IO.hSetEncoding Handle
fileHandle TextEncoding
textEncoding
Tag
contents <- Handle -> IO Tag
System.IO.hGetContents Handle
fileHandle
Tag -> IO Tag
forall (m :: * -> *) a. Monad m => a -> m a
return (Tag -> IO Tag) -> Tag -> IO Tag
forall a b. NFData a => (a -> b) -> a -> b
$!! Tag
contents
) (
\Tag
decompressor -> do
(ExitCode
exitCode, Tag
stdOut, Tag
stdErr) <- Tag -> [Tag] -> Tag -> IO (ExitCode, Tag, Tag)
System.Process.readProcessWithExitCode Tag
decompressor [Tag
filePath] []
IsStrictlySequential -> IO () -> IO ()
forall (f :: * -> *).
Applicative f =>
IsStrictlySequential -> f () -> f ()
Control.Monad.unless (ExitCode
exitCode ExitCode -> ExitCode -> IsStrictlySequential
forall a. Eq a => a -> a -> IsStrictlySequential
== ExitCode
System.Exit.ExitSuccess) (IO () -> IO ()) -> (Tag -> IO ()) -> Tag -> IO ()
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Exception -> IO ()
forall e a. Exception e => e -> IO a
Control.Exception.throwIO (Exception -> IO ()) -> (Tag -> Exception) -> Tag -> IO ()
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Tag -> Exception
Data.Exception.mkRequestFailure (Tag -> IO ()) -> Tag -> IO ()
forall a b. (a -> b) -> a -> b
$ Tag -> ShowS
showString Tag
"BishBosh.ContextualNotation.PGNDatabase.decompress:\t" Tag
stdErr
Tag -> IO Tag
forall (m :: * -> *) a. Monad m => a -> m a
return Tag
stdOut
) Maybe Tag
maybeDecompressionCommand