itp_server.ml 36.6 KB
Newer Older
1
open Format
Sylvain Dailler's avatar
Sylvain Dailler committed
2
open Stdlib
3
4
5
open Call_provers
open Session_itp
open Controller_itp
Sylvain Dailler's avatar
Sylvain Dailler committed
6
open Server_utils
7

Sylvain Dailler's avatar
Sylvain Dailler committed
8
exception Bad_prover_name of string
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

(**********************************)
(* list unproven goal and related *)
(**********************************)

(* If the transformation is proved, return acc.
   Else, return the concatenation of the reversed list of unproven
   goals below the transformation and acc *)
let rec unproven_goals_below_tn cont acc tn =
  if tn_proved cont tn then
    acc                         (* we ignore "dead" goals *)
  else
    let sub_tasks = get_sub_tasks cont.controller_session tn in
    List.fold_left (unproven_goals_below_pn cont) acc sub_tasks

(* Same as unproven_goals_below_tn; note that if goal is not proved
   and there is no transformation, goal is returned (else it is not) *)
and unproven_goals_below_pn cont acc goal =
  if pn_proved cont goal then
    acc                         (* we ignore "dead" transformations *)
  else
    match get_transformations cont.controller_session goal with
    | [] -> goal :: acc
    | tns -> List.fold_left (unproven_goals_below_tn cont) acc tns

(* Same as unproven_goals_below_tn *)
let unproven_goals_below_th cont acc th =
  if th_proved cont th then
    acc
  else
    let goals = theory_goals th in
    List.fold_left (unproven_goals_below_pn cont) acc goals

(* Same as unproven_goals_below_tn *)
let unproven_goals_below_file cont file =
  if file_proved cont file then
    []
  else
    let theories = file.file_theories in
    List.fold_left (unproven_goals_below_th cont) [] theories

(* returns the list of unproven goals in the controller session *)
let unproven_goals_in_session cont =
  let files = get_files cont.controller_session in
  Stdlib.Hstr.fold (fun _key file acc ->
      let file_goals = unproven_goals_below_file cont file in
      List.rev_append file_goals acc)
    files []

