2022-03-04 20:32:03 +03:00
|
|
|
(* This file is part of the Catala compiler, a specification language for tax
|
|
|
|
and social benefits computation rules. Copyright (C) 2022 Inria, OCamlPro;
|
|
|
|
contributors: Louis Gesbert <louis.gesbert@ocamlpro.com>
|
|
|
|
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License"); you may not
|
|
|
|
use this file except in compliance with the License. You may obtain a copy of
|
|
|
|
the License at
|
|
|
|
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
|
|
distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
|
|
WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
|
|
License for the specific language governing permissions and limitations under
|
|
|
|
the License. *)
|
|
|
|
|
2022-11-21 12:46:17 +03:00
|
|
|
open Catala_utils
|
|
|
|
|
2023-06-15 18:37:52 +03:00
|
|
|
type t = Cmdliner.Cmd.Exit.code Cmdliner.Cmd.t
|
2023-04-13 22:49:16 +03:00
|
|
|
|
2022-03-04 20:32:03 +03:00
|
|
|
let backend_plugins : (string, t) Hashtbl.t = Hashtbl.create 17
|
|
|
|
|
|
|
|
let register t =
|
2023-06-15 18:37:52 +03:00
|
|
|
Hashtbl.replace backend_plugins
|
|
|
|
(String.lowercase_ascii (Cmdliner.Cmd.name t))
|
|
|
|
t
|
|
|
|
|
|
|
|
let list () = Hashtbl.to_seq_values backend_plugins |> List.of_seq
|
2022-03-04 20:32:03 +03:00
|
|
|
|
|
|
|
module PluginAPI = struct
|
2023-06-15 18:37:52 +03:00
|
|
|
open Cmdliner
|
|
|
|
|
|
|
|
let register_generic info term = register (Cmd.v info term)
|
2023-04-13 22:49:16 +03:00
|
|
|
|
2023-06-15 18:37:52 +03:00
|
|
|
(* For plugins relying on the standard [Driver] *)
|
2022-03-04 20:32:03 +03:00
|
|
|
|
2023-06-15 18:37:52 +03:00
|
|
|
type 'ast plugin_apply_fun_typ =
|
|
|
|
source_file:Pos.input_file ->
|
|
|
|
output_file:string option ->
|
|
|
|
scope:Shared_ast.ScopeName.t option ->
|
|
|
|
'ast ->
|
|
|
|
Scopelang.Dependency.TVertex.t list ->
|
|
|
|
unit
|
2022-03-04 20:32:03 +03:00
|
|
|
end
|
|
|
|
|
2023-06-15 18:37:52 +03:00
|
|
|
type 'ast gen = {
|
|
|
|
name : string;
|
|
|
|
extension : string;
|
|
|
|
apply : 'ast PluginAPI.plugin_apply_fun_typ;
|
|
|
|
}
|
|
|
|
|
|
|
|
type handler =
|
|
|
|
| Dcalc of Shared_ast.untyped Dcalc.Ast.program gen
|
|
|
|
| Lcalc of Shared_ast.untyped Lcalc.Ast.program gen
|
|
|
|
| Scalc of Scalc.Ast.program gen
|
2022-03-04 20:32:03 +03:00
|
|
|
|
|
|
|
let load_file f =
|
|
|
|
try
|
|
|
|
Dynlink.loadfile f;
|
2023-06-13 12:27:45 +03:00
|
|
|
Message.emit_debug "Plugin %S loaded" f
|
2022-03-04 20:32:03 +03:00
|
|
|
with e ->
|
2023-06-15 11:51:31 +03:00
|
|
|
Message.emit_warning "Could not load plugin %S: %s" f (Printexc.to_string e)
|
2022-03-04 20:32:03 +03:00
|
|
|
|
2023-01-17 15:46:21 +03:00
|
|
|
let rec load_dir d =
|
2022-03-04 20:32:03 +03:00
|
|
|
let dynlink_exts =
|
|
|
|
if Dynlink.is_native then [".cmxs"] else [".cmo"; ".cma"]
|
|
|
|
in
|
|
|
|
Array.iter
|
|
|
|
(fun f ->
|
2023-01-17 15:46:21 +03:00
|
|
|
if f.[0] = '.' then ()
|
|
|
|
else
|
|
|
|
let f = Filename.concat d f in
|
|
|
|
if Sys.is_directory f then load_dir f
|
|
|
|
else if List.exists (Filename.check_suffix f) dynlink_exts then
|
|
|
|
load_file f)
|
2022-03-04 20:32:03 +03:00
|
|
|
(Sys.readdir d)
|