diff options
| -rw-r--r-- | lyt.cabal | 8 | ||||
| -rw-r--r-- | src/Fragment.hs | 52 | ||||
| -rw-r--r-- | src/Main.hs | 16 |
3 files changed, 72 insertions, 4 deletions
@@ -28,13 +28,15 @@ tested-with: GHC==7.6.3 executable lyt hs-source-dirs: src default-language: Haskell2010 - -- other-extensions: + default-extensions: DeriveDataTypeable + , DeriveGeneric + -- other-extensions: ghc-options: -Wall -fwarn-tabs -fwarn-incomplete-uni-patterns -fwarn-monomorphism-restriction -fwarn-auto-orphans main-is: Main.hs - -- other-modules: + other-modules: Fragment build-depends: base >=4.6 && <4.7 - , vector >=0.10 && <0.11 + , parsec >=3.1.3 && <3.2 diff --git a/src/Fragment.hs b/src/Fragment.hs new file mode 100644 index 0000000..2d45ff4 --- /dev/null +++ b/src/Fragment.hs @@ -0,0 +1,52 @@ +{- Copyright © 2015 Benjamin Barenblat + +This program is free software: you can redistribute it and/or modify it under +the terms of the GNU General Public License as published by the Free Software +Foundation, either version 3 of the License, or (at your option) any later +version. + +This program is distributed in the hope that it will be useful, but WITHOUT ANY +WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A +PARTICULAR PURPOSE. See the GNU General Public License for more details. + +You should have received a copy of the GNU General Public License along with +this program. If not, see <http://www.gnu.org/licenses/>. -} + +module Fragment ( Fragment + , parseStdin + , parseFile + , ParseError) where + +import Data.Data (Data) +import Data.Typeable (Typeable) +import GHC.Generics (Generic) + +import Control.Applicative ((<$>), (<*>), pure) +import Control.Monad (void) +import System.IO (hGetContents, stdin) +import Text.Parsec +import Text.Parsec.String + +data Fragment = Documentation String + | BlockCode String String + deriving (Eq, Show, Data, Typeable, Generic) + +parseStdin :: IO (Either ParseError [Fragment]) +parseStdin = parse literateFile "<stdin>" <$> hGetContents stdin + +parseFile :: FilePath -> IO (Either ParseError [Fragment]) +parseFile = parseFromFile literateFile + +literateFile :: Parser [Fragment] +literateFile = alternate documentation blockCode + +documentation :: Parser Fragment +documentation = Documentation <$> manyTill anyChar (void (string "<<") <|> eof) + +blockCode :: Parser Fragment +blockCode = + BlockCode <$> manyTill anyChar (void $ string ">>=") + <*> manyTill anyChar (void $ char '@') + +alternate :: Parser a -> Parser a -> Parser [a] +alternate x y = (:) <$> x <*> (alternate y x <|> pure []) diff --git a/src/Main.hs b/src/Main.hs index f569b1a..d69710c 100644 --- a/src/Main.hs +++ b/src/Main.hs @@ -14,5 +14,19 @@ this program. If not, see <http://www.gnu.org/licenses/>. -} module Main where +import System.Environment (getArgs) +import System.Exit (exitFailure) + +import Fragment (parseFile, parseStdin) + main :: IO () -main = putStrLn "Hello, world!" +main = do + args <- getArgs + parsed <- case args of + [] -> parseStdin + [f] -> parseFile f + _ -> usage >> exitFailure + print parsed + +usage :: IO () +usage = putStrLn "usage: lyt [file]" |