summaryrefslogtreecommitdiff
path: root/miscutils/tree.c
blob: fa55696c6c4953d525e42c2ebe0da4be25a01349 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
/* vi: set sw=4 ts=4: */
/*
 * Copyright (C) 2022 Roger Knecht <rknecht@pm.me>
 *
 * Licensed under GPLv2, see file LICENSE in this source tree.
 */
//config:config TREE
//config:	bool "tree (0.6 kb)"
//config:	default y
//config:	help
//config:	List files and directories in a tree structure.

//applet:IF_TREE(APPLET(tree, BB_DIR_USR_BIN, BB_SUID_DROP))

//kbuild:lib-$(CONFIG_TREE) += tree.o

//usage:#define tree_trivial_usage NOUSAGE_STR
//usage:#define tree_full_usage ""

#include "libbb.h"
#include "common_bufsiz.h"
#include "unicode.h"

#define prefix_buf bb_common_bufsiz1

static void tree_print(unsigned count[2], const char* directory_name, char* prefix_pos)
{
	struct dirent **entries;
	int index, size;
	const char *bar = "|   ";
	const char *mid = "|-- ";
	const char *end = "`-- ";

#if ENABLE_UNICODE_SUPPORT
	if (unicode_status == UNICODE_ON) {
		bar = "│   ";
		mid = "├── ";
		end = "└── ";
	}
#endif

	// read directory entries
	size = scandir(directory_name, &entries, NULL, alphasort);

	if (size < 0) {
		fputs_stdout(directory_name);
		puts(" [error opening dir]");
		return;
	}

	// print directory name
	puts(directory_name);

	// switch to sub directory
	xchdir(directory_name);

	// print all directory entries
	for (index = 0; index < size;) {
		struct dirent *dirent = entries[index++];

		// filter hidden files and directories
		if (dirent->d_name[0] != '.') {
			int status;
			struct stat statBuf;

//TODO: when -l is implemented, use stat, not lstat, if -l
			status = lstat(dirent->d_name, &statBuf);

			if (index == size) {
				strcpy(prefix_pos, end);
			} else {
				strcpy(prefix_pos, mid);
			}
			fputs_stdout(prefix_buf);

			if (status == 0 && S_ISLNK(statBuf.st_mode)) {
				// handle symlink
				char* symlink_path = xmalloc_readlink(dirent->d_name);
				printf("%s -> %s\n", dirent->d_name, symlink_path);
				free(symlink_path);
				count[1]++;
			} else if (status == 0 && S_ISDIR(statBuf.st_mode)
			 && (prefix_pos - prefix_buf) < (COMMON_BUFSIZE - 16)
			) {
				// handle directory
				char* pos;
				if (index == size) {
					pos = stpcpy(prefix_pos, "    ");
				} else {
					pos = stpcpy(prefix_pos, bar);
				}
				tree_print(count, dirent->d_name, pos);
				count[0]++;
			} else {
				// handle file
				puts(dirent->d_name);
				count[1]++;
			}
		}

		// release directory entry
		free(dirent);
	}

	// release directory array
	free(entries);

	// switch to parent directory
	xchdir("..");
}

int tree_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int tree_main(int argc UNUSED_PARAM, char **argv)
{
	unsigned count[2] = { 0, 0 };

	setup_common_bufsiz();
	init_unicode();

	if (!argv[1])
		*argv-- = (char*)".";

	// list directories given as command line arguments
	while (*(++argv))
		tree_print(count, *argv, prefix_buf);

	// print statistic
	printf("\n%u directories, %u files\n", count[0], count[1]);

	return EXIT_SUCCESS;
}