source: trunk/lib/Directory.cc @ 91

// $Id: Directory.cc 91 2006-03-23 22:56:17Z jari $

/*
  Copyright (C) 2005, 2006 Jari Häkkinen, Peter Johansson

  This file is part of svnstat, http://lev.thep.lu.se/trac/svnstat

  svnstat is free software; you can redistribute it and/or modify it
  under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2 of the License, or
  (at your option) any later version.

  svnstat is distributed in the hope that it will be useful, but
  WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
  02111-1307, USA.
*/

#include "Directory.h"
#include "File.h"
#include "Node.h"
#include "SVN.h"

#include <algorithm>
#include <fstream>
#include <functional>
#include <iostream>
#include <iterator>
#include <list>

#include <cerrno> // Needed to check error state below.
#include <dirent.h>
#include <sys/stat.h>

namespace theplu{
namespace svnstat{


  Directory::Directory(const std::string& path, const std::string& output)
    : Node(path,output)
  {
    using namespace std;
    DIR* directory=opendir(path.c_str());    // C API from dirent.h
    if (!directory)
      throw NodeException("ERROR: opendir() failed; " + path +
                          " is not a directory");
    list<string> entries;
    struct dirent* entry;
    errno=0;  // Global variable used by C to track errors, from errno.h
    while ((entry=readdir(directory)))       // C API from dirent.h
      entries.push_back(string(entry->d_name));
    if (errno)
      throw NodeException("ERROR: readdir() failed on " + path);
    closedir(directory);

    SVN* svn=SVN::instance();
    for (list<string>::iterator i=entries.begin(); i!=entries.end(); i++)
      if ((*i)!=string(".") && (*i)!=string("..") && (*i)!=string(".svn")) {
        string fullpath(path_+'/'+(*i));
        switch (svn->version_controlled(fullpath)) {
        case SVN::uptodate:
          struct stat nodestat;                // C api from sys/stat.h
          lstat(fullpath.c_str(),&nodestat);   // C api from sys/stat.h
          if (S_ISDIR(nodestat.st_mode))       // C api from sys/stat.h
            daughters_.push_back(new Directory(fullpath, output_name()+"_"));
          else
            daughters_.push_back(new File(fullpath,output_name()+"_"));
          break;
        case SVN::unresolved:
          throw NodeException(fullpath+" is not up to date");
        case SVN::unversioned: ; // do nothing
        }
      }
  }


  Directory::~Directory(void)
  {
    for (NodeIterator i=daughters_.begin(); i!=daughters_.end(); i++)
      delete *i;
  }


  const Stats& Directory::parse(const bool verbose)
  {
    stats_.reset();

    // Directories give no contribution to statistics.
    for (NodeIterator i=daughters_.begin(); i!=daughters_.end(); i++)
      stats_ += (*i)->parse(verbose);
    return stats_;
  }


  void Directory::print(const bool verbose) const
  {
    std::string output(output_name() + ".html");
    if (verbose)
      std::cout << "Printing output for " << path_ << std::endl;
    std::ofstream os(output.c_str());
    print_header(os);
    os << "<p align=center>\n<img src='" << stats_.plot(output_name()+".png")
       << "' alt='[plot]' border=0><br>\n";
    os << "<table>\n";
    os << "<tr><td><strong>Node</strong></td>\n";
    os << "<td><strong>Count</strong></td></tr>\n";
    os << "<tr><td>Total</td>\n";
    os << "<td align=right>" << stats_.rows() << "</td></tr>\n";
    // print html links to daughter nodes
    transform(daughters_.begin(), daughters_.end(), 
              std::ostream_iterator<std::string>(os," "), 
              std::mem_fun(&Node::html_tablerow));
    os << "</table>\n";
    os << "</p>";
    print_footer(os);
    os.close(); 

    // print daughter nodes, i.e, this function is recursive
    std::for_each(daughters_.begin(), daughters_.end(),
                  std::bind2nd(std::mem_fun(&Node::print),verbose));
  }

}} // end of namespace svnstat and namespace theplu