USING: combinators io io.encodings.binary io.files kernel math
ranges sbufs sequences strings ;
IN: ulz

: copy-forward ( buffer to from -- )
    [a..b) dupd [ swap nth swap push ] 2with each ;

: cpy-start ( decoder offset -- i ) neg swap length + ;

: (cpy) ( buffer length offset -- )
    swap [ dup ] [ cpy-start ] [ over + ] tri* copy-forward ;

: lit ( buffer op -- buffer )
    dupd 1 + [ read1 suffix! ] times drop ;

: cpy ( buffer op -- buffer )
    dupd 0x3F 0xc0 [ bitand ] bi-curry@ bi 0xc0 = [
        8 shift read1 bitor
    ] when 4 + read1 1 + (cpy) ;

: interpret ( buffer -- buffer )
    read1 [ 
        dup 0xc0 bitand {
            { 0x00 [ lit ] }
            [ drop cpy ]
        } case interpret
    ] when* ; recursive 

: decode ( file -- string )
    binary <file-reader> [ 
        0 <sbuf> interpret 
    ] with-input-stream >string ;