如何用C实现带管道缩进与末尾文件标识的Linux tree命令？

嘿，这个需求我之前捣鼓原生tree的时候研究过，要实现带管道符的层级缩进，核心是跟踪每一层的层级状态——得记住哪些位置该留管道符，哪些该用空格，还要准确判断当前条目是不是同目录下的最后一个，这样才能替换成拐角的符号。我给你拆解下关键逻辑和代码实现：

1. 用状态数组记录层级缩进标记 #

每一层目录的缩进样式会直接影响下一级的显示：如果某一层的父目录是该目录下的最后一个条目，那么它下面的子项对应的缩进位置就不能显示管道符，得用空格代替。

我们可以定义一个字符数组（比如indent_flags），每个元素对应一个层级的标记：

• indent_flags[i] = '|'：表示第i层需要保留管道符，后续子项的对应位置要显示|
• indent_flags[i] = ' '：表示第i层是空格，后续子项的对应位置要显示

2. 判断当前条目是否为目录下的最后一个 #

要做到这一点，不能边读边判断，得先把当前目录的所有有效条目（跳过.和..）先存到数组/链表中，统计总数后，再遍历数组——当当前索引等于总数减1时，就说明是最后一个条目。

3. 递归遍历的缩进打印逻辑 #

在打印每个条目之前，先根据indent_flags数组打印对应层级的缩进符号，然后根据是否是最后一个条目，选择打印├── 还是└── ：

• 如果是最后一个条目：打印└── ，同时把下一层的标记设为空格（因为它后面没有兄弟项了）
• 如果不是最后一个：打印├── ，同时把下一层的标记设为|（因为后面还有兄弟项，需要保留管道符）

#include <stdio.h>
#include <stdlib.h>
#include <dirent.h>
#include <string.h>
#include <sys/stat.h>
#include <errno.h>

// 最大支持的目录层级，可根据需求调整
#define MAX_DEPTH 100

// 比较函数，用于排序目录条目（和原生tree行为一致）
int compare_entries(const void *a, const void *b) {
    const struct dirent **entry_a = (const struct dirent **)a;
    const struct dirent **entry_b = (const struct dirent **)b;
    return strcmp((*entry_a)->d_name, (*entry_b)->d_name);
}

void traverse_directory(const char *path, int current_depth, char indent_flags[]) {
    DIR *dir = opendir(path);
    struct dirent *entry;
    struct stat stat_info;
    char full_path[1024];
    int entry_count = 0;
    struct dirent **entry_list = NULL;

    if (!dir) {
        fprintf(stderr, "无法打开目录 %s: %s\n", path, strerror(errno));
        return;
    }

    // 先读取所有有效条目，跳过.和..
    while ((entry = readdir(dir)) != NULL) {
        if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0) {
            continue;
        }
        entry_list = realloc(entry_list, sizeof(struct dirent*) * (entry_count + 1));
        entry_list[entry_count++] = entry;
    }
    closedir(dir);

    // 对条目排序，和原生tree的输出顺序一致
    qsort(entry_list, entry_count, sizeof(struct dirent*), compare_entries);

    // 遍历所有条目
    for (int i = 0; i < entry_count; i++) {
        entry = entry_list[i];
        snprintf(full_path, sizeof(full_path), "%s/%s", path, entry->d_name);

        if (lstat(full_path, &stat_info) == -1) {
            fprintf(stderr, "无法获取文件信息 %s: %s\n", full_path, strerror(errno));
            continue;
        }

        // 打印当前层级的缩进符号
        for (int j = 0; j < current_depth; j++) {
            printf("%s", indent_flags[j] == '|' ? "|   " : "    ");
        }

        // 判断是否为当前目录的最后一个条目
        int is_last_entry = (i == entry_count - 1);
        if (is_last_entry) {
            printf("└── ");
            indent_flags[current_depth] = ' ';
        } else {
            printf("├── ");
            indent_flags[current_depth] = '|';
        }

        // 打印文件名，目录末尾加/
        printf("%s", entry->d_name);
        if (S_ISDIR(stat_info.st_mode)) {
            printf("/\n");
            // 递归遍历子目录
            traverse_directory(full_path, current_depth + 1, indent_flags);
        } else {
            printf("\n");
        }
    }

    free(entry_list);
}

int main(int argc, char *argv[]) {
    char *start_path = ".";
    if (argc > 1) {
        start_path = argv[1];
    }

    char indent_flags[MAX_DEPTH] = {0};
    printf("%s/\n", start_path);
    traverse_directory(start_path, 0, indent_flags);

    return 0;
}

• 排序处理：用qsort对条目按名称排序，和原生tree的输出顺序一致
• 符号链接处理：用lstat而不是stat，避免跟随符号链接进入其他目录
• Unicode支持：├──和└──是UTF-8字符，确保你的终端支持UTF-8编码，否则会显示乱码
• 错误处理：加入了基本的错误提示，比如无法打开目录、无法获取文件信息的情况

内容的提问来源于stack exchange，提问作者DamienzOnly