aboutsummaryrefslogtreecommitdiff
path: root/stdlib/source/library/lux/control/function/contract.lux
blob: 1490532307d0289f963896505431d469803d5fab (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
(.module:
  [library
   [lux #*
    [control
     ["." exception (#+ exception:)]]
    [data
     [text
      ["%" format (#+ format)]]]
    [macro (#+ with_gensyms)
     [syntax (#+ syntax:)]
     ["." code]]
    [math
     [number
      ["i" int]]]]])

(template [<name>]
  [(exception: (<name> {condition Code})
     (exception.report
      ["Condition" (%.code condition)]))]

  [pre_condition_failed]
  [post_condition_failed]
  )

(def: (assert! message test)
  (-> Text Bit [])
  (if test
    []
    (error! message)))

(syntax: #export (pre test expr)
  {#.doc (doc "Pre-conditions."
              "Given a test and an expression to run, only runs the expression if the test passes."
              "Otherwise, an error is raised."
              (pre (i.= +4 (i.+ +2 +2))
                   (foo +123 +456 +789)))}
  (wrap (list (` (exec ((~! ..assert!) (~ (code.text (exception.construct ..pre_condition_failed test)))
                        (~ test))
                   (~ expr))))))

(syntax: #export (post test expr)
  {#.doc (doc "Post-conditions."
              "Given a predicate and an expression to run, evaluates the expression and then tests the output with the predicate."
              "If the predicate returns #1, returns the value of the expression."
              "Otherwise, an error is raised."
              (post i.even?
                    (i.+ +2 +2)))}
  (with_gensyms [g!output]
    (wrap (list (` (let [(~ g!output) (~ expr)]
                     (exec ((~! ..assert!) (~ (code.text (exception.construct ..post_condition_failed test)))
                            ((~ test) (~ g!output)))
                       (~ g!output))))))))