disarm and rearm syntax in syntax-srcloc-lens

fixes part of https://github.com/jackfirth/lens/issues/287 see also https://github.com/racket/racket/pull/1448
2016-09-09 20:25:53 -04:00 · 2016-09-09 20:25:53 -04:00 · c7b5783088
commit c7b5783088
parent 04667e02ac
1 changed files with 7 additions and 4 deletions
--- a/lens-data/lens/private/syntax/srcloc.rkt
+++ b/lens-data/lens/private/syntax/srcloc.rkt
@ -39,10 +39,13 @@ module+ test

 ;; syntax-set-source-location : Syntax Source-Location -> Syntax
 (define (syntax-set-source-location stx src)
-  (datum->syntax stx
-                 (syntax-e stx)
-                 (source-location->list src)
-                 stx))
+  (define stx* (syntax-disarm stx #f))
+  (syntax-rearm
+   (datum->syntax stx*
+                  (syntax-e stx*)
+                  (source-location->list src)
+                  stx*)
+   stx))

 (define syntax-srcloc-lens
  (make-lens