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)))))
|