(* COCANWIKI scripts.
 * Written by Richard W.M. Jones <rich@merjis.com>.
 * Copyright (C) 2004 Merjis Ltd.
 * $Id: undelete_file.ml,v 1.3 2004/09/07 14:58:34 rich Exp $
 *)

open Apache
open Registry
open Cgi
open Printf

open Cocanwiki
open Cocanwiki_template
open Cocanwiki_ok

let run r (q : cgi) (dbh : Dbi.connection) (hostid, hostname, _) _ =
  let id = int_of_string (q#param "id") in

  if q#param_true "yes" then (
    (* Get the name of the file. *)
    let sth = dbh#prepare_cached "select name_deleted from files
                                   where hostid = ? and id = ?" in
    sth#execute [`Int hostid; `Int id];

    let name = sth#fetch1string () in

    (* First delete any more recent versions of this file. *)
    let sth = dbh#prepare_cached "update files
                                     set name_deleted = name, name = null
                                   where hostid = ? and name = ?" in
    sth#execute [`Int hostid; `String name];

    (* Now copy the old row, changing name_deleted back to name so the file
     * becomes live.
     *)
    let sth = dbh#prepare_cached "insert into files
                                  (hostid, name, content, title, mime_type,
                                   upload_date)
                                  select hostid, name_deleted, content,
                                         title, mime_type, upload_date
                                    from files
                                   where hostid = ? and id = ?" in
    sth#execute [`Int hostid; `Int id];

    dbh#commit ();

    (* Done. *)
    let buttons = [ ok_button "/_files" ] in
    ok ~title:"File restored" ~buttons
      q "File was restored successfully."
  ) else
    q#redirect ("http://" ^ hostname ^ "/_files")

let () =
  register_script run