// 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 <errno.h>

#include "vpwtable.h"

#include <gdbm.h>

class gdbm_vpwtable_writer : public vpwtable_writer

{

private:

  const mystring& tmpname;

  const mystring& destname;

  GDBM_FILE out;

  bool opened;

public:

  gdbm_vpwtable_writer(const mystring& filename);

  ~gdbm_vpwtable_writer();

  bool operator!() const;

  bool put(const vpwentry& vpw);

  bool end();

  bool abort();

};

vpwtable_writer* vpwtable::start_write() const

{

  return new gdbm_vpwtable_writer(filename);

}

gdbm_vpwtable_writer::gdbm_vpwtable_writer(const mystring& filename)

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

    out(gdbm_open((char*)filename.c_str(), 0, GDBM_NEWDB|GDBM_FAST, 0600, 0)),

    opened(true)

{

}

gdbm_vpwtable_writer::~gdbm_vpwtable_writer()

{

  end();

}

bool gdbm_vpwtable_writer::operator!() const

{

  return opened;

}

bool gdbm_vpwtable_writer::put(const vpwentry& vpw)

{

  mystring name = vpw->name.lower();

  datum key = { (char*)name.c_str(), name.length() };

  mystring binary = vpw->to_record();

  datum data = { (char*)binary.c_str(), binary.length() };

  return gdbm_store(out, key, data, GDBM_INSERT) == 0;

}

bool gdbm_vpwtable_writer::end() 

{

  if(!opened)

    return false;

  gdbm_sync(out);

  opened = false;

  return gdbm_close(out) == 0 &&

    rename(tmpname.c_str(), destname.c_str()) == 0;

}

bool gdbm_vpwtable_writer::abort()

{

  if(!opened)

    return false;

  opened = false;

  gdbm_close(out);

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

}