diff --git a/bin/main.ml b/bin/main.ml index d720843..9f37f0a 100644 --- a/bin/main.ml +++ b/bin/main.ml @@ -1,110 +1,3 @@ -(* open Lwt.Syntax -open Yojson.Safe.Util -open Pgn_logic - -let is_dev_mode () = - Array.to_list Sys.argv - |> List.exists (fun arg -> arg = "DREAM_ENV=development") - -let cors_middleware next_handler request = - let allowed_origin = - if is_dev_mode () then "*" else "https://chess-scribe.org" - in - - if Dream.method_ request = `OPTIONS then ( - let response = Dream.response "" in - Dream.add_header response "Access-Control-Allow-Origin" allowed_origin; - Dream.add_header response "Access-Control-Allow-Methods" - "POST, GET, OPTIONS"; - Dream.add_header response "Access-Control-Allow-Headers" - "Content-Type, Authorization"; - Dream.add_header response "Access-Control-Max-Age" "86400"; - Dream.set_status response `No_Content; - Lwt.return response) - else - let* response = next_handler request in - Dream.add_header response "Access-Control-Allow-Origin" allowed_origin; - Lwt.return response - -let handle_convert request = - let start_time = Unix.gettimeofday () in - - let temp_id = - Int64.to_string (Int64.of_float (Unix.gettimeofday () *. 1000.0)) - in - - let temp_dir = Filename.get_temp_dir_name () in - - let tex_file_path = - Filename.concat temp_dir (Printf.sprintf "game-%s.tex" temp_id) - in - - let pdf_file_path = - Filename.concat temp_dir (Printf.sprintf "game-%s.pdf" temp_id) - in - - let log_file_path = - Filename.concat temp_dir (Printf.sprintf "game-%s.log" temp_id) - in - - let aux_file_path = - Filename.concat temp_dir (Printf.sprintf "game-%s.aux" temp_id) - in - - let cleanup () = - [ tex_file_path; pdf_file_path; log_file_path; aux_file_path ] - |> List.iter (fun path -> if Sys.file_exists path then Sys.remove path) - in - - let* body = Dream.body request in - - try - let json = Yojson.Safe.from_string body in - - let pgn = json |> member "pgn" |> to_string in - let diagram_clock = json |> member "diagramClock" |> to_bool in - let diagrams_json = json |> member "diagrams" in - - let diagrams_data = - Yojson.Safe.to_string diagrams_json |> Pgn2tex.parse_diagrams_json - in - - let game_tex = - Pgn2tex.to_tex pgn ~diagram_data:diagrams_data ~clock:diagram_clock - in - - print_endline game_tex; - - Dream.json {|{"status": "ok", "message": "Payload parsed successfully!"}|} - with - | Type_error (msg, _) -> - Dream.json ~status:`Bad_Request - (Printf.sprintf - {|{"type": "error", "message": "JSON structure error: %s"}|} msg) - | Yojson.Json_error msg -> - Dream.json ~status:`Bad_Request - (Printf.sprintf - {|{"type": "error", "message": "Invalid JSON syntax: %s"}|} msg) - | exn -> - Dream.json ~status:`Internal_Server_Error - (Printf.sprintf {|{"type": "error", "message": "%s"}|} - (Printexc.to_string exn)) - -let () = - Dream.run ~interface:"0.0.0.0" ~port:8080 - @@ Dream.logger @@ cors_middleware - @@ Dream.router - [ - Dream.get "/" (fun request -> - Dream.from_filesystem "static" "doc.html" request); - Dream.get "/health" (fun _ -> Dream.respond "OK"); - Dream.post "/api/v1/pdf" handle_convert; - Dream.get "/api/v1" (fun request -> - Dream.from_filesystem "static" "doc.html" request); - Dream.get "/swagger.json" (fun request -> - Dream.from_filesystem "static" "swagger.json" request); - ] *) - open Lwt.Syntax open Yojson.Safe.Util open Pgn_logic