在mac环境下使用shell脚本实现tree命令

使用ls实现tree

实现思路

1. 使用ls -F 打印文件类型，如果是目录后面跟/，如果是可执行文件后面跟*；
2. 使用grep -v /$ 筛选文件排除目录，-v为反向筛选；
3. 使用grep /$ 仅筛选目录；
4. ${files[@]} 是获取数组的全部元素；
5. 获取文件直接打印，获取目录后，打印目录，拼接父目录给当前目录（因为shell执行总是在当前目录，如果只传目录，会报错找不到目录），然后递归打印。
   代码如下：

#!/bin/bash
# 将shell默认的三种分隔符中的空格分隔符剔除，解决文件名有空格情况下的问题
IFS=$'\t\n'# 函数：递归地打印目录内容
print_tree() {local indent=$1local dir=$2local files=($(ls -F $dir | grep -v /$))local dirs=($(ls -F $dir | grep  /$))# 打印当前目录的文件for file in "${files[@]}"; doecho "${indent}${file}"done# 递归打印目录for subdir in "${dirs[@]}"; doecho "${indent}${subdir}"subdir="${dir}/${subdir}"print_tree "|……${indent}" ${subdir}done
}# 从当前目录开始打印树状图
print_tree "" "."

测试一下，输出如下：

a/
|……b/
|……|……c/
|……|……d/
|……|……f/
|……c/
|……|……f/
|……|……|……test*
|……|……|……tt.txt

当然这种方法虽然简单，但是文件是可执行文件，那么在文件名后会跟一个*，风格不统一，下面再介绍一种打印方式。

使用find实现tree

实现思路：
find 命令加参数-type f 可以仅查找文件排除掉目录，但是查出来的文件带父目录，maxdepth可以指定查找深度，1表示只在本层目录中查找；
basename 命令可以删除指定结尾的后缀，也能打印除了/的最后一部分字符。比如basename /usr/local/Centos输出结果为Centos，但是如果目录名有空格就不会得到预期的效果；
tr 命令可以替换字符，比如可以将目录名中的空格替换为/，使得basename可以得到预期效果；
代码如下：

#!/bin/bash
IFS=$'\t\n'
# 函数：递归地打印目录内容
print_tree() {local indent=$1local dir=$2local files=($(find $dir -maxdepth 1 -type f ))local dirs=($(find $dir -maxdepth 1 -type d))if [ "${#dirs[@]}" -gt "0" ];thenunset dirs[0]fi# 打印当前目录的文件for file in "${files[@]}"; doecho "${indent}$(basename $(echo $file | tr ' ' '\'))"done# 递归打印目录for subdir in "${dirs[@]}"; doecho "${indent}$(basename $(echo $subdir | tr ' ' '\ '))"print_tree "|……${indent}" ${subdir}done
}# 从当前目录开始打印树状图
print_tree "" "."

测试一下，输出如下：

a
|……c
|……|……f
|……|……|……test
|……|……|……tt.txt
|……b
|……|……f
|……|……c
|……|……d

快一起试试吧～