factor/library/cli.factor

! Copyright (C) 2003, 2004 Slava Pestov.
! See http://factor.sf.net/license.txt for BSD license.
IN: kernel
USING: errors hashtables io kernel-internals lists namespaces
parser sequences strings ;

: ?run-file ( file -- )
    dup exists? [ [ dup run-file ] try ] when drop ;

: run-user-init ( -- )
    #! Run user init file if it exists
    "user-init" get
    [ "~" get "/.factor-rc" append ?run-file ] when ;

: set-path ( value seq -- )
    unswons over [ nest [ set-path ] bind ] [ nip set ] if ;

: cli-var-param ( name value -- ) swap ":" split set-path ;

: cli-bool-param ( name -- ) "no-" ?head not cli-var-param ;

: cli-param ( param -- )
    #! Handle a command-line argument starting with '-' by
    #! setting that variable to t, or if the argument is
    #! prefixed with 'no-', setting the variable to f.
    #!
    #! Arguments containing = are handled differently; they
    #! set the object path.
    "=" split1 [ cli-var-param ] [ cli-bool-param ] if* ;

: cli-arg ( argument -- argument )
    #! Handle a command-line argument. If the argument was
    #! consumed, returns f. Otherwise returns the argument.
    #! Parameters that start with + are runtime parameters.
    dup empty? [
        "-" ?head [ cli-param f ] when
        dup [ "+" ?head [ drop f ] when ] when
    ] unless ;

: cli-args ( -- args ) 10 getenv ;

: default-cli-args
    #! Some flags are *on* by default, unless user specifies
    #! -no-<flag> CLI switch
    "user-init" on
    "compile" on
    "null-stdio" off
    os "win32" = "ui" "tty" ? "shell" set ;

: parse-command-line ( -- )
    cli-args [ cli-arg ] subset [ run-file ] each  ;