Open code for workspace if available, file if configured, new workspace if nothing else

.profile

# A function to replace the `code` command
#
# With args: does what's intended.
# ```
# > code ~/.bash_history
# ```
#
# With no args, opens the first workspace it founds
# in the current folder, or a new workspace based
# on the current folder if none exists.
# ```
# > code
# ```

function code-ws() {
    if [ ! $# -eq 0 ]; then
        code "$@"
    else
        files=$(find . -maxdepth 1 -type f -name "*.code-workspace" | wc -l)
        if [ $files -eq 0 ]; then
            code -n .
        else 
        code *.code-workspace
        fi
    fi
}
alias code=code-ws