#include <unistd.h>
#include <sys/types.h>
#include <dirent.h>
#include <string.h>
#include <stdio.h>
#include <sys/stat.h>
#include <unordered_map>
using namespace std;

int file_cnt = 0, link_cnt = 0, dir_cnt = 0;
unsigned long space_used = 0;
struct stat buf;
off_t space_used_bytes;
unordered_map<ino_t, bool> ht;

void listdir (const char *name) {
  DIR *dir;
  struct dirent *entry;

  if (!(dir = opendir(name))) return;
  if (!(entry = readdir(dir))) return;

  do {
    if (strcmp(entry->d_name, ".") == 0 || strcmp(entry->d_name, "..") == 0) continue;
    if (ht[entry->d_ino]) continue;
    if (entry->d_type == DT_DIR) {
      dir_cnt++;
      char path[4096];
      int len = snprintf(path, sizeof(path)-1, "%s/%s", name, entry->d_name);
      path[len] = 0;
      printf("[d] %s/%s\n", name, entry->d_name);
      listdir(path);
    }
    else {
      if (entry->d_type == DT_LNK) {
        link_cnt++;
        printf("[l] %s/%s\n", name, entry->d_name);
      }
      else{
        file_cnt++;
        char path[4096];
        int len = snprintf(path, sizeof(path) - 1, "%s/%s", name, entry->d_name);
        path[len] = 0;
        stat(path, &buf);
        space_used += buf.st_blocks;
        space_used_bytes += buf.st_size;
        printf("[f] %lu %lu\t%s/%s\n", buf.st_blocks, buf.st_size, name, entry->d_name);
      }
    }
    ht[entry->d_ino] = true;
  } while ((entry = readdir(dir)));
  closedir(dir);
}

int main (int argc, char** argv) {
  char buf[PATH_MAX + 1];
  char *dirpath = (argc > 1) ? realpath(argv[1], buf) : realpath(".", buf);
  listdir(dirpath);
  printf("\ntotals\nfile count: %d\tdir count: %d\tsymbolic link count: %d\nspace used: %lu blocks\n\t%lu bytes(blocks*512)\n\t%lu bytes(size on disk)\n\n", file_cnt, dir_cnt, link_cnt, space_used, space_used * 512, space_used_bytes);
  return 0;
}