helm/ocaml/getter/http_getter_map.ml

   1 (*
   2  * Copyright (C) 2003-2004:
   3  *    Stefano Zacchiroli <zack@cs.unibo.it>
   4  *    for the HELM Team http://helm.cs.unibo.it/
   5  *
   6  *  This file is part of HELM, an Hypertextual, Electronic
   7  *  Library of Mathematics, developed at the Computer Science
   8  *  Department, University of Bologna, Italy.
   9  *
  10  *  HELM is free software; you can redistribute it and/or
  11  *  modify it under the terms of the GNU General Public License
  12  *  as published by the Free Software Foundation; either version 2
  13  *  of the License, or (at your option) any later version.
  14  *
  15  *  HELM is distributed in the hope that it will be useful,
  16  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  *  GNU General Public License for more details.
  19  *
  20  *  You should have received a copy of the GNU General Public License
  21  *  along with HELM; if not, write to the Free Software
  22  *  Foundation, Inc., 59 Temple Place - Suite 330, Boston,
  23  *  MA  02111-1307, USA.
  24  *
  25  *  For details, see the HELM World-Wide-Web page,
  26  *  http://helm.cs.unibo.it/
  27  *)
  28
  29 open Http_getter_types
  30
  31 class map dbname =
  32   let perm = 420 in (* permission 644 in decimal notation *)
  33   let open_dbm () =
  34     Http_getter_misc.mkdir ~parents:true (Filename.dirname dbname);
  35     Dbm.opendbm dbname [ Dbm.Dbm_rdwr; Dbm.Dbm_create ] perm
  36   in
  37
  38   object (self)
  39
  40     inherit ThreadSafe.threadSafe
  41
  42     val mutable db = open_dbm ()
  43     val index_RE = Pcre.regexp "^theory:/.*/index.theory$"
  44
  45     (*initializer Gc.finalise Dbm.close db  (* close db on GC *)*)
  46
  47     (* Backward compatibility code to map
  48        theory:/path/t.theory into theory:/path/t/index.theory
  49        when cic:/path/t/ exists *)
  50     method private normalize_key key =
  51      if Pcre.pmatch ~rex:index_RE key &&
  52         (try ignore (Dbm.find db key); false with Not_found -> true)
  53      then
  54       (* we substitute /index.theory with .theory *)
  55       String.sub key 0 (String.length key - 13) ^ ".theory"
  56      else key
  57
  58     method add key value =
  59       self#doWriter (lazy (
  60         try
  61           Dbm.add db key value
  62         with Dbm.Dbm_error "Entry already exists" -> raise (Key_already_in key)
  63       ))
  64
  65     method replace key value =
  66       self#doWriter (lazy (
  67         Dbm.replace db key value
  68       ))
  69
  70     method remove key =
  71       self#doWriter (lazy (
  72         try
  73           Dbm.remove db key
  74         with Dbm.Dbm_error "dbm_delete" -> raise (Key_not_found key)
  75       ))
  76
  77     method resolve key =
  78       self#doReader (lazy (
  79         try
  80          Dbm.find db (self#normalize_key key)
  81         with Not_found -> raise (Key_not_found key)
  82       ))
  83
  84     method iter (f: string -> string -> unit) =
  85       self#doReader (lazy (
  86         Dbm.iter f db
  87       ))
  88
  89     method sync =
  90       self#doWriter (lazy (
  91         Dbm.close db;
  92         db <- open_dbm ()
  93       ))
  94
  95     method clear =
  96       self#doWriter (lazy (
  97         Dbm.close db;
  98         List.iter
  99           (fun ext ->
 100             let file = dbname ^ ext in
 101             if Sys.file_exists file then Sys.remove file)
 102           [".dir"; ".pag"; ".db"];
 103         db <- open_dbm ()
 104       ))
 105
 106     method close =
 107       self#doWriter (lazy (
 108         Dbm.close db
 109       ))
 110
 111   end
 112