(* [get_first_unproven_goal_around_pn cont pn]
   returns the `first unproven goal' 'after' [pn]. Precisely:
   (1) it finds the youngest ancestor a of [pn] that is not proved
   (2) it returns the first unproved leaf of a
   it returns None if all ancestors are proved *)
let _get_first_unproven_goal_around_pn cont pn =
  let ses = cont.controller_session in
  let rec look_around pn =
    match get_proof_parent ses pn with
    | Trans tn  ->
      if tn_proved cont tn
      then look_around (get_trans_parent ses tn)
      else unproven_goals_below_tn cont [] tn
    | Theory th ->
      if th_proved cont th then begin
        let parent = (theory_parent ses th) in
        if file_proved cont parent
        then unproven_goals_in_session cont
        else unproven_goals_below_file cont parent
      end else
        unproven_goals_below_th cont [] th
  in
  match look_around pn with
  | [] -> None
  | l -> Some (List.hd (List.rev l))

Sylvain Dailler's avatar
Sylvain Dailler committed
84
85
86
(****************)
(* Command list *)
(****************)
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
let commands =
  [
    "list-transforms", "list available transformations", Qnotask list_transforms_query;
    "list-provers", "list available provers", Qnotask list_provers;
(*
    "list-strategies", "list available strategies", list_strategies;
*)
    "print", "<s> print the declaration where s was defined", Qtask print_id;
    "search", "<s> print some declarations where s appear", Qtask search_id;
(*
    "r", "reload the session (test only)", test_reload;
    "rp", "replay", test_replay;
    "s", "save the current session", test_save_session;
    "ng", "go to the next goal", then_print (move_to_goal_ret_p next_node);
    "pg", "go to the prev goal", then_print (move_to_goal_ret_p prev_node);
    "gu", "go to the goal up",  then_print (move_to_goal_ret_p zipper_up);
    "gd", "go to the goal down",  then_print (move_to_goal_ret_p zipper_down);
    "gr", "go to the goal right",  then_print (move_to_goal_ret_p zipper_right);
    "gl", "go to the goal left",  then_print (move_to_goal_ret_p zipper_left)
 *)
  ]

let commands_table = Stdlib.Hstr.create 17
let () =
  List.iter
    (fun (c,_,f) -> Stdlib.Hstr.add commands_table c f)
    commands

Sylvain Dailler's avatar
Sylvain Dailler committed
115
116
117
118
(*******************)
(* Strategies list *)
(*******************)
let loaded_strategies = ref []
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273

(****** Exception handling *********)

let print_term s id fmt t =
  let tables = match (Session_itp.get_tables s id) with
  | None -> Args_wrapper.build_name_tables (Session_itp.get_task s id)
  | Some tables -> tables in
  Why3printer.print_term tables fmt t

let print_type s id fmt t =
  let tables = match (Session_itp.get_tables s id) with
  | None -> Args_wrapper.build_name_tables (Session_itp.get_task s id)
  | Some tables -> tables in
  Why3printer.print_ty tables fmt t

let print_ts s id fmt t =
  let tables = match (Session_itp.get_tables s id) with
  | None -> Args_wrapper.build_name_tables (Session_itp.get_task s id)
  | Some tables -> tables in
  Why3printer.print_ts tables fmt t

let print_ls s id fmt t =
  let tables = match (Session_itp.get_tables s id) with
  | None -> Args_wrapper.build_name_tables (Session_itp.get_task s id)
  | Some tables -> tables in
  Why3printer.print_ls tables fmt t

let print_tv s id fmt t =
  let tables = match (Session_itp.get_tables s id) with
  | None -> Args_wrapper.build_name_tables (Session_itp.get_task s id)
  | Some tables -> tables in
  Why3printer.print_tv tables fmt t

let print_vsty s id fmt t =
  let tables = match (Session_itp.get_tables s id) with
  | None -> Args_wrapper.build_name_tables (Session_itp.get_task s id)
  | Some tables -> tables in
  Why3printer.print_forget_vsty tables fmt t

let print_pr s id fmt t =
  let tables = match (Session_itp.get_tables s id) with
  | None -> Args_wrapper.build_name_tables (Session_itp.get_task s id)
  | Some tables -> tables in
  Why3printer.print_pr tables fmt t

let print_pat s id fmt t =
  let tables = match (Session_itp.get_tables s id) with
  | None -> Args_wrapper.build_name_tables (Session_itp.get_task s id)
  | Some tables -> tables in
  Why3printer.print_pat tables fmt t

(* Exception reporting *)

(* TODO remove references to id.id_string in this function *)
let bypass_pretty s id =
  begin fun fmt exn -> match exn with
  | Ty.TypeMismatch (t1,t2) ->
      fprintf fmt "Type mismatch between %a and %a"
        (print_type s id) t1 (print_type s id) t2
  | Ty.BadTypeArity ({Ty.ts_args = []} as ts, _) ->
      fprintf fmt "Type symbol %a expects no arguments" (print_ts s id) ts
  | Ty.BadTypeArity (ts, app_arg) ->
      let i = List.length ts.Ty.ts_args in
      fprintf fmt "Type symbol %a expects %i argument%s but is applied to %i"
        (print_ts s id) ts i (if i = 1 then "" else "s") app_arg
  | Ty.DuplicateTypeVar tv ->
      fprintf fmt "Type variable %a is used twice" (print_tv s id) tv
  | Ty.UnboundTypeVar tv ->
      fprintf fmt "Unbound type variable: %a" (print_tv s id) tv
  | Ty.UnexpectedProp ->
      fprintf fmt "Unexpected propositional type"
  | Term.BadArity ({Term.ls_args = []} as ls, _) ->
      fprintf fmt "%s %a expects no arguments"
        (if ls.Term.ls_value = None then "Predicate" else "Function") (print_ls s id) ls
  | Term.BadArity (ls, app_arg) ->
      let i = List.length ls.Term.ls_args in
      fprintf fmt "%s %a expects %i argument%s but is applied to %i"
        (if ls.Term.ls_value = None then "Predicate" else "Function")
        (print_ls s id) ls i (if i = 1 then "" else "s") app_arg
  | Term.EmptyCase ->
      fprintf fmt "Empty match expression"
  | Term.DuplicateVar vs ->
      fprintf fmt "Variable %a is used twice" (print_vsty s id) vs
  | Term.UncoveredVar vs ->
      fprintf fmt "Variable %a uncovered in \"or\"-pattern" (print_vsty s id) vs
  | Term.FunctionSymbolExpected ls ->
      fprintf fmt "Not a function symbol: %a" (print_ls s id) ls
  | Term.PredicateSymbolExpected ls ->
      fprintf fmt "Not a predicate symbol: %a" (print_ls s id) ls
  | Term.ConstructorExpected ls ->
      fprintf fmt "%s %a is not a constructor"
        (if ls.Term.ls_value = None then "Predicate" else "Function") (print_ls s id) ls
  | Term.TermExpected t ->
      fprintf fmt "Not a term: %a" (print_term s id) t
  | Term.FmlaExpected t ->
      fprintf fmt "Not a formula: %a" (print_term s id) t
  | Pattern.ConstructorExpected (ls,ty) ->
      fprintf fmt "%s %a is not a constructor of type %a"
        (if ls.Term.ls_value = None then "Predicate" else "Function") (print_ls s id) ls
        (print_type s id) ty
  | Pattern.NonExhaustive pl ->
      fprintf fmt "Pattern not covered by a match:@\n  @[%a@]"
        (print_pat s id) (List.hd pl)
  | Decl.BadConstructor ls ->
      fprintf fmt "Bad constructor: %a" (print_ls s id) ls
  | Decl.BadRecordField ls ->
      fprintf fmt "Not a record field: %a" (print_ls s id) ls
  | Decl.RecordFieldMissing (_cs,ls) ->
      fprintf fmt "Field %a is missing" (print_ls s id) ls
  | Decl.DuplicateRecordField (_cs,ls) ->
      fprintf fmt "Field %a is used twice in the same constructor" (print_ls s id) ls
  | Decl.IllegalTypeAlias ts ->
      fprintf fmt
        "Type symbol %a is a type alias and cannot be declared as algebraic"
        (print_ts s id) ts
  | Decl.NonFoundedTypeDecl ts ->
      fprintf fmt "Cannot construct a value of type %a" (print_ts s id) ts
  | Decl.NonPositiveTypeDecl (_ts, ls, ty) ->
      fprintf fmt "Constructor %a \
          contains a non strictly positive occurrence of type %a"
        (print_ls s id) ls (print_type s id) ty
  | Decl.InvalidIndDecl (_ls, pr) ->
      fprintf fmt "Ill-formed inductive clause %a"
        (print_pr s id) pr
  | Decl.NonPositiveIndDecl (_ls, pr, ls1) ->
      fprintf fmt "Inductive clause %a contains \
          a non strictly positive occurrence of symbol %a"
        (print_pr s id) pr (print_ls s id) ls1
  | Decl.BadLogicDecl (ls1,ls2) ->
      fprintf fmt "Ill-formed definition: symbols %a and %a are different"
        (print_ls s id) ls1 (print_ls s id) ls2
  | Decl.UnboundVar vs ->
      fprintf fmt "Unbound variable: %a" (print_vsty s id) vs
  | Decl.ClashIdent id ->
      fprintf fmt "Ident %s is defined twice" id.Ident.id_string
  | Decl.EmptyDecl ->
      fprintf fmt "Empty declaration"
  | Decl.EmptyAlgDecl ts ->
      fprintf fmt "Algebraic type %a has no constructors" (print_ts s id) ts
  | Decl.EmptyIndDecl ls ->
      fprintf fmt "Inductive predicate %a has no constructors" (print_ls s id) ls
  | Decl.KnownIdent id ->
      fprintf fmt "Ident %s is already declared" id.Ident.id_string
  | Decl.UnknownIdent id ->
      fprintf fmt "Ident %s is not yet declared" id.Ident.id_string
  | Decl.RedeclaredIdent id ->
      fprintf fmt "Ident %s is already declared, with a different declaration"
        id.Ident.id_string
  | Decl.NoTerminationProof ls ->
      fprintf fmt "Cannot prove the termination of %a" (print_ls s id) ls
  | _ -> Format.fprintf fmt "Uncaught: %a" Exn_printer.exn_printer exn
  end

let get_exception_message ses id fmt e =
  match e with
274
275
  | Controller_itp.Noprogress ->
      Format.fprintf fmt "Transformation made no progress\n"
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
  | Case.Arg_trans_type (s, ty1, ty2) ->
      Format.fprintf fmt "Error in transformation %s during unification of the following terms:\n %a \n %a"
        s (print_type ses id) ty1 (print_type ses id) ty2
  | Case.Arg_trans_term (s, t1, t2) ->
      Format.fprintf fmt "Error in transformation %s during unification of following two terms:\n %a \n %a" s
        (print_term ses id) t1 (print_term ses id) t2
  | Case.Arg_trans (s) ->
      Format.fprintf fmt "Error in transformation function: %s \n" s
  | Case.Arg_hyp_not_found (s) ->
      Format.fprintf fmt "Following hypothesis was not found: %s \n" s
  | Args_wrapper.Arg_theory_not_found (s) ->
      Format.fprintf fmt "Theory not found: %s" s
  | e ->
      bypass_pretty ses id fmt e



Clément Fumex's avatar
Clément Fumex committed
293

Sylvain Dailler's avatar
Sylvain Dailler committed
294
(* Information that the IDE may want to have *)
Clément Fumex's avatar
Clément Fumex committed
295
296
297
298
type prover = string
type transformation = string
type strategy = string

Sylvain Dailler's avatar
Sylvain Dailler committed
299

300
301
type node_ID = int
let root_node : node_ID = 0
Sylvain Dailler's avatar
Sylvain Dailler committed
302

Clément Fumex's avatar
Clément Fumex committed
303

304
305
type global_information =
    {
Clément Fumex's avatar
Clément Fumex committed
306
307
308
309
     provers         : prover list;
     transformations : transformation list;
     strategies      : strategy list;
     commands        : string list;
310
311
312
313
314
315
316
     (* hidden_provers       : string list; *)
     (* session_time_limit   : int; *)
     (* session_mem_limit    : int; *)
     (* session_nb_processes : int; *)
     (* session_cntexample   : bool; *)
     (* main_dir             : string *)
    }
Clément Fumex's avatar
Clément Fumex committed
317

318
type message_notification =
319
320
321
322
323
324
325
326
327
328
329
  | Proof_error           of node_ID * string
  | Transf_error          of node_ID * string
  | Strat_error           of node_ID * string
  | Replay_Info           of string
  | Query_Info            of node_ID * string
  | Query_Error           of node_ID * string
  | Help                  of string
  | Information           of string
  | Task_Monitor          of int * int * int
  | Error                 of string
  | Open_File_Error       of string
Clément Fumex's avatar
Clément Fumex committed
330

331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
type node_type =
  | NRoot
  | NFile
  | NTheory
  | NTransformation
  | NGoal
  | NProofAttempt

type update_info =
  | Proved of bool
  | Proof_status_change of
      Controller_itp.proof_attempt_status
      * bool   (* obsolete or not *)
      * Call_provers.resource_limit

Clément Fumex's avatar
Clément Fumex committed
346
type notification =
347
348
349
350
351
  | New_node     of node_ID * node_ID * node_type * string
  (* Notification of creation of new_node:
     New_node (new_node, parent_node, node_type, name). *)
  | Node_change  of node_ID * update_info
  (* inform that the data of the given node changed *)
Clément Fumex's avatar
Clément Fumex committed
352
  | Remove       of node_ID
353
  (* the given node was removed *)
Clément Fumex's avatar
Clément Fumex committed
354
  | Initialized  of global_information
355
  (* initial global data *)
Sylvain Dailler's avatar
Sylvain Dailler committed
356
  | Saved
357
  (* the session was saved on disk *)
Clément Fumex's avatar
Clément Fumex committed
358
  | Message      of message_notification
359
  (* an informative message, can be an error message *)
Clément Fumex's avatar
Clément Fumex committed
360
  | Dead         of string
361
  (* server exited *)
Clément Fumex's avatar
Clément Fumex committed
362
  | Task         of node_ID * string
363
364
365
366
367
368
369
370
371
  (* the node_ID's task *)

type ide_request =
  | Command_req       of node_ID * string
  | Prove_req         of node_ID * prover * resource_limit
  | Transform_req     of node_ID * transformation * string list
  | Strategy_req      of node_ID * strategy
  | Open_session_req  of string
  | Add_file_req      of string
Clément Fumex's avatar
Clément Fumex committed
372
  | Set_max_tasks_req of int
373
  | Get_task          of node_ID
374
  | Remove_subtree    of node_ID
Sylvain Dailler's avatar
Sylvain Dailler committed
375
376
  | Copy_paste        of node_ID * node_ID
  | Copy_detached     of node_ID
377
378
379
380
381
  | Get_Session_Tree_req
  | Save_req
  | Reload_req
  | Replay_req
  | Exit_req
Clément Fumex's avatar
Clément Fumex committed
382

Sylvain Dailler's avatar
Sylvain Dailler committed
383
(* Debugging functions *)
384
385
let print_request fmt r =
  match r with
386
387
388
389
390
391
392
393
  | Command_req (_nid, s)           -> fprintf fmt "command \"%s\"" s
  | Prove_req (_nid, prover, _rl)   -> fprintf fmt "prove with %s" prover
  | Transform_req (_nid, tr, _args) -> fprintf fmt "transformation :%s" tr
  | Strategy_req (_nid, st)         -> fprintf fmt "strategy %s" st
  | Open_session_req f              -> fprintf fmt "open session file %s" f
  | Add_file_req f                  -> fprintf fmt "open file %s" f
  | Set_max_tasks_req i             -> fprintf fmt "set max tasks %i" i
  | Get_task _nid                   -> fprintf fmt "get task"
394
  | Remove_subtree _nid             -> fprintf fmt "remove subtree"
Sylvain Dailler's avatar
Sylvain Dailler committed
395
396
  | Copy_paste _                    -> fprintf fmt "copy paste"
  | Copy_detached _                 -> fprintf fmt "copy detached"
397
398
399
400
401
  | Get_Session_Tree_req            -> fprintf fmt "get session tree"
  | Save_req                        -> fprintf fmt "save"
  | Reload_req                      -> fprintf fmt "reload"
  | Replay_req                      -> fprintf fmt "replay"
  | Exit_req                        -> fprintf fmt "exit"
402

Sylvain Dailler's avatar
Sylvain Dailler committed
403
404
405
406
407
408
409
410
411
412
413
414
let print_msg fmt m =
  match m with
  | Proof_error (_ids, s)  -> fprintf fmt "proof error %s" s
  | Transf_error (_ids, s) -> fprintf fmt "transf error %s" s
  | Strat_error (_ids, s)  -> fprintf fmt "start error %s" s
  | Replay_Info s          -> fprintf fmt "replay info %s" s
  | Query_Info (_ids, s)   -> fprintf fmt "query info %s" s
  | Query_Error (_ids, s)  -> fprintf fmt "query error %s" s
  | Help _s                -> fprintf fmt "help"
  | Information s          -> fprintf fmt "info %s" s
  | Task_Monitor _         -> fprintf fmt "task montor"
  | Error s                -> fprintf fmt "%s" s
415
  | Open_File_Error s        -> fprintf fmt "%s" s
Sylvain Dailler's avatar
Sylvain Dailler committed
416
417
418
419

let print_notify fmt n =
  match n with
  | Node_change (_ni, _nf)          -> fprintf fmt "node change"
420
  | New_node (ni, _pni, _nt,  _nf)  -> fprintf fmt "new node %d" ni
Sylvain Dailler's avatar
Sylvain Dailler committed
421
422
423
424
425
426
427
428
  | Remove _ni                      -> fprintf fmt "remove"
  | Initialized _gi                 -> fprintf fmt "initialized"
  | Saved                           -> fprintf fmt "saved"
  | Message msg                     ->
      print_msg fmt msg
  | Dead s                          -> fprintf fmt "dead :%s" s
  | Task (_ni, _s)                  -> fprintf fmt "task"

Clément Fumex's avatar
Clément Fumex committed
429
module type Protocol = sig
Clément Fumex's avatar
Clément Fumex committed
430
431
  val get_requests : unit -> ide_request list
  val notify : notification -> unit
Clément Fumex's avatar
Clément Fumex committed
432
433
end

434
435
436
module Make (S:Controller_itp.Scheduler) (P:Protocol) = struct

  module C = Controller_itp.Make(S)
Clément Fumex's avatar
Clément Fumex committed
437

438
  let debug = Debug.register_flag "itp_server" ~desc:"ITP server"
Clément Fumex's avatar
Clément Fumex committed
439

440
441
442
  type server_data =
    { config : Whyconf.config;
      task_driver : Driver.driver;
443
      sd_provers : Whyconf.config_prover Whyconf.Mprover.t;
444
445
      cont : Controller_itp.controller;
    }
Clément Fumex's avatar
Clément Fumex committed
446

447
  let server_data = ref None
Clément Fumex's avatar
Clément Fumex committed
448

449
  let get_server_data () =
450
451
452
453
454
    match !server_data with
    | None ->
       Format.eprintf "[ITP server] not yet initialized@.";
       exit 1
    | Some x -> x
Clément Fumex's avatar
Clément Fumex committed
455

456
  let task_driver config env =
457
458
459
460
461
462
463
464
465
    try
      let main = Whyconf.get_main config in
      let d = Filename.concat (Whyconf.datadir main)
                              (Filename.concat "drivers" "why3_itp.drv")
      in
      let d = Driver.load_driver env d [] in
      Debug.dprintf debug "[ITP server] driver for task printing loaded@.";
      d
    with e ->
466
467
      Format.eprintf "Fatal error while loading itp driver: %a@." Exn_printer.exn_printer e;
      exit 1
Clément Fumex's avatar
Clément Fumex committed
468

469
470
  let get_prover_list (config: Whyconf.config) =
    Mstr.fold (fun x _ acc -> x :: acc) (Whyconf.get_prover_shortcuts config) []
Sylvain Dailler's avatar
Sylvain Dailler committed
471

472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
  let init_server config env =
    let provers = Whyconf.get_provers config in
    let c = create_controller env in
    let task_driver = task_driver config env in
    Whyconf.Mprover.iter
      (fun _ p ->
       try
         let d = Driver.load_driver c.controller_env p.Whyconf.driver [] in
         Whyconf.Hprover.add c.controller_provers p.Whyconf.prover (p,d)
       with e ->
         Format.eprintf
           "[ITP server] error loading driver for prover %a: %a@."
           Whyconf.print_prover p.Whyconf.prover
           Exn_printer.exn_printer e)
      provers;
    server_data := Some
                     { config = config;
                       task_driver = task_driver;
490
                       sd_provers = provers;
491
                       cont = c }
Sylvain Dailler's avatar
Sylvain Dailler committed
492

Clément Fumex's avatar
Clément Fumex committed
493
494
  (* ------------ init controller ------------ *)

495
496
497
  (* Init cont on file or directory. It is called only when an
     Open_session_req is requested *)
  let init_cont f =
498
    let d = get_server_data () in
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
    let prover_list = get_prover_list d.config in
    let transformation_list = List.map fst (list_transforms ()) in
    let strategies_list =
      let l = strategies d.cont.controller_env d.config loaded_strategies in
      List.map (fun (a,_,_,_) -> a) l
    in
    let infos =
      {
        provers = prover_list;
        transformations = transformation_list;
        strategies = strategies_list;
        commands =
          List.map (fun (c,_,_) -> c) commands
      }
    in
514
    try (
Sylvain Dailler's avatar
Sylvain Dailler committed
515
      if (Sys.file_exists f) then
516
      begin
Sylvain Dailler's avatar
Sylvain Dailler committed
517
518
        if (Sys.is_directory f) then
        begin
519
          cont_from_session_dir d.cont f;
Sylvain Dailler's avatar
Sylvain Dailler committed
520
521
522
523
          P.notify (Initialized infos)
        end
        else
        begin
524
          cont_from_file d.cont f;
Sylvain Dailler's avatar
Sylvain Dailler committed
525
526
          P.notify (Initialized infos)
        end
527
528
      end
      else
Sylvain Dailler's avatar
Sylvain Dailler committed
529
        P.notify (Message (Open_File_Error ("Not a file: " ^ f))))
530
    with
Sylvain Dailler's avatar
Sylvain Dailler committed
531
532
533
534
      | NotADirectory f ->
          P.notify (Message (Open_File_Error ("Not a directory: " ^ f)))
      | BadFileName f ->
          P.notify (Message (Open_File_Error ("Bad file name: " ^ f)))
535
536
537
538
      | e ->
          Format.eprintf "%a@." Exn_printer.exn_printer e;
          P.notify (Dead (Pp.string_of Exn_printer.exn_printer e));
          exit 1
Clément Fumex's avatar
Clément Fumex committed
539

Sylvain Dailler's avatar
Sylvain Dailler committed
540
  (* -----------------------------------   ------------------------------------- *)
Clément Fumex's avatar
Clément Fumex committed
541

542
543
544
  let get_node_type (node: any) =
    match node with
    | AFile _ -> NFile
Sylvain Dailler's avatar
Sylvain Dailler committed
545
546
547
548
    | ATh _   -> NTheory
    | ATn _   -> NTransformation
    | APn _   -> NGoal
    | APa _   -> NProofAttempt
549
550

  let get_node_name (node: any) =
551
    let d = get_server_data () in
552
553
554
555
556
557
    match node with
    | AFile file ->
      file.file_name
    | ATh th ->
      (theory_name th).Ident.id_string
    | ATn tn ->
558
      get_transf_name d.cont.controller_session tn
559
    | APn pn ->
560
      (get_proof_name d.cont.controller_session pn).Ident.id_string
561
    | APa pa ->
562
      let pa = get_proof_attempt_node d.cont.controller_session pa in
563
564
565
566
567
568
569
570
571
572
      Pp.string_of Whyconf.print_prover pa.prover

(*
  let get_node_proved (node: any) =
    match node with
    | Afile file ->
      file_proved cont file
    | ATh th ->
      th_proved cont th

Sylvain Dailler's avatar
Sylvain Dailler committed
573
574
575
576
577
  let get_info_and_type ses (node: any) =
    match node with
    | AFile file ->
        let name = file.file_name in
        let proved = file_proved cont file in
578
        NFile, {name; proved}
Sylvain Dailler's avatar
Sylvain Dailler committed
579
580
581
    | ATh th     ->
        let name = (theory_name th).Ident.id_string in
        let proved = th_proved cont th in
582
        NTheory, {name; proved}
Sylvain Dailler's avatar
Sylvain Dailler committed
583
584
585
    | ATn tn     ->
        let name = get_transf_name ses tn in
        let proved = tn_proved cont tn in
586
        NTransformation, {name; proved}
Sylvain Dailler's avatar
Sylvain Dailler committed
587
588
589
    | APn pn     ->
        let name = (get_proof_name ses pn).Ident.id_string in
        let proved = pn_proved cont pn in
590
          NGoal, {name; proved}
Sylvain Dailler's avatar
Sylvain Dailler committed
591
592
593
    | APa pan    ->
        let pa = get_proof_attempt_node ses pan in
        let name = Pp.string_of Whyconf.print_prover pa.prover in
Clément Fumex's avatar
Clément Fumex committed
594
595
596
        let pr, proved = match pa.Session_itp.proof_state with
        | Some pr -> Some pr.pr_answer, pr.pr_answer = Valid
        | None -> None, false
Sylvain Dailler's avatar
Sylvain Dailler committed
597
        in
Clément Fumex's avatar
Clément Fumex committed
598
599
        (NProofAttempt (pr, pa.proof_obsolete)),
        {name; proved}
600
*)
Sylvain Dailler's avatar
Sylvain Dailler committed
601

602
603
604
605
606
607
608
609
610
611
(* fresh gives new fresh "names" for node_ID using a counter.
   reset resets the counter so that we can regenerate node_IDs as if session
   was fresh *)
  let reset, fresh =
    let count = ref 0 in
    (fun () ->
      count := 0),
    fun () ->
      count := !count + 1;
      !count
Sylvain Dailler's avatar
Sylvain Dailler committed
612

613
  let model_any : any Hint.t = Hint.create 17
Clément Fumex's avatar
Clément Fumex committed
614

615
  let any_from_node_ID (nid:node_ID) : any = Hint.find model_any nid
Clément Fumex's avatar
Clément Fumex committed
616
617
618
619
620
621
622
623
624
625
626
627

  let pan_to_node_ID  : node_ID Hpan.t = Hpan.create 17
  let pn_to_node_ID   : node_ID Hpn.t = Hpn.create 17
  let tn_to_node_ID   : node_ID Htn.t = Htn.create 17
  let th_to_node_ID   : node_ID Ident.Hid.t = Ident.Hid.create 7
  let file_to_node_ID : node_ID Hstr.t = Hstr.create 3

  let node_ID_from_pan  pan  = Hpan.find pan_to_node_ID pan
  let node_ID_from_pn   pn   = Hpn.find pn_to_node_ID pn
  let node_ID_from_tn   tn   = Htn.find tn_to_node_ID tn
  let node_ID_from_th   th   = Ident.Hid.find th_to_node_ID (theory_name th)
  let node_ID_from_file file = Hstr.find file_to_node_ID (file.file_name)
Clément Fumex's avatar
Clément Fumex committed
628

629
630
631
632
633
634
635
636
  let node_ID_from_any  any  =
    match any with
    | AFile file -> node_ID_from_file file
    | ATh th     -> node_ID_from_th th
    | ATn tn     -> node_ID_from_tn tn
    | APn pn     -> node_ID_from_pn pn
    | APa pan    -> node_ID_from_pan pan

Clément Fumex's avatar
Clément Fumex committed
637

Sylvain Dailler's avatar
Sylvain Dailler committed
638
  let get_prover p =
639
    let d = get_server_data () in
640
    match return_prover p d.config with
Sylvain Dailler's avatar
Sylvain Dailler committed
641
642
    | None -> raise (Bad_prover_name p)
    | Some c -> c
Clément Fumex's avatar
Clément Fumex committed
643

Sylvain Dailler's avatar
Sylvain Dailler committed
644
645
646
647
648
649
650
  let add_node_to_table node new_id =
    match node with
    | AFile file -> Hstr.add file_to_node_ID file.file_name new_id
    | ATh th     -> Ident.Hid.add th_to_node_ID (theory_name th) new_id
    | ATn tn     -> Htn.add tn_to_node_ID tn new_id
    | APn pn     -> Hpn.add pn_to_node_ID pn new_id
    | APa pan    -> Hpan.add pan_to_node_ID pan new_id
Clément Fumex's avatar
Clément Fumex committed
651

Sylvain Dailler's avatar
Sylvain Dailler committed
652
  (* Create a new node in the_tree, update the tables and send a
Clément Fumex's avatar
Clément Fumex committed
653
     notification about it *)
Sylvain Dailler's avatar
Sylvain Dailler committed
654
  let new_node ~parent node : node_ID =
655
656
    let new_id = fresh () in
      Hint.add model_any new_id node;
657
658
      let node_type = get_node_type node in
      let node_name = get_node_name node in
Sylvain Dailler's avatar
Sylvain Dailler committed
659
      add_node_to_table node new_id;
660
      P.notify (New_node (new_id, parent, node_type, node_name));
Clément Fumex's avatar
Clément Fumex committed
661
662
      new_id

Clément Fumex's avatar
Clément Fumex committed
663
  let root = 0
Sylvain Dailler's avatar
Sylvain Dailler committed
664

Sylvain Dailler's avatar
Sylvain Dailler committed
665
666
667
  (****************************)
  (* Iter on the session tree *)
  (****************************)
Sylvain Dailler's avatar
Sylvain Dailler committed
668

669
670
671
672
  (* Iter on the session tree with a function [f parent current] with type
     node_ID -> any -> unit *)
  let iter_subtree_proof_attempt_from_goal
    (f: parent:node_ID -> any -> unit) parent id =
673
    let d = get_server_data () in
Clément Fumex's avatar
Clément Fumex committed
674
    Whyconf.Hprover.iter
675
      (fun _pa panid -> f ~parent (APa panid))
676
      (get_proof_attempt_ids d.cont.controller_session id)
Clément Fumex's avatar
Clément Fumex committed
677

678
679
  let rec iter_subtree_from_goal
    (f: parent:node_ID -> any -> unit) parent id =
680
681
    let d = get_server_data () in
    let ses = d.cont.controller_session in
682
683
    f ~parent (APn id);
    let nid = node_ID_from_pn id in
Clément Fumex's avatar
Clément Fumex committed
684
    List.iter
685
      (fun trans_id -> iter_subtree_from_trans f nid trans_id)
Clément Fumex's avatar
Clément Fumex committed
686
      (get_transformations ses id);
687
    iter_subtree_proof_attempt_from_goal f nid id
Clément Fumex's avatar
Clément Fumex committed
688

689
690
  and iter_subtree_from_trans
    (f: parent:node_ID -> any -> unit) parent trans_id =
691
692
    let d = get_server_data () in
    let ses = d.cont.controller_session in
693
694
    f ~parent (ATn trans_id);
    let nid = node_ID_from_tn trans_id in
Clément Fumex's avatar
Clément Fumex committed
695
    List.iter
696
      (fun goal_id -> (iter_subtree_from_goal f nid goal_id))
Sylvain Dailler's avatar
Sylvain Dailler committed
697
      (get_sub_tasks ses trans_id)
Clément Fumex's avatar
Clément Fumex committed
698

699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
  let iter_subtree_from_theory
    (f: parent:node_ID -> any -> unit) parent theory_id =
    f ~parent (ATh theory_id);
    let nid = node_ID_from_th theory_id in
    List.iter (iter_subtree_from_goal f nid)
               (theory_goals theory_id)

  let iter_subtree_from_file
    (f: parent:node_ID -> any -> unit) parent file =
    f ~parent (AFile file);
    let nid = node_ID_from_file file in
    List.iter (iter_subtree_from_theory f nid)
      file.file_theories

  let iter_the_files (f: parent:node_ID -> any -> unit) parent : unit =
714
715
    let d = get_server_data () in
    let ses = d.cont.controller_session in
Clément Fumex's avatar
Clément Fumex committed
716
717
718
    let files = get_files ses in
    Stdlib.Hstr.iter
      (fun _ file ->
719
        iter_subtree_from_file f parent file)
Sylvain Dailler's avatar
Sylvain Dailler committed
720
      files
Clément Fumex's avatar
Clément Fumex committed
721

Sylvain Dailler's avatar
Sylvain Dailler committed
722
723
724
725
  (**********************************)
  (* Initialization of session tree *)
  (**********************************)

726
727
728
729
730
  let _init_the_tree (): unit =
    let f ~parent node_id = ignore (new_node ~parent node_id) in
    iter_the_files f root

  let init_and_send_subtree_from_trans parent trans_id : unit =
Clément Fumex's avatar
Clément Fumex committed
731
732
    iter_subtree_from_trans
      (fun ~parent id -> ignore (new_node ~parent id)) parent trans_id
733

Sylvain Dailler's avatar
Sylvain Dailler committed
734
735
736
737
  let init_and_send_file f =
    iter_subtree_from_file (fun ~parent id -> ignore (new_node ~parent id))
      root f

738
  let init_and_send_the_tree (): unit =
739
    P.notify (New_node (0, 0, NRoot, "root"));
Clément Fumex's avatar
Clément Fumex committed
740
    iter_the_files (fun ~parent id -> ignore (new_node ~parent id)) root
741
742
743
744

  let resend_the_tree (): unit =
    let send_node ~parent any =
      let node_id = node_ID_from_any any in
745
746
747
748
      let node_name = get_node_name any in
      let node_type = get_node_type any in
      P.notify (New_node (node_id, parent, node_type, node_name)) in
    P.notify (New_node (0, 0, NRoot, "root"));
749
750
    iter_the_files send_node root

Clément Fumex's avatar
Clément Fumex committed
751
752
  (* -- send the task -- *)
  let send_task nid =
753
    let d = get_server_data () in
Clément Fumex's avatar
Clément Fumex committed
754
755
    match any_from_node_ID nid with
    | APn id ->
756
757
      let task = get_task d.cont.controller_session id in
      let tables = get_tables d.cont.controller_session id in
Clément Fumex's avatar
Clément Fumex committed
758
      let s = Pp.string_of
759
          (fun fmt -> Driver.print_task ~cntexample:false ?name_table:tables d.task_driver fmt)
Clément Fumex's avatar
Clément Fumex committed
760
761
762
763
          task in
      P.notify (Task (nid,s))
    | _ ->
      P.notify (Task (nid, "can not associate a task to a node that is not a goal."))
764

Sylvain Dailler's avatar
Sylvain Dailler committed
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
  (* -------------------- *)

  (* Add a file into the session when (Add_file_req f) is sent *)
  (* Note that f is the path from execution directory to the file and fn is the
     path from the session directory to the file. *)
  let add_file_to_session cont f =
    let fn = Sysutil.relativize_filename
      (Session_itp.get_dir cont.controller_session) f in
    let files = get_files cont.controller_session in
    if (not (Stdlib.Hstr.mem files fn)) then
      if (Sys.file_exists f) then
      begin
        Controller_itp.add_file cont f;
        let file = Stdlib.Hstr.find files fn in
        init_and_send_file file
      end
      else
        P.notify (Message (Open_File_Error ("Not a file: " ^ f)))
783
    else
Sylvain Dailler's avatar
Sylvain Dailler committed
784
      P.notify (Message (Open_File_Error ("File already in session: " ^ fn)))
785
786


Clément Fumex's avatar
Clément Fumex committed
787
788
789
790
791
  (* ----------------- Schedule proof attempt -------------------- *)

  (* Callback of a proof_attempt *)
  let callback_update_tree_proof cont panid pa_status =
    let ses = cont.controller_session in
Clément Fumex's avatar
Clément Fumex committed
792
    begin match pa_status with
Clément Fumex's avatar
Clément Fumex committed
793
794
795
796
797
798
799
800
801
802
    | Scheduled ->
      begin
        try
          ignore (node_ID_from_pan panid)
        (* TODO: do we notify here ? *)
        with Not_found ->
          let parent_id = get_proof_attempt_parent ses panid in
          let parent = node_ID_from_pn parent_id in
          ignore (new_node ~parent (APa panid))
      end
Sylvain Dailler's avatar
Sylvain Dailler committed
803
804
805
    (* | Done pr -> *)
    (*   P.notify (Node_change (node_ID_from_pan panid, Proved true)) *)
                             (*{proved=(pr.pr_answer=Valid); name=""})); *)
Clément Fumex's avatar
Clément Fumex committed
806
      (* TODO: we don't want to resend the name every time, separate
Clément Fumex's avatar
Clément Fumex committed
807
808
         updatable from the rest *)
    | _  -> () (* TODO ? *)
Clément Fumex's avatar
Clément Fumex committed
809
    end;
810
811
812
    let limit = (get_proof_attempt_node cont.controller_session panid).limit in
    let new_status = Proof_status_change (pa_status, false, limit) in
    P.notify (Node_change (node_ID_from_pan panid, new_status))
Clément Fumex's avatar
Clément Fumex committed
813

814
  let notify_change_proved x b =
815
816
    try (
      let node_ID = node_ID_from_any x in
817
      P.notify (Node_change (node_ID, Proved b))
818
819
820
     )
    with Not_found -> ()

Clément Fumex's avatar
Clément Fumex committed
821
  let schedule_proof_attempt nid (p: Whyconf.config_prover) limit =
822
    let d = get_server_data () in
Clément Fumex's avatar
Clément Fumex committed
823
    let prover = p.Whyconf.prover in
824
    let callback = callback_update_tree_proof d.cont in
Clément Fumex's avatar
Clément Fumex committed
825
826
827
828
    let goals =
      match any_from_node_ID nid with
      | APn pnid   -> [pnid]
      | APa panid  ->
829
          let ses = d.cont.controller_session in
Clément Fumex's avatar
Clément Fumex committed
830
831
          [get_proof_attempt_parent ses panid]
      | ATn tn     ->
832
        List.rev (unproven_goals_below_tn d.cont [] tn)
Clément Fumex's avatar
Clément Fumex committed
833
      | AFile file ->
834
        List.rev (unproven_goals_below_file d.cont file)
Clément Fumex's avatar
Clément Fumex committed
835
      | ATh th     ->
836
        List.rev (unproven_goals_below_th d.cont [] th)
Clément Fumex's avatar
Clément Fumex committed
837
    in
838
    List.iter (fun id -> C.schedule_proof_attempt d.cont id prover
Sylvain Dailler's avatar
Sylvain Dailler committed
839
                ~limit ~callback ~notification:notify_change_proved)
Clément Fumex's avatar
Clément Fumex committed
840
841
842
843
844
845
      goals

  (* ----------------- Schedule transformation -------------------- *)

  (* Callback of a transformation *)
  let callback_update_tree_transform status =
846
    let d = get_server_data () in
Clément Fumex's avatar
Clément Fumex committed
847
848
    match status with
    | TSdone trans_id ->
849
      let ses = d.cont.controller_session in
Clément Fumex's avatar
Clément Fumex committed
850
851
      let id = get_trans_parent ses trans_id in
      let nid = node_ID_from_pn id in
852
      init_and_send_subtree_from_trans nid trans_id
Clément Fumex's avatar
Clément Fumex committed
853
854
    | TSfailed (id, e) ->
      let msg =
855
        Pp.sprintf "%a" (get_exception_message d.cont.controller_session id) e
Clément Fumex's avatar
Clément Fumex committed
856
      in
857
      P.notify (Message (Strat_error(node_ID_from_pn id, msg)))
Clément Fumex's avatar
Clément Fumex committed
858
859
860
    | _ -> ()

  let rec apply_transform nid t args =
861
    let d = get_server_data () in
Clément Fumex's avatar
Clément Fumex committed
862
863
864
    match any_from_node_ID nid with
    | APn id ->
      let callback = callback_update_tree_transform in
865
      C.schedule_transformation d.cont id t args ~callback ~notification:notify_change_proved
Clément Fumex's avatar
Clément Fumex committed
866
    | APa panid ->
867
      let parent_id = get_proof_attempt_parent d.cont.controller_session panid in
Clément Fumex's avatar
Clément Fumex committed
868
869
870
871
872
873
874
875
876
      let parent = node_ID_from_pn parent_id in
      apply_transform parent t args
    | ATn _ | AFile _ | ATh _ ->
      (* TODO: propagate trans to all subgoals, just the first one, do nothing ... ?  *)
      ()

  (* ----------------- run strategy -------------------- *)

  let run_strategy_on_task nid s =
877
    let d = get_server_data () in
Clément Fumex's avatar
Clément Fumex committed
878
879
    match any_from_node_ID nid with
    | APn id ->
880
      let l = strategies d.cont.controller_env d.config loaded_strategies in
Clément Fumex's avatar
Clément Fumex committed
881
882
883
884
885
886
887
888
      let st = List.filter (fun (_,c,_,_) -> c=s) l in
      begin
        match st with
        | [(n,_,_,st)] ->
          Format.printf "running strategy '%s'@." n;
          let callback sts =
            Format.printf "Strategy status: %a@." print_strategy_status sts
          in
889
          let callback_pa = callback_update_tree_proof d.cont in
Clément Fumex's avatar
Clément Fumex committed
890
          let callback_tr st = callback_update_tree_transform st in
891
          C.run_strategy_on_goal d.cont id st ~callback_pa ~callback_tr ~callback ~notification:notify_change_proved
Clément Fumex's avatar
Clément Fumex committed
892
893
894
        | _ -> Format.printf "Strategy '%s' not found@." s
      end
    | _ -> ()
Clément Fumex's avatar
Clément Fumex committed
895

Sylvain Dailler's avatar
Sylvain Dailler committed
896
897
  (* ----------------- Save session --------------------- *)
  let save_session () =
898
899
    let d = get_server_data () in
    Session_itp.save_session d.cont.controller_session;
Sylvain Dailler's avatar
Sylvain Dailler committed
900
901
902
903
    P.notify Saved

  (* ----------------- Reload session ------------------- *)
  let clear_tables () : unit =
904
905
    reset ();
    Hint.clear model_any;
Sylvain Dailler's avatar
Sylvain Dailler committed
906
907
908
909
910
911
912
    Hpan.clear pan_to_node_ID;
    Hpn.clear pn_to_node_ID;
    Htn.clear tn_to_node_ID;
    Ident.Hid.clear th_to_node_ID;
    Hstr.clear file_to_node_ID

  let reload_session () : unit =
913
    let d = get_server_data () in
Sylvain Dailler's avatar
Sylvain Dailler committed
914
    clear_tables ();
915
    reload_files d.cont ~use_shapes:true;
916
    init_and_send_the_tree ()
Sylvain Dailler's avatar
Sylvain Dailler committed
917
918

  let replay_session () : unit =
919
    let d = get_server_data () in
920
921
    let callback = fun lr ->
      P.notify (Message (Replay_Info (Pp.string_of C.replay_print lr))) in
Sylvain Dailler's avatar
Sylvain Dailler committed
922
    (* TODO make replay print *)
923
    C.replay ~use_steps:false d.cont ~callback:callback ~remove_obsolete:false
Sylvain Dailler's avatar
Sylvain Dailler committed
924

Clément Fumex's avatar
Clément Fumex committed
925
  (* ----------------- treat_request -------------------- *)
Clément Fumex's avatar
Clément Fumex committed
926

Sylvain Dailler's avatar
Sylvain Dailler committed
927
928
929
930
931
932
933
934
  let get_proof_node_id nid =
    try
      match any_from_node_ID nid with
      | APn pn_id -> Some pn_id
      | _ -> None
    with
      Not_found -> None

935
  let rec treat_request r =
936
    let d = get_server_data () in
937
938
    try (
    match r with
939
    | Prove_req (nid,p,limit)      ->
Sylvain Dailler's avatar
Sylvain Dailler committed
940
941
942
943
944
945
946
      let p = try Some (get_prover p) with
      | Bad_prover_name p -> P.notify (Message (Proof_error (nid, "Bad prover name" ^ p))); None
      in
      begin match p with
      | None -> ()
      | Some p -> schedule_proof_attempt nid p limit
      end
947
948
949
    | Transform_req (nid, t, args) -> apply_transform nid t args
    | Strategy_req (nid, st)       -> run_strategy_on_task nid st
    | Save_req                     -> save_session ()
Sylvain Dailler's avatar
Sylvain Dailler committed
950
    | Reload_req                   -> reload_session ()
951
    | Get_Session_Tree_req         -> resend_the_tree ()
952
953
954
955
956
957
958
959
960
961
    | Remove_subtree nid           ->
        let n = any_from_node_ID nid in
        begin
        try
          Session_itp.remove_subtree d.cont.controller_session n ~notification:(fun x ->
            let nid = node_ID_from_any x in
            P.notify (Remove nid))
        with RemoveError -> (* TODO send an error instead of information *)
          P.notify (Message (Information "Cannot remove a proof node or theory"))
        end
Sylvain Dailler's avatar
Sylvain Dailler committed
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
    | Copy_paste (from_id, to_id)    ->
        let from_any = any_from_node_ID from_id in
        let to_any = any_from_node_ID to_id in
        C.copy_paste ~notification:notify_change_proved
          ~callback_pa:(callback_update_tree_proof d.cont)
          ~callback_tr:(callback_update_tree_transform)
          d.cont from_any to_any

    | Copy_detached from_id        ->
        let from_any = any_from_node_ID from_id in
        let copy ~parent p =
          let parent = node_ID_from_any parent in
          ignore (new_node ~parent p)
        in
        C.copy_detached ~copy d.cont from_any
977
978
979
    | Get_task nid                 -> send_task nid
    | Replay_req                   -> replay_session (); resend_the_tree ()
    | Command_req (nid, cmd)       ->
Sylvain Dailler's avatar
Sylvain Dailler committed
980
      begin
Sylvain Dailler's avatar
Sylvain Dailler committed
981
        let snid = get_proof_node_id nid in
982
        match (interp commands commands_table d.config d.cont snid cmd) with
983
        | Transform (s, _t, args) -> treat_request (Transform_req (nid, s, args))
Sylvain Dailler's avatar
Sylvain Dailler committed
984
985
986
987
988
989
        | Query s                 -> P.notify (Message (Query_Info (nid, s)))
        | Prove (p, limit)        -> schedule_proof_attempt nid p limit
        | Strategies st           -> run_strategy_on_task nid st
        | Help_message s          -> P.notify (Message (Help s))
        | QError s                -> P.notify (Message (Query_Error (nid, s)))
        | Other (s, _args)        ->
Sylvain Dailler's avatar
Sylvain Dailler committed
990
            P.notify (Message (Information ("Unknown command: "^s)))
Sylvain Dailler's avatar
Sylvain Dailler committed
991
      end
Sylvain Dailler's avatar
Sylvain Dailler committed
992
    | Add_file_req f ->
993
      add_file_to_session d.cont f
994
995
996
    | Open_session_req file_or_dir_name      ->
      init_cont file_or_dir_name;
      reload_session ()
Clément Fumex's avatar
Clément Fumex committed
997
    | Set_max_tasks_req i     -> C.set_max_tasks i
998
    | Exit_req                -> exit 0
999
1000
     )
    with e -> P.notify (Message (Error (Pp.string_of
1001
1002
1003
      (fun fmt (r,e) -> Format.fprintf fmt
          "There was an unrecoverable error during treatment of request:\n %a\nwith exception: %a"
    print_request r Exn_printer.exn_printer e ) (r, e))))
Clément Fumex's avatar
Clément Fumex committed
1004
1005
1006
1007
1008

  let treat_requests () : bool =
    List.iter treat_request (P.get_requests ());
    true

1009
1010
1011
1012
1013
1014
1015
1016
1017
1018
1019
1020
  let update_monitor =
    let tr = ref 0 in
    let sr = ref 0 in
    let rr = ref 0 in
    fun t s r ->
      if (not (t = !tr && s = !sr && r = !rr)) then
        begin
          P.notify (Message (Task_Monitor (t,s,r)));
          tr := t;
          sr := s;
          rr := r
        end
Clément Fumex's avatar
Clément Fumex committed
1021

Clément Fumex's avatar
Clément Fumex committed
1022
1023
1024
1025
  let _ =
    S.timeout ~ms:100 treat_requests;
    (* S.idle ~prio:1 treat_requests; *)
    C.register_observer update_monitor
Clément Fumex's avatar
Clément Fumex committed
1026
1027

end