fa1da7d2ee
Add the possibility of giving flags to dir_iterator_begin to initialize a dir-iterator with special options. Currently possible flags are: - DIR_ITERATOR_PEDANTIC, which makes dir_iterator_advance abort immediately in the case of an error, instead of keep looking for the next valid entry; - DIR_ITERATOR_FOLLOW_SYMLINKS, which makes the iterator follow symlinks and include linked directories' contents in the iteration. These new flags will be used in a subsequent patch. Also add tests for the flags' usage and adjust refs/files-backend.c to the new dir_iterator_begin signature. Signed-off-by: Matheus Tavares <matheus.bernardino@usp.br> Signed-off-by: Junio C Hamano <gitster@pobox.com>
59 lines
1.4 KiB
C
59 lines
1.4 KiB
C
#include "test-tool.h"
|
|
#include "git-compat-util.h"
|
|
#include "strbuf.h"
|
|
#include "iterator.h"
|
|
#include "dir-iterator.h"
|
|
|
|
/*
|
|
* usage:
|
|
* tool-test dir-iterator [--follow-symlinks] [--pedantic] directory_path
|
|
*/
|
|
int cmd__dir_iterator(int argc, const char **argv)
|
|
{
|
|
struct strbuf path = STRBUF_INIT;
|
|
struct dir_iterator *diter;
|
|
unsigned int flags = 0;
|
|
int iter_status;
|
|
|
|
for (++argv, --argc; *argv && starts_with(*argv, "--"); ++argv, --argc) {
|
|
if (strcmp(*argv, "--follow-symlinks") == 0)
|
|
flags |= DIR_ITERATOR_FOLLOW_SYMLINKS;
|
|
else if (strcmp(*argv, "--pedantic") == 0)
|
|
flags |= DIR_ITERATOR_PEDANTIC;
|
|
else
|
|
die("invalid option '%s'", *argv);
|
|
}
|
|
|
|
if (!*argv || argc != 1)
|
|
die("dir-iterator needs exactly one non-option argument");
|
|
|
|
strbuf_add(&path, *argv, strlen(*argv));
|
|
diter = dir_iterator_begin(path.buf, flags);
|
|
|
|
if (!diter) {
|
|
printf("dir_iterator_begin failure: %d\n", errno);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
while ((iter_status = dir_iterator_advance(diter)) == ITER_OK) {
|
|
if (S_ISDIR(diter->st.st_mode))
|
|
printf("[d] ");
|
|
else if (S_ISREG(diter->st.st_mode))
|
|
printf("[f] ");
|
|
else if (S_ISLNK(diter->st.st_mode))
|
|
printf("[s] ");
|
|
else
|
|
printf("[?] ");
|
|
|
|
printf("(%s) [%s] %s\n", diter->relative_path, diter->basename,
|
|
diter->path.buf);
|
|
}
|
|
|
|
if (iter_status != ITER_DONE) {
|
|
printf("dir_iterator_advance failure\n");
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|