factor/basis/json/writer/writer.factor

85 lines
2.4 KiB
Factor

! Copyright (C) 2006 Chris Double.
! See http://factorcode.org/license.txt for BSD license.
USING: accessors kernel io.streams.string io strings splitting
sequences math math.parser assocs classes words namespaces make
prettyprint hashtables mirrors tr json fry combinators ;
IN: json.writer
#! Writes the object out to a stream in JSON format
GENERIC# stream-json-print 1 ( obj stream -- )
: json-print ( obj -- )
output-stream get stream-json-print ;
: >json ( obj -- string )
#! Returns a string representing the factor object in JSON format
[ json-print ] with-string-writer ;
M: f stream-json-print
[ drop "false" ] [ stream-write ] bi* ;
M: t stream-json-print
[ drop "true" ] [ stream-write ] bi* ;
M: json-null stream-json-print
[ drop "null" ] [ stream-write ] bi* ;
M: string stream-json-print
CHAR: " over stream-write1 swap [
{
{ CHAR: " [ "\\\"" over stream-write ] }
{ CHAR: \r [ ] }
{ CHAR: \n [ "\\r\\n" over stream-write ] }
[ over stream-write1 ]
} case
] each CHAR: " swap stream-write1 ;
M: integer stream-json-print
[ number>string ] [ stream-write ] bi* ;
M: real stream-json-print
[ >float number>string ] [ stream-write ] bi* ;
M: sequence stream-json-print
CHAR: [ over stream-write1 swap [
over '[ CHAR: , _ stream-write1 ]
pick '[ _ stream-json-print ] interleave
] unless-empty CHAR: ] swap stream-write1 ;
SYMBOL: jsvar-encode?
t jsvar-encode? set-global
TR: jsvar-encode "-" "_" ;
<PRIVATE
: json-print-assoc ( assoc stream -- )
CHAR: { over stream-write1 swap >alist [
jsvar-encode? get [
over '[ CHAR: , _ stream-write1 ]
pick dup '[
first2
[ jsvar-encode _ stream-json-print ]
[ _ CHAR: : over stream-write1 stream-json-print ]
bi*
] interleave
] [
over '[ CHAR: , _ stream-write1 ]
pick dup '[
first2
[ _ stream-json-print ]
[ _ CHAR: : over stream-write1 stream-json-print ]
bi*
] interleave
] if
] unless-empty CHAR: } swap stream-write1 ;
PRIVATE>
M: tuple stream-json-print
[ <mirror> ] dip json-print-assoc ;
M: hashtable stream-json-print json-print-assoc ;
M: word stream-json-print
[ name>> ] dip stream-json-print ;