Changes done on the Mac.
[cocanwiki.git] / scripts / page.ml
index d9b85ec..b6a4185 100644 (file)
@@ -1,7 +1,7 @@
 (* COCANWIKI - a wiki written in Objective CAML.
  * Written by Richard W.M. Jones <rich@merjis.com>.
  * Copyright (C) 2004 Merjis Ltd.
- * $Id: page.ml,v 1.21 2004/09/24 16:30:07 rich Exp $
+ * $Id: page.ml,v 1.44 2006/03/28 13:20:00 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
@@ -25,18 +25,18 @@ open Cgi
 open Printf
 
 open ExtString
+open ExtList
 
 open Cocanwiki
 open Cocanwiki_template
 open Cocanwiki_ok
 open Cocanwiki_date
 open Cocanwiki_server_settings
-
-(* Maximum level of redirection. *)
-let max_redirect = 4
+open Cocanwiki_links
 
 type fp_status = FPOK of int * string * string * Dbi.datetime * bool
-              | FPRedirect of string
+              | FPInternalRedirect of string
+              | FPExternalRedirect of string
               | FPNotFound
 
 (* Referer strings which help us decide if the user came from
@@ -49,33 +49,46 @@ let search_engines = [
 ]
 let split_words = Pcre.regexp "\\W+"
 
-let xhtml_re = Pcre.regexp "<.*?>|[^<>]+"
+let split_qs_re = Pcre.regexp "\\?"
 
-let run r (q : cgi) (dbh : Dbi.connection) hostid {edit_anon=edit_anon} user =
-  let template_page = get_template dbh hostid "page.html" in
-  let template_404  = get_template dbh hostid "page_404.html" in
+let xhtml_re = Pcre.regexp "<.*?>|[^<>]+"
 
+let run r (q : cgi) dbh hostid
+    ({ edit_anon = edit_anon; view_anon = view_anon } as host)
+    user =
   let page = q#param "page" in
   let page = if page = "" then "index" else page in
 
+  (* The main "page" template is split in two to improve the speed of
+   * delivery of the page.  The very first part ("page_header.html")
+   * contains the page <head>, crucially including all the links to the
+   * stylesheets.  We send this first and flush it out to the client so
+   * that the client can begin requesting stylesheets, background images
+   * and so on.  After this we compose the main page ("page.html") and
+   * send it out second.
+   *)
+
+  let template_page_header =
+    get_template ~page dbh hostid "page_header.html" in
+  let template_page = get_template ~page dbh hostid "page.html" in
+
+  (* This is the simpler template for 404 pages. *)
+  let template_404  = get_template dbh hostid "page_404.html" in
+
   (* Host-specific fields. *)
-  let sth = dbh#prepare_cached "select css is not null,
-                                       feedback_email is not null,
-                                       mailing_list
-                                  from hosts where id = ?" in
-  sth#execute [`Int hostid];
-  let has_host_css, has_feedback_email, mailing_list =
-    match sth#fetch1 () with
-      | [ `Bool has_host_css; `Bool has_feedback_email; `Bool mailing_list ] ->
-         has_host_css, has_feedback_email, mailing_list
+  let rows = PGSQL(dbh)
+    "select css is not null, feedback_email is not null, mailing_list, navigation
+       from hosts where id = $hostid" in
+  let has_host_css, has_feedback_email, mailing_list, navigation =
+    match rows with
+      | [Some has_host_css, Some has_feedback_email,
+        mailing_list, navigation] ->
+         has_host_css, has_feedback_email, mailing_list, navigation
       | _ -> assert false in
 
-  (* Can the user edit?  Manage users?  etc. *)
-  let can_edit = can_edit edit_anon user in
-  let can_manage_users = can_manage_users user in
-  let can_manage_contacts = can_manage_contacts user in
-  let can_manage_site = can_manage_site user in
-  let can_edit_global_css = can_edit_global_css user in
+  (* User permissions. *)
+  let can_edit = can_edit host user in
+  let can_manage_users = can_manage_users host user in
 
   (* Do we have a stats page set up? *)
   let has_stats = server_settings_stats_page dbh <> None in
@@ -86,7 +99,12 @@ let run r (q : cgi) (dbh : Dbi.connection) hostid {edit_anon=edit_anon} user =
   let search_terms_from_referer referer =
     let _, argnames =
       List.find (fun (rex, _) -> Pcre.pmatch ~rex referer) search_engines in
-    let _, _, args = Cocanwiki_cgi_args.parse referer in
+    let url, qs =
+      match Pcre.split ~rex:split_qs_re ~max:2 referer with
+       | [url] | [url;""] -> url, ""
+       | [url;qs] -> url, qs
+       | _ -> assert false in
+    let args = Cgi_args.parse qs in
     let argname =
       List.find (fun argname -> List.mem_assoc argname args) argnames in
     let search_string = List.assoc argname args in
@@ -145,90 +163,197 @@ let run r (q : cgi) (dbh : Dbi.connection) hostid {edit_anon=edit_anon} user =
     String.concat "" (List.concat [ head ; body ; tail ])
   in
 
+  (* Check the templates table for extensions. *)
+  let get_extension url =
+    try
+      let name =
+       List.hd (
+         PGSQL(dbh) "select extension from templates
+                       where $url ~ url_regexp
+                       order by ordering
+                       limit 1"
+       ) in
+      Some (List.assoc name !extensions)
+    with
+      Not_found | ExtList.List.Empty_list -> None
+  in
+
   (* This code generates ordinary pages. *)
   let make_page title description pageid last_modified_date has_page_css
-      version page page' =
+      version page page' extension =
     let t = template_page in
+    let th = template_page_header in
     t#set "title" title;
-    t#set "description" description;
-    t#set "pageid" (string_of_int pageid);
-    t#set "last_modified_date" (printable_date last_modified_date);
+    th#set "title" title;
+    t#set "last_modified_date" last_modified_date;
+
+    (match description with
+        None -> th#conditional "has_description" false
+       | Some description ->
+          th#conditional "has_description" true;
+          th#set "description" description);
 
     if page <> page' then (* redirection *) (
       t#set "page" page';
+      th#set "page" page';
       t#set "original_page" page; (* XXX title - get it from database *)
       t#conditional "redirected" true
     ) else (
       t#set "page" page;
+      th#set "page" page;
       t#conditional "redirected" false
     );
 
-    t#conditional "has_host_css" has_host_css;
-    t#conditional "has_page_css" has_page_css;
+    th#conditional "has_host_css" has_host_css;
+    th#conditional "has_page_css" has_page_css;
+
+    (* Are we showing an old version of the page?  If so, warn. *)
+    (match version with
+        None ->
+          t#conditional "is_old_version" false;
+          th#conditional "is_old_version" false
+       | Some pageid ->
+          t#conditional "is_old_version" true;
+          th#conditional "is_old_version" true;
+          t#set "old_version" (Int32.to_string pageid);
+          th#set "old_version" (Int32.to_string pageid));
+
+    (* At this point, we can print out the header and flush it back to
+     * the user, allowing the browser to start fetching stylesheets
+     * and background images while we compose the page.
+     *)
+    q#header ();
+    print_string r th#to_string;
+    Request.rflush r;
 
     t#conditional "has_feedback_email" has_feedback_email;
     t#conditional "mailing_list" mailing_list;
+    t#conditional "navigation" navigation;
 
     t#conditional "can_edit" can_edit;
     t#conditional "can_manage_users" can_manage_users;
-    t#conditional "can_manage_contacts" can_manage_contacts;
-    t#conditional "can_manage_site" can_manage_site;
-    t#conditional "can_edit_global_css" can_edit_global_css;
-
     t#conditional "has_stats" has_stats;
 
     (* Pull out the sections in this page. *)
-    let sth = dbh#prepare_cached
-               "select ordering, sectionname, content, divname
-                   from contents
-                  where pageid = ?
-                  order by ordering" in
-    sth#execute [`Int pageid];
-
     let sections =
-      sth#map
-       (function [`Int ordering;
-                  (`Null | `String _) as sectionname;
-                  `String content;
-                  (`Null | `String _) as divname] ->
-          let divname, has_divname =
-            match divname with
-                `Null -> "", false
-              | `String divname -> divname, true in
-          let sectionname, has_sectionname =
-            match sectionname with
-                `Null -> "", false
-              | `String sectionname -> sectionname, true in
-          let linkname = linkname_of_sectionname sectionname in
-          [ "ordering", Template.VarString (string_of_int ordering);
-            "has_sectionname", Template.VarConditional has_sectionname;
-            "sectionname", Template.VarString sectionname;
-            "linkname", Template.VarString linkname;
-            "content",
-              Template.VarString
-                (Wikilib.xhtml_of_content dbh hostid content);
-            "has_divname", Template.VarConditional has_divname;
-            "divname", Template.VarString divname ]
-          | _ -> assert false) in
+      match pageid with
+         None -> []
+       | Some pageid ->
+           let rows = PGSQL(dbh)
+             "select ordering, sectionname, content, divname
+                 from contents where pageid = $pageid order by ordering" in
+
+           List.map
+             (fun (ordering, sectionname, content, divname) ->
+                let divname, has_divname =
+                  match divname with
+                  | None -> "", false
+                  | Some divname -> divname, true in
+                let sectionname, has_sectionname =
+                  match sectionname with
+                  | None -> "", false
+                  | Some sectionname -> sectionname, true in
+                let linkname = linkname_of_sectionname sectionname in
+                [ "ordering", Template.VarString (Int32.to_string ordering);
+                  "has_sectionname",
+                    Template.VarConditional has_sectionname;
+                  "sectionname", Template.VarString sectionname;
+                  "linkname", Template.VarString linkname;
+                  "content",
+                    Template.VarString
+                      (Wikilib.xhtml_of_content dbh hostid content);
+                  "has_divname", Template.VarConditional has_divname;
+                  "divname", Template.VarString divname ]) rows in
+
+    (* Call an extension to generate the first section in this page? *)
+    let sections =
+      match extension with
+         None -> sections
+       | Some extension ->
+           let content = extension dbh hostid page' in
+           let section = [
+             "ordering", Template.VarString "0";
+             "has_sectionname", Template.VarConditional false;
+             "linkname", Template.VarString "";
+             "content", Template.VarString content;
+             "has_divname", Template.VarConditional true;
+             "divname", Template.VarString "form_div";
+           ] in
+           section :: sections in
 
     t#table "sections" sections;
 
-    (* Are we showing an old version of the page?  If so, warn. *)
-    (match version with
-        None ->
-          t#conditional "is_old_version" false
-       | Some pageid ->
-          t#conditional "is_old_version" true;
-          t#set "old_version" (string_of_int pageid));
-
     (* Login status. *)
     (match user with
         Anonymous ->
           t#conditional "user_logged_in" false
-       | User (_, username, _) ->
+       | User (_, username, _, _) ->
           t#conditional "user_logged_in" true;
           t#set "username" username);
 
+    (* If logged in, we want to update the recently_visited table. *)
+    if pageid <> None then (
+      match user with
+       | User (userid, _, _, _) ->
+           PGSQL(dbh)
+             "delete from recently_visited
+                where hostid = $hostid and userid = $userid and url = $page'";
+           PGSQL(dbh)
+             "insert into recently_visited (hostid, userid, url)
+               values ($hostid, $userid, $page')";
+           PGOCaml.commit dbh;
+           PGOCaml.begin_work dbh;
+       | _ -> ()
+    );
+
+    (* Navigation links. *)
+    if navigation then (
+      let max_links = 18 in            (* Show no more links than this. *)
+
+      (* What links here. *)
+      let wlh = what_links_here dbh hostid page' in
+      let wlh = List.take max_links wlh in
+      let wlh_urls = List.map fst wlh in (* Just the URLs ... *)
+
+      let rv =
+       match user with
+         | User (userid, _, _, _) ->
+             (* Recently visited URLs, but don't repeat any from the 'what
+              * links here' section, and don't link to self.
+              *)
+             let not_urls = page' :: wlh_urls in
+             let limit = Int32.of_int (max_links - List.length wlh_urls) in
+             let rows =
+               PGSQL(dbh)
+                 "select rv.url, p.title, rv.visit_time
+                     from recently_visited rv, pages p
+                    where rv.hostid = $hostid and rv.userid = $userid
+                      and rv.url not in $@not_urls
+                      and rv.hostid = p.hostid and rv.url = p.url
+                    order by 3 desc
+                    limit $limit" in
+             List.map (
+               fun (url, title, _) -> url, title
+             ) rows
+         | _ -> [] in
+
+      (* Links to page. *)
+      let f (page, title) = [ "page", Template.VarString page;
+                             "title", Template.VarString title ] in
+      let table = List.map f wlh in
+      t#table "what_links_here" table;
+      t#conditional "has_what_links_here" (wlh <> []);
+
+      let table = List.map f rv in
+      t#table "recently_visited" table;
+      t#conditional "has_recently_visited" (rv <> []);
+
+      (* If both lists are empty (ie. an empty navigation box would
+       * appear), then disable navigation altogether.
+       *)
+      if wlh = [] && rv = [] then t#conditional "navigation" false
+    );
+
     (* If we are coming from a search engine then we want to highlight
      * search terms throughout the whole page ...
      *)
@@ -241,12 +366,11 @@ let run r (q : cgi) (dbh : Dbi.connection) hostid {edit_anon=edit_anon} user =
       let xhtml = highlight_search_terms xhtml search_terms "search_term" in
 
       (* Deliver the page. *)
-      q#header ();
-      print_string r xhtml
+      ignore (print_string r xhtml)
     with
        Not_found ->
          (* No referer / no search terms / not a search engine referer. *)
-         q#template t
+         ignore (print_string r t#to_string)
   in
 
   (* This code generates 404 pages. *)
@@ -264,14 +388,8 @@ let run r (q : cgi) (dbh : Dbi.connection) hostid {edit_anon=edit_anon} user =
 
     t#set "search_terms" search_terms;
 
-    t#conditional "has_host_css" has_host_css;
-
     t#conditional "can_edit" can_edit;
     t#conditional "can_manage_users" can_manage_users;
-    t#conditional "can_manage_contacts" can_manage_contacts;
-    t#conditional "can_manage_site" can_manage_site;
-    t#conditional "can_edit_global_css" can_edit_global_css;
-
     t#conditional "has_stats" has_stats;
 
     q#template t
@@ -280,68 +398,59 @@ let run r (q : cgi) (dbh : Dbi.connection) hostid {edit_anon=edit_anon} user =
   (* Fetch a page by name.  This function can give three answers:
    * (1) Page fetched OK (fetches some details of the page).
    * (2) Page is a redirect (fetches the name of the redirect page).
-   * (3) Page not found in database, ie. 404 error.
-   *)
-  (* XXX Should do a case-insensitive matching of URLs, and if the URL differs
-   * in case only should redirect to the lowercase version.
+   * (3) Page not found in database, could be template or 404 error.
    *)
   let fetch_page page version allow_redirect =
     match version with
       | None ->
          if allow_redirect then (
-           let sth =
-             dbh#prepare_cached
-               "select redirect, id, title, description, last_modified_date,
-                        css is not null
-                   from pages where hostid = ? and url = ?" in
-           sth#execute [`Int hostid; `String page];
-           (try
-              (match sth#fetch1 () with
-                 | [ `Null; `Int id; `String title; `String description;
-                     `Timestamp last_modified_date; `Bool has_page_css ] ->
-                     FPOK (id, title, description, last_modified_date,
-                           has_page_css)
-                 | `String redirect :: _ ->
-                     FPRedirect redirect
-                 | _ -> assert false)
-            with
-                Not_found -> FPNotFound)
+           let rows = PGSQL(dbh)
+             "select url, redirect, id, title, description,
+                      last_modified_date, css is not null
+                 from pages
+                where hostid = $hostid and lower (url) = lower ($page)" in
+           match rows with
+           | [page', _, _, _, _, _, _]
+               when page <> page' -> (* different case *)
+               FPExternalRedirect page'
+           | [ _, None, id, title, description,
+               last_modified_date, has_page_css ] ->
+               let has_page_css = Option.get has_page_css in
+               FPOK (id, title, description, last_modified_date,
+                     has_page_css)
+           | [_, Some redirect, _, _, _, _, _] ->
+               FPInternalRedirect redirect
+           | [] -> FPNotFound
+           | _ -> assert false
          ) else (* redirects not allowed ... *) (
-           let sth =
-             dbh#prepare_cached
-               "select id, title, description, last_modified_date,
-                        css is not null
-                   from pages where hostid = ? and url = ?" in
-           sth#execute [`Int hostid; `String page];
-           (try
-              (match sth#fetch1 () with
-                 | [ `Int id; `String title; `String description;
-                     `Timestamp last_modified_date; `Bool has_page_css ] ->
-                     FPOK (id, title, description, last_modified_date,
-                           has_page_css)
-                 | _ -> assert false)
-            with
-                Not_found -> FPNotFound)
+           let rows = PGSQL(dbh)
+             "select id, title, description, last_modified_date,
+                      css is not null
+                 from pages where hostid = $hostid and url = $page" in
+           match rows with
+           | [ id, title, description,
+               last_modified_date, has_page_css ] ->
+               let has_page_css = Option.get has_page_css in
+               FPOK (id, title, description, last_modified_date,
+                     has_page_css)
+           | [] -> FPNotFound
+           | _ -> assert false
          )
       | Some version ->
-           let sth =
-             dbh#prepare_cached
-               "select id, title, description, last_modified_date,
-                        css is not null
-                   from pages
-                  where hostid = ? and id = ? and
-                        (url = ? or url_deleted = ?)" in
-           sth#execute [`Int hostid; `Int version;
-                        `String page; `String page];
-           (try
-              (match sth#fetch1 () with
-                 | [ `Int id; `String title; `String description;
-                     `Timestamp last_modified_date; `Bool has_page_css ] ->
-                     FPOK (id, title, description, last_modified_date,
-                           has_page_css)
-                 | _ -> assert false)
-            with
-                Not_found -> FPNotFound)
+         let rows = PGSQL(dbh)
+           "select id, title, description, last_modified_date,
+                    css is not null
+               from pages
+              where hostid = $hostid and id = $version and
+                    (url = $page or url_deleted = $page)" in
+         match rows with
+         | [ id, title, description,
+             last_modified_date, has_page_css ] ->
+             let has_page_css = Option.get has_page_css in
+             FPOK (id, title, description, last_modified_date,
+                   has_page_css)
+         | [] -> FPNotFound
+         | _ -> assert false
   in
 
   (* Here we deal with the complex business of redirects and versions. *)
@@ -349,27 +458,46 @@ let run r (q : cgi) (dbh : Dbi.connection) hostid {edit_anon=edit_anon} user =
   let allow_redirect, version =
     if can_edit then (
       not (q#param_true "no_redirect"),
-      try Some (int_of_string (q#param "version")) with Not_found -> None
+      try Some (Int32.of_string (q#param "version")) with Not_found -> None
     ) else
       (true, None) in
 
   let rec loop page' i =
     if i > max_redirect then (
       error ~title:"Too many redirections" ~back_button:true
-        q ("Too many redirects between pages.  This may happen because " ^
-          "of a cycle of redirections.");
+        dbh hostid q
+       ("Too many redirects between pages.  This may happen because " ^
+        "of a cycle of redirections.");
       return ()
     ) else
       match fetch_page page' version allow_redirect with
        | FPOK (pageid, title, description, last_modified_date, has_page_css)->
-           make_page title description pageid last_modified_date has_page_css
-             version page page'
-       | FPRedirect page' ->
+           (* Check if the page is also a template. *)
+           let extension = get_extension page' in
+           make_page title (Some description) (Some pageid)
+             (printable_date last_modified_date) has_page_css
+             version page page' extension
+       | FPInternalRedirect page' ->
            loop page' (i+1)
+       | FPExternalRedirect page' ->
+           (* This normally happens when a user has request an uppercase
+            * page name.  We redirect to the true (lowercase) version.
+            *)
+           q#redirect ("http://" ^ host.hostname ^ "/" ^ page');
+           return ()
        | FPNotFound ->
-           make_404 ()
+           (* Might be a templated page with no content in it. *)
+           let extension = get_extension page' in
+           (match extension with
+              | (Some _) as extension ->
+                  let title = page' in
+                  make_page title None None
+                    "Now" false None page page'
+                    extension
+              | None ->
+                  make_404 ())
   in
   loop page 0
 
 let () =
-  register_script run
+  register_script ~restrict:[CanView] run