source: code/trunk/cmd/txt/conversion.ml@ 75

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 = []
}