forked from GitHub/gf-core
71 lines
2.2 KiB
Haskell
71 lines
2.2 KiB
Haskell
module GF.Command.Importing (importGrammar, importSource) where
|
|
|
|
import PGF
|
|
import PGF.Data
|
|
|
|
import GF.Compile
|
|
import GF.Compile.Multi (readMulti)
|
|
import GF.Compile.GetGrammar (getCFRules, getEBNFRules)
|
|
import GF.Grammar (identS, SourceGrammar) -- for cc command
|
|
import GF.Grammar.CFG
|
|
import GF.Grammar.EBNF
|
|
import GF.Compile.CFGtoPGF
|
|
import GF.Infra.UseIO
|
|
import GF.Infra.Option
|
|
import GF.Data.ErrM
|
|
|
|
import System.FilePath
|
|
import qualified Data.Set as Set
|
|
|
|
-- import a grammar in an environment where it extends an existing grammar
|
|
importGrammar :: PGF -> Options -> [FilePath] -> IO PGF
|
|
importGrammar pgf0 _ [] = return pgf0
|
|
importGrammar pgf0 opts files =
|
|
case takeExtensions (last files) of
|
|
".cf" -> importCF opts files getCFRules id
|
|
".ebnf" -> importCF opts files getEBNFRules ebnf2cf
|
|
".gfm" -> do
|
|
ascss <- mapM readMulti files
|
|
let cs = concatMap snd ascss
|
|
importGrammar pgf0 opts cs
|
|
s | elem s [".gf",".gfo"] -> do
|
|
res <- appIOE $ compileToPGF opts files
|
|
case res of
|
|
Ok pgf2 -> ioUnionPGF pgf0 pgf2
|
|
Bad msg -> do putStrLn ('\n':'\n':msg)
|
|
return pgf0
|
|
".pgf" -> do
|
|
pgf2 <- mapM readPGF files >>= return . foldl1 unionPGF
|
|
ioUnionPGF pgf0 pgf2
|
|
ext -> die $ "Unknown filename extension: " ++ show ext
|
|
|
|
ioUnionPGF :: PGF -> PGF -> IO PGF
|
|
ioUnionPGF one two = case msgUnionPGF one two of
|
|
(pgf, Just msg) -> putStrLn msg >> return pgf
|
|
(pgf,_) -> return pgf
|
|
|
|
importSource :: SourceGrammar -> Options -> [FilePath] -> IO SourceGrammar
|
|
importSource src0 opts files = do
|
|
src <- appIOE $ batchCompile opts files
|
|
case src of
|
|
Ok (_,_,gr) -> return gr
|
|
Bad msg -> do
|
|
putStrLn msg
|
|
return src0
|
|
|
|
-- for different cf formats
|
|
importCF opts files get convert = do
|
|
res <- appIOE impCF
|
|
case res of
|
|
Ok pgf -> return pgf
|
|
Bad s -> error s
|
|
where
|
|
impCF = do
|
|
rules <- fmap (convert . concat) $ mapM (get opts) files
|
|
startCat <- case rules of
|
|
(CFRule cat _ _ : _) -> return cat
|
|
_ -> fail "empty CFG"
|
|
let gf = cf2gf (last files) (uniqueFuns (mkCFG startCat Set.empty rules))
|
|
gr <- compileSourceGrammar opts gf
|
|
link opts (identS (justModuleName (last files) ++ "Abs"), (), gr)
|