// 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 <string.h>
#include "cdb++/cdb++.h"
#include "cli/cli.h"
#include "config/configrc.h"
#include "fdbuf/fdbuf.h"
#include "mystring/mystring.h"
#include "vpwentry/vpwentry.h"
#include "vcommand.h"

const char* cli_program = "vpasswd2cdb";
const char* cli_help_prefix = "Converts text password tables to CDB format\n";
const char* cli_help_suffix =
"Reads in a standard virtual password table in the current directory,
and writes it out to a CDB table.  The file names for the text and CDB
tables are determined from the configuration file.";
const char* cli_args_usage = "";
const int cli_args_min = 0;
const int cli_args_max = 0;

static int o_quiet = false;

cli_option cli_options[] = {
  { 0, "quiet", cli_option::flag, true, &o_quiet,
    "Suppress all status messages", 0 },
  {0}
};

static bool getpw(fdibuf& in, vpwentry& out)
{
  mystring buf;
  if(!in.getline(buf))
    return false;

  int first = buf.find_first(':');
  if(first < 0) return false;

  return out.from_record(buf.left(first), buf.c_str() + first);
}

int cli_main(int, char* [])
{
  if(!go_home())
    return 1;

  fdibuf in(password_file.c_str());
  if(!in) {
    if(!o_quiet)
      ferr << "Can't open password table named '"
	   << password_file << "'." << endl;
    return 1;
  }
  mystring cdbname = password_file + ".cdb";
  mystring cdbtmp = cdbname + ".tmp";
  cdb_writer cdb(cdbtmp, 0600);
  if(!cdb) {
    if(!o_quiet)
      ferr << "Can't open CDB temporary file named '" << cdbtmp << "'."
	   << endl;
    return 1;
  }
  vpwentry vpw;
  while(getpw(in, vpw)) {
    if(!cdb.put(vpw.name, vpw.to_record())) {
      if(!o_quiet)
	ferr << "Failed to add record to CDB table." << endl;
      return 1;
    }
  }
  if(!cdb.end(cdbname)) {
    if(!o_quiet)
      ferr << "Failed to finish CDB table into '"
	   << password_file << ".cdb'." << endl;
    return 1;
  }
  return 0;
}