! Copyright (C) 2009 Slava Pestov.
! See http://factorcode.org/license.txt for BSD license.
USING: threads kernel namespaces continuations combinators
sequences math namespaces.private continuations.private
concurrency.messaging quotations kernel.private words
sequences.private assocs models models.arrow arrays accessors
generic generic.single definitions make sbufs tools.crossref fry ;
IN: tools.continuations

<PRIVATE

: after-break ( object -- )
    {
        { [ dup continuation? ] [ (continue) ] }
        { [ dup not ] [ "Single stepping abandoned" rethrow ] }
    } cond ;

PRIVATE>

SYMBOL: break-hook

: break ( -- )
    continuation callstack >>call
    break-hook get call( continuation -- continuation' )
    after-break ;

\ break t "break?" set-word-prop

GENERIC: add-breakpoint ( quot -- quot' )

<PRIVATE

M: callable add-breakpoint
    dup [ break ] head? [ \ break prefix ] unless ;

M: array add-breakpoint
    [ add-breakpoint ] map ;

M: object add-breakpoint ;

: (step-into-quot) ( quot -- ) add-breakpoint call ;

: (step-into-dip) ( quot -- ) add-breakpoint dip ;

: (step-into-2dip) ( quot -- ) add-breakpoint 2dip ;

: (step-into-3dip) ( quot -- ) add-breakpoint 3dip ;

: (step-into-if) ( true false ? -- ) ? (step-into-quot) ;

: (step-into-dispatch) ( array n -- ) nth (step-into-quot) ;

: (step-into-execute) ( word -- )
    {
        { [ dup "step-into" word-prop ] [ "step-into" word-prop call ] }
        { [ dup single-generic? ] [ effective-method (step-into-execute) ] }
        { [ dup uses \ suspend swap member? ] [ execute break ] }
        { [ dup primitive? ] [ execute break ] }
        [ def>> (step-into-quot) ]
    } cond ;

\ (step-into-execute) t "step-into?" set-word-prop

: (step-into-continuation) ( -- )
    continuation callstack >>call break ;

: (step-into-call-next-method) ( method -- )
    next-method-quot (step-into-quot) ;

<< {
    (step-into-quot)
    (step-into-dip)
    (step-into-2dip)
    (step-into-3dip)
    (step-into-if)
    (step-into-dispatch)
    (step-into-execute)
    (step-into-continuation)
    (step-into-call-next-method)
} [ t "no-compile" set-word-prop ] each >>

: >innermost-frame< ( callstack -- n quot )
    [ innermost-frame-scan 1 + ] [ innermost-frame-executing ] bi ;

: (change-frame) ( callstack quot -- callstack' )
    [ dup innermost-frame-executing quotation? ] dip '[
        clone [ >innermost-frame< @ ] [ set-innermost-frame-quot ] [ ] tri
    ] when ; inline

: change-frame ( continuation quot -- continuation' )
    #! Applies quot to innermost call frame of the
    #! continuation.
    [ clone ] dip '[ _ (change-frame) ] change-call ; inline

PRIVATE>

: continuation-step ( continuation -- continuation' )
    [
        2dup length = [ nip [ break ] append ] [
            2dup nth \ break = [ nip ] [
                swap 1 + cut [ break ] glue 
            ] if
        ] if
    ] change-frame ;

: continuation-step-out ( continuation -- continuation' )
    [ nip \ break suffix ] change-frame ;

{
    { call [ (step-into-quot) ] }
    { dip [ (step-into-dip) ] }
    { 2dip [ (step-into-2dip) ] }
    { 3dip [ (step-into-3dip) ] }
    { execute [ (step-into-execute) ] }
    { if [ (step-into-if) ] }
    { dispatch [ (step-into-dispatch) ] }
    { continuation [ (step-into-continuation) ] }
    { (call-next-method) [ (step-into-call-next-method) ] }
} [ "step-into" set-word-prop ] assoc-each

! Never step into these words
: don't-step-into ( word -- )
    dup '[ _ execute break ] "step-into" set-word-prop ;

{
    >n ndrop >c c>
    continue continue-with
    stop suspend (spawn)
} [ don't-step-into ] each

\ break [ break ] "step-into" set-word-prop

: continuation-step-into ( continuation -- continuation' )
    [
        swap cut [
            swap %
            [ \ break , ] [
                unclip {
                    { [ dup \ break eq? ] [ , ] }
                    { [ dup quotation? ] [ add-breakpoint , \ break , ] }
                    { [ dup array? ] [ add-breakpoint , \ break , ] }
                    { [ dup word? ] [ literalize , \ (step-into-execute) , ] }
                    [ , \ break , ]
                } cond %
            ] if-empty
        ] [ ] make
    ] change-frame ;

: continuation-current ( continuation -- obj )
    call>> >innermost-frame< ?nth ;