Hex Artifact Content
Not logged in

Artifact f1f072612971a3b6e4240be9251862e4d604a94f:


0000: 3b 3b 3b 3b 20 66 6d 74 2d 6a 73 2e 73 63 6d 20  ;;;; fmt-js.scm 
0010: 2d 2d 20 6a 61 76 61 73 63 72 69 70 74 20 66 6f  -- javascript fo
0020: 72 6d 61 74 74 69 6e 67 20 75 74 69 6c 69 74 69  rmatting utiliti
0030: 65 73 0a 3b 3b 0a 3b 3b 20 43 6f 70 79 72 69 67  es.;;.;; Copyrig
0040: 68 74 20 28 63 29 20 32 30 31 31 2d 32 30 31 32  ht (c) 2011-2012
0050: 20 41 6c 65 78 20 53 68 69 6e 6e 2e 20 20 41 6c   Alex Shinn.  Al
0060: 6c 20 72 69 67 68 74 73 20 72 65 73 65 72 76 65  l rights reserve
0070: 64 2e 0a 3b 3b 20 42 53 44 2d 73 74 79 6c 65 20  d..;; BSD-style 
0080: 6c 69 63 65 6e 73 65 3a 20 68 74 74 70 3a 2f 2f  license: http://
0090: 73 79 6e 74 68 63 6f 64 65 2e 63 6f 6d 2f 6c 69  synthcode.com/li
00a0: 63 65 6e 73 65 2e 74 78 74 0a 0a 3b 3b 3b 3b 3b  cense.txt..;;;;;
00b0: 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b  ;;;;;;;;;;;;;;;;
00c0: 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b  ;;;;;;;;;;;;;;;;
00d0: 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b  ;;;;;;;;;;;;;;;;
00e0: 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b  ;;;;;;;;;;;;;;;;
00f0: 3b 3b 3b 0a 0a 28 64 65 66 69 6e 65 20 28 6a 73  ;;;..(define (js
0100: 2d 65 78 70 72 20 78 29 0a 20 20 28 66 6d 74 2d  -expr x).  (fmt-
0110: 6c 65 74 20 27 67 65 6e 20 6a 73 2d 65 78 70 72  let 'gen js-expr
0120: 2f 73 65 78 70 0a 20 20 20 20 20 20 20 20 20 20  /sexp.          
0130: 20 28 6c 61 6d 62 64 61 20 28 73 74 29 20 28 28   (lambda (st) ((
0140: 28 6f 72 20 28 66 6d 74 2d 67 65 6e 20 73 74 29  (or (fmt-gen st)
0150: 20 6a 73 2d 65 78 70 72 2f 73 65 78 70 29 20 78   js-expr/sexp) x
0160: 29 20 73 74 29 29 29 29 0a 0a 28 64 65 66 69 6e  ) st))))..(defin
0170: 65 20 28 6a 73 2d 65 78 70 72 2f 73 65 78 70 20  e (js-expr/sexp 
0180: 78 29 0a 20 20 28 63 6f 6e 64 0a 20 20 20 28 28  x).  (cond.   ((
0190: 70 72 6f 63 65 64 75 72 65 3f 20 78 29 0a 20 20  procedure? x).  
01a0: 20 20 78 29 0a 20 20 20 28 28 70 61 69 72 3f 20    x).   ((pair? 
01b0: 78 29 0a 20 20 20 20 28 63 61 73 65 20 28 63 61  x).    (case (ca
01c0: 72 20 78 29 0a 20 20 20 20 20 20 28 28 25 66 75  r x).      ((%fu
01d0: 6e 20 66 75 6e 63 74 69 6f 6e 29 20 28 61 70 70  n function) (app
01e0: 6c 79 20 6a 73 2d 66 75 6e 63 74 69 6f 6e 20 28  ly js-function (
01f0: 63 64 72 20 78 29 29 29 0a 20 20 20 20 20 20 28  cdr x))).      (
0200: 28 25 76 61 72 20 76 61 72 29 20 28 61 70 70 6c  (%var var) (appl
0210: 79 20 6a 73 2d 76 61 72 20 28 63 64 72 20 78 29  y js-var (cdr x)
0220: 29 29 0a 20 20 20 20 20 20 28 28 65 71 3f 20 3d  )).      ((eq? =
0230: 3d 3d 29 20 28 61 70 70 6c 79 20 6a 73 3d 3d 3d  ==) (apply js===
0240: 20 28 63 64 72 20 78 29 29 29 0a 20 20 20 20 20   (cdr x))).     
0250: 20 28 28 3e 3e 3e 29 20 28 61 70 70 6c 79 20 6a   ((>>>) (apply j
0260: 73 3e 3e 3e 20 28 63 64 72 20 78 29 29 29 0a 20  s>>> (cdr x))). 
0270: 20 20 20 20 20 28 28 25 61 72 72 61 79 29 20 28       ((%array) (
0280: 6a 73 2d 61 72 72 61 79 20 78 29 29 0a 20 20 20  js-array x)).   
0290: 20 20 20 28 28 25 6f 62 6a 65 63 74 29 20 28 6a     ((%object) (j
02a0: 73 2d 6f 62 6a 65 63 74 20 28 63 64 72 20 78 29  s-object (cdr x)
02b0: 29 29 0a 20 20 20 20 20 20 28 28 25 63 6f 6d 6d  )).      ((%comm
02c0: 65 6e 74 29 20 28 6a 73 2d 63 6f 6d 6d 65 6e 74  ent) (js-comment
02d0: 20 78 29 29 0a 20 20 20 20 20 20 28 65 6c 73 65   x)).      (else
02e0: 20 28 63 2d 65 78 70 72 2f 73 65 78 70 20 78 29   (c-expr/sexp x)
02f0: 29 29 29 0a 20 20 20 28 28 76 65 63 74 6f 72 3f  ))).   ((vector?
0300: 20 78 29 20 28 6a 73 2d 61 72 72 61 79 20 78 29   x) (js-array x)
0310: 29 0a 20 20 20 28 28 62 6f 6f 6c 65 61 6e 3f 20  ).   ((boolean? 
0320: 78 29 20 28 63 61 74 20 28 69 66 20 78 20 22 74  x) (cat (if x "t
0330: 72 75 65 22 20 22 66 61 6c 73 65 22 29 29 29 0a  rue" "false"))).
0340: 20 20 20 28 28 63 68 61 72 3f 20 78 29 20 28 6a     ((char? x) (j
0350: 73 2d 65 78 70 72 2f 73 65 78 70 20 28 73 74 72  s-expr/sexp (str
0360: 69 6e 67 20 78 29 29 29 0a 20 20 20 28 65 6c 73  ing x))).   (els
0370: 65 20 28 63 2d 65 78 70 72 2f 73 65 78 70 20 78  e (c-expr/sexp x
0380: 29 29 29 29 0a 0a 28 64 65 66 69 6e 65 20 28 6a  ))))..(define (j
0390: 73 2d 66 75 6e 63 74 69 6f 6e 20 2e 20 78 29 0a  s-function . x).
03a0: 20 20 28 6c 65 74 2a 20 28 28 6e 61 6d 65 20 28    (let* ((name (
03b0: 61 6e 64 20 28 73 79 6d 62 6f 6c 3f 20 28 63 61  and (symbol? (ca
03c0: 72 20 78 29 29 20 28 63 61 72 20 78 29 29 29 0a  r x)) (car x))).
03d0: 20 20 20 20 20 20 20 20 20 28 70 61 72 61 6d 73           (params
03e0: 20 28 69 66 20 6e 61 6d 65 20 28 63 61 64 72 20   (if name (cadr 
03f0: 78 29 20 28 63 61 72 20 78 29 29 29 0a 20 20 20  x) (car x))).   
0400: 20 20 20 20 20 20 28 62 6f 64 79 20 28 69 66 20        (body (if 
0410: 6e 61 6d 65 20 28 63 64 64 72 20 78 29 20 28 63  name (cddr x) (c
0420: 64 72 20 78 29 29 29 29 0a 20 20 20 20 28 63 2d  dr x)))).    (c-
0430: 62 6c 6f 63 6b 0a 20 20 20 20 20 28 63 61 74 20  block.     (cat 
0440: 22 66 75 6e 63 74 69 6f 6e 20 22 20 28 64 73 70  "function " (dsp
0450: 20 28 6f 72 20 6e 61 6d 65 20 22 22 29 29 20 20   (or name ""))  
0460: 22 28 22 0a 20 20 20 20 20 20 20 20 20 20 28 66  "(".          (f
0470: 6d 74 2d 6a 6f 69 6e 20 64 73 70 20 70 61 72 61  mt-join dsp para
0480: 6d 73 20 22 2c 20 22 29 20 22 29 22 29 0a 20 20  ms ", ") ")").  
0490: 20 20 20 28 66 6d 74 2d 6c 65 74 20 27 72 65 74     (fmt-let 'ret
04a0: 75 72 6e 3f 20 23 74 20 28 63 2d 69 6e 2d 73 74  urn? #t (c-in-st
04b0: 6d 74 20 28 61 70 70 6c 79 20 63 2d 62 65 67 69  mt (apply c-begi
04c0: 6e 20 62 6f 64 79 29 29 29 29 29 29 0a 0a 28 64  n body))))))..(d
04d0: 65 66 69 6e 65 20 28 6a 73 2d 76 61 72 20 2e 20  efine (js-var . 
04e0: 61 72 67 73 29 0a 20 20 28 61 70 70 6c 79 20 63  args).  (apply c
04f0: 2d 76 61 72 20 27 76 61 72 20 61 72 67 73 29 29  -var 'var args))
0500: 0a 0a 28 64 65 66 69 6e 65 20 28 6a 73 3d 3d 3d  ..(define (js===
0510: 20 2e 20 61 72 67 73 29 0a 20 20 28 61 70 70 6c   . args).  (appl
0520: 79 20 63 2d 6f 70 20 22 3d 3d 3d 22 20 61 72 67  y c-op "===" arg
0530: 73 29 29 0a 0a 28 64 65 66 69 6e 65 20 28 6a 73  s))..(define (js
0540: 3e 3e 3e 20 2e 20 61 72 67 73 29 0a 20 20 28 61  >>> . args).  (a
0550: 70 70 6c 79 20 63 2d 6f 70 20 22 3e 3e 3e 22 20  pply c-op ">>>" 
0560: 61 72 67 73 29 29 0a 0a 28 64 65 66 69 6e 65 20  args))..(define 
0570: 28 6a 73 2d 63 6f 6d 6d 65 6e 74 20 2e 20 61 72  (js-comment . ar
0580: 67 73 29 0a 20 20 28 63 6f 6c 75 6d 6e 61 72 20  gs).  (columnar 
0590: 22 2f 2f 20 22 20 28 61 70 70 6c 79 2d 63 61 74  "// " (apply-cat
05a0: 20 61 72 67 73 29 29 29 0a 0a 28 64 65 66 69 6e   args)))..(defin
05b0: 65 20 28 6a 73 2d 61 72 72 61 79 20 78 29 0a 20  e (js-array x). 
05c0: 20 28 6c 65 74 20 28 28 6c 73 20 28 76 65 63 74   (let ((ls (vect
05d0: 6f 72 2d 3e 6c 69 73 74 20 78 29 29 29 0a 20 20  or->list x))).  
05e0: 20 20 28 63 2d 77 72 61 70 2d 73 74 6d 74 0a 20    (c-wrap-stmt. 
05f0: 20 20 20 20 28 66 6d 74 2d 74 72 79 2d 66 69 74      (fmt-try-fit
0600: 0a 20 20 20 20 20 20 28 66 6d 74 2d 6c 65 74 20  .      (fmt-let 
0610: 27 6e 6f 2d 77 72 61 70 3f 20 23 74 20 28 63 61  'no-wrap? #t (ca
0620: 74 20 22 5b 22 20 28 66 6d 74 2d 6a 6f 69 6e 20  t "[" (fmt-join 
0630: 6a 73 2d 65 78 70 72 20 6c 73 20 22 2c 20 22 29  js-expr ls ", ")
0640: 20 22 5d 22 29 29 0a 20 20 20 20 20 20 28 6c 61   "]")).      (la
0650: 6d 62 64 61 20 28 73 74 29 0a 20 20 20 20 20 20  mbda (st).      
0660: 20 20 28 6c 65 74 2a 20 28 28 63 6f 6c 20 28 66    (let* ((col (f
0670: 6d 74 2d 63 6f 6c 20 73 74 29 29 0a 20 20 20 20  mt-col st)).    
0680: 20 20 20 20 20 20 20 20 20 20 20 28 73 65 70 20             (sep 
0690: 28 73 74 72 69 6e 67 2d 61 70 70 65 6e 64 20 22  (string-append "
06a0: 2c 22 20 28 6d 61 6b 65 2d 6e 6c 2d 73 70 61 63  ," (make-nl-spac
06b0: 65 20 63 6f 6c 29 29 29 29 0a 20 20 20 20 20 20  e col)))).      
06c0: 20 20 20 20 28 28 63 61 74 20 22 5b 22 20 28 66      ((cat "[" (f
06d0: 6d 74 2d 6a 6f 69 6e 20 6a 73 2d 65 78 70 72 20  mt-join js-expr 
06e0: 6c 73 20 73 65 70 29 20 22 5d 22 20 6e 6c 29 20  ls sep) "]" nl) 
06f0: 73 74 29 29 29 29 29 29 29 0a 0a 28 64 65 66 69  st)))))))..(defi
0700: 6e 65 20 28 6a 73 2d 70 61 69 72 20 78 29 0a 20  ne (js-pair x). 
0710: 20 28 63 61 74 20 28 6a 73 2d 65 78 70 72 20 28   (cat (js-expr (
0720: 63 61 72 20 78 29 29 20 22 3a 20 22 20 28 6a 73  car x)) ": " (js
0730: 2d 65 78 70 72 20 28 63 64 72 20 78 29 29 29 29  -expr (cdr x))))
0740: 0a 0a 28 64 65 66 69 6e 65 20 28 6a 73 2d 6f 62  ..(define (js-ob
0750: 6a 65 63 74 20 6c 73 29 0a 20 20 28 63 2d 69 6e  ject ls).  (c-in
0760: 2d 65 78 70 72 0a 20 20 20 28 66 6d 74 2d 74 72  -expr.   (fmt-tr
0770: 79 2d 66 69 74 0a 20 20 20 20 28 66 6d 74 2d 6c  y-fit.    (fmt-l
0780: 65 74 20 27 6e 6f 2d 77 72 61 70 3f 20 23 74 20  et 'no-wrap? #t 
0790: 28 63 61 74 20 22 7b 22 20 28 66 6d 74 2d 6a 6f  (cat "{" (fmt-jo
07a0: 69 6e 20 6a 73 2d 70 61 69 72 20 6c 73 20 22 2c  in js-pair ls ",
07b0: 20 22 29 20 22 7d 22 29 29 0a 20 20 20 20 28 6c   ") "}")).    (l
07c0: 61 6d 62 64 61 20 28 73 74 29 0a 20 20 20 20 20  ambda (st).     
07d0: 20 28 6c 65 74 2a 20 28 28 63 6f 6c 20 28 66 6d   (let* ((col (fm
07e0: 74 2d 63 6f 6c 20 73 74 29 29 0a 20 20 20 20 20  t-col st)).     
07f0: 20 20 20 20 20 20 20 20 28 73 65 70 20 28 73 74          (sep (st
0800: 72 69 6e 67 2d 61 70 70 65 6e 64 20 22 2c 22 20  ring-append "," 
0810: 28 6d 61 6b 65 2d 6e 6c 2d 73 70 61 63 65 20 63  (make-nl-space c
0820: 6f 6c 29 29 29 29 0a 20 20 20 20 20 20 20 20 28  ol)))).        (
0830: 28 63 61 74 20 22 7b 22 20 28 66 6d 74 2d 6a 6f  (cat "{" (fmt-jo
0840: 69 6e 20 6a 73 2d 70 61 69 72 20 6c 73 20 73 65  in js-pair ls se
0850: 70 29 20 22 7d 22 20 6e 6c 29 20 73 74 29 29 29  p) "}" nl) st)))
0860: 29 29 29 0a 0a 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b  )))..;;;;;;;;;;;
0870: 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b  ;;;;;;;;;;;;;;;;
0880: 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b  ;;;;;;;;;;;;;;;;
0890: 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b  ;;;;;;;;;;;;;;;;
08a0: 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 3b 0a        ;;;;;;;;;;;;;.