(* This file is part of SKS.  SKS 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 *)

(* Copyright 2002, 2003, 2004 Yaron M. Minsky *)
(* Copyright 2004 Peter Palfrader *)
(***********************************************************************)

(** creates keydump consisting of recently added keys *)

open StdLabels
open MoreLabels
open Printf
open Common
open Packet
module Set = PSet.Set

let settings = {
  Keydb.withtxn = !Settings.transactions;
  Keydb.cache_bytes = !Settings.cache_bytes;
  Keydb.pagesize = !Settings.pagesize;
  Keydb.dbdir = !Settings.dbdir;
  Keydb.dumpdir = !Settings.dumpdir;
}

module Keydb = Keydb.Unsafe

let dump_database timestamp fname =
  let maxsize = 250_000 in
  let log = Keydb.reverse_logquery ~maxsize timestamp in
  if List.length log = 0 then
    printf "No changes since timestamp\n"
  else
    let file = open_out fname in
    let run () = 
      let newkeys = List.fold_left log ~init:Set.empty 
		      ~f:(fun set (_,change) -> match change with
			      Add hash -> Set.add hash set
			    | Delete hash -> Set.remove hash set)
      in
      printf "%d new keys in log.\n%!" (Set.cardinal newkeys);
      Set.iter newkeys 
	~f:(fun hash ->
	      try
		let keystring = Keydb.get_keystring_by_hash hash in
		output_string file keystring;
	      with
		  e -> 
		    eprintf "Error fetching keystring from hash %s: %s\n%!"
		    (Utils.hexstring hash)
		    (Printexc.to_string e)
	   )
    in
    protect ~f:run ~finally:(fun () -> close_out file)

let run () =
  List.iter !Settings.anonlist
    ~f:(fun x -> printf "\"%s\" " x);
  printf "\n%!";
  match !Settings.anonlist with
    | timestamp::tl ->
	let name = match tl with
	  | [] -> "incdump.pgp"
	  | [name] -> name
	  | _ -> raise (Argument_error "too many arguments")
	in
	printf "saving to file %s\n%!" name;
	set_logfile ".incdump";
	Keydb.open_dbs settings;
	protect ~f:(fun () -> 
		      let timestamp = float_of_string timestamp in
		      dump_database timestamp name )
	  ~finally:(fun () -> Keydb.close_dbs ())

    | _ ->
	raise (Argument_error "no timestamp provided")