allow whitespace around infix op

1 files changed, 4 insertions, 2 deletions

diff --git a/plugins/extraction/ocaml.ml b/plugins/extraction/ocaml.ml
index eea5968ce..fb8dfb459 100644
--- a/plugins/extraction/ocaml.ml
+++ b/plugins/extraction/ocaml.ml
@@ -122,11 +122,13 @@ let is_infix r =
    len >= 3 &&
      (* parenthesized *)
      (s.[0] == '(' && s.[len-1] == ')' &&
+        let inparens = String.trim (String.sub s 1 (len - 2)) in
+        let inparens_len = String.length inparens in
         (* either, begins with infix symbol, any remainder is all operator chars *)
-        (List.mem s.[1] infix_symbols && substring_all_opchars s 2 (len-1))
+        (List.mem inparens.[0] infix_symbols && substring_all_opchars inparens 1 inparens_len)
       ||
         (* or, starts with #, at least one more char, all are operator chars *)
-        (s.[1] == '#' && len >= 4 && substring_all_opchars s 2 (len-1))))
+        (inparens.[0] == '#' && inparens_len >= 2 && substring_all_opchars inparens 1 inparens_len)))
 
 let get_infix r =
   let s = find_custom r in