simple_ls.cpp

//  simple_ls program  -------------------------------------------------------//

//  Copyright Jeff Garland and Beman Dawes, 2002

//  Use, modification, and distribution is subject to the Boost Software
//  License, Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at
//  http://www.boost.org/LICENSE_1_0.txt)

//  See http://www.boost.org/libs/filesystem for documentation.

#define BOOST_FILESYSTEM_VERSION 3

//  As an example program, we don't want to use any deprecated features
#ifndef BOOST_FILESYSTEM_NO_DEPRECATED
#  define BOOST_FILESYSTEM_NO_DEPRECATED
#endif
#ifndef BOOST_SYSTEM_NO_DEPRECATED
#  define BOOST_SYSTEM_NO_DEPRECATED
#endif

#include "boost/filesystem/operations.hpp"
#include "boost/filesystem/path.hpp"
#include "boost/progress.hpp"
#include <iostream>
#include <vector>
#include <string>

namespace fs = boost::filesystem;

std::vector<std::string> dirFiles(std::string path, std::string format){

  std::vector<std::string> files;
  fs::path p;

  if(path == ""){
    p = fs::current_path();
  }else{
    fs::path dest(path);
    p = dest;
  }

  unsigned long file_count = 0;
  unsigned long dir_count = 0;
  unsigned long other_count = 0;
  unsigned long err_count = 0;
  unsigned long file_format_count = 0;

  if (!fs::exists(p))
  {
    std::cout << "## folder not found: " << p << std::endl;
    exit(1);
  }

  if (fs::is_directory(p))
  {
    //std::cout << "\nIn directory: " << p << "\n\n";
    fs::directory_iterator end_iter;
    for (fs::directory_iterator dir_itr(p); dir_itr != end_iter; ++dir_itr)
    {
      try
      {
        if (fs::is_directory(dir_itr->status()))
        {
          ++dir_count;
          //std::cout << dir_itr->path().filename() << " [directory]\n";
        }
        else if (fs::is_regular_file(dir_itr->status()))
        {
          ++file_count;

          std::string current_file = dir_itr->path().filename().string();
          std::string current_format = current_file.substr(current_file.length()-4, current_file.length());
          //std::cout << current_format << std::endl;

          if(current_format == format){
            ++file_format_count;
            std::string full_path = (path == "") ? current_file : path + "/" + current_file;
            files.push_back(full_path);
          }

        }
        else
        {
          ++other_count;
          //std::cout << dir_itr->path().filename() << " [other]\n";
        }

      }
      catch (const std::exception & ex)
      {
        ++err_count;
        //std::cout << dir_itr->path().filename() << " " << ex.what() << std::endl;
      }
    }
    //std::cout << "\n" << file_count << " files\n"
    //          << dir_count << " directories\n"
    //          << other_count << " others\n"
    //          << err_count << " errors\n";
  }
  else // must be a file
  {
    std::cout << "## not a directory: " << p << std::endl;
    exit(1);
  }

  return files;

}