USING: alien alien.c-types arrays continuations destructors io
io.windows libc io.nonblocking io.streams.duplex windows.types
math windows.kernel32 windows namespaces io.launcher kernel
sequences windows.errors assocs splitting system ;
IN: io.windows.launcher

! From MSDN: "Handles in PROCESS_INFORMATION must be closed with CloseHandle when they are no longer needed."

TUPLE: CreateProcess-args
       lpApplicationName
       lpCommandLine
       lpProcessAttributes
       lpThreadAttributes
       bInheritHandles
       dwCreateFlags
       lpEnvironment
       lpCurrentDirectory
       lpStartupInfo
       lpProcessInformation
       stdout-pipe stdin-pipe ;

: dispose-CreateProcess-args ( args -- )
    CreateProcess-args-lpProcessInformation dup
    PROCESS_INFORMATION-hProcess [ CloseHandle drop ] when*
    PROCESS_INFORMATION-hThread [ CloseHandle drop ] when* ;

: default-CreateProcess-args ( -- obj )
    0
    0
    "STARTUPINFO" <c-object>
    "STARTUPINFO" heap-size over set-STARTUPINFO-cb
    "PROCESS_INFORMATION" <c-object>
    {
        set-CreateProcess-args-bInheritHandles
        set-CreateProcess-args-dwCreateFlags
        set-CreateProcess-args-lpStartupInfo
        set-CreateProcess-args-lpProcessInformation
    } \ CreateProcess-args construct ;

: call-CreateProcess ( CreateProcess-args -- )
    {
        CreateProcess-args-lpApplicationName
        CreateProcess-args-lpCommandLine
        CreateProcess-args-lpProcessAttributes
        CreateProcess-args-lpThreadAttributes
        CreateProcess-args-bInheritHandles
        CreateProcess-args-dwCreateFlags
        CreateProcess-args-lpEnvironment
        CreateProcess-args-lpCurrentDirectory
        CreateProcess-args-lpStartupInfo
        CreateProcess-args-lpProcessInformation
    } get-slots CreateProcess win32-error=0/f ;

: join-arguments ( args -- cmd-line )
    [ "\"" swap "\"" 3append ] map " " join ;

: app-name/cmd-line ( -- app-name cmd-line )
    +command+ get [
        " " split1
    ] [
        +arguments+ get unclip swap join-arguments
    ] if* ;

: cmd-line ( -- cmd-line )
    +command+ get [ +arguments+ get join-arguments ] unless* ;

: fill-lpApplicationName
    app-name/cmd-line
    pick set-CreateProcess-args-lpCommandLine
    over set-CreateProcess-args-lpApplicationName ;

: fill-lpCommandLine
    cmd-line over set-CreateProcess-args-lpCommandLine ;

: fill-dwCreateFlags
    0
    pass-environment? [ CREATE_UNICODE_ENVIRONMENT bitor ] when
    +detached+ get [ DETACHED_PROCESS bitor ] when
    over set-CreateProcess-args-dwCreateFlags ;

: fill-lpEnvironment
    pass-environment? [
        [
            get-environment
            [ swap % "=" % % "\0" % ] assoc-each
            "\0" %
        ] "" make >c-ushort-array
        over set-CreateProcess-args-lpEnvironment
    ] when ;

: wait-for-process ( args -- )
    CreateProcess-args-lpProcessInformation
    PROCESS_INFORMATION-hProcess INFINITE
    WaitForSingleObject drop ;

M: windows-io run-process* ( desc -- )
    [
        default-CreateProcess-args
        wince? [
            fill-lpApplicationName
        ] [
            fill-lpCommandLine
        ] if
        fill-dwCreateFlags
        fill-lpEnvironment
        dup call-CreateProcess
        +detached+ get [ dup wait-for-process ] unless
        dispose-CreateProcess-args
    ] with-descriptor ;

