Shell script that emulates the tree command in Linux

[DarkLordoftheMonkeys]

#!/bin/bash

# This shell program prints a tree diagram of
# the files in a directory, using recursion.
# It is an emulation of the Linux tree command,
# which I don't have on my Mac.

olddir=$PWD;
color="off";
for option in "$@"
do
    case $option in
        --color ) color="on";
              shift;;
        -*    ) shift;;
        *       )      ;;
    esac
done
declare -i stackheight=0;
# stackheight determines the directory depth.
function listfiles {
    cd "$1";
    for file in *
    do
        for ((i=0; $i < $stackheight; i++))
        do
            printf "\t";
        done
        # indentation shows the depth of the file
        if [ -d "$file" ]
        then
            if [ $color == "on" ]
            then
                printf "\e[34m";
                # display directory names in blue
            fi
        fi
        printf "$file\e[0m\n";
        if [ -d "$file" ]
        then
            stackheight=$stackheight+1;
            # more indentation
            listfiles "$file";
            # recursive listing of files
            cd ..;
        fi
    done
    let stackheight=$stackheight-1;
    # less indentation
}
listfiles "$1";
cd $olddir;
unset i color stackheight;

[dargueta]

Nice! +rep. I think this'll fix your problem with the spaces:

Instead of:

for file in *
do
    ...
done

Try this:

declare -a  FILELIST
FILELIST="*"

for file in ${FILELIST[@]}
do
    ...
done

Can't test it right now, but you can try it.

EDIT: Change all your if [ ... ] to if [[ ... ]]; and it'll work for spaces.

[DarkLordoftheMonkeys]

Doesn't matter. I already fixed it weeks ago. Thanks though.

[dargueta]

Ah, well you should've posted the solution then.

[DarkLordoftheMonkeys]

I've just edited the original post with the script that works. All I did was put quotation marks around the arguments, and use "$@" instead of $*.