1 files changed, 26 insertions, 1 deletions

diff --git a/tprint/src/Main.hs b/tprint/src/Main.hs
index bc5ee2f..f6008ac 100644
--- a/tprint/src/Main.hs
+++ b/tprint/src/Main.hs
@@ -1,5 +1,6 @@
 {-# LANGUAGE RecordWildCards #-}
 {-# LANGUAGE ViewPatterns #-}
+{-# LANGUAGE TupleSections #-}
 
 import Data.Map (Map)
 import qualified Data.Map as Map
@@ -8,7 +9,9 @@ import qualified Data.Sequence as Seq
 import Data.Text (Text)
 import qualified Data.Text as T
 import qualified Data.Text.IO as T
+import qualified Data.Text.Encoding as T
 import qualified Data.ByteString.Lazy.Char8 as Lazy (ByteString)
+import qualified Data.ByteString.Lazy as LBS
 import qualified Data.ByteString.Lazy.Char8 as LCBS
 import Data.Time
 
@@ -26,7 +29,9 @@ import Control.Concurrent (threadDelay)
 
 import Text.Show.Pretty (dumpStr)
 import Data.Aeson.Encode.Pretty (encodePretty)
-import Data.Aeson (eitherDecode')
+import Data.Aeson (eitherDecode', eitherDecodeStrict')
+
+import Text.Editor (runUserEditorDWIM, jsonTemplate, plainTemplate)
 
 import System.IO
 
@@ -135,6 +140,26 @@ tprint TPrint{ operation = DraftCreate{..}, ..} Client{..} out = withPrintout in
 tprint TPrint{ operation = DraftReplace{..}, ..} Client{..} _ = withPrintout input $ \p -> do
   unless dryRun $ draftReplace draftId draftTitle p
 
+tprint TPrint{ operation = DraftEdit{..}, ..} Client{..} _ = do
+  d@(fromMaybe T.empty -> dTitle, dContent) <- draft draftId
+  let
+    content
+      | (JSON, _) <- output = return . LBS.toStrict $ encodePretty d
+      | otherwise           = do
+          c <- either throwM return (cobbcode dContent)
+          return . T.encodeUtf8 $ dTitle <> T.pack "\n" <> c
+    tmpl
+      | (JSON, _) <- output = jsonTemplate
+      | otherwise           = plainTemplate
+    parse
+      | (JSON, _) <- output = either (throwM . userError) return . eitherDecodeStrict'
+      | otherwise           = (\(t:(T.unlines -> c)) -> (massage t, ) <$> either throwM return (bbcode c)) . T.lines . T.decodeUtf8
+    massage (T.strip -> t)
+      | T.null t = Nothing
+      | otherwise = Just t
+  (dTitle', dContent') <- parse =<< runUserEditorDWIM tmpl =<< content
+  unless dryRun $ draftReplace draftId dTitle' dContent'
+
 tprint TPrint{ operation = DraftDelete{..}, ..} Client{..} _ = unless dryRun $ draftDelete draftId
 
 tprint TPrint{ operation = DraftPrint{..}, ..} client@Client{..} out = do