Http_daemon.respond
~body:(pp_fun msg) ~headers:["Content-Type", contype] outchan
let pp_error s =
- sprintf "<html><body>Http Getter error: %s</body></html>" s
+ sprintf "<html><body>Http Getter error: <span style=\"color:red\">%s</span></body></html>" s
let pp_internal_error s =
- sprintf "<html><body>Http Getter Internal error: %s</body></html>" s
+ sprintf "<html><body>Http Getter Internal error: <span style=\"color:red\">%s</span></body></html>" s
let pp_msg s = sprintf "<html><body>%s</body></html>" s
let null_pp s = s
let return_html_error = mk_return_fun pp_error "text/html"
Http_getter_env.reload ();
print_string (Http_getter_env.env_to_string ());
flush stdout;
- at_exit Http_getter.close_maps;
- Sys.catch_break true;
- try
- Http_daemon.start' ~mode:`Thread
- ~timeout:(Some 600) ~port:(Helm_registry.get_int "getter.port")
- callback
- with Sys.Break -> () (* 'close_maps' already registered with 'at_exit' *)
+ let batch_update =
+ try Sys.argv.(1) = "-update" with Invalid_argument _ -> false
+ in
+ if batch_update then (* batch mode: performs update and exit *)
+ Http_getter.update ~logger:Http_getter.stdout_logger ()
+ else begin (* daemon mode: start http daemon *)
+ at_exit Http_getter.close_maps;
+ Sys.catch_break true;
+ try
+ Http_daemon.start' ~mode:`Thread
+ ~timeout:(Some 600) ~port:(Helm_registry.get_int "getter.port")
+ callback
+ with Sys.Break -> () (* 'close_maps' already registered with 'at_exit' *)
+ end
let _ = main ()