FindJpeg.cc

#include "FindJpeg.hh"
#include <cstdio>
#include <cstring>
#include <sys/stat.h>

static const bool debug=0;


bool FindJpeg::scan () 
{
  struct dirent entry;
  struct dirent *de;
  while (readdir_r(dh_,&entry,&de) == 0) 
    {
      if (de == 0)
	break;
      
      if (de->d_name[0] == '.')
	continue;

      if (debug) fprintf(stderr, "scanning: %s\n", de->d_name);
      
      
#ifdef _DIRENT_HAVE_D_TYPE
      switch (de->d_type) 
	{
	case DT_DIR:
	  dirs_.push_back( dir_ + "/" + de->d_name );
	  break;
	case DT_REG:
	  {
	    if (debug) fprintf(stderr, "reg file\n");
	    
	    int len = strlen(de->d_name);
	    if (len < 5) break;
	    const char * ext = de->d_name + len - 4;
	    if (strncasecmp(ext,".jpg",4)) break;
	    file_ = dir_ + "/" + de->d_name;
	    if (debug) fprintf(stderr, "dir=%s, name=%s, file='%s'\n", dir_.c_str(), de->d_name, file_.c_str());
	    
	    return true;
	  }
	  
	case DT_UNKNOWN:
#endif
	  {
	    
	    //TODO - check for file type & repeat above logic...
	    std::string path = dir_ + "/" + de->d_name;
	    struct stat buf;
	    stat(path.c_str(), &buf);
	    if (S_ISDIR(buf.st_mode)) 
	      {
		dirs_.push_back( dir_ + "/" + de->d_name );
	      }
	    else if (S_ISREG(buf.st_mode)) 
	      {
		int len = strlen(de->d_name);
		if (len < 5) break;
		const char * ext = de->d_name + len - 4;
		if (strncasecmp(ext,".jpg",4)) break;
		file_ = path;
		return true;
	      }
	  }
	  
#ifdef _DIRENT_HAVE_D_TYPE
	}
#endif
    }
  return false;
}

bool FindJpeg::next () 
{
  if (dh_) 
    {
      if ( scan() )
	return true;
      else
	{
	  closedir( dh_ );
	  dh_ = 0;
	}
      
    }
  while( dirs_.size() ) 
    {
      dir_ = dirs_.back();
      dirs_.pop_back();
      dh_ = opendir(dir_.c_str());
      if (dh_ == NULL)
	{
	  perror(dir_.c_str());
	}
      else
	{
	  if ( scan() ) return true;
	}
    }
  return false;
  
}