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

文章目录

• 使用ls实现tree
• 使用find实现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=1 local dir=2
   local files=((ls -F dir | grep -v /)) local dirs=((ls -F dir | grep /))
   # 打印当前目录的文件
   for file in "{files[@]}"; do echo "{indent}{file}" done # 递归打印目录 for subdir in "{dirs[@]}"; do
   echo "{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=$1
    local dir=$2
    local files=($(find $dir -maxdepth 1 -type f ))
    local dirs=($(find $dir -maxdepth 1 -type d))
    if [ "${#dirs[@]}" -gt "0" ];then
	unset dirs[0]
    fi
    # 打印当前目录的文件
    for file in "${files[@]}"; do
        echo "${indent}$(basename $(echo $file | tr ' ' '\'))"
    done

    # 递归打印目录
    for subdir in "${dirs[@]}"; do
	echo "${indent}$(basename $(echo $subdir | tr ' ' '\ '))"
	print_tree "|......${indent}" ${subdir}
    done
}

# 从当前目录开始打印树状图
print_tree "" "."

测试一下，输出如下：

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

快一起试试吧～