src/lua_stdlib.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
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
77
78
79
80
81
82
83
84
85
86

open Ast
open Ast.Value
open Env
open Syntax
module LibMap = Map.Make (String)

let () = Random.self_init ()

let lib () =
  let add_func lib_name func_name func lib =
    match LibMap.find_opt lib_name lib with
    | Some l ->
      let l = LibMap.add func_name func l in
      LibMap.add lib_name l lib
    | None -> assert false
  in
  let lib_basic = LibMap.empty in
  let lib_basic = LibMap.add "assert" Lua_stdlib_basic.asert lib_basic in
  let lib_basic = LibMap.add "ipairs" Lua_stdlib_basic.ipairs lib_basic in
  let lib_basic = LibMap.add "next" Lua_stdlib_basic.next lib_basic in
  let lib_basic = LibMap.add "pairs" Lua_stdlib_basic.pairs lib_basic in
  let lib_basic = LibMap.add "print" Lua_stdlib_basic.print lib_basic in
  let lib_basic = LibMap.add "type" Lua_stdlib_basic.typ lib_basic in
  let lib_basic = LibMap.add "tostring" Lua_stdlib_basic.tostring lib_basic in
  let lib_basic =
    LibMap.add "getmetatable" Lua_stdlib_basic.getmetatable lib_basic
  in
  let lib_basic =
    LibMap.add "setmetatable" Lua_stdlib_basic.setmetatable lib_basic
  in
  let lib_basic = LibMap.add "require" Lua_stdlib_basic.require lib_basic in
  let lib = LibMap.empty in
  let lib = LibMap.add "math" LibMap.empty lib in
  let lib = add_func "math" "abs" Lua_stdlib_math.abs lib in
  let lib = add_func "math" "floor" Lua_stdlib_math.floor lib in
  let lib = add_func "math" "cos" Lua_stdlib_math.cos lib in
  let lib = add_func "math" "sin" Lua_stdlib_math.sin lib in
  let lib = add_func "math" "tan" Lua_stdlib_math.tan lib in
  let lib = add_func "math" "random" Lua_stdlib_math.random lib in
  let lib = LibMap.add "io" LibMap.empty lib in
  let lib = add_func "io" "flush" Lua_stdlib_io.flush lib in
  let lib = add_func "io" "write" Lua_stdlib_io.write lib in
  let lib = LibMap.add "os" LibMap.empty lib in
  let lib = add_func "os" "execute" Lua_stdlib_os.execute lib in
  let lib = LibMap.add "package" LibMap.empty lib in
  let lib = add_func "package" "loaded" Lua_stdlib_package.loaded lib in
  (lib_basic, lib)

let load env =
  let lib_basic, lib = lib () in
  let add_basic func_name fct env =
    let vfct = VfunctionStdLib (Random.bits32 (), fct) in
    add_global_force func_name vfct env
  in
  let add_empty_lib name env =
    let vtbl = Vtable (LuaTable.empty ()) in
    add_global_force name vtbl env
  in
  let add_function_lib lib_name fct_name fct env =
    let* v = get_value lib_name env in
    match v with
    | Vtable tbl ->
      let tbl =
        LuaTable.add (Vstring fct_name)
          (VfunctionStdLib (Random.bits32 (), fct))
          tbl
      in
      add_value lib_name (Vtable tbl) env
    | _ -> assert false
  in
  let env =
    LibMap.fold (fun func_name f e -> add_basic func_name f e) lib_basic env
  in
  let env =
    LibMap.fold
      (fun lib_name l e ->
        let e = add_empty_lib lib_name e in
        LibMap.fold
          (fun func_name f e ->
            match add_function_lib lib_name func_name f e with
            | Ok v -> v
            | Error _ -> assert false )
          l e )
      lib env
  in
  Ok env