summary refs log tree commit diff stats
path: root/tests/compilerapi/tcompilerapi.nim
blob: 2d6cfa0ca3b58100556b7e42c3f5cc1e45a32b18 (plain) (blame)
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
87
88
89
90
91
92
93
94
95
96
97
98
discard """
  output: '''top level statements are executed!
(ival: 10, fval: 2.0)
2.0
my secret
11
12
raising VMQuit
'''
  joinable: "false"
"""

## Example program that demonstrates how to use the
## compiler as an API to embed into your own projects.

import "../../compiler" / [ast, vmdef, vm, nimeval, llstream, lineinfos, options]
import std / [os]

proc initInterpreter(script: string): Interpreter =
  let std = findNimStdLibCompileTime()
  result = createInterpreter(script, [std, parentDir(currentSourcePath),
    std / "pure", std / "core"])

proc main() =
  let i = initInterpreter("myscript.nim")
  i.implementRoutine("nim", "exposed", "addFloats", proc (a: VmArgs) =
    setResult(a, getFloat(a, 0) + getFloat(a, 1) + getFloat(a, 2))
  )
  i.evalScript()
  let foreignProc = i.selectRoutine("hostProgramRunsThis")
  if foreignProc == nil:
    quit "script does not export a proc of the name: 'hostProgramRunsThis'"
  let res = i.callRoutine(foreignProc, [newFloatNode(nkFloatLit, 0.9),
                                        newFloatNode(nkFloatLit, 0.1)])
  doAssert res.kind == nkFloatLit
  echo res.floatVal

  let foreignValue = i.selectUniqueSymbol("hostProgramWantsThis")
  if foreignValue == nil:
    quit "script does not export a global of the name: hostProgramWantsThis"
  let val = i.getGlobalValue(foreignValue)
  doAssert val.kind in {nkStrLit..nkTripleStrLit}
  echo val.strVal
  i.destroyInterpreter()

main()

block issue9180:
  proc evalString(code: string, moduleName = "script.nim") =
    let stream = llStreamOpen(code)
    let std = findNimStdLibCompileTime()
    var intr = createInterpreter(moduleName, [std, std / "pure", std / "core"])
    intr.evalScript(stream)
    destroyInterpreter(intr)
    llStreamClose(stream)

  evalString("echo 10+1")
  evalString("echo 10+2")

block error_hook:
  type VMQuit = object of CatchableError

  let i = initInterpreter("invalid.nim")
  i.registerErrorHook proc(config: ConfigRef; info: TLineInfo; msg: string;
                           severity: Severity) {.gcsafe.} =
    if severity == Error and config.errorCounter >= config.errorMax:
      echo "raising VMQuit"
      raise newException(VMQuit, "Script error")

  doAssertRaises(VMQuit):
    i.evalScript()

block resetmacrocache:
  let std = findNimStdLibCompileTime()
  let intr = createInterpreter("script.nim", [std, std / "pure", std / "core"])
  proc evalString(intr: Interpreter; code: string) =
    let stream = llStreamOpen(code)
    intr.evalScript(stream)
    llStreamClose(stream)
  let code = """
import std/[macrocache, macros]
static:
  let counter = CacheCounter"valTest"
  inc counter
  assert counter.value == 1

  const mySeq = CacheSeq"addTest"
  mySeq.add(newLit(5))
  mySeq.add(newLit("hello ic"))
  assert mySeq.len == 2

  const mcTable = CacheTable"subTest"
  mcTable["toAdd"] = newStmtList() #would crash if not empty
  assert mcTable.len == 1
"""
  intr.evalString(code)
  intr.evalString(code)
  destroyInterpreter(intr)