mirror of
https://github.com/zap-proto/haskell.git
synced 2026-10-30 15:05:00 +00:00
Ian Denhardt
f238ad7a67
This makes things a little more robust, but we have a fundamental problem: if a node is defined in a file not in requestedFiles, it may be referenced but not actually in the cgr in the first place. This means needing to rethink how we generate instances of interfaces classes.
62 lines
1.8 KiB
Haskell
62 lines
1.8 KiB
Haskell
-- This module is the main entry point for the capnpc-haskell code
|
|
-- generator plugin.
|
|
module Main (main) where
|
|
|
|
import Data.Foldable (for_)
|
|
import System.Directory (createDirectoryIfMissing)
|
|
import System.FilePath (takeDirectory)
|
|
import System.IO (IOMode(WriteMode), withFile)
|
|
|
|
import qualified Data.Text.Lazy as LT
|
|
import qualified Data.Text.Lazy.IO as TIO
|
|
|
|
import Capnp (defaultLimit, getValue)
|
|
import Capnp.Gen.Capnp.Schema.Pure (CodeGeneratorRequest)
|
|
|
|
import qualified Check
|
|
import qualified IR.Flat as Flat
|
|
import qualified IR.Haskell as Haskell
|
|
import qualified Trans.CgrToStage1
|
|
import qualified Trans.FlatToPure
|
|
import qualified Trans.FlatToRaw
|
|
import qualified Trans.HaskellToText
|
|
import qualified Trans.PureToHaskell
|
|
import qualified Trans.RawToHaskell
|
|
import qualified Trans.Stage1ToFlat
|
|
|
|
main :: IO ()
|
|
main = do
|
|
cgr <- getValue defaultLimit
|
|
Check.reportIssues cgr
|
|
for_ (handleCGR cgr) $ \(path, contents) -> do
|
|
createDirectoryIfMissing True (takeDirectory path)
|
|
withFile path WriteMode $ \h ->
|
|
TIO.hPutStr h contents
|
|
|
|
-- | Convert a 'CodeGeneratorRequest' to a list of files to create.
|
|
handleCGR :: CodeGeneratorRequest -> [(FilePath, LT.Text)]
|
|
handleCGR cgr =
|
|
let flat =
|
|
Trans.Stage1ToFlat.cgrToCgr $
|
|
Trans.CgrToStage1.cgrToCgr cgr
|
|
modules =
|
|
handleFlatRaw flat ++ handleFlatPure flat
|
|
in
|
|
map
|
|
(\mod ->
|
|
( Haskell.modFilePath mod
|
|
, Trans.HaskellToText.moduleToText mod
|
|
)
|
|
)
|
|
modules
|
|
|
|
handleFlatPure, handleFlatRaw :: Flat.CodeGenReq -> [Haskell.Module]
|
|
|
|
handleFlatPure =
|
|
concatMap Trans.PureToHaskell.fileToModules
|
|
. Trans.FlatToPure.cgrToFiles
|
|
|
|
handleFlatRaw =
|
|
concatMap Trans.RawToHaskell.fileToModules
|
|
. Trans.FlatToRaw.cgrToFiles
|