]> matita.cs.unibo.it Git - helm.git/blob - components/hmysql/hSqlite3.ml
fixed an escaping error, added more infos to the generic error, callback catches...
[helm.git] / components / hmysql / hSqlite3.ml
1 (* Copyright (C) 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://cs.unibo.it/helm/.
24  *)
25
26 (* $Id: hMysql.ml 5769 2006-01-08 18:00:22Z sacerdot $ *)
27
28 (*
29 type result = Mysql.result option
30 *)
31
32 type result = Sqlite3.row list
33 type dbd = Sqlite3.db option
34
35 type error_code = 
36  | OK
37  | Table_exists_error
38  | Dup_keyname
39  | No_such_table
40  | No_such_index
41  | Bad_table_error
42  | GENERIC_ERROR of string
43
44 exception Error of string
45
46 let prerr_endline s = () (*HLog.debug s*);;
47
48 let profiler = HExtlib.profile "Sqlite3"
49
50 let quick_connect 
51   is_library
52   ?(host = Helm_registry.get "matita.basedir") 
53   ?(database = "sqlite") ?port ?password ?user () 
54
55   let username = Helm_registry.get "user.name" in
56   let host_mangled = Pcre.replace ~pat:"[/ .]" ~templ:"_" host in
57   let tmp_db_name = 
58     "/dev/shm/matita.db." ^ username ^ "." ^ host_mangled ^ "." ^
59     string_of_int (Unix.getpid ())
60   in
61   let db_name = host ^ "/" ^ database in
62   let db_to_open =
63     if HExtlib.is_dir "/dev/shm/" && HExtlib.writable_dir "/dev/shm/" 
64        && (not is_library) 
65     then
66       (let cp_to_ram_cmd = "cp " ^ db_name ^ " " ^ tmp_db_name in
67       ignore (Sys.command cp_to_ram_cmd);
68       let mv_to_disk_cmd _ = 
69         ignore (Sys.command ("mv " ^ tmp_db_name ^ " " ^ db_name)) 
70       in
71       at_exit mv_to_disk_cmd;
72       tmp_db_name)
73     else
74       db_name
75   in
76   let db = Sqlite3.db_open db_to_open in
77   (* attach the REGEX function *)
78   Sqlite3.create_fun2 db "REGEXP"
79       (fun s rex -> 
80         try
81          match rex, s with
82          | Sqlite3.Data.TEXT rex, Sqlite3.Data.BLOB s
83          | Sqlite3.Data.TEXT rex, Sqlite3.Data.TEXT s ->
84              let r = Str.regexp rex in
85              if Str.string_match r s 0 then 
86                Sqlite3.Data.INT 1L 
87              else 
88                Sqlite3.Data.INT 0L
89          | _ -> raise (Error "wrong types to 'REGEXP'")
90         with exn -> HLog.error (Printexc.to_string exn); raise exn);
91   Some db
92 ;;
93
94 let disconnect db =
95   match db with
96   | None -> ()
97   | Some db -> 
98       let b = Sqlite3.db_close db in
99       if b=false then prerr_endline "No Closed DataBase"
100 ;;
101
102 (* XXX hack, sqlite has a print "%q" that should be used, but is not bound *)
103 let escape s = 
104   let s_escaped = Pcre.replace ~pat:"'" ~templ:"''" s in
105   (*let s_escaped = Pcre.replace ~pat:"([^'])'([^'])" ~templ:"$1''$2" s in*)
106   prerr_endline s;
107   prerr_endline s_escaped;
108   s_escaped
109 ;;
110
111 let string_of_rc = function
112   |Sqlite3.Rc.OK -> "Sqlite3.Rc.OK" 
113   |Sqlite3.Rc.ERROR -> "Sqlite3.Rc.ERROR" 
114   |Sqlite3.Rc.INTERNAL -> "Sqlite3.Rc.INTERNAL" 
115   |Sqlite3.Rc.PERM -> "Sqlite3.Rc.PERM" 
116   |Sqlite3.Rc.ABORT -> "Sqlite3.Rc.ABORT" 
117   |Sqlite3.Rc.BUSY -> "Sqlite3.Rc.BUSY" 
118   |Sqlite3.Rc.LOCKED -> "Sqlite3.Rc.LOCKED" 
119   |Sqlite3.Rc.NOMEM -> "Sqlite3.Rc.NOMEM" 
120   |Sqlite3.Rc.READONLY -> "Sqlite3.Rc.READONLY" 
121   |Sqlite3.Rc.INTERRUPT -> "Sqlite3.Rc.INTERRUPT" 
122   |Sqlite3.Rc.IOERR -> "Sqlite3.Rc.IOERR" 
123   |Sqlite3.Rc.CORRUPT -> "Sqlite3.Rc.CORRUPT" 
124   |Sqlite3.Rc.NOTFOUND -> "Sqlite3.Rc.NOTFOUND" 
125   |Sqlite3.Rc.FULL -> "Sqlite3.Rc.FULL" 
126   |Sqlite3.Rc.CANTOPEN -> "Sqlite3.Rc.CANTOPEN" 
127   |Sqlite3.Rc.PROTOCOL -> "Sqlite3.Rc.PROTOCOL" 
128   |Sqlite3.Rc.EMPTY -> "Sqlite3.Rc.EMPTY" 
129   |Sqlite3.Rc.SCHEMA -> "Sqlite3.Rc.SCHEMA" 
130   |Sqlite3.Rc.TOOBIG -> "Sqlite3.Rc.TOOBIG" 
131   |Sqlite3.Rc.CONSTRAINT -> "Sqlite3.Rc.CONSTRAINT" 
132   |Sqlite3.Rc.MISMATCH -> "Sqlite3.Rc.MISMATCH" 
133   |Sqlite3.Rc.MISUSE -> "Sqlite3.Rc.MISUSE" 
134   |Sqlite3.Rc.NOFLS -> "Sqlite3.Rc.NOFLS" 
135   |Sqlite3.Rc.AUTH -> "Sqlite3.Rc.AUTH" 
136   |Sqlite3.Rc.FORMAT -> "Sqlite3.Rc.FORMAT" 
137   |Sqlite3.Rc.RANGE -> "Sqlite3.Rc.RANGE" 
138   |Sqlite3.Rc.NOTADB -> "Sqlite3.Rc.NOTADB" 
139   |Sqlite3.Rc.ROW -> "Sqlite3.Rc.ROW" 
140   |Sqlite3.Rc.DONE -> "Sqlite3.Rc.DONE" 
141   |Sqlite3.Rc.UNKNOWN n -> 
142     "Sqlite3.Rc.UNKNOWN " ^ string_of_int (Sqlite3.Rc.int_of_unknown n)
143 ;;
144
145 let pp_rc rc = prerr_endline (string_of_rc rc);;
146
147 let exec s db =
148  prerr_endline s;
149   let stored_result = ref [] in
150   let store row =
151     stored_result := row :: !stored_result
152   in
153   match db with
154   | None -> [] 
155   | Some db ->  
156       let rc = 
157         profiler.HExtlib.profile (Sqlite3.exec_no_headers db ~cb:store) s 
158       in
159       match rc with
160       | Sqlite3.Rc.OK -> !stored_result
161       | _ -> raise (Error (string_of_rc rc ^ ": " ^ Sqlite3.errmsg db ))
162 ;;
163
164 let rec map res ~f = 
165   let map f = List.map f res in
166   profiler.HExtlib.profile map f
167 ;;
168
169 let iter res ~f =
170   let iter f = List.iter f res in
171   profiler.HExtlib.profile iter f
172 ;;
173
174 let errno = function 
175   | None -> OK
176   | Some db -> 
177       match Sqlite3.errcode db with
178       |Sqlite3.Rc.OK -> OK
179       |Sqlite3.Rc.ERROR ->
180          let errmsg = (Sqlite3.errmsg db) in
181          if Pcre.pmatch errmsg ~pat:"^table .* already exists" then
182            Table_exists_error
183          else
184          if Pcre.pmatch errmsg ~pat:"^index .* already exists" then Dup_keyname
185          else if Pcre.pmatch errmsg ~pat:"^no such table: .*" then No_such_table
186          else if Pcre.pmatch errmsg ~pat:"^no such index: .*" then No_such_index
187          else GENERIC_ERROR errmsg
188       |Sqlite3.Rc.INTERNAL |Sqlite3.Rc.PERM |Sqlite3.Rc.ABORT
189       |Sqlite3.Rc.BUSY |Sqlite3.Rc.LOCKED |Sqlite3.Rc.NOMEM
190       |Sqlite3.Rc.READONLY |Sqlite3.Rc.INTERRUPT |Sqlite3.Rc.IOERR
191       |Sqlite3.Rc.CORRUPT |Sqlite3.Rc.NOTFOUND |Sqlite3.Rc.FULL
192       |Sqlite3.Rc.CANTOPEN |Sqlite3.Rc.PROTOCOL |Sqlite3.Rc.EMPTY 
193       |Sqlite3.Rc.SCHEMA |Sqlite3.Rc.TOOBIG |Sqlite3.Rc.CONSTRAINT
194       |Sqlite3.Rc.MISMATCH |Sqlite3.Rc.MISUSE |Sqlite3.Rc.NOFLS
195       |Sqlite3.Rc.AUTH |Sqlite3.Rc.FORMAT |Sqlite3.Rc.RANGE
196       |Sqlite3.Rc.NOTADB |Sqlite3.Rc.ROW |Sqlite3.Rc.DONE
197       |Sqlite3.Rc.UNKNOWN _ -> GENERIC_ERROR "Sqlite3_generic_error"
198 ;;
199
200 let isMysql = false
201
202 let escape_string_for_like = ("ESCAPE \"\\\"" : ('a,'b,'c,'a) format4);;