sql2cpp.ml

   1 (*
   2   Main
   3 *)
   4
   5 open Printf
   6 open Operators
   7 open ListMore
   8 open ExtString
   9
  10 module L = List
  11 module S = String
  12
  13 let statements s =
  14    let lexbuf = Lexing.from_string s in
  15    let rec loop l =
  16     match (try Sql_lexer.ruleStatement Props.empty lexbuf with exn -> None) with
  17     | Some x -> loop (x::l)
  18     | None -> l
  19    in
  20     List.rev (loop [])
  21
  22 let parse_one (stmt,props) =
  23   try
  24     print_endline stmt;
  25     Some ((Parser.parse_stmt stmt), Props.set props "sql" stmt)
  26   with
  27   | exn ->
  28     begin
  29       print_endline (Printexc.to_string exn);
  30       None
  31     end
  32
  33 let show_one ((s,p),props) =
  34   RA.Scheme.print s;
  35   print_endline (Show.show<Stmt.Raw.params>(p))
  36
  37 let catch f x = try Some (f x) with _ -> None
  38
  39 let with_file filename f =
  40   match catch Std.input_file filename with
  41   | None -> Error.log "cannot open file : %s" filename
  42   | Some s -> f s
  43
  44 let tee f x = f x; x
  45
  46 let work filename =
  47   with_file filename (fun s -> s
  48   >> statements >> L.map parse_one >> L.filter_valid
  49   >> tee (L.iter show_one) >> Gen.process)
  50
  51 let show_help () =
  52   Error.log "SQL to C++ Code Generator Version %s (%s)" Version.version (Version.revision >> S.explode >> L.take 8 >> S.implode);
  53   Error.log "";
  54   Error.log " Usage: %s file_with_statements.sql" (Filename.basename Sys.executable_name);
  55   Error.log "";
  56   Error.log " Parse given file (treating content as SQL statements) and emit corresponding code to stdout"
  57
  58 let main () =
  59   match Array.to_list Sys.argv with
  60   | _::"-test"::_ -> Test.run ()
  61   | _::[file] -> work file
  62   | _ -> show_help ()
  63
  64 let _ = Printexc.print main ()