! : default-security-attributes ( -- obj )
!     "SECURITY_ATTRIBUTES" <c-object>
!     "SECURITY_ATTRIBUTES" heap-size over set-SECURITY_ATTRIBUTES-nLength ;
!
! : security-attributes-inherit ( -- obj )
!     default-security-attributes
!     TRUE over set-SECURITY_ATTRIBUTES-bInheritHandle ;
!
! : set-inherit ( handle ? -- )
!     >r HANDLE_FLAG_INHERIT r> >BOOLEAN SetHandleInformation win32-error=0/f ;
!
! ! http://msdn2.microsoft.com/en-us/library/ms682499.aspx
!
! TUPLE: pipe hRead hWrite ;
!
! C: <pipe> pipe
!
! : factor-pipe-name
!     "\\\\.\\pipe\\Factor" ;
!
! : create-named-pipe ( str -- handle )
!     PIPE_ACCESS_DUPLEX FILE_FLAG_OVERLAPPED bitor
!     PIPE_TYPE_BYTE PIPE_READMODE_BYTE PIPE_NOWAIT bitor bitor
!     PIPE_UNLIMITED_INSTANCES
!     default-buffer-size get
!     default-buffer-size get
!     0
!     security-attributes-inherit
!     CreateNamedPipe dup invalid-handle? ;
!
! : ERROR_PIPE_CONNECT 535 ; inline
!
! : pipe-connect-error? ( n -- ? )
!     ERROR_SUCCESS ERROR_PIPE_CONNECT 2array member? not ;
!
! clear "ls" <process-stream> contents
! M: windows-nt-io <process-stream> ( command -- stream )
!     [
!         [
!             default-CreateProcess-args
!             fill-lpCommandLine
!             TRUE over set-CreateProcess-args-bInheritHandles
!
!             dup CreateProcess-args-lpStartupInfo
!             STARTF_USESTDHANDLES over set-STARTUPINFO-dwFlags
!
!             factor-pipe-name create-named-pipe
!             global [ "Named pipe: " write dup . ] bind
!             dup t set-inherit
!             [ add-completion ] keep
!
!             ! CreateFile
!             ! factor-pipe-name open-pipe-r/w
!             factor-pipe-name GENERIC_READ GENERIC_WRITE bitor
!             0 f OPEN_EXISTING FILE_FLAG_OVERLAPPED f
!             CreateFile
!             global [ "Created File: " write dup . ] bind
!             dup invalid-handle? dup close-later
!             dup add-completion
!
!             swap (make-overlapped) ConnectNamedPipe zero? [
!                 GetLastError pipe-connect-error? [
!                     win32-error-string throw
!                 ] when
!             ] when
!             dup t set-inherit
!
!             ! ERROR_PIPE_CONNECTED
!             [ pick set-CreateProcess-args-stdin-pipe ] keep
!             global [ "Setting the stdios to: " write dup . ] bind
!             [ over set-STARTUPINFO-hStdOutput ] keep
!             [ over set-STARTUPINFO-hStdInput ] keep
!             swap set-STARTUPINFO-hStdError
!             !
!             [ call-CreateProcess ] keep
!             [ CreateProcess-args-stdin-pipe f <win32-file> dup handle>duplex-stream ] keep
!             drop ! TODO: close handles instead of drop
!         ] with-destructors
!     ] with-descriptor ;
!
! : create-pipe ( -- pipe )
!     "HANDLE" <c-object>
!     "HANDLE" <c-object>
!     [
!         security-attributes-inherit
!         0
!         CreatePipe win32-error=0/f
!     ] 2keep
!     [ *void* dup close-later ] 2apply <pipe> ;
!
! M: windows-ce-io process-stream*
!     [
!         default-CreateProcess-args
!         TRUE over set-CreateProcess-args-bInheritHandles
!
!         create-pipe  ! for child's STDOUT
!         dup pipe-hRead f set-inherit
!         over set-CreateProcess-args-stdout-pipe
!
!         create-pipe  ! for child's STDIN
!         dup pipe-hWrite f set-inherit
!         over set-CreateProcess-args-stdin-pipe
!
!         dup CreateProcess-args-lpStartupInfo
!         STARTF_USESTDHANDLES over set-STARTUPINFO-dwFlags
!
!         over CreateProcess-args-stdout-pipe
!             pipe-hWrite over set-STARTUPINFO-hStdOutput
!         over CreateProcess-args-stdout-pipe
!             pipe-hWrite over set-STARTUPINFO-hStdError
!         over CreateProcess-args-stdin-pipe
!             pipe-hRead swap set-STARTUPINFO-hStdInput
!
!         [ call-CreateProcess ] keep
!         [ CreateProcess-args-stdin-pipe pipe-hRead f <win32-file> <reader> ] keep
!         [ CreateProcess-args-stdout-pipe pipe-hWrite f <win32-file> <writer> <duplex-stream> ] keep
!         drop ! TODO: close handles instead of drop
!     ] with-destructors ;
!