aboutsummaryrefslogtreecommitdiff
path: root/rego-proofs/compiler/passes/compile-reference.scm
blob: 0d282069665d0dc0322f9cf3f18164d19eaca3d1 (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
;;; Copyright © 2023 Jakob L. Kreuze <zerodaysfordays@sdf.org>
;;;
;;; This program is free software; you can redistribute it and/or
;;; modify it under the terms of the GNU General Public License as
;;; published by the Free Software Foundation; either version 3 of the
;;; License, or (at your option) any later version.
;;;
;;; This program is distributed in the hope that it will be useful,
;;; but WITHOUT ANY WARRANTY; without even the implied warranty of
;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
;;; General Public License for more details.
;;;
;;; You should have received a copy of the GNU General Public License
;;; along with this program. If not, see
;;; <http://www.gnu.org/licenses/>.

(define-module (compiler passes compile-reference)
  #:use-module ((rnrs base) #:select (assert))
  #:use-module (compiler lean)
  #:use-module (compiler type-inference)
  #:use-module (compiler types)
  #:use-module (compiler utilities)
  #:use-module (ice-9 match)
  #:use-module (oop goops)
  #:export (compile-reference))

(define (compile-reference policy)

  (define (specialize breadcrumb scope)
    (define (recur input-identifier form new-identifiers)
      (match form
        ((target rest ...)
         (with-anonymous-identifiers (new-identifiers target-cast-id deref-id)
           (cond ((and (infer-is-a? input-identifier 'array breadcrumb scope)
                       (infer-is-a? input-identifier 'number breadcrumb scope))
                  `((,target-cast-id ,(make <lean-funcall>
                                        #:name "Rego.expectArray"
                                        #:args `(,input-identifier)))
                    (,deref-id       ,(make <lean-funcall>
                                        #:name "Array.get?"
                                        #:args `(,target-cast-id ,target)))
                    ,@(recur deref-id rest new-identifiers)))
                 ((and (infer-is-a? input-identifier 'set breadcrumb scope)
                       (infer-is-a? input-identifier 'number breadcrumb scope))
                  `((,target-cast-id ,(make <lean-funcall>
                                        #:name "Rego.expectSet"
                                        #:args `(,input-identifier)))
                    (,deref-id       ,(make <lean-funcall>
                                        #:name "Array.get?"
                                        ;; Yes, you're reading this right. Set indices start at `1',
                                        ;; despite array indices starting at `0'.
                                        #:args `(,target-cast-id ,target)))
                    ,@(recur deref-id rest new-identifiers)))
                 (else
                  `((,target-cast-id ,(make <lean-funcall>
                                        #:name "Rego.expectObject"
                                        #:args `(,input-identifier)))
                    (,deref-id       ,(make <lean-funcall>
                                        #:name "Rego.Object.ref"
                                        #:args `(,target-cast-id ,target)))
                    ,@(recur deref-id rest new-identifiers))))))
        (() '())))
    recur)

  (policy-map
   policy
   (lambda (scope breadcrumb form)
     (match form
       (($ <reference> (singleton))
        singleton)
       (($ <reference> (first rest ...))
        (make <lean-assignment-chain> #:identifiers ((specialize breadcrumb scope) first rest '())))
       (_ form)))))

© 2015 - 2026 Jakob L. Kreuze