45 lines
1.2 KiB
Factor
45 lines
1.2 KiB
Factor
! Copyright (C) 2004, 2007 Slava Pestov.
|
|
! See http://factorcode.org/license.txt for BSD license.
|
|
IN: io
|
|
USING: errors generic io kernel math namespaces sequences
|
|
vectors ;
|
|
|
|
TUPLE: line-reader cr ;
|
|
|
|
C: line-reader ( stream -- new-stream ) [ set-delegate ] keep ;
|
|
|
|
: cr+ t swap set-line-reader-cr ; inline
|
|
|
|
: cr- f swap set-line-reader-cr ; inline
|
|
|
|
: line-ends/eof ( stream str -- str ) f like swap cr- ; inline
|
|
|
|
: line-ends\r ( stream str -- str ) swap cr+ ; inline
|
|
|
|
: line-ends\n ( stream str -- str )
|
|
over line-reader-cr over empty? and
|
|
[ drop dup cr- stream-readln ] [ swap cr- ] if ; inline
|
|
|
|
: handle-readln ( stream str ch -- str )
|
|
{
|
|
{ f [ line-ends/eof ] }
|
|
{ CHAR: \r [ line-ends\r ] }
|
|
{ CHAR: \n [ line-ends\n ] }
|
|
} case ;
|
|
|
|
M: line-reader stream-readln ( stream -- str )
|
|
"\r\n" over delegate stream-read-until handle-readln ;
|
|
|
|
: fix\r ( stream string -- string )
|
|
"\n" ?head [ swap stream-read1 [ add ] when* ] [ nip ] if ;
|
|
|
|
M: line-reader stream-read
|
|
tuck delegate stream-read over line-reader-cr
|
|
[ over cr- fix\r ] [ nip ] if ;
|
|
|
|
: lines-loop ( -- ) readln [ , lines-loop ] when* ;
|
|
|
|
: (lines) ( -- seq ) [ lines-loop ] { } make ;
|
|
|
|
: lines ( stream -- seq ) [ (lines) ] with-stream ;
|