// Copyright (C) 1999,2000 Bruce Guenter <bruceg@em.ca>

//

// This program 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.

//

// This program 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 <config.h>

#include <stdlib.h>

#include "daemon.h"

#include "misc/pwentry.h"

#include "misc/lookup.h"

#include "misc/utoa.h"

#include "misc/stat_fns.h"

#include "ac/dirent.h"

struct count_size

{

  unsigned long count;

  unsigned long size;

  count_size() : count(0), size(0) { }

};

struct stats

{

  count_size unseen_new;

  count_size unseen;

  count_size seen;

  mystring rep() const;

};

mystring stats::rep() const

{

  return utoa(unseen_new.count) + mystring::NUL +

    utoa(unseen_new.size)  + mystring::NUL + 

    utoa(unseen.count) + mystring::NUL +

    utoa(unseen.size) + mystring::NUL +

    utoa(seen.count) + mystring::NUL +

    utoa(seen.size) + mystring::NUL;

}

class statdir

{

  mystring path;

  DIR* dir;

  struct dirent* curr;

  struct stat statbuf;

public:

  statdir(const mystring& dirname);

  ~statdir() { close(); }

  void close();

  operator void*() const { return (void*)dir; }

  bool operator!() const { return !dir; }

  const struct stat* operator->() const { return &statbuf; }

  const char* currname() const { return curr->d_name; }

  void operator++() { advance(); }

  void advance();

};

statdir::statdir(const mystring& dirname)

  : path(dirname),

    dir(opendir(dirname.c_str()))

{

  advance();

}

void statdir::close()

{

  if(dir)

    closedir(dir);

  dir = NULL;

}

void statdir::advance()

{

  if(dir) {

    while((curr = readdir(dir)) != 0) {

      if(curr->d_name[0] == '.')

	continue;

      break;

    }

  }

  if(!curr)

    close();

  else {

    mystring fullpath = path + "/" + curr->d_name;

    if(stat(fullpath.c_str(), &statbuf) == -1)

      close();

  }

}

bool stat_new_dir(const mystring& basename, stats& stats)

{

  statdir dir(basename + "/new");

  if(!dir)

    return false;

  while(dir) {

    if(S_ISREG(dir->st_mode)) {

      ++stats.unseen_new.count;

      stats.unseen_new.size += dir->st_blocks * 512;

    }

    ++dir;

  }

  return true;

}

bool stat_cur_dir(const mystring& basename, stats& stats)

{

  statdir dir(basename + "/cur");

  if(!dir)

    return false;

  while(dir) {

    if(S_ISREG(dir->st_mode)) {

      count_size* stat = &stats.unseen;

      const char* colon = strchr(dir.currname(), ':');

      if(colon) {

	++colon;

	if(*colon++ == '2' && *colon++ == ',' && !strchr(colon, 'S'))

	  stat = &stats.seen;

      }

      ++stat->count;

      stat->size += dir->st_blocks * 512;

    }

    ++dir;

  }

  return true;

}

bool stat_dir(const mystring& basename, stats& stats)

{

  return stat_new_dir(basename, stats) &&

    stat_cur_dir(basename, stats);

}

CMD(stat)

  // Usage: stat baseuser-virtuser pass

{

  mystring user = args[0];

  mystring pass = args[1];

  pwentry* pw;

  vpwentry* vpw;

  OK_RESPONSE(lookup_and_validate(user, pw, vpw, pass, true, false));

  if(!vpw->has_mailbox)

    RETURN(err, "User is alias");

  mystring dirname = pw->home + "/" + vpw->directory;

  stats stats;

  if(!stat_dir(dirname, stats))

    RETURN(err, "Failed to stat maildir");

  RETURN(ok, mystring::NUL + stats.rep());

}