config.ml

open Mirage

let port =
  let doc = Key.Arg.info ~doc:"HTTPS listen port." [ "port" ] in
  Key.(create "port" Arg.(opt int 8080 doc))

let service_name =
  let doc =
    Key.Arg.info ~doc:"Service name (Used as tab title)" [ "service-name" ]
  in
  Key.(create "service-name" Arg.(required string doc))

let default_host =
  let doc = Key.Arg.info ~doc:"The default host proxied" [ "default-host" ] in
  Key.(create "default-host" Arg.(required string doc))

let about_url =
  let doc = Key.Arg.info ~doc:"URL of about page" [ "about-url" ] in
  Key.(create "about-url" Arg.(required string doc))

let random_banner_url =
  let doc =
    Key.Arg.info ~doc:"Gemini URL of random banner page" [ "random-banner-url" ]
  in
  Key.(create "random-banner-url" Arg.(required string doc))

let nameservers =
  let doc = Key.Arg.info ~doc:"Nameserver" [ "nameserver" ] in
  Key.(create "nameserver" Arg.(opt_all string doc))

let main =
  main
    ~keys:
      [
        Key.v port;
        Key.v service_name;
        Key.v default_host;
        Key.v about_url;
        Key.v random_banner_url;
      ]
    ~packages:
      [
        package "dream-mirage"
          ~pin:"git+https://github.com/tmcgilchrist/dream.git#patch-1";
        package "razzia" ~pin:"git+https://github.com/Psi-Prod/Razzia.git#dev";
        package "razzia-mirage"
          ~pin:"git+https://github.com/Psi-Prod/Razzia.git#dev";
        package "tyxml";
        package "hilite";
      ]
    "Unikernel.Main"
    (kv_ro @-> pclock @-> stackv4v6 @-> time @-> dns_client @-> job)

let static_key = Key.(value @@ kv_ro ~group:"static" ())
let static = generic_kv_ro ~key:static_key "static"
let stack = generic_stackv4v6 default_network

let () =
  register "proximite"
    [
      main $ static $ default_posix_clock $ stack $ default_time
      $ generic_dns_client ~nameservers stack;
    ]