1 files changed, 23 insertions, 19 deletions

diff --git a/tprint/src/Main.hs b/tprint/src/Main.hs
index f6008ac..aed0388 100644
--- a/tprint/src/Main.hs
+++ b/tprint/src/Main.hs
@@ -140,25 +140,9 @@ 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 = DraftEdit{..}, ..} Client{..} out = do
+  (dTitle, dContent) <- editDraft (fst output) =<< maybe (return mempty) draft mDraftId
+  unless dryRun $ maybe (\t c -> hPutStrLn out . show =<< draftCreate t c) draftReplace mDraftId dTitle dContent
 
 tprint TPrint{ operation = DraftDelete{..}, ..} Client{..} _ = unless dryRun $ draftDelete draftId
 
@@ -170,3 +154,23 @@ tprint TPrint{ operation = DraftPrint{..}, ..} client@Client{..} out = do
 
 
 tprint _ _ _ = undefined
+
+
+editDraft :: Format -> (Maybe DraftTitle, Printout) -> IO (Maybe DraftTitle, Printout)
+editDraft fmt d@(fromMaybe T.empty -> dTitle, dContent) = do
+  let
+    content
+      | JSON <- fmt = return . LBS.toStrict $ encodePretty d
+      | otherwise   = do
+          c <- either throwM return (cobbcode dContent)
+          return . T.encodeUtf8 $ dTitle <> T.pack "\n--- ^ Title ---\n" <> c
+    tmpl
+      | JSON <- fmt = jsonTemplate
+      | otherwise   = plainTemplate
+    parse
+      | JSON <- fmt = 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
+  parse =<< runUserEditorDWIM tmpl =<< content