simon/ocaml-containers
1
0
Fork 0
mirror of https://github.com/c-cube/ocaml-containers.git synced 2026-01-24 01:56:41 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

CCFormat: fix support of unrecognized styles

Browse source
This commit is contained in:
Simon Cruanes 2018-02-05 08:58:32 -06:00
parent 3ab9cd58e1
commit 24592bf926
1 changed files with 16 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
src/core/CCFormat.ml
View file

@ -238,6 +238,8 @@ let ansi_l_to_str_ = function
Buffer.add_string buf "m";
Buffer.contents buf
exception No_such_style
(* parse a tag *)
let style_of_tag_ s = match String.trim s with
| "reset" -> [`Reset]
@ -258,7 +260,7 @@ let style_of_tag_ s = match String.trim s with
| "Magenta" -> [`FG `Magenta; `Bold]
| "Cyan" -> [`FG `Cyan; `Bold]
| "White" -> [`FG `White; `Bold]
| s -> failwith ("unknown style: " ^ s)
| _ -> raise No_such_style
let color_enabled = ref false
@ -268,20 +270,21 @@ let mark_open_tag st ~or_else s =
let style = style_of_tag_ s in
Stack.push style st;
if !color_enabled then ansi_l_to_str_ style else ""
with Not_found -> or_else s
with No_such_style -> or_else s
let mark_close_tag st ~or_else s =
try
let _ = style_of_tag_ s in (* check if it's indeed about color *)
let style =
try
ignore (Stack.pop st); (* pop current style (if well-scoped...) *)
Stack.top st (* look at previous style *)
with Stack.Empty ->
[`Reset]
in
if !color_enabled then ansi_l_to_str_ style else ""
with Not_found -> or_else s
(* check if it's indeed about color *)
match style_of_tag_ s with
| _ ->
let style =
try
ignore (Stack.pop st); (* pop current style (if well-scoped...) *)
Stack.top st (* look at previous style *)
with Stack.Empty ->
[`Reset]
in
if !color_enabled then ansi_l_to_str_ style else ""
| exception No_such_style -> or_else s
(* add color handling to formatter [ppf] *)
let set_color_tag_handling ppf =