mirror of
https://github.com/c-cube/tiny_httpd.git
synced 2025-12-07 03:35:34 -05:00
41 lines
1.5 KiB
OCaml
41 lines
1.5 KiB
OCaml
|
|
module S = Tiny_httpd
|
|
|
|
let () =
|
|
let port_ = ref 8080 in
|
|
let j = ref 32 in
|
|
Arg.parse (Arg.align [
|
|
"--port", Arg.Set_int port_, " set port";
|
|
"-p", Arg.Set_int port_, " set port";
|
|
"--debug", Arg.Unit (fun () -> S._enable_debug true), " enable debug";
|
|
"-j", Arg.Set_int j, " maximum number of connections";
|
|
]) (fun _ -> raise (Arg.Bad "")) "echo [option]*";
|
|
let server = S.create ~port:!port_ ~max_connections:!j () in
|
|
Tiny_httpd_zlib.setup server;
|
|
(* say hello *)
|
|
S.add_path_handler ~meth:`GET server
|
|
"/hello/%s@/" (fun name _req -> S.Response.make_string (Ok ("hello " ^name ^"!\n")));
|
|
(* echo request *)
|
|
S.add_path_handler server
|
|
"/echo" (fun req ->
|
|
let q =
|
|
S.Request.query req |> List.map (fun (k,v) -> Printf.sprintf "%S = %S" k v)
|
|
|> String.concat ";"
|
|
in
|
|
S.Response.make_string
|
|
(Ok (Format.asprintf "echo:@ %a@ (query: %s)@." S.Request.pp req q)));
|
|
S.add_path_handler ~meth:`PUT server
|
|
"/upload/%s" (fun path req ->
|
|
S._debug (fun k->k "start upload %S\n%!" path);
|
|
try
|
|
let oc = open_out @@ "/tmp/" ^ path in
|
|
output_string oc req.S.Request.body;
|
|
flush oc;
|
|
S.Response.make_string (Ok "uploaded file")
|
|
with e ->
|
|
S.Response.fail ~code:500 "couldn't upload file: %s" (Printexc.to_string e)
|
|
);
|
|
Printf.printf "listening on http://%s:%d\n%!" (S.addr server) (S.port server);
|
|
match S.run server with
|
|
| Ok () -> ()
|
|
| Error e -> raise e
|