103 lines
3.7 KiB
OCaml
103 lines
3.7 KiB
OCaml
module String_map = Map.Make (String)
|
|
type t = {
|
|
id: Id.t;
|
|
title: string;
|
|
authors: Person.Set.t;
|
|
date: Date.t;
|
|
string_map: string String_map.t;
|
|
stringset_map: String_set.t String_map.t;
|
|
body: string;
|
|
}
|
|
|
|
let blank ?(id=(Id.generate ())) () = {
|
|
id;
|
|
title = "";
|
|
authors = Person.Set.empty;
|
|
date = Date.({ created = now (); edited = ""});
|
|
string_map = String_map.empty;
|
|
stringset_map = String_map.empty;
|
|
body = "";
|
|
}
|
|
|
|
let compare = Stdlib.compare
|
|
let newest a b = Date.(compare a.date b.date)
|
|
let oldest a b = Date.(compare b.date a.date)
|
|
let str key m = try String_map.find (String.lowercase_ascii key) m.string_map with Not_found -> ""
|
|
let set key m = try String_map.find (String.lowercase_ascii key) m.stringset_map with Not_found -> String_set.empty
|
|
let str_set key m = String_set.to_string @@ set key m
|
|
let with_str_set m key str = { m with stringset_map = String_map.add (String.lowercase_ascii key) (String_set.of_string str) m.stringset_map }
|
|
|
|
let with_kv x (k,v) =
|
|
let trim = String.trim in
|
|
match String.lowercase_ascii k with
|
|
| "body" -> { x with body = String.trim v }
|
|
| "title"-> { x with title = trim v }
|
|
| "id" -> (match v with "" -> x | s -> { x with id = s })
|
|
| "author"
|
|
| "authors" -> { x with authors = Person.Set.of_string (trim v)}
|
|
| "date" -> { x with date = Date.{ x.date with created = Date.of_string v }}
|
|
| "date-edited"-> { x with date = Date.{ x.date with edited = Date.of_string v }}
|
|
| "licences" | "topics" | "keywords" | "series" as k -> with_str_set x k v
|
|
| k -> { x with string_map = String_map.add k (trim v) x.string_map }
|
|
|
|
let kv_of_string line = match Str.(bounded_split (regexp ": *")) line 2 with
|
|
| [ key; value ] -> Str.(replace_first (regexp "^#\\+") "" key), value
|
|
| [ key ] -> Str.(replace_first (regexp "^#\\+") "" key), ""
|
|
| _ -> "",""
|
|
|
|
let of_header front_matter =
|
|
let fields = List.map kv_of_string (Str.(split (regexp "\n")) front_matter) in
|
|
List.fold_left with_kv (blank ~id:Id.nil ()) fields
|
|
|
|
let front_matter_body_split s =
|
|
if Str.(string_match (regexp ".*:.*")) s 0
|
|
then match Str.(bounded_split (regexp "^$")) s 2 with
|
|
| front::body::[] -> (front, body)
|
|
| _ -> ("", s)
|
|
else ("", s)
|
|
|
|
let of_string s =
|
|
let front_matter, body = front_matter_body_split s in
|
|
try
|
|
let note = { (of_header front_matter) with body } in
|
|
if note.id <> Id.nil then Ok note else Error "Missing ID header"
|
|
with _ -> Error ("Failed parsing" ^ s)
|
|
|
|
let to_string x =
|
|
let has_len v = String.length v > 0 in
|
|
let s field value = if has_len value then field ^ ": " ^ value ^ "\n" else "" in
|
|
let a value = if Person.Set.is_empty value then "" else "Authors: " ^ Person.Set.to_string value ^ "\n" in
|
|
let d field value = match value with "" -> "" | s -> field ^ ": " ^ Date.rfc_string s ^ "\n" in
|
|
let rows = [
|
|
s "ID" x.id;
|
|
d "Date" x.date.Date.created;
|
|
d "Edited" x.date.Date.edited;
|
|
s "Title" x.title;
|
|
a x.authors;
|
|
s "Licences" (str_set "licences" x);
|
|
s "Topics" (str_set "topics" x);
|
|
s "Keywords" (str_set "keywords" x);
|
|
s "Series" (str_set "series" x);
|
|
s "Abstract" (str "abstract" x);
|
|
s "Alias" (str "Alias" x)
|
|
] in
|
|
String.concat "" rows ^ "\n" ^ x.body
|
|
|
|
let string_alias t =
|
|
let is_reserved = function
|
|
| '!' | '*' | '\'' | '(' | ')' | ';' | ':' | '@' | '&' | '=' | '+' | '$'
|
|
| ',' | '/' | '?' | '#' | '[' | ']' | ' ' | '\t' | '\x00' -> true
|
|
| _ -> false
|
|
in
|
|
let b = Buffer.create (String.length t) in
|
|
let filter char =
|
|
let open Buffer in
|
|
if is_reserved char then (try (if nth b (pred (length b)) <> '-' then add_char b '-') with Invalid_argument _ -> prerr_endline "reserved")
|
|
else add_char b char
|
|
in
|
|
String.(iter filter (lowercase_ascii t));
|
|
Buffer.contents b
|
|
|
|
let alias t = match str "alias" t with "" -> string_alias t.title | x -> x
|
|
let short_id t = Id.short t.id
|