758d1d6d47
Signed-off-by: Izuru Yakumo <yakumo.izuru@chaotic.ninja> git-svn-id: file:///srv/svn/repo/kosuzu/trunk@73 eb64cd80-c68d-6f47-b6a3-0ada418499da
75 lines
2.3 KiB
OCaml
75 lines
2.3 KiB
OCaml
open Kosuzu
|
|
|
|
module Rel = struct
|
|
|
|
module Rel_set = Set.Make(String)
|
|
module Id_map = Map.Make(String)
|
|
|
|
type t = { last_rel: string; ref_set: String_set.t; rep_set: String_set.t }
|
|
type map_t = t Id_map.t
|
|
|
|
let empty = { last_rel = ""; ref_set = Rel_set.empty; rep_set = Rel_set.empty }
|
|
let empty_map = Id_map.empty
|
|
|
|
let acc_ref date source target = Id_map.update target (function
|
|
| None -> Some { last_rel = date;
|
|
ref_set = Rel_set.singleton source;
|
|
rep_set = Rel_set.empty }
|
|
| Some rel -> Some { rel with
|
|
last_rel = if Date.compare date rel.last_rel > 0 then date else rel.last_rel;
|
|
ref_set = Rel_set.add source rel.ref_set })
|
|
|
|
let acc_rep date source target = Id_map.update target (function
|
|
| None -> Some { last_rel = date;
|
|
rep_set = Rel_set.singleton source;
|
|
ref_set = Rel_set.empty }
|
|
| Some rel -> Some { rel with
|
|
last_rel = if Date.compare date rel.last_rel > 0 then date else rel.last_rel;
|
|
rep_set = Rel_set.add source rel.rep_set })
|
|
|
|
let acc_txt rels (text, _paths) =
|
|
let acc_ref = acc_ref (Date.listing text.Text.date) text.Text.id in
|
|
let acc_rep = acc_rep (Date.listing text.Text.date) text.Text.id in
|
|
let rels = String_set.fold acc_ref (Text.set "references" text) rels in
|
|
let rels = String_set.fold acc_rep (Text.set "in-reply-to" text) rels in
|
|
rels
|
|
|
|
let acc_pck rels peer =
|
|
let path = try List.hd peer.Peers.pack.Header_pack.info.locations with Failure _->"" in
|
|
try Header_pack.fold
|
|
(fun rels id t _title _authors _topics refs_ls reps_ls ->
|
|
let acc_ref = acc_ref (Date.of_secs @@ Int32.to_int t) (Filename.concat path id) in
|
|
let acc_rep = acc_rep (Date.of_secs @@ Int32.to_int t) (Filename.concat path id) in
|
|
let rels = String_set.fold acc_ref (String_set.of_list refs_ls) rels in
|
|
let rels = String_set.fold acc_rep (String_set.of_list reps_ls) rels in
|
|
rels)
|
|
rels peer.Peers.pack
|
|
with e -> prerr_endline "acc_pck"; raise e
|
|
end
|
|
|
|
|
|
type t = {
|
|
id: string;
|
|
dir: string;
|
|
kv: string Store.KV.t;
|
|
topic_roots: string list;
|
|
topics: (String_set.t * String_set.t) Topic_set.Map.t;
|
|
relations: Rel.map_t;
|
|
texts: Text.t list
|
|
}
|
|
|
|
type fn_t = {
|
|
ext: string;
|
|
page: (t -> Kosuzu.Text.t -> string) option;
|
|
indices: (t -> unit) option;
|
|
}
|
|
|
|
let empty () = {
|
|
id = ""; dir = "";
|
|
kv = Store.KV.empty;
|
|
topic_roots = [];
|
|
topics = Topic_set.Map.empty;
|
|
relations = Rel.Id_map.empty;
|
|
texts = []
|
|
}
|