scripts/undelete_file.ml

   1 (* COCANWIKI - a wiki written in Objective CAML.
   2  * Written by Richard W.M. Jones <rich@merjis.com>.
   3  * Copyright (C) 2004 Merjis Ltd.
   4  * $Id: undelete_file.ml,v 1.9 2006/03/28 16:24:08 rich Exp $
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; see the file COPYING.  If not, write to
  18  * the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  19  * Boston, MA 02111-1307, USA.
  20  *)
  21
  22 open Apache
  23 open Registry
  24 open Cgi
  25 open Printf
  26
  27 open Cocanwiki
  28 open Cocanwiki_template
  29 open Cocanwiki_ok
  30
  31 let run r (q : cgi) dbh hostid { hostname = hostname } _ =
  32   let id = Int32.of_string (q#param "id") in
  33
  34   if q#param_true "yes" then (
  35     (* Get the name of the file. *)
  36     let name = Option.get (
  37       List.hd (
  38         PGSQL(dbh)
  39           "select name_deleted from files
  40             where hostid = $hostid and id = $id"
  41       )
  42     ) in
  43
  44     (* First delete any more recent versions of this file. *)
  45     PGSQL(dbh) "update files
  46                    set name_deleted = name, name = null
  47                  where hostid = $hostid and name = $name";
  48
  49     (* Now copy the old row, changing name_deleted back to name so the file
  50      * becomes live.
  51      *)
  52     PGSQL(dbh) "insert into files
  53                   (hostid, name, content, title, mime_type, upload_date)
  54                 select hostid, name_deleted, content,
  55                        title, mime_type, upload_date
  56                   from files
  57                  where hostid = $hostid and id = $id";
  58
  59     PGOCaml.commit dbh;
  60
  61     (* Done. *)
  62     let buttons = [ ok_button "/_files" ] in
  63     ok ~title:"File restored" ~buttons
  64       dbh hostid q "File was restored successfully."
  65   ) else
  66     q#redirect ("http://" ^ hostname ^ "/_files")
  67
  68 let () =
  69   register_script ~restrict:[CanEdit] run