Andrew's git - gitweb.git/blob - t/helper/test-dir-iterator.c

   1#include "test-tool.h"
   2#include "git-compat-util.h"
   3#include "strbuf.h"
   4#include "iterator.h"
   5#include "dir-iterator.h"
   6
   7static const char *error_name(int error_number)
   8{
   9        switch (error_number) {
  10        case ENOENT: return "ENOENT";
  11        case ENOTDIR: return "ENOTDIR";
  12        default: return "ESOMETHINGELSE";
  13        }
  14}
  15
  16/*
  17 * usage:
  18 * tool-test dir-iterator [--follow-symlinks] [--pedantic] directory_path
  19 */
  20int cmd__dir_iterator(int argc, const char **argv)
  21{
  22        struct strbuf path = STRBUF_INIT;
  23        struct dir_iterator *diter;
  24        unsigned int flags = 0;
  25        int iter_status;
  26
  27        for (++argv, --argc; *argv && starts_with(*argv, "--"); ++argv, --argc) {
  28                if (strcmp(*argv, "--follow-symlinks") == 0)
  29                        flags |= DIR_ITERATOR_FOLLOW_SYMLINKS;
  30                else if (strcmp(*argv, "--pedantic") == 0)
  31                        flags |= DIR_ITERATOR_PEDANTIC;
  32                else
  33                        die("invalid option '%s'", *argv);
  34        }
  35
  36        if (!*argv || argc != 1)
  37                die("dir-iterator needs exactly one non-option argument");
  38
  39        strbuf_add(&path, *argv, strlen(*argv));
  40        diter = dir_iterator_begin(path.buf, flags);
  41
  42        if (!diter) {
  43                printf("dir_iterator_begin failure: %s\n", error_name(errno));
  44                exit(EXIT_FAILURE);
  45        }
  46
  47        while ((iter_status = dir_iterator_advance(diter)) == ITER_OK) {
  48                if (S_ISDIR(diter->st.st_mode))
  49                        printf("[d] ");
  50                else if (S_ISREG(diter->st.st_mode))
  51                        printf("[f] ");
  52                else if (S_ISLNK(diter->st.st_mode))
  53                        printf("[s] ");
  54                else
  55                        printf("[?] ");
  56
  57                printf("(%s) [%s] %s\n", diter->relative_path, diter->basename,
  58                       diter->path.buf);
  59        }
  60
  61        if (iter_status != ITER_DONE) {
  62                printf("dir_iterator_advance failure\n");
  63                return 1;
  64        }
  65
  66        return 0;
  67}