1
 2
 3
 4
 5
 6
 7
 8
 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

type filepath = string

module Set = Set.Make(String)

type env = {
  msgs : Diag.msg_store;
  base : filepath Lazy.t;
  imported : Set.t ref;
}

open Syntax
open Source

let rec decs env = List.iter (dec env)
and dec env d = match d.it with
  | TypD _ -> ()
  | ImportD (f, fp) ->
     let f = if Filename.is_relative f
             then Filename.concat (Lazy.force env.base) f
             else f in
     fp := f;
     env.imported := Set.add f !(env.imported)
         
let prog env p = decs env p.it.decs
   
let resolve : Syntax.prog -> filepath -> filepath list Diag.result = fun p base ->
  Diag.with_message_store (fun msgs ->
      let base = lazy (if Sys.is_directory base then base else Filename.dirname base) in
      let env = { msgs; base; imported = ref Set.empty } in
      prog env p;
      Some (Set.elements !(env.imported))
    )