// Copyright (C) 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 "vpwtable.h"

#include <stdlib.h>

#include <unistd.h>

#include "cdb++/cdb++.h"

class cdb_vpwtable_writer : public vpwtable_writer

{

private:

  const mystring& tmpname;

  const mystring& cdbname;

  cdb_writer out;

  bool opened;

public:

  cdb_vpwtable_writer(const mystring& filename);

  ~cdb_vpwtable_writer();

  bool operator!() const;

  bool put(const vpwentry& vpw);

  bool end();

  bool abort();

};

vpwtable_writer* vpwtable::start_write() const

{

  return new cdb_vpwtable_writer(filename);

}

cdb_vpwtable_writer::cdb_vpwtable_writer(const mystring& filename)

  : tmpname(filename + ".tmp"), cdbname(filename),

    out(filename, 0600), opened(true)

{

}

cdb_vpwtable_writer::~cdb_vpwtable_writer()

{

  end();

}

bool cdb_vpwtable_writer::operator!() const

{

  return opened;

}

bool cdb_vpwtable_writer::put(const vpwentry& vpw)

{

  return out.put(vpw.name.lower(), vpw.to_record());

}

bool cdb_vpwtable_writer::end() 

{

  if(!opened)

    return false;

  opened = false;

  return out.end(cdbname);

}

bool cdb_vpwtable_writer::abort()

{

  if(!opened)

    return false;

  opened = false;

  return unlink(tmpname.c_str()) == 0;

}