print.nu

# A print command that concatenates arguments together with an optional separator
#    By default there will be no newline
def print [
    --separator(-s):any     # Optional separator (not yet flagged as optional?)
    ...rest                 # All of the parameters
    ] {
    let is_empty = ($separator | empty?)
    let num_of_rest = ($rest | length)
    $rest | each --numbered { |param|
        if $is_empty {
            $param.item
        } {
            if $num_of_rest > ($param.index + 1) {
                $"($param.item)($separator)"
            } {
                $param.item
            }
        }
    } | into string | str collect
}

# > print 1 2 3 "four" -s '--'
# 1--2--3--four

# > print 1 2 3 "four"
# 123four

# An alternate print command that concatenates arguments together with an optional separator.
# This one uses str collect instead of build-string.
#    By default there will be no newline
def print2 [
    --separator(-s):any     # Optional separator (not yet flagged as optional?)
    ...rest                 # All of the parameters
    ] {
    let is_empty = ($separator | empty?)
    let num_of_rest = ($rest | length)
    if $is_empty {
        $rest | into string | str collect
    } {
        $rest | into string | str collect $separator
    }
}

# Bring in the logging command
#source logging.nu

# A print command that concatenates arguments together with an optional separator.
# This print command will also concatenate tables like [1 2 3] as well as most other primitives
# since the into string command has been updated with wider support.
def print3 [
    --separator(-s):any     # Optional separator (not yet flagged as optional?)
    --flat(-f)              # If tables are found, flatten them
    ...rest                 # All of the parameters
    ] {
    let sep_empty = ($separator | empty?)
    let num_of_rest = ($rest | length)
    let flat = ($flat | empty?)
    $rest | each --numbered { |param|
        if $sep_empty {
            #log 'sep is empty'
            if (echo $param.item | length) > 1 && $flat {
                #log 'flatten please'
                let flatter = ($param.item | flatten | into string | str collect)
                $flatter
            } {
                #log 'no flat'
                $param.item
            }
        } {
            if $num_of_rest > ($param.index + 1) {
                if ($param.item | length) > 1 && $flat {
                    let flatter = ($param.item | flatten | into string | str collect $separator)
                    $"($flatter)($separator)"
                } {
                    $"($param.item)($separator)"
                }
            } {
                if ($param.item | length) > 1 && $flat {
                    let flatter = ($param.item | flatten | into string | str collect $separator)
                    $flatter
                } {
                    $param.item
                }
            }
        }
    } | str collect
}