Don't forget the date\!

[cocanwiki.git] / scripts / files.ml

diff --git a/scripts/files.ml b/scripts/files.ml
index f9b7dee..c99d6df 100644 (file)
--- a/scripts/files.ml
+++ b/scripts/files.ml
@@ -1,7 +1,22 @@
-(* COCANWIKI scripts.
+(* COCANWIKI - a wiki written in Objective CAML.
  * Written by Richard W.M. Jones <rich@merjis.com>.
  * Copyright (C) 2004 Merjis Ltd.
- * $Id: files.ml,v 1.1 2004/09/07 10:14:09 rich Exp $
+ * $Id: files.ml,v 1.10 2006/12/06 09:46:57 rich Exp $
+ *
+ * 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; see the file COPYING.  If not, write to
+ * the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ * Boston, MA 02111-1307, USA.
  *)
 
 open Apache
@@ -10,43 +25,48 @@ open Cgi
 open Printf
 
 open Cocanwiki
+open Cocanwiki_template
 
-open Merjisforwiki
+let run r (q : cgi) dbh hostid _ _ =
+  let template = get_template r dbh hostid "files.html" in
 
-let template = get_template "files.html"
-
-let run r (q : cgi) (dbh : Dbi.connection) (hostid, _) _ =
   let deleted = q#param_true "deleted" in
   template#conditional "deleted" deleted;
 
-  let sql =
-    "select id, name, name_deleted, octet_length (content)
-       from files
-      where hostid = ? and " ^
-    (if not deleted then "name is not null"
-     else "name_deleted is not null") ^
-    " order by 2, 3" in
-  let sth = dbh#prepare_cached sql in
-  sth#execute [`Int hostid];
+  let rows =
+    if not deleted then
+      PGSQL(dbh)
+       "select id, name, name_deleted, octet_length (content)
+           from files
+          where hostid = $hostid and name is not null
+          order by 2, 3"
+    else
+      PGSQL(dbh)
+       "select id, name, name_deleted, octet_length (content)
+           from files
+          where hostid = $hostid and name_deleted is not null
+          order by 2, 3" in
 
   let table =
-    sth#map
+    List.map
       (fun row ->
         let id, name, size, is_deleted =
           match row with
-            | [`Int id; `String name; `Null; `Int size] ->
+            | (id, Some name, None, size) ->
                 id, name, size, false
-            | [`Int id; `Null; `String name; `Int size] ->
+            | (id, None, Some name, size) ->
                 id, name, size, true
             | _ -> assert false in
-        [ "id", Template.VarString (string_of_int id);
+        let size = Int32.to_int (Option.get size) in
+        [ "id", Template.VarString (Int32.to_string id);
           "name", Template.VarString name;
           "ksize", Template.VarString (string_of_int (size / 1024));
-          "is_deleted", Template.VarConditional is_deleted ]) in
+          "is_deleted", Template.VarConditional is_deleted ]
+      ) rows in
 
   template#table "files" table;
 
   q#template template
 
 let () =
-  register_script run
+  register_script ~restrict:[CanEdit] run