diff options
| -rw-r--r-- | gemini/mimeType.ml | 43 |
1 files changed, 40 insertions, 3 deletions
diff --git a/gemini/mimeType.ml b/gemini/mimeType.ml index 0559168..1682416 100644 --- a/gemini/mimeType.ml +++ b/gemini/mimeType.ml @@ -24,9 +24,46 @@ struct exception ParseError of string - let from_string _ = { mime_type = "text"; - mime_subtype = "plain"; - mime_param = None } + let create_mimetype mime_type mime_subtype mime_param = + { mime_type; mime_subtype; mime_param; } + + let from_string (s : string) = + (* RFC 2045 § 5.1 *) + (* RFC 6838 § 4.2 *) + (* FIXME: parameter does not handle quoted strings and + does not try to ensure that everything is well-formed *) + let open Angstrom in + let lift_or p q b = p b || q b in + let is_letter = function + | 'a' .. 'z' | 'A' .. 'Z' -> true + | _ -> false in + let is_digit = function + | '0' .. '9' -> true + | _ -> false in + let is_symbol = function + | '!' | '#' | '$' | '&' | '-' | '^' | '_' | '.' | '+' -> true + | _ -> false in + let is_space = function + | ' ' | '\t' -> true + | _ -> false in + let take_all = take_while (fun _ -> true) in + let maybe p = + option None (lift (fun s -> Some s) p) in + let restricted_name = + lift2 (fun c -> fun s -> (Char.escaped c) ^ s) + (satisfy (lift_or is_letter is_digit)) + (take_while (lift_or is_letter (lift_or is_digit is_symbol))) in + let parameter = lift2 (fun p -> fun v -> (p, v)) + (take_till (fun c -> c = '=')) + (char '=' *> take_all) <?> "parameter" in + let parse = lift3 create_mimetype + (restricted_name <?> "type") + ((char '/' *> restricted_name) <?> "subtype") + (maybe (char ';' *> skip_while is_space *> parameter)) + <?> "mimetype" + in match Angstrom.parse_string ~consume:All parse s with + | Ok x -> x + | Error msg -> raise (ParseError msg) let to_type m = m.mime_type |