aboutsummaryrefslogtreecommitdiff
path: root/src/Lib/CSV.ml
blob: f0366f731e5de684813c1dd4b1372060fdd23813 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
let to_string lines =
  let
    cell_to_string cell =
      if Js.String.includes "\"" cell then
        "\"" ^ (Js.String.replaceByRe [%re "/\"/g"] "\"\"" cell) ^ "\""
      else
        cell
  in let
    line_to_string line =
      line
        |> Js.Array.map cell_to_string
        |> Js.Array.joinWith ","
  in lines
    |> Js.Array.map line_to_string
    |> Js.Array.joinWith "\n"

let parse str =
  let lines = [| |] in
  let current_line = ref [| |] in
  let current_cell = ref "" in
  let in_quote = ref false in
  let i = ref 0 in
  let l = Js.String.length str in
  let () = while !i < l do
    let cc = Js.String.get str !i in
    let nc = Js.String.get str (!i + 1) in
    let () =
      if !in_quote && cc == "\"" && nc == "\"" then
        let () = current_cell := !current_cell ^ cc in
        i := !i + 1
      else if cc == "\"" then
        in_quote := not !in_quote
      else if not !in_quote && cc == "," then
        let _ = Js.Array.push !current_cell !current_line in
        current_cell := ""
      else if not !in_quote && ((cc == "\r" && nc == "\n") || cc == "\n" || cc == "\r") then
        let _ = Js.Array.push !current_cell !current_line in
        let _ = Js.Array.push !current_line lines in
        let _ = current_line := [| |] in
        current_cell := ""
      else
        current_cell := !current_cell ^ cc
    in
    i := !i + 1
  done
  in
  let _ =
    if Js.String.length !current_cell > 0 then
      let _ = Js.Array.push !current_cell !current_line in ()
    else
      ()
  in
  let _ =
    if Js.Array.length !current_line > 0 then
      let _ = Js.Array.push !current_line lines in ()
    else
      ()
  in
  lines

let to_dicts lines =
  let res = [| |] in
  let () =
    if Js.Array.length lines > 0 then
      let header = Js.Array.unsafe_get lines 0 in
      for i = 1 to Js.Array.length lines - 1 do
        let line = Js.Array.unsafe_get lines i in
        let dict = Js.Dict.empty() in
        let () =
          Js.Array.forEachi
            (fun key j -> Js.Dict.set dict key (Js.Array.unsafe_get line j))
            header
        in
        ignore (Js.Array.push dict res)
      done
  in res