package luv

  1. Overview
  2. Docs

Source file env.ml

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
(* This file is part of Luv, released under the MIT license. See LICENSE.md for
   details, or visit https://github.com/aantron/luv/blob/master/LICENSE.md. *)



let getenv variable =
  let length = 1024 in
  let buffer = Bytes.create length in
  C.Functions.Env.getenv
    (Ctypes.ocaml_string_start variable)
    (Ctypes.ocaml_bytes_start buffer)
    (Ctypes.(allocate size_t) (Unsigned.Size_t.of_int length))
  |> Error.to_result_f @@ fun () ->
  let length = Bytes.index buffer '\000' in
  Bytes.sub_string buffer 0 length

let setenv variable ~value =
  C.Functions.Env.setenv
    (Ctypes.ocaml_string_start variable) (Ctypes.ocaml_string_start value)
  |> Error.to_result ()

let unsetenv variable =
  C.Functions.Env.unsetenv (Ctypes.ocaml_string_start variable)
  |> Error.to_result ()

let environ () =
  let env_items = Ctypes.(allocate_n (ptr C.Types.Env_item.t) ~count:1) in
  let count = Ctypes.(allocate_n int ~count:1) in
  C.Functions.Env.environ env_items count
  |> Error.to_result_f @@ fun () ->
  let env_items = Ctypes.(!@) env_items in
  let count = Ctypes.(!@) count in
  let converted_env_items =
    Ctypes.(CArray.fold_left (fun env_items c_env_item ->
      let name = getf c_env_item C.Types.Env_item.name in
      let value = getf c_env_item C.Types.Env_item.value in
      (name, value)::env_items) [] (CArray.from_ptr env_items count))
    |> List.rev
  in
  C.Functions.Env.free_environ env_items count;
  converted_env_items