Source file res.ml

open Base
open Http_core

(* TODO remove all opium references *)
module OpiumResponse = struct
  (* We want to be able to derive show and eq from our own response type t *)
  type t = Opium_kernel.Response.t

  let equal _ _ = false

  let pp _ _ = ()
end

type body = String of string | File_path of string [@@deriving show, eq]

type t = {
  content_type : content_type;
  redirect : string option;
  body : body option;
  headers : headers;
  opium_res : OpiumResponse.t option;
  cookies : (string * string) list;
  status : int;
}
[@@deriving show, eq]

let file path content_type =
  {
    content_type;
    redirect = None;
    body = Some (File_path path);
    headers = [];
    opium_res = None;
    cookies = [];
    status = 200;
  }

let html =
  {
    content_type = Html;
    redirect = None;
    body = None;
    headers = [];
    opium_res = None;
    cookies = [];
    status = 200;
  }

let json =
  {
    content_type = Json;
    redirect = None;
    body = None;
    headers = [];
    opium_res = None;
    cookies = [];
    status = 200;
  }

let redirect path =
  {
    content_type = Html;
    redirect = Some path;
    body = None;
    headers = [];
    opium_res = None;
    cookies = [];
    status = 302;
  }

let redirect_path res = res.redirect

let body res = res.body

let set_body str res = { res with body = Some (String str) }

let headers res = res.headers

let set_headers headers res = { res with headers }

let content_type res = res.content_type

let set_content_type content_type res = { res with content_type }

(* TODO this is a hack and has to be removed *)
let opium_res res = res.opium_res

(* TODO this is a hack and has to be removed *)
let set_opium_res opium_res res = { res with opium_res = Some opium_res }

let cookies res = res.cookies

let set_cookie ~key ~data res =
  { res with cookies = List.cons (key, data) res.cookies }

let status res = res.status

let set_status status res = { res with status }