helm/software/components/library/libraryDb.ml

   1 (* Copyright (C) 2004-2005, HELM Team.
   2  *
   3  * This file is part of HELM, an Hypertextual, Electronic
   4  * Library of Mathematics, developed at the Computer Science
   5  * Department, University of Bologna, Italy.
   6  *
   7  * HELM is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU General Public License
   9  * as published by the Free Software Foundation; either version 2
  10  * of the License, or (at your option) any later version.
  11  *
  12  * HELM is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with HELM; if not, write to the Free Software
  19  * Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  20  * MA  02111-1307, USA.
  21  *
  22  * For details, see the HELM World-Wide-Web page,
  23  * http://helm.cs.unibo.it/
  24  *)
  25
  26 (* $Id$ *)
  27
  28 open Printf ;;
  29
  30 let instance =
  31   let dbd = lazy (
  32     HSql.quick_connect
  33       ~host:(Helm_registry.get "db.host")
  34       ~user:(Helm_registry.get "db.user")
  35       ~database:(Helm_registry.get "db.database")
  36       ())
  37   in
  38   fun () -> Lazy.force dbd
  39
  40
  41 let xpointer_RE = Pcre.regexp "#.*$"
  42 let file_scheme_RE = Pcre.regexp "^file://"
  43
  44 let clean_owner_environment () =
  45   let dbd = instance () in
  46   let obj_tbl = MetadataTypes.obj_tbl () in
  47   let sort_tbl = MetadataTypes.sort_tbl () in
  48   let rel_tbl = MetadataTypes.rel_tbl () in
  49   let name_tbl =  MetadataTypes.name_tbl () in
  50   let count_tbl = MetadataTypes.count_tbl () in
  51   let tbls = [
  52     (obj_tbl,`RefObj) ; (sort_tbl,`RefSort) ; (rel_tbl,`RefRel) ;
  53     (name_tbl,`ObjectName) ; (count_tbl,`Count) ]
  54   in
  55   let statements =
  56     (SqlStatements.drop_tables tbls) @ (SqlStatements.drop_indexes tbls)
  57   in
  58   let owned_uris =
  59     try
  60       MetadataDb.clean ~dbd
  61     with (HSql.Error _) as exn ->
  62       match HSql.errno dbd with
  63       | HSql.No_such_table -> []
  64       | _ -> raise exn
  65   in
  66   List.iter
  67     (fun uri ->
  68       let uri = Pcre.replace ~rex:xpointer_RE ~templ:"" uri in
  69       List.iter
  70         (fun suffix ->
  71           try
  72            HExtlib.safe_remove
  73              (Http_getter.resolve ~writable:true (uri ^ suffix))
  74           with Http_getter_types.Key_not_found _ -> ())
  75         [""; ".body"; ".types"])
  76     owned_uris;
  77   List.iter (fun statement ->
  78     try
  79       ignore (HSql.exec dbd statement)
  80     with (HSql.Error _) as exn ->
  81       match HSql.errno dbd with
  82       | HSql.No_such_table
  83       | HSql.Bad_table_error
  84       | HSql.No_such_index -> prerr_endline statement; ()
  85       | _ -> raise exn
  86     ) statements;
  87 ;;
  88
  89 let create_owner_environment () =
  90   let dbd = instance () in
  91   let obj_tbl = MetadataTypes.obj_tbl () in
  92   let sort_tbl = MetadataTypes.sort_tbl () in
  93   let rel_tbl = MetadataTypes.rel_tbl () in
  94   let name_tbl =  MetadataTypes.name_tbl () in
  95   let count_tbl = MetadataTypes.count_tbl () in
  96   let l_obj_tbl = MetadataTypes.library_obj_tbl  in
  97   let l_sort_tbl = MetadataTypes.library_sort_tbl  in
  98   let l_rel_tbl = MetadataTypes.library_rel_tbl  in
  99   let l_name_tbl =  MetadataTypes.library_name_tbl  in
 100   let l_count_tbl = MetadataTypes.library_count_tbl  in
 101   let tbls = [
 102     (obj_tbl,`RefObj) ; (sort_tbl,`RefSort) ; (rel_tbl,`RefRel) ;
 103     (name_tbl,`ObjectName) ; (count_tbl,`Count) ]
 104   in
 105   let system_tbls = [
 106     (l_obj_tbl,`RefObj) ; (l_sort_tbl,`RefSort) ; (l_rel_tbl,`RefRel) ;
 107     (l_name_tbl,`ObjectName) ; (l_count_tbl,`Count) ]
 108   in
 109   let statements =
 110     (SqlStatements.create_tables system_tbls) @
 111     (SqlStatements.create_tables tbls) @
 112     (SqlStatements.create_indexes system_tbls) @
 113     (SqlStatements.create_indexes tbls)
 114   in
 115   List.iter (fun statement ->
 116     try
 117       ignore (HSql.exec dbd statement)
 118     with
 119       (HSql.Error _) as exc ->
 120       let status = HSql.errno dbd in
 121       match status with
 122       | HSql.Table_exists_error -> ()
 123       | HSql.Dup_keyname -> ()
 124       | HSql.GENERIC_ERROR _ ->
 125           prerr_endline statement;
 126           raise exc
 127       | _ -> ()
 128
 129
 130       ) statements
 131 ;;
 132
 133 (* removes uri from the ownerized tables, and returns the list of other objects
 134  * (theyr uris) that ref the one removed.
 135  * AFAIK there is no need to return it, since the MatitaTypes.staus should
 136  * contain all defined objects. but to double check we do not garbage the
 137  * metadata...
 138  *)
 139 let remove_uri uri =
 140   let obj_tbl = MetadataTypes.obj_tbl () in
 141   let sort_tbl = MetadataTypes.sort_tbl () in
 142   let rel_tbl = MetadataTypes.rel_tbl () in
 143   let name_tbl =  MetadataTypes.name_tbl () in
 144   (*let conclno_tbl = MetadataTypes.conclno_tbl () in
 145   let conclno_hyp_tbl = MetadataTypes.fullno_tbl () in*)
 146   let count_tbl = MetadataTypes.count_tbl () in
 147
 148   let dbd = instance () in
 149   let suri = UriManager.string_of_uri uri in
 150   let query table suri =
 151     if HSql.isMysql then
 152       sprintf "DELETE QUICK LOW_PRIORITY FROM %s WHERE source='%s'" table (HSql.escape suri)
 153     else sprintf "DELETE FROM %s WHERE source='%s'" table (HSql.escape suri)
 154   in
 155   List.iter (fun t ->
 156     try
 157       ignore (HSql.exec dbd (query t suri))
 158     with
 159       exn -> raise exn (* no errors should be accepted *)
 160     )
 161   [obj_tbl;sort_tbl;rel_tbl;name_tbl;(*conclno_tbl;conclno_hyp_tbl*)count_tbl];
 162 ;;
 163
 164 let xpointers_of_ind uri =
 165   let dbd = instance () in
 166   let name_tbl =  MetadataTypes.name_tbl () in
 167   let query = sprintf
 168     "SELECT source FROM %s WHERE source LIKE '%s#xpointer%%'" name_tbl
 169       (HSql.escape (UriManager.string_of_uri uri))
 170   in
 171   let rc = HSql.exec dbd query in
 172   let l = ref [] in
 173   HSql.iter rc (fun a ->  match a.(0) with None ->()|Some a -> l := a:: !l);
 174   List.map UriManager.uri_of_string !l
 175