Write weave

2 files changed, 34 insertions, 1 deletions

diff --git a/src/Main.hs b/src/Main.hs
index bf81ae2..814e26f 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -30,7 +30,9 @@ main = do
               _ -> usage >> exitFailure
   case parsed of
     Left err -> print err >> exitFailure
-    Right ok -> print $ tangle ok
+    Right result -> case weave result of
+      Left err -> print err >> exitFailure
+      Right ok -> putStrLn ok
 
 usage :: IO ()
 usage = putStrLn "usage: lyt [file]"
diff --git a/src/Weave.hs b/src/Weave.hs
new file mode 100644
index 0000000..f10617f
--- /dev/null
+++ b/src/Weave.hs
@@ -0,0 +1,31 @@
+{- 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 Weave (weave) where
+
+import Fragment
+
+weave :: [Fragment] -> Either String String
+weave = Right . concatMap weaveFragment
+
+weaveFragment :: Fragment -> String
+weaveFragment (Documentation text) = text
+weaveFragment (BlockCode name body) =
+  "\\begin{LytBlockCode}{" ++ name ++ "}\n"
+  ++ concatMap weaveCodeOrReference body
+  ++ "\\end{LytBlockCode}\n"
+
+weaveCodeOrReference :: CodeOrReference -> String
+weaveCodeOrReference (Code text) = text
+weaveCodeOrReference (Reference name) = "\\LytFragmentReference{" ++ name ++ "}"