add (obscure) option to avoid pretty printing notebooks (used with online toplevels)

Author: andrewray

files.ml

@@ -166,7 +166,7 @@ let diffable_pretty_to_string json =
   in
   Pretty.to_string (f (Yojson.Basic.pretty_format ~std:true json))
 
-let prepare_ipynb_for_saving data = 
+let prepare_ipynb_for_saving no_split_lines data = 
     let open Yojson.Basic in
     let json = from_string data in
 
@@ -176,9 +176,12 @@ let prepare_ipynb_for_saving data =
 
     (* rewrite the json with an empty notebook name *) 
     let json = replace_dict "metadata" (replace_dict "name" (`String "") metadata) json in
-    let json = process_lines split json in
+    let json = 
+      if no_split_lines then to_string ~std:true json 
+      else diffable_pretty_to_string (process_lines split json) 
+    in
 
-    filename, diffable_pretty_to_string json
+    filename, json
 
 let load_ipynb_for_serving path nbname = 
     let open Yojson.Basic in

files.mli

@@ -19,7 +19,7 @@ val empty_notebook : string -> string
 val file_or_path : string -> string * string
 
 (** extract notebook name from json, then remove it *)
-val prepare_ipynb_for_saving : string -> string * string
+val prepare_ipynb_for_saving : bool -> string -> string * string
 
 val load_ipynb_for_serving : (string -> string) -> string -> string Lwt.t
 

iocamlserver.ml

@@ -56,6 +56,7 @@ let configure_js_serve () =
       !serve_file_path
   end
 
+let no_split_lines = ref false
 
 let () = 
     Arg.(parse (align [
@@ -75,6 +76,7 @@ let () =
         "-completion", Set(Kernel.(kernel_args.completion)), " enable tab completion";
         "-object-info", Set(Kernel.(kernel_args.object_info)), " enable introspection";
         "-browser", Set_string(browser), "<exe> browser command [xdg-open]";
+        "-no-split-lines", Set(no_split_lines), " dont split lines when saving";
         "-v", Unit(fun () -> incr verbose), " increase verbosity";
     ])
     (fun s -> file_or_path := s)
@@ -260,7 +262,7 @@ let get_filename_of_ipynb s =
 let save_notebook guid body = 
     let old_filename = Kernel.M.filename_of_notebook_guid guid in
     (*lwt new_filename = get_filename_of_ipynb body in*)
-    let new_filename, body = Files.prepare_ipynb_for_saving body in
+    let new_filename, body = Files.prepare_ipynb_for_saving !no_split_lines body in
     lwt () = Lwt_io.(with_file ~mode:output 
         (filename (new_filename ^ ".ipynb"))
         (fun f -> write f body))