summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--lyt.cabal8
-rw-r--r--src/Fragment.hs52
-rw-r--r--src/Main.hs16
3 files changed, 72 insertions, 4 deletions
diff --git a/lyt.cabal b/lyt.cabal
index c3905a8..01c150f 100644
--- a/lyt.cabal
+++ b/lyt.cabal
@@ -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]"