Fix export with relative line

3 files changed, 18 insertions, 8 deletions

diff --git a/lib/export.ml b/lib/export.ml
index 3346195..0468e64 100644
--- a/lib/export.ml
+++ b/lib/export.ml
@@ -1,4 +1,14 @@
 let to_path path ~content =
-  let oc = open_out path in
-  Printf.fprintf oc "%s" content;
-  close_out oc
+  try
+    let home = Sys.getenv_opt "HOME" in
+    let path =
+      match home with
+      | Some home -> Str.replace_first (Str.regexp "~") home path
+      | None -> path
+    in
+    let oc = open_out path in
+    Printf.fprintf oc "%s\n" content;
+    close_out oc;
+    "Successfully wrote to " ^ path
+  with
+  | exn -> "Failed to write to path: " ^ Printexc.to_string exn
diff --git a/lib/headlines.ml b/lib/headlines.ml
index 5c740ed..cefe31d 100644
--- a/lib/headlines.ml
+++ b/lib/headlines.ml
@@ -118,8 +118,8 @@ let rec render
             if String.equal (String.trim path) String.empty |> not
             then (
               let export = String.concat "\n" (Array.to_list content_pretty) in
-              Export.to_path path ~content:export;
-              render t { state with output = Some ("Export written to: " ^ path) })
+              let result = Export.to_path path ~content:export in
+              render t { state with output = Some result })
             else render t { state with output = Some "Export Blank; skipping" })
       ; on_cancel =
           (fun _ ->
diff --git a/lib/stitched_article.ml b/lib/stitched_article.ml
index deedb43..86a22e9 100644
--- a/lib/stitched_article.ml
+++ b/lib/stitched_article.ml
@@ -95,13 +95,13 @@ let rec render
                   (Array.to_list content_pretty
                    |> List.map (fun (_, b) -> Grep.display_type_line b))
               in
-              Export.to_path path ~content:export;
-              render t { state with output = Some ("Export written to: " ^ path ) })
+              let result = Export.to_path path ~content:export in
+              render t { state with output = Some result })
             else render t { state with output = Some "Export Blank; skipping" })
       ; on_cancel =
           (fun _ ->
             Common.Term.cursor t None;
-            render t {state with output = Some "Export Cancelled"})
+            render t { state with output = Some "Export Cancelled" })
       }
     in
     Input_prompt.render t input